gitea-tea/main.go

// Copyright 2020 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

// Tea is a scriptable command line tool for Gitea.
package main

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

	"gitea.dev/tea/internal/app"
	"gitea.dev/tea/internal/clierr"
)

func main() {
	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
	// Restore the default handling once the first signal arrived, so that a second Ctrl-C or SIGTERM
	// ends tea even while it blocks on something that ignores ctx, such as reading stdin.
	go func() {
		<-ctx.Done()
		stop()
	}()
	code := run(ctx, app.System(), os.Args[1:])
	stop()
	os.Exit(code)
}

// run executes tea, reports a failure on stderr and returns the exit code. An interrupt exits
// silently, and a declined confirmation prints only "Canceled.".
func run(ctx context.Context, deps app.Dependencies, args []string) int {
	err := app.Execute(ctx, deps, args)
	if err != nil && ctx.Err() != nil {
		// Whatever failed after Ctrl-C or SIGTERM failed because of it; the user knows why.
		err = clierr.Wrap(clierr.Interrupted, err)
	}
	code := clierr.ExitCode(err)
	switch {
	case err == nil, clierr.IsQuiet(err):
	case code == clierr.Interrupted:
		reportDraft(deps, err)
	case code == clierr.Canceled && !errors.Is(err, context.DeadlineExceeded):
		fmt.Fprintln(deps.Streams.ErrOut, "Canceled.")
		reportDraft(deps, err)
	default:
		fmt.Fprintf(deps.Streams.ErrOut, "Error: %v\n", app.Explain(ctx, deps, err))
	}
	return code
}

// reportDraft tells where the text of a canceled command was saved; the error is otherwise not shown.
func reportDraft(deps app.Dependencies, err error) {
	if path := app.DraftPath(err); path != "" {
		fmt.Fprintf(deps.Streams.ErrOut, "Draft saved at %s\n", path)
	}
}