used flag for command line arguments

db.go

@@ -4,8 +4,6 @@ import (
 	"encoding/json"
 	"fmt"
 	"log"
-	"path"
-	"runtime"
 	"time"
 
 	"github.com/boltdb/bolt"
@@ -15,12 +13,10 @@ var db *bolt.DB
 var open bool
 
 // Open to create the database and open
-func Open() error {
+func Open(filename string) error {
 	var err error
-	_, filename, _, _ := runtime.Caller(0) // get full path of this file
-	dbfile := path.Join(path.Dir(filename), "data.db")
 	config := &bolt.Options{Timeout: 30 * time.Second}
-	db, err = bolt.Open(dbfile, 0600, config)
+	db, err = bolt.Open(filename, 0600, config)
 	if err != nil {
 		fmt.Println("Opening BoltDB timed out")
 		log.Fatal(err)

main.go

@@ -1,26 +1,56 @@
 package main
 
 import (
+	"flag"
 	"fmt"
 	"log"
-	"os"
+	"path"
+	"runtime"
 
 	"github.com/gin-gonic/gin"
 )
 
-var ExternalIP string
+// AllowedIPs is a white/black list of
-var AllowedIPs string
+// IP addresses allowed to access cowyo
+var AllowedIPs = map[string]bool{
+	"192.168.1.13": true,
+	"192.168.1.12": true,
+	"192.168.1.2":  true,
+}
 
-func init() {
+// RuntimeArgs contains all runtime
-	AllowedIPs = "192.168.1.13,192.168.1.12,192.168.1.2"
+// arguments available
+var RuntimeArgs struct {
+	ExternalIP       string
+	Port             string
+	DatabaseLocation string
+	ServerCRT        string
+	ServerKey        string
 }
 
 func main() {
-	if len(os.Args) == 1 {
+	_, executableFile, _, _ := runtime.Caller(0) // get full path of this file
+	databaseFile := path.Join(path.Dir(executableFile), "data.db")
+	flag.StringVar(&RuntimeArgs.Port, "p", ":12312", "port to bind")
+	flag.StringVar(&RuntimeArgs.DatabaseLocation, "db", databaseFile, "location of database file")
+	flag.StringVar(&RuntimeArgs.ServerCRT, "crt", "", "location of ssl crt")
+	flag.StringVar(&RuntimeArgs.ServerKey, "key", "", "location of ssl key")
+	flag.CommandLine.Usage = func() {
+		fmt.Println(`cowyo: a websocket notepad
+run this to start the server and then visit localhost at the port you specify
+(see parameters).
+Example: 'cowyo localhost'
+Example: 'cowyo -p :8080 localhost'
+Example: 'cowyo -db /var/lib/cowyo/db.bolt localhost'
+Options:`)
+		flag.CommandLine.PrintDefaults()
+	}
+	flag.Parse()
+	RuntimeArgs.ExternalIP = flag.Arg(0)
+	if RuntimeArgs.ExternalIP == "" {
 		log.Fatal("You need to specify the external IP address")
 	}
-	ExternalIP = os.Args[1]
+	Open(RuntimeArgs.DatabaseLocation)
-	Open()
 	defer Close()
 
 	// Default page
@@ -34,5 +64,5 @@ func main() {
 	r.GET("/:title", editNote)
 	r.GET("/:title/*option", everythingElse)
 	r.DELETE("/listitem", deleteListItem)
-	r.Run(":12312")
+	r.Run(RuntimeArgs.Port)
 }

routes.go

@@ -26,7 +26,7 @@ func editNote(c *gin.Context) {
 	} else {
 		c.HTML(http.StatusOK, "index.tmpl", gin.H{
 			"Title":      title,
-			"ExternalIP": ExternalIP,
+			"ExternalIP": RuntimeArgs.ExternalIP,
 		})
 	}
 }