diff --git a/cmd/banforge/main.go b/cmd/banforge/main.go index 3e3dff6..b877b8c 100644 --- a/cmd/banforge/main.go +++ b/cmd/banforge/main.go @@ -81,6 +81,19 @@ var initCmd = &cobra.Command{ fmt.Println(err) os.Exit(1) } + db, err := storage.NewDB() + if err != nil { + fmt.Println(err) + os.Exit(1) + } + db.CreateTable() + defer func() { + err = db.Close() + if err != nil { + fmt.Println(err) + os.Exit(1) + } + }() fmt.Println("Firewall detected and configured") fmt.Println("BanForge initialized successfully!")