fj/cmd/root.go

package cmd

import (
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strconv"
	"strings"

	"forgejo.zerova.net/public/fj/internal/config"
	"forgejo.zerova.net/public/fj/internal/git"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

var cfgFile string
var jsonErrors bool

var rootCmd = &cobra.Command{
	Use:   "fj",
	Short: "Forgejo CLI tool - work seamlessly with Forgejo from the command line",
	Long: `fj is a command line tool for Forgejo instances (including Codeberg).
It brings pull requests, issues, and other Forgejo concepts to the terminal.`,
	Version:       "0.4.0",
	SilenceErrors: true,
}

// JSONErrors reports whether the --json-errors flag is set.
func JSONErrors() bool {
	return jsonErrors
}

func Execute() error {
	return rootCmd.Execute()
}

func init() {
	cobra.OnInitialize(initConfig)

	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.config/fj/config.yaml)")
	rootCmd.PersistentFlags().BoolVar(&jsonErrors, "json-errors", false, "output errors as structured JSON to stderr")
	rootCmd.PersistentFlags().String("hostname", "", "Forgejo instance hostname")
	_ = viper.BindPFlag("hostname", rootCmd.PersistentFlags().Lookup("hostname"))
}

func initConfig() {
	if cfgFile != "" {
		// Tell viper to load this file for env-style overrides AND make
		// internal/config.Load()/.Save() use it (this is the load-bearing
		// half — without SetExplicitConfigPath, --config was silently
		// ignored by every auth-touching command).
		viper.SetConfigFile(cfgFile)
		config.SetExplicitConfigPath(cfgFile)
	} else {
		home, err := os.UserHomeDir()
		if err != nil {
			fmt.Fprintln(ios.ErrOut, err)
			os.Exit(1)
		}

		configDir := home + "/.config/fj"
		legacyDir := home + "/.config/fgj"

		// Migrate from ~/.config/fgj/ if the new dir doesn't exist yet.
		if _, err := os.Stat(configDir); os.IsNotExist(err) {
			if info, err := os.Stat(legacyDir); err == nil && info.IsDir() {
				if copyErr := migrateConfigDir(legacyDir, configDir); copyErr == nil {
					fmt.Fprintln(ios.ErrOut, "notice: migrated config from ~/.config/fgj/ to ~/.config/fj/")
					fmt.Fprintln(ios.ErrOut, "        you can remove ~/.config/fgj/ when ready")
				}
			}
		}

		_ = os.MkdirAll(configDir, 0700)

		viper.AddConfigPath(configDir)
		viper.SetConfigType("yaml")
		viper.SetConfigName("config")
	}

	viper.AutomaticEnv()
	viper.SetEnvPrefix("FJ")

	_ = viper.ReadInConfig()

	// If the resolved config exists with overly permissive mode, warn — the
	// file holds API tokens. Don't fail-close; just nudge the user.
	if path, err := config.GetConfigPath(); err == nil {
		if info, statErr := os.Stat(path); statErr == nil && info.Mode()&0o077 != 0 {
			fmt.Fprintf(ios.ErrOut, "warning: %s mode %o is world/group readable; tokens may leak. chmod 600 it.\n", path, info.Mode().Perm())
		}
	}
}

// parseRepo parses the repository string in the format "owner/name".
// If not provided, it attempts to auto-detect from the git repository.
func parseRepo(repo string) (string, string, error) {
	// If repo flag is provided, use it
	if repo != "" {
		parts := strings.Split(repo, "/")
		if len(parts) != 2 {
			return "", "", fmt.Errorf("invalid repository format: %s (expected: owner/name)", repo)
		}
		return parts[0], parts[1], nil
	}

	// Try to auto-detect from git
	owner, name, err := git.DetectRepo()
	if err != nil {
		return "", "", fmt.Errorf("repository flag is required (use -R owner/name) or run from a git repository: %w", err)
	}

	return owner, name, nil
}

// getDetectedHost attempts to auto-detect the Forgejo instance hostname.
// Returns empty string if detection fails, which will fall back to other methods.
func getDetectedHost() string {
	host, err := git.DetectHost()
	if err != nil {
		return ""
	}
	return host
}

// getCwd returns the current working directory, or "" on error.
func getCwd() string {
	cwd, err := os.Getwd()
	if err != nil {
		return ""
	}
	return cwd
}

// promptLine prints a prompt to stderr and reads a line from stdin.
func promptLine(prompt string) (string, error) {
	fmt.Fprint(ios.ErrOut, prompt)
	var buf [1024]byte
	n, err := ios.In.Read(buf[:])
	if err != nil {
		return "", fmt.Errorf("reading input: %w", err)
	}
	return strings.TrimSpace(string(buf[:n])), nil
}

// parseIssueArg parses an issue/PR number from various formats:
// "123", "#123", "https://host/owner/repo/pulls/123", "https://host/owner/repo/issues/123"
func parseIssueArg(arg string) (int64, error) {
	arg = strings.TrimPrefix(arg, "#")
	// Try URL format
	if strings.HasPrefix(arg, "http") {
		parts := strings.Split(strings.TrimRight(arg, "/"), "/")
		arg = parts[len(parts)-1]
	}
	return strconv.ParseInt(arg, 10, 64)
}

// migrateConfigDir copies all files from src to dst (one level, no subdirs).
// Uses O_TRUNC so a partially-pre-existing dst file is fully replaced rather
// than having the legacy contents overwrite a prefix and leaving stale tail
// bytes — which for a YAML token store would silently corrupt config.
func migrateConfigDir(src, dst string) error {
	if err := os.MkdirAll(dst, 0700); err != nil {
		return err
	}
	entries, err := os.ReadDir(src)
	if err != nil {
		return err
	}
	for _, e := range entries {
		if e.IsDir() {
			continue
		}
		if err := copyOneConfigFile(filepath.Join(src, e.Name()), filepath.Join(dst, e.Name())); err != nil {
			return err
		}
	}
	return nil
}

func copyOneConfigFile(srcPath, dstPath string) (retErr error) {
	in, err := os.Open(srcPath)
	if err != nil {
		return err
	}
	defer func() {
		if cerr := in.Close(); retErr == nil {
			retErr = cerr
		}
	}()

	out, err := os.OpenFile(dstPath, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0600)
	if err != nil {
		return err
	}
	defer func() {
		if cerr := out.Close(); retErr == nil {
			retErr = cerr
		}
	}()

	_, err = io.Copy(out, in)
	return err
}