54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/cli/cli/v2/cmd/ghcs"
|
|
"github.com/cli/cli/v2/cmd/ghcs/output"
|
|
"github.com/cli/cli/v2/internal/api"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func main() {
|
|
token := os.Getenv("GITHUB_TOKEN")
|
|
rootCmd := ghcs.NewRootCmd(ghcs.NewApp(
|
|
output.NewLogger(os.Stdout, os.Stderr, false),
|
|
api.New(token, http.DefaultClient),
|
|
))
|
|
|
|
// Require GITHUB_TOKEN through a Cobra pre-run hook so that Cobra's help system for commands can still
|
|
// function without the token set.
|
|
oldPreRun := rootCmd.PersistentPreRunE
|
|
rootCmd.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
|
|
if token == "" {
|
|
return errTokenMissing
|
|
}
|
|
if oldPreRun != nil {
|
|
return oldPreRun(cmd, args)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
if cmd, err := rootCmd.ExecuteC(); err != nil {
|
|
explainError(os.Stderr, err, cmd)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
var errTokenMissing = errors.New("GITHUB_TOKEN is missing")
|
|
|
|
func explainError(w io.Writer, err error, cmd *cobra.Command) {
|
|
if errors.Is(err, errTokenMissing) {
|
|
fmt.Fprintln(w, "The GITHUB_TOKEN environment variable is required. Create a Personal Access Token at https://github.com/settings/tokens/new?scopes=repo")
|
|
fmt.Fprintln(w, "Make sure to enable SSO for your organizations after creating the token.")
|
|
return
|
|
}
|
|
if errors.Is(err, ghcs.ErrTooManyArgs) {
|
|
_ = cmd.Usage()
|
|
return
|
|
}
|
|
}
|