From b82cfcadd9bd0365cbc3d1e377c27d9f69672cf4 Mon Sep 17 00:00:00 2001
From: Kahvi-0xFF <46513413+Kahvi-0@users.noreply.github.com>
Date: Fri, 20 Sep 2024 09:21:42 -0400
Subject: [PATCH 1/4] Update cmd.go

---
 teamserver/cmd/cmd.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/teamserver/cmd/cmd.go b/teamserver/cmd/cmd.go
index 7f8319a7..ddd173de 100644
--- a/teamserver/cmd/cmd.go
+++ b/teamserver/cmd/cmd.go
@@ -32,6 +32,7 @@ func init() {
 	// server flags
 	CobraServer.Flags().SortFlags = false
 	CobraServer.Flags().StringVarP(&flags.Server.Profile, "profile", "", "", "set havoc teamserver profile")
+	CobraServer.Flags().StringVarP(&flags.Server.Database, "database", "", "false", "set havoc teamserver database (default data/teamserver.db)")
 	CobraServer.Flags().BoolVarP(&flags.Server.Debug, "debug", "", false, "enable debug mode")
 	CobraServer.Flags().BoolVarP(&flags.Server.DebugDev, "debug-dev", "", false, "enable debug mode for developers (compiles the agent with the debug mode/macro enabled)")
 	CobraServer.Flags().BoolVarP(&flags.Server.SendLogs, "send-logs", "", false, "the agent will send logs over http(s) to the teamserver")

From be0e9154b74c03efc309a069770dac8cd8f464e6 Mon Sep 17 00:00:00 2001
From: Kahvi-0xFF <46513413+Kahvi-0@users.noreply.github.com>
Date: Fri, 20 Sep 2024 09:29:19 -0400
Subject: [PATCH 2/4] Update server.go

---
 teamserver/cmd/server.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/teamserver/cmd/server.go b/teamserver/cmd/server.go
index 0fcefab2..f7c16f03 100644
--- a/teamserver/cmd/server.go
+++ b/teamserver/cmd/server.go
@@ -33,7 +33,10 @@ var CobraServer = &cobra.Command{
 			}
 			os.Exit(0)
 		}
-
+		
+		if flags.Server.Database != "" {
+			DatabasePath = (flags.Server.Database)
+		}
 		Server = server.NewTeamserver(DatabasePath)
 		Server.SetServerFlags(flags)
 

From 8d704964da016ce2234808fd9b9aaf3d760ae6b8 Mon Sep 17 00:00:00 2001
From: Kahvi-0xFF <46513413+Kahvi-0@users.noreply.github.com>
Date: Fri, 20 Sep 2024 09:30:37 -0400
Subject: [PATCH 3/4] Update types.go

---
 teamserver/cmd/server/types.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/teamserver/cmd/server/types.go b/teamserver/cmd/server/types.go
index 7fd2b556..02372781 100644
--- a/teamserver/cmd/server/types.go
+++ b/teamserver/cmd/server/types.go
@@ -42,6 +42,7 @@ type serverFlags struct {
 	Host string
 	Port string
 
+	Database string
 	Profile  string
 	Verbose  bool
 	Debug    bool

From f70bcb173c84061c5b1f4e365942d0608962d586 Mon Sep 17 00:00:00 2001
From: Kahvi-0xFF <46513413+Kahvi-0@users.noreply.github.com>
Date: Fri, 20 Sep 2024 11:46:24 -0400
Subject: [PATCH 4/4] Update cmd.go

---
 teamserver/cmd/cmd.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/teamserver/cmd/cmd.go b/teamserver/cmd/cmd.go
index ddd173de..d892ca2e 100644
--- a/teamserver/cmd/cmd.go
+++ b/teamserver/cmd/cmd.go
@@ -32,7 +32,7 @@ func init() {
 	// server flags
 	CobraServer.Flags().SortFlags = false
 	CobraServer.Flags().StringVarP(&flags.Server.Profile, "profile", "", "", "set havoc teamserver profile")
-	CobraServer.Flags().StringVarP(&flags.Server.Database, "database", "", "false", "set havoc teamserver database (default data/teamserver.db)")
+	CobraServer.Flags().StringVarP(&flags.Server.Database, "database", "", "data/teamserver.db", "set havoc teamserver database")
 	CobraServer.Flags().BoolVarP(&flags.Server.Debug, "debug", "", false, "enable debug mode")
 	CobraServer.Flags().BoolVarP(&flags.Server.DebugDev, "debug-dev", "", false, "enable debug mode for developers (compiles the agent with the debug mode/macro enabled)")
 	CobraServer.Flags().BoolVarP(&flags.Server.SendLogs, "send-logs", "", false, "the agent will send logs over http(s) to the teamserver")