Optimise docker (#34)

* Smaller dockerfile, better error handling

* Add config dir

2 files changed, 20 insertions, 8 deletions

diff --git a/src/remote/database.rs b/src/remote/database.rs
index ddcffda0..03973ca1 100644
--- a/src/remote/database.rs
+++ b/src/remote/database.rs
@@ -1,5 +1,6 @@
 use diesel::pg::PgConnection;
 use diesel::prelude::*;
+use eyre::{eyre, Result};
 
 use crate::settings::Settings;
 
@@ -7,8 +8,15 @@ use crate::settings::Settings;
 pub struct AtuinDbConn(diesel::PgConnection);
 
 // TODO: connection pooling
-pub fn establish_connection(settings: &Settings) -> PgConnection {
-    let database_url = &settings.server.db_uri;
-    PgConnection::establish(database_url)
-        .unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
+pub fn establish_connection(settings: &Settings) -> Result<PgConnection> {
+    if settings.server.db_uri == "default_uri" {
+        Err(eyre!(
+            "Please configure your database! Set db_uri in config.toml"
+        ))
+    } else {
+        let database_url = &settings.server.db_uri;
+        let conn = PgConnection::establish(database_url)?;
+
+        Ok(conn)
+    }
 }
diff --git a/src/remote/server.rs b/src/remote/server.rs
index de58397d..ee481ca4 100644
--- a/src/remote/server.rs
+++ b/src/remote/server.rs
@@ -1,5 +1,3 @@
-use rocket::config::{Config, Environment, LoggingLevel, Value};
-
 use std::collections::HashMap;
 
 use crate::remote::database::establish_connection;
@@ -7,6 +5,9 @@ use crate::settings::Settings;
 
 use super::database::AtuinDbConn;
 
+use eyre::Result;
+use rocket::config::{Config, Environment, LoggingLevel, Value};
+
 // a bunch of these imports are generated by macros, it's easier to wildcard
 #[allow(clippy::clippy::wildcard_imports)]
 use super::views::*;
@@ -16,7 +17,7 @@ use super::auth::*;
 
 embed_migrations!("migrations");
 
-pub fn launch(settings: &Settings, host: String, port: u16) {
+pub fn launch(settings: &Settings, host: String, port: u16) -> Result<()> {
     let settings: Settings = settings.clone(); // clone so rocket can manage it
 
     let mut database_config = HashMap::new();
@@ -25,7 +26,8 @@ pub fn launch(settings: &Settings, host: String, port: u16) {
     database_config.insert("url", Value::from(settings.server.db_uri.clone()));
     databases.insert("atuin", Value::from(database_config));
 
-    let connection = establish_connection(&settings);
+    let connection = establish_connection(&settings)?;
+
     embedded_migrations::run(&connection).expect("failed to run migrations");
 
     let config = Config::build(Environment::Production)
@@ -54,4 +56,6 @@ pub fn launch(settings: &Settings, host: String, port: u16) {
     .attach(AtuinDbConn::fairing())
     .register(catchers![internal_error, bad_request])
     .launch();
+
+    Ok(())
 }