mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-08-31 00:44:53 +02:00
148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
|
// Copyright The Forgejo Authors.
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package forgejo
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
"io"
|
||
|
"os"
|
||
|
"os/signal"
|
||
|
"syscall"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
"code.gitea.io/gitea/modules/log"
|
||
|
"code.gitea.io/gitea/modules/private"
|
||
|
"code.gitea.io/gitea/modules/setting"
|
||
|
|
||
|
"github.com/urfave/cli/v2"
|
||
|
)
|
||
|
|
||
|
type key int
|
||
|
|
||
|
const (
|
||
|
noInitKey key = iota + 1
|
||
|
noExitKey
|
||
|
stdoutKey
|
||
|
stderrKey
|
||
|
stdinKey
|
||
|
)
|
||
|
|
||
|
func CmdForgejo(ctx context.Context) *cli.Command {
|
||
|
return &cli.Command{
|
||
|
Name: "forgejo-cli",
|
||
|
Usage: "Forgejo CLI",
|
||
|
Flags: []cli.Flag{},
|
||
|
Subcommands: []*cli.Command{
|
||
|
CmdActions(ctx),
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func ContextSetNoInit(ctx context.Context, value bool) context.Context {
|
||
|
return context.WithValue(ctx, noInitKey, value)
|
||
|
}
|
||
|
|
||
|
func ContextGetNoInit(ctx context.Context) bool {
|
||
|
value, ok := ctx.Value(noInitKey).(bool)
|
||
|
return ok && value
|
||
|
}
|
||
|
|
||
|
func ContextSetNoExit(ctx context.Context, value bool) context.Context {
|
||
|
return context.WithValue(ctx, noExitKey, value)
|
||
|
}
|
||
|
|
||
|
func ContextGetNoExit(ctx context.Context) bool {
|
||
|
value, ok := ctx.Value(noExitKey).(bool)
|
||
|
return ok && value
|
||
|
}
|
||
|
|
||
|
func ContextSetStderr(ctx context.Context, value io.Writer) context.Context {
|
||
|
return context.WithValue(ctx, stderrKey, value)
|
||
|
}
|
||
|
|
||
|
func ContextGetStderr(ctx context.Context) io.Writer {
|
||
|
value, ok := ctx.Value(stderrKey).(io.Writer)
|
||
|
if !ok {
|
||
|
return os.Stderr
|
||
|
}
|
||
|
return value
|
||
|
}
|
||
|
|
||
|
func ContextSetStdout(ctx context.Context, value io.Writer) context.Context {
|
||
|
return context.WithValue(ctx, stdoutKey, value)
|
||
|
}
|
||
|
|
||
|
func ContextGetStdout(ctx context.Context) io.Writer {
|
||
|
value, ok := ctx.Value(stderrKey).(io.Writer)
|
||
|
if !ok {
|
||
|
return os.Stdout
|
||
|
}
|
||
|
return value
|
||
|
}
|
||
|
|
||
|
func ContextSetStdin(ctx context.Context, value io.Reader) context.Context {
|
||
|
return context.WithValue(ctx, stdinKey, value)
|
||
|
}
|
||
|
|
||
|
func ContextGetStdin(ctx context.Context) io.Reader {
|
||
|
value, ok := ctx.Value(stdinKey).(io.Reader)
|
||
|
if !ok {
|
||
|
return os.Stdin
|
||
|
}
|
||
|
return value
|
||
|
}
|
||
|
|
||
|
// copied from ../cmd.go
|
||
|
func initDB(ctx context.Context) error {
|
||
|
setting.MustInstalled()
|
||
|
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: %w", setting.CustomConf, err)
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// copied from ../cmd.go
|
||
|
func installSignals(ctx context.Context) (context.Context, context.CancelFunc) {
|
||
|
ctx, cancel := context.WithCancel(ctx)
|
||
|
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 handleCliResponseExtra(ctx context.Context, extra private.ResponseExtra) error {
|
||
|
if false && extra.UserMsg != "" {
|
||
|
if _, err := fmt.Fprintf(ContextGetStdout(ctx), "%s", extra.UserMsg); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
if ContextGetNoExit(ctx) {
|
||
|
return extra.Error
|
||
|
}
|
||
|
return cli.Exit(extra.Error, 1)
|
||
|
}
|