Save messages in a database

1 files changed, 14 insertions, 1 deletions

diff --git a/cmd/daemon/main.go b/cmd/daemon/main.go
index bad350f..242a45a 100644
--- a/cmd/daemon/main.go
+++ b/cmd/daemon/main.go
@@ -11,11 +11,15 @@ import (
 	"strings"
 
 	"go.rctt.net/solec/server"
+	"go.rctt.net/solec/storage"
+
+	_ "github.com/mattn/go-sqlite3"
 )
 
 var (
 	cmds = map[string]func(args []string){
 		"broadcast": sendBroadcast,
+		"exit":      exit,
 	}
 	serv *server.Server
 )
@@ -28,7 +32,12 @@ func main() {
 	name := flag.String("n", "localhost", "server name")
 	flag.Parse()
 
-	serv = server.NewServer(*addr, *name)
+	db, err := storage.InitDb("test.db")
+	if err != nil {
+		panic(err)
+	}
+
+	serv = server.NewServer(*addr, *name, db)
 	serv.AddChannel("test")
 
 	go func() {
@@ -64,3 +73,7 @@ func sendBroadcast(args []string) {
 	msg := strings.Join(args[0:], " ")
 	serv.SendBroadcast(msg)
 }
+
+func exit(args []string) {
+	os.Exit(0)
+}