teai/internal/cli/login.go

// login.go — teai login 指令：管理 teai 自己的登入組態。
//
// 組態由 teai 自行讀寫（README 設計原則 5），不依賴 tea。login 的職責是：
// 以 token 向站點驗證（GET /user），成功後才寫入組態並記下帳號。
// token 只走記憶體與組態檔，不進輸出、日誌或錯誤訊息。
//
// 子命令：list／add／default／remove。
package cli

import (
	"bufio"
	"context"
	"errors"
	"flag"
	"fmt"
	"io"
	"net/url"
	"os"
	"path/filepath"
	"strings"

	"gitea.alterminal.com/alterminal/teai/internal/gitea"
)

// loginStdin 是讀 token 與確認的輸入（測試可替換）。
var loginStdin io.Reader = os.Stdin

// loginConfigPath 回傳組態檔路徑：--config／TEAI_CONFIG 優先，其次
// ~/.config/teai/config.yml。找不到可用路徑回空字串。
func loginConfigPath(g Globals) string {
	if p := strings.TrimSpace(g.ConfigPath); p != "" {
		return expandHome(p)
	}
	if p := os.Getenv("TEAI_CONFIG"); strings.TrimSpace(p) != "" {
		return expandHome(strings.TrimSpace(p))
	}
	home, err := os.UserHomeDir()
	if err != nil || home == "" {
		return ""
	}
	return filepath.Join(home, ".config", "teai", "config.yml")
}

// expandHome 展開路徑開頭的 ~。
func expandHome(p string) string {
	if p == "~" || strings.HasPrefix(p, "~/") {
		if home, err := os.UserHomeDir(); err == nil {
			return filepath.Join(home, strings.TrimPrefix(p, "~"))
		}
	}
	return p
}

// runLogin 是 teai login 的進入點：分派子命令。
func runLogin(env *Env, args []string) error {
	if len(args) == 0 {
		return &ErrUsage{Msg: "login requires a subcommand: list, add, default, remove"}
	}
	sub, rest := args[0], args[1:]
	switch sub {
	case "list":
		return runLoginList(env, rest)
	case "add":
		return runLoginAdd(env, rest)
	case "default":
		return runLoginDefault(env, rest)
	case "remove":
		return runLoginRemove(env, rest)
	default:
		return &ErrUsage{Msg: fmt.Sprintf("unknown login subcommand %q (want list, add, default, remove)", sub)}
	}
}

// loginFields 是輸出欄位（不含 token）。
var loginFields = []gitea.Field{
	{Name: "name"},
	{Name: "url"},
	{Name: "user"},
	{Name: "default"},
}

// printLoginList 依格式輸出登入清單。
func printLoginList(env *Env, logins []gitea.LoginEntry) error {
	p := gitea.Printer{Format: env.Globals.Output, Out: env.Out}
	rows := make([]gitea.Values, 0, len(logins))
	for _, l := range logins {
		rows = append(rows, gitea.Values{l.Name, l.URL, l.User, l.Default})
	}
	return p.PrintList(loginFields, rows)
}

// runLoginList 列出組態中的登入（一律不含 token）。
func runLoginList(env *Env, args []string) error {
	fs := flag.NewFlagSet("teai login list", flag.ContinueOnError)
	fs.SetOutput(env.Err)
	if err := parseFlags(fs, args); err != nil {
		return err
	}
	if fs.NArg() > 0 {
		return &ErrUsage{Msg: fmt.Sprintf("unexpected argument %q", fs.Arg(0))}
	}
	path := loginConfigPath(env.Globals)
	if path == "" {
		return fmt.Errorf("cannot determine config path")
	}
	logins, err := gitea.ListLoginsFile(path)
	if err != nil {
		return fmt.Errorf("read config %s: %w", path, err)
	}
	return printLoginList(env, logins)
}

// runLoginAdd 新增或更新登入：先驗證 token（GET /user），成功才寫入。
//
// token 來源：--token 旗標，或 stdin（預設未給旗標時讀一行，
// 避免 token 進 shell 歷史）。
func runLoginAdd(env *Env, args []string) error {
	fs := flag.NewFlagSet("teai login add", flag.ContinueOnError)
	fs.SetOutput(env.Err)
	urlFlag := fs.String("url", env.Globals.URL, "Gitea 站點網址")
	nameFlag := fs.String("name", "", "登入名稱（省略時取 URL 主機）")
	tokenFlag := fs.String("token", "", "API token（省略時自 stdin 讀一行）")
	defaultFlag := fs.Bool("default", true, "設為預設登入")
	if err := parseFlags(fs, args); err != nil {
		return err
	}
	if fs.NArg() > 0 {
		return &ErrUsage{Msg: fmt.Sprintf("unexpected argument %q", fs.Arg(0))}
	}
	siteURL := strings.TrimSuffix(strings.TrimSpace(*urlFlag), "/")
	if siteURL == "" {
		return &ErrUsage{Msg: "--url must not be empty"}
	}

	token := strings.TrimSpace(*tokenFlag)
	if token == "" {
		t, err := readLineFrom(loginStdin)
		if err != nil {
			return fmt.Errorf("read token from stdin: %w", err)
		}
		token = t
	}
	if token == "" {
		return &ErrUsage{Msg: "token must not be empty"}
	}

	path := loginConfigPath(env.Globals)
	if path == "" {
		return fmt.Errorf("cannot determine config path")
	}
	data, err := os.ReadFile(path)
	if err != nil && !os.IsNotExist(err) {
		return fmt.Errorf("read config %s: %w", path, err)
	}

	name := strings.TrimSpace(*nameFlag)
	if name == "" {
		// 未指定名稱：同站（主機相同）已有登入就就地更新，避免重複項目。
		if existing, ok := findLoginByHost(data, siteURL); ok {
			name = existing.Name
		} else {
			name = defaultLoginName(siteURL)
		}
	}

	// 先驗證再寫入：token 無效時不動組態檔。
	user, err := verifyToken(env, siteURL, token)
	if err != nil {
		return err
	}
	out, final, err := gitea.UpsertLogin(data, gitea.LoginEntry{
		Name:    name,
		URL:     siteURL,
		Token:   token,
		Default: *defaultFlag,
		User:    user,
	}, *defaultFlag)
	if err != nil {
		return err
	}
	if err := gitea.SaveConfigFile(path, out); err != nil {
		return err
	}
	p := gitea.Printer{Format: env.Globals.Output, Out: env.Out}
	return p.PrintOne(loginFields, gitea.Values{final.Name, final.URL, final.User, final.Default})
}

// readLineFrom 讀一行並去除頭尾空白；EOF 且無內容回空字串。
func readLineFrom(r io.Reader) (string, error) {
	line, err := bufio.NewReader(r).ReadString('\n')
	if err != nil && err != io.EOF {
		return "", err
	}
	return strings.TrimSpace(line), nil
}

// verifyToken 以 token 呼叫 GET /user，回傳帳號名稱。
// 401 特別標明是 token 遭拒，其餘依 *gitea.ErrAPI 對應結束碼 3。
func verifyToken(env *Env, siteURL, token string) (string, error) {
	c, err := gitea.New(siteURL, token, env.Globals.Timeout)
	if err != nil {
		return "", &ErrUsage{Msg: err.Error()}
	}
	ctx, cancel := context.WithTimeout(context.Background(), env.Globals.Timeout)
	defer cancel()
	var out struct {
		Login string `json:"login"`
	}
	if err := c.GetJSON(ctx, "/user", &out); err != nil {
		var api *gitea.ErrAPI
		if errors.As(err, &api) && api.StatusCode == 401 {
			// 包 %w 保留 *ErrAPI：結束碼對應 3（API 錯誤）。
			return "", fmt.Errorf("token rejected by %s: %w", siteURL, api)
		}
		return "", err
	}
	if out.Login == "" {
		return "", fmt.Errorf("empty account name returned by %s", siteURL)
	}
	return out.Login, nil
}

// runLoginDefault 設定預設登入。
func runLoginDefault(env *Env, args []string) error {
	fs := flag.NewFlagSet("teai login default", flag.ContinueOnError)
	fs.SetOutput(env.Err)
	if err := parseFlags(fs, args); err != nil {
		return err
	}
	if fs.NArg() != 1 {
		return &ErrUsage{Msg: "login default requires exactly one login name"}
	}
	name := fs.Arg(0)
	data, err := readLoginConfig(env, "login default")
	if err != nil {
		return err
	}
	entry, found := findLogin(data, name)
	if !found {
		return fmt.Errorf("gitea: login %q not found", name)
	}
	out, err := gitea.SetDefaultLogin(data, name)
	if err != nil {
		return err
	}
	if err := saveLoginConfig(env, out); err != nil {
		return err
	}
	p := gitea.Printer{Format: env.Globals.Output, Out: env.Out}
	return p.PrintOne(loginFields, gitea.Values{entry.Name, entry.URL, entry.User, true})
}

// runLoginRemove 移除登入；未加 --yes 時先確認。
func runLoginRemove(env *Env, args []string) error {
	fs := flag.NewFlagSet("teai login remove", flag.ContinueOnError)
	fs.SetOutput(env.Err)
	yesFlag := fs.Bool("yes", false, "跳過確認")
	if err := parseFlags(fs, args); err != nil {
		return err
	}
	if fs.NArg() != 1 {
		return &ErrUsage{Msg: "login remove requires exactly one login name"}
	}
	name := fs.Arg(0)
	data, err := readLoginConfig(env, "login remove")
	if err != nil {
		return err
	}
	entry, found := findLogin(data, name)
	if !found {
		return fmt.Errorf("gitea: login %q not found", name)
	}
	if !*yesFlag {
		confirmed, err := confirmRemove(env, entry)
		if err != nil {
			return err
		}
		if !confirmed {
			fmt.Fprintln(env.Out, "cancelled")
			return nil
		}
	}
	out, promoted, err := gitea.RemoveLogin(data, name)
	if err != nil {
		return err
	}
	if err := saveLoginConfig(env, out); err != nil {
		return err
	}
	if promoted != "" {
		fmt.Fprintf(env.Err, "default login is now %q\n", promoted)
	}
	p := gitea.Printer{Format: env.Globals.Output, Out: env.Out}
	return p.PrintOne(loginFields, gitea.Values{entry.Name, entry.URL, entry.User, false})
}

// confirmRemove 印出待移除項目（不含 token）並讀 y/yes 確認；
// 其他輸入（含 EOF）一律視為取消。
func confirmRemove(env *Env, entry gitea.LoginEntry) (bool, error) {
	fmt.Fprintf(env.Err, "remove login %q (url %s, user %s)? [y/N] ", entry.Name, entry.URL, entry.User)
	line, err := readLineFrom(loginStdin)
	if err != nil {
		return false, fmt.Errorf("read confirmation: %w", err)
	}
	switch strings.ToLower(line) {
	case "y", "yes":
		return true, nil
	default:
		return false, nil
	}
}

// readLoginConfig 讀組態檔內容；路徑無法判定或讀取失敗時回錯。
func readLoginConfig(env *Env, op string) ([]byte, error) {
	path := loginConfigPath(env.Globals)
	if path == "" {
		return nil, fmt.Errorf("cannot determine config path")
	}
	data, err := os.ReadFile(path)
	if err != nil {
		return nil, fmt.Errorf("%s: read config %s: %w", op, path, err)
	}
	return data, nil
}

// saveLoginConfig 原子寫回組態檔。
func saveLoginConfig(env *Env, data []byte) error {
	path := loginConfigPath(env.Globals)
	if path == "" {
		return fmt.Errorf("cannot determine config path")
	}
	return gitea.SaveConfigFile(path, data)
}

// findLoginByHost 在組態內容中找 URL 主機與 siteURL 相同的登入。
func findLoginByHost(data []byte, siteURL string) (gitea.LoginEntry, bool) {
	for _, l := range gitea.ListLogins(data) {
		if gitea.HostOf(l.URL) == gitea.HostOf(siteURL) {
			return l, true
		}
	}
	return gitea.LoginEntry{}, false
}

// findLogin 在組態內容中找指定名稱的登入。
func findLogin(data []byte, name string) (gitea.LoginEntry, bool) {
	for _, l := range gitea.ListLogins(data) {
		if l.Name == name {
			return l, true
		}
	}
	return gitea.LoginEntry{}, false
}

// defaultLoginName 以 URL 主機作為登入名稱。
func defaultLoginName(rawURL string) string {
	s := rawURL
	if !strings.Contains(s, "://") {
		s = "https://" + s
	}
	if u, err := url.Parse(s); err == nil && u.Host != "" {
		return u.Host
	}
	return rawURL
}