forkjo/cmd/forgejo/forgejo.go

// Copyright The Forgejo Authors.
// SPDX-License-Identifier: MIT

package forgejo

import (
	"context"
	"fmt"
	"io"
	"os"
	"os/signal"
	"syscall"

	"code.gitea.io/gitea/modules/private"

	"github.com/urfave/cli"
)

type key int

const (
	noInstallSignalsKey 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{},
	}
}

func ContextSetNoInstallSignals(ctx context.Context, value bool) context.Context {
	return context.WithValue(ctx, noInstallSignalsKey, value)
}

func ContextGetNoInstallSignals(ctx context.Context) bool {
	value, ok := ctx.Value(noInstallSignalsKey).(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
}

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.NewExitError(extra.Error, 1)
}