tool.go

package main

import "strings"

// Tool 提供 main 所需的统一工具集合。
type Tool struct {
	args   []string
	tokens TokenPair
}

// NewTool 返回工具集合的实例，供 main 统一使用。
func NewTool(args []string) *Tool {
	return &Tool{args: args}
}

// optionValue 读取简单的命令行选项：
//
//	--name value / --name=value / -name value / -name=value
func (t *Tool) optionValue(name string) string {
	keyShort := "-" + name
	keyLong := "--" + name
	prefixShort := keyShort + "="
	prefixLong := keyLong + "="

	for i := 0; i < len(t.args); i++ {
		arg := t.args[i]

		if strings.HasPrefix(arg, prefixShort) {
			return strings.TrimPrefix(arg, prefixShort)
		}
		if strings.HasPrefix(arg, prefixLong) {
			return strings.TrimPrefix(arg, prefixLong)
		}

		if arg == keyShort || arg == keyLong {
			if i+1 < len(t.args) {
				return t.args[i+1]
			}
			break
		}
	}

	return ""
}

// Tokens 返回当前缓存的 token 对。
func (t *Tool) Tokens() TokenPair {
	return t.tokens
}

// setTokens 只用于在登录或刷新后更新缓存。
func (t *Tool) setTokens(pair TokenPair) {
	t.tokens = pair
}