mirror of
https://github.com/github/gh-stack.git
synced 2026-09-14 20:26:28 +08:00
Sameen Karim
126 lines
3.3 KiB
Go
126 lines
3.3 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/cli/go-gh/v2/pkg/repository"
|
|
"github.com/cli/go-gh/v2/pkg/term"
|
|
"github.com/mgutz/ansi"
|
|
|
|
ghapi "github.com/github/gh-stack/internal/github"
|
|
)
|
|
|
|
// Config holds shared state for all commands.
|
|
type Config struct {
|
|
Terminal term.Term
|
|
Out *os.File
|
|
Err *os.File
|
|
In *os.File
|
|
|
|
ColorSuccess func(string) string
|
|
ColorError func(string) string
|
|
ColorWarning func(string) string
|
|
ColorBold func(string) string
|
|
ColorBlue func(string) string
|
|
ColorMagenta func(string) string
|
|
ColorCyan func(string) string
|
|
ColorGray func(string) string
|
|
|
|
// GitHubClientOverride, when non-nil, is returned by GitHubClient()
|
|
// instead of creating a real client. Used in tests to inject a MockClient.
|
|
GitHubClientOverride ghapi.ClientOps
|
|
}
|
|
|
|
// New creates a new Config with terminal-aware output and color support.
|
|
func New() *Config {
|
|
terminal := term.FromEnv()
|
|
cfg := &Config{
|
|
Terminal: terminal,
|
|
Out: os.Stdout,
|
|
Err: os.Stderr,
|
|
In: os.Stdin,
|
|
}
|
|
|
|
if terminal.IsColorEnabled() {
|
|
cfg.ColorSuccess = ansi.ColorFunc("green")
|
|
cfg.ColorError = ansi.ColorFunc("red")
|
|
cfg.ColorWarning = ansi.ColorFunc("yellow")
|
|
cfg.ColorBold = ansi.ColorFunc("default+b")
|
|
cfg.ColorBlue = ansi.ColorFunc("blue")
|
|
cfg.ColorMagenta = ansi.ColorFunc("magenta")
|
|
cfg.ColorCyan = ansi.ColorFunc("cyan")
|
|
cfg.ColorGray = ansi.ColorFunc("default+d")
|
|
} else {
|
|
noop := func(s string) string { return s }
|
|
cfg.ColorSuccess = noop
|
|
cfg.ColorError = noop
|
|
cfg.ColorWarning = noop
|
|
cfg.ColorBold = noop
|
|
cfg.ColorBlue = noop
|
|
cfg.ColorMagenta = noop
|
|
cfg.ColorCyan = noop
|
|
cfg.ColorGray = noop
|
|
}
|
|
|
|
return cfg
|
|
}
|
|
|
|
func (c *Config) Successf(format string, args ...any) {
|
|
fmt.Fprintf(c.Err, "%s %s\n", c.ColorSuccess("\u2713"), fmt.Sprintf(format, args...))
|
|
}
|
|
|
|
func (c *Config) Errorf(format string, args ...any) {
|
|
fmt.Fprintf(c.Err, "%s %s\n", c.ColorError("\u2717"), fmt.Sprintf(format, args...))
|
|
}
|
|
|
|
func (c *Config) Warningf(format string, args ...any) {
|
|
fmt.Fprintf(c.Err, "%s %s\n", c.ColorWarning("\u26a0"), fmt.Sprintf(format, args...))
|
|
}
|
|
|
|
func (c *Config) Infof(format string, args ...any) {
|
|
fmt.Fprintf(c.Err, "%s %s\n", c.ColorCyan("\u2139"), fmt.Sprintf(format, args...))
|
|
}
|
|
|
|
func (c *Config) Printf(format string, args ...any) {
|
|
fmt.Fprintf(c.Err, format+"\n", args...)
|
|
}
|
|
|
|
func (c *Config) Outf(format string, args ...any) {
|
|
fmt.Fprintf(c.Out, format, args...)
|
|
}
|
|
|
|
// PRLink formats a PR number as a clickable, underlined terminal hyperlink.
|
|
// Falls back to plain "#N" when color is disabled.
|
|
func (c *Config) PRLink(number int, url string) string {
|
|
label := fmt.Sprintf("#%d", number)
|
|
if c.Terminal.IsColorEnabled() {
|
|
if url != "" {
|
|
// OSC 8 hyperlink
|
|
label = fmt.Sprintf("\033]8;;%s\033\\%s\033]8;;\033\\", url, label)
|
|
}
|
|
// Underline
|
|
label = fmt.Sprintf("\033[4m%s\033[24m", label)
|
|
}
|
|
return label
|
|
}
|
|
|
|
func (c *Config) IsInteractive() bool {
|
|
return c.Terminal.IsTerminalOutput()
|
|
}
|
|
|
|
func (c *Config) Repo() (repository.Repository, error) {
|
|
return repository.Current()
|
|
}
|
|
|
|
func (c *Config) GitHubClient() (ghapi.ClientOps, error) {
|
|
if c.GitHubClientOverride != nil {
|
|
return c.GitHubClientOverride, nil
|
|
}
|
|
repo, err := c.Repo()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("determining repository: %w", err)
|
|
}
|
|
return ghapi.NewClient(repo.Host, repo.Owner, repo.Name)
|
|
}
|