From 24337588a0ad482a8f44ead453db4c6aacae272d Mon Sep 17 00:00:00 2001
From: sora233 <sora@sora233.me>
Date: Sat, 16 Jan 2021 19:25:31 +0800
Subject: [PATCH] show err when db invalid & update exit

---
 main.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index c1efe6f..58d591c 100644
--- a/main.go
+++ b/main.go
@@ -8,6 +8,7 @@ import (
 	"github.com/c-bata/go-prompt"
 	"os"
 	"path"
+	"strings"
 )
 
 var CLI struct {
@@ -18,7 +19,6 @@ var CLI struct {
 
 func main() {
 	kong.Parse(&CLI, kong.UsageOnError(), kong.Name("buntdb-cli"))
-	defer os.Exit(0)
 
 	if CLI.Debug {
 		cli.Debug = true
@@ -28,7 +28,11 @@ func main() {
 		CLI.Path = db.GetTempDbPath("buntdb-cli")
 	}
 
-	db.InitBuntDB(CLI.Path)
+	err := db.InitBuntDB(CLI.Path)
+	if err != nil {
+		fmt.Printf("ERR: %v\n", err)
+		os.Exit(1)
+	}
 	defer db.Close()
 
 	p := prompt.New(
@@ -44,7 +48,7 @@ func main() {
 			}
 		}),
 		prompt.OptionSetExitCheckerOnInput(func(in string, breakline bool) bool {
-			return in == "exit" && breakline
+			return strings.TrimSpace(in) == "exit" && breakline
 		}),
 	)
 	p.Run()