new .db file be obend as argument

1 files changed, 25 insertions, 3 deletions

diff --git a/main.go b/main.go
index 0d73a7a..488e2b3 100644
--- a/main.go
+++ b/main.go
@@ -3,6 +3,7 @@ package main
 import (
     "fmt"
     "flag"
+    "path/filepath"
     _ "log"
     //"bufio"
     "os"
@@ -92,11 +93,30 @@ func stdOut() {
 }
 
 func main() {
+    fmt.Println("Laboravi Started")
+    dbname := "./.mytimes.db"
     if len(flag.Args())>0 {
-        fmt.Println("Unknown Commands: ",flag.Args())
+
+        aar := flag.Args()
+        var uar []string
+        haveone := false
+        for _,car := range aar {
+            if filepath.Ext(car) == ".db" {
+                if haveone {
+                    fmt.Println("Only One DB Allowed! Will ignore",car)
+                } else {
+                    dbname=car
+                    haveone=true
+                }
+            }else{
+                uar = append(uar,car)
+            }
+        }
+        if len(uar)>0 {
+            fmt.Println("Unknown Commands: ",uar)
+        }
     }
-    fmt.Println("Laboravi Started")
-    initDB("./.mytimes.db")
+    initDB(dbname)
     getLastProject()
     //getProjects()
     getOpenTask()
@@ -116,6 +136,7 @@ func main() {
         addCustomer()
         os.Exit(0)
     } else if newtask {
+        showLastProject()
         newTask(projectid)
         stdOut()
         //showLastProject()
@@ -129,6 +150,7 @@ func main() {
         //getClosedTasks(0)
         //showOpenTask()
     } else if (starttime != "now") {
+        showLastProject()
         newTaskTime(projectid,starttime)
         os.Exit(0)
     } else if (stoptime != "now") {