89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func installSignals() (context.Context, context.CancelFunc) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go func() {
|
|
// install notify
|
|
signalChannel := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(
|
|
signalChannel,
|
|
syscall.SIGINT,
|
|
syscall.SIGTERM,
|
|
)
|
|
select {
|
|
case <-signalChannel:
|
|
case <-ctx.Done():
|
|
}
|
|
cancel()
|
|
signal.Reset()
|
|
}()
|
|
|
|
return ctx, cancel
|
|
}
|
|
|
|
func initDB(ctx context.Context) error {
|
|
setting.LoadCommonSettings()
|
|
setting.LoadDBSetting()
|
|
setting.InitSQLLoggersForCli(log.INFO)
|
|
|
|
if setting.Database.Type == "" {
|
|
log.Fatal(`Database settings are missing from the configuration file: %q.
|
|
Ensure you are running in the correct environment or set the correct configuration file with -c.
|
|
If this is the intended configuration file complete the [database] section.`, setting.CustomConf)
|
|
}
|
|
if err := db.InitEngine(ctx); err != nil {
|
|
return fmt.Errorf("unable to initialize the database using the configuration in %q. Error: %v", setting.CustomConf, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CmdMigrate represents the available migrate sub-command.
|
|
var CmdMigrate = &cli.Command{
|
|
Name: "migrate",
|
|
Usage: "Migrate the database",
|
|
Description: "This is a command for migrating the database, so that you can run gitea admin create-user before starting the server.",
|
|
Action: runMigrate,
|
|
}
|
|
|
|
func runMigrate(ctx *cli.Context) error {
|
|
stdCtx, cancel := installSignals()
|
|
defer cancel()
|
|
|
|
if err := initDB(stdCtx); err != nil {
|
|
return err
|
|
}
|
|
|
|
log.Info("AppPath: %s", setting.AppPath)
|
|
log.Info("AppWorkPath: %s", setting.AppWorkPath)
|
|
log.Info("Custom path: %s", setting.CustomPath)
|
|
log.Info("Log path: %s", setting.Log.RootPath)
|
|
log.Info("Configuration file: %s", setting.CustomConf)
|
|
|
|
// if err := db.InitEngineWithMigration(context.Background(), migrations.Migrate); err != nil {
|
|
// log.Fatal("Failed to initialize ORM engine: %v", err)
|
|
// return err
|
|
// }
|
|
|
|
// if err := db.InitEngineWithMigration(context.Background(), hat_migrations.Migrate); err != nil {
|
|
// log.Fatal("Failed to initialize hat ORM engine: %v", err)
|
|
// fmt.Println(err)
|
|
// return err
|
|
// }
|
|
|
|
return nil
|
|
}
|