86 lines
2.3 KiB
Go
86 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"git.haelnorr.com/h/golib/hlog"
|
|
"git.haelnorr.com/h/oslstats/internal/config"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func main() {
|
|
flags, err := config.SetupFlags()
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error parsing flags: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
cfg, loader, err := config.GetConfig(flags)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", errors.Wrap(err, "Failed to load config"))
|
|
os.Exit(1)
|
|
}
|
|
// Handle utility flags
|
|
if flags.EnvDoc || flags.ShowEnv {
|
|
if err = loader.PrintEnvVarsStdout(flags.ShowEnv); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", errors.Wrap(err, "Failed to print env doc"))
|
|
}
|
|
return
|
|
}
|
|
|
|
if flags.GenEnv != "" {
|
|
if err = loader.GenerateEnvFile(flags.GenEnv, true); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", errors.Wrap(err, "Failed to generate env file"))
|
|
}
|
|
return
|
|
}
|
|
|
|
// Setup the logger
|
|
logger, err := hlog.NewLogger(cfg.HLOG, os.Stdout)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%s\n", errors.Wrap(err, "Failed to init logger"))
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Handle migration file creation (doesn't need DB connection)
|
|
if flags.MigrateCreate != "" {
|
|
if err := createMigration(flags.MigrateCreate); err != nil {
|
|
logger.Fatal().Err(err).Str("stacktrace", fmt.Sprintf("%+v", errors.Wrap(err, "createMigration"))).Msg("Error creating migration")
|
|
}
|
|
return
|
|
}
|
|
|
|
// Handle commands that need database connection
|
|
if flags.MigrateUp != "" || flags.MigrateRollback != "" ||
|
|
flags.MigrateStatus || flags.MigrateDryRun ||
|
|
flags.ResetDB {
|
|
|
|
// Route to appropriate command
|
|
if flags.MigrateUp != "" {
|
|
err = runMigrations(ctx, cfg, "up", flags.MigrateUp)
|
|
} else if flags.MigrateRollback != "" {
|
|
err = runMigrations(ctx, cfg, "rollback", flags.MigrateRollback)
|
|
} else if flags.MigrateStatus {
|
|
err = runMigrations(ctx, cfg, "status", "")
|
|
} else if flags.MigrateDryRun {
|
|
err = runMigrations(ctx, cfg, "dry-run", "")
|
|
} else if flags.ResetDB {
|
|
err = resetDatabase(ctx, cfg)
|
|
}
|
|
|
|
if err != nil {
|
|
logger.Fatal().Err(err).Str("stacktrace", fmt.Sprintf("%+v", errors.Wrap(err, "dbFlags"))).Msg("Error migrating database")
|
|
}
|
|
return
|
|
}
|
|
|
|
// Normal server startup
|
|
if err := run(ctx, logger, cfg); err != nil {
|
|
logger.Fatal().Err(err).Str("stacktrace", fmt.Sprintf("%+v", errors.Wrap(err, "run"))).Msg("Error starting server")
|
|
}
|
|
}
|