feat(db): add database seeding command (#1310)

- create db command for seeding
- implement seed command logic
- check for unapplied migrations before seeding
- integrate db command into root command

Author: zhendi

cmd/csghub-portal/cmd/db/db.go

@@ -0,0 +1,51 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/spf13/cobra"
+	"opencsg.com/portal/config"
+	"opencsg.com/portal/pkg/database"
+	"opencsg.com/portal/pkg/database/migrations"
+)
+
+func init() {
+	Cmd.AddCommand(seedCmd)
+}
+
+var Cmd = &cobra.Command{
+	Use:   "db",
+	Short: "start db seed",
+	PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+		ctx, cancel := context.WithTimeout(cmd.Context(), 10*time.Second)
+		defer cancel()
+
+		config, err := config.LoadConfig()
+		if err != nil {
+			return err
+		}
+
+		db, err := database.NewDB(config)
+		if err != nil {
+			return fmt.Errorf("initializing DB: %w", err)
+		}
+
+		migrator := migrations.NewMigrator(db)
+		status, err := migrator.MigrationsWithStatus(ctx)
+		if err != nil {
+			return fmt.Errorf("checking migrations status: %w", err)
+		}
+		unapplied := status.Unapplied()
+		if len(unapplied) > 0 {
+			return fmt.Errorf("there are %d migrations to apply: %s", len(unapplied), unapplied)
+		}
+
+		return nil
+	},
+
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return cmd.Help()
+	},
+}

cmd/csghub-portal/cmd/db/seed.go

@@ -0,0 +1,20 @@
+package db
+
+import (
+	"log/slog"
+
+	"github.com/gin-gonic/gin"
+	"github.com/spf13/cobra"
+)
+
+var seedCmd = &cobra.Command{
+	Use:   "seed",
+	Short: "start db seed",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		gin.SetMode(gin.ReleaseMode)
+
+		slog.Info("Database seeded successfully")
+
+		return nil
+	},
+}

cmd/csghub-portal/cmd/root.go

@@ -6,6 +6,7 @@ import (
 	"os"
 
 	"github.com/spf13/cobra"
+	"opencsg.com/portal/cmd/csghub-portal/cmd/db"
 	"opencsg.com/portal/cmd/csghub-portal/cmd/migration"
 	"opencsg.com/portal/cmd/csghub-portal/cmd/start"
 )
@@ -40,6 +41,7 @@ func init() {
 	RootCmd.AddCommand(
 		migration.Cmd,
 		start.Cmd,
+		db.Cmd,
 	)
 }