Monorepo for Tangled
Something went wrong. Try again.
1.8 kB · 83 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384package main
import ( "context" "log/slog" "os" "os/signal" "syscall"
"github.com/carlmjohnson/versioninfo" "github.com/urfave/cli/v3" "tangled.org/core/knotmirror" "tangled.org/core/knotmirror/config" "tangled.org/core/knotmirror/db" "tangled.org/core/knotmirror/migrate" "tangled.org/core/log")
func main() { if err := run(os.Args); err != nil { slog.Error("error running knotmirror", "err", err) os.Exit(-1) }}
func run(args []string) error { ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM) defer cancel()
logger := log.New("knotmirror") slog.SetDefault(logger) ctx = log.IntoContext(ctx, logger)
app := cli.Command{ Name: "knotmirror", Usage: "knot mirroring service", Version: versioninfo.Short(), } app.Flags = []cli.Flag{} app.Commands = []*cli.Command{ { Name: "serve", Usage: "run the knotmirror daemon", Action: runKnotMirror, Flags: []cli.Flag{}, }, { Name: "migrate-disk", Usage: "rename mirror dirs from {did}/{rkey} -> {repo_did}; daemon must be stopped", Action: runMigrateDisk, Flags: []cli.Flag{}, }, } return app.Run(ctx, args)}
func runKnotMirror(ctx context.Context, cmd *cli.Command) error { logger := log.FromContext(ctx) cfg, err := config.Load(ctx) if err != nil { return err }
logger.Debug("config loaded:", "config", cfg) return knotmirror.Run(ctx, cfg)}
func runMigrateDisk(ctx context.Context, cmd *cli.Command) error { logger := log.FromContext(ctx) cfg, err := config.Load(ctx) if err != nil { return err } database, err := db.Make(ctx, cfg.DbUrl, 4) if err != nil { return err } defer database.Close()
stats, err := migrate.RenameDisk(ctx, cfg.GitRepoBasePath, database, logger) logger.Info("migrate-disk complete", "stats", stats.String()) return err}