From fbaeedbba441a89be7778be5cffa89a421007d17 Mon Sep 17 00:00:00 2001
From: Ellie Huxtable <e@elm.sh>
Date: Sat, 13 Feb 2021 20:54:35 +0000
Subject: Minor fixes

---
 src/command/import.rs | 2 +-
 src/main.rs           | 7 +++++++
 2 files changed, 8 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/command/import.rs b/src/command/import.rs
index 5ce0e749..a593ef61 100644
--- a/src/command/import.rs
+++ b/src/command/import.rs
@@ -112,7 +112,7 @@ impl ImportCmd {
                 }
             }
 
-            ImportCmd::Zsh => Ok(()),
+            ImportCmd::Zsh => self.import_zsh(db),
         }
     }
 }
diff --git a/src/main.rs b/src/main.rs
index 22c8b21d..86540a97 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,4 @@
+use std::env;
 use std::path::PathBuf;
 
 use directories::ProjectDirs;
@@ -84,5 +85,11 @@ impl Atuin {
 fn main() -> Result<()> {
     pretty_env_logger::init();
 
+    let session = env::var("ATUIN_SESSION");
+
+    if session.is_err() {
+        panic!("Please follow the atuin setup! Missing session.")
+    }
+
     Atuin::from_args().run()
 }
-- 
cgit v1.3.1