1 files changed, 62 insertions, 0 deletions
diff --git a/cmd/admin-cli/main.go b/cmd/admin-cli/main.go
new file mode 100644
index 0000000..fdcfe2e
--- /dev/null
+++ b/cmd/admin-cli/main.go
@@ -0,0 +1,62 @@
+package main
+
+import (
+	"auth-proxy/database"
+	"auth-proxy/logging"
+	"auth-proxy/models"
+	"auth-proxy/utils"
+	"flag"
+	"fmt"
+	"os"
+
+	"github.com/joho/godotenv"
+)
+
+func main() {
+	// Initialize logging
+	logging.InitLoggers()
+
+	// Load .env file
+	if err := godotenv.Load(); err != nil {
+		logging.AppLog.Info("No .env file found, using environment variables")
+	}
+
+	username := flag.String("username", "", "Username for the new admin user")
+	password := flag.String("password", "", "Password for the new admin user")
+	flag.Parse()
+
+	if *username == "" || *password == "" {
+		logging.AppLog.Error("Both --username and --password flags are required")
+		os.Exit(1)
+	}
+
+	// Validate password policy
+	if ok, message := utils.ValidatePassword(*password); !ok {
+		logging.AppLog.Error("Password does not meet the policy", "reason", message)
+		os.Exit(1)
+	}
+
+	// Initialize DB connection
+	database.InitDB()
+	defer database.CloseDB()
+
+	// Check if user already exists
+	existingUser, err := models.GetUserByUsername(*username)
+	if err != nil {
+		logging.AppLog.Error("Error checking for existing user", "error", err)
+		os.Exit(1)
+	}
+	if existingUser != nil {
+		logging.AppLog.Error(fmt.Sprintf("User '%s' already exists.", *username))
+		os.Exit(1)
+	}
+
+	// Create admin user
+	_, err = models.CreateUser(*username, *password, "admin")
+	if err != nil {
+		logging.AppLog.Error("Failed to create admin user", "error", err)
+		os.Exit(1)
+	}
+
+	fmt.Printf("Admin user '%s' created successfully.\n", *username)
+}