ChineseSubFinder/pkg/settings/proxy_settings.go

package settings

type ProxySettings struct {
	UseProxy                 bool   `json:"use_proxy"`                                    // 是否使用代理
	UseWhichProxyProtocol    string `json:"use_which_proxy_protocol"`                     // 是使用 socks5 还是 http 代理
	LocalHttpProxyServerPort string `json:"local_http_proxy_server_port" default:"19036"` // 本地代理服务器端口
	InputProxyAddress        string `json:"input_proxy_address"`                          // 输入的代理地址
	InputProxyPort           string `json:"input_proxy_port"`                             // 输入的代理端口
	NeedPWD                  bool   `json:"need_pwd"`                                     // 是否使用用户名密码
	InputProxyUsername       string `json:"input_proxy_username"`                         // 输入的代理用户名
	InputProxyPassword       string `json:"input_proxy_password"`                         // 输入的代理密码
	Referer                  string `json:"-"`                                            // 可能下载文件的时候需要设置
}

func NewProxySettings(useProxy bool, useWhichProxyProtocol string,
	localHttpProxyServerPort string,
	inputProxyAddress string, inputProxyPort string,
	inputProxyUsername string, inputProxyPassword string) *ProxySettings {

	set := ProxySettings{UseProxy: useProxy, UseWhichProxyProtocol: useWhichProxyProtocol,
		LocalHttpProxyServerPort: localHttpProxyServerPort,
		InputProxyAddress:        inputProxyAddress, InputProxyPort: inputProxyPort,
		InputProxyUsername: inputProxyUsername, InputProxyPassword: inputProxyPassword}

	if inputProxyUsername != "" && inputProxyPassword != "" {
		set.NeedPWD = true
	}

	return &set
}

func (p *ProxySettings) CopyOne() *ProxySettings {

	nowSettings := NewProxySettings(
		p.UseProxy, p.UseWhichProxyProtocol, p.LocalHttpProxyServerPort,
		p.InputProxyAddress, p.InputProxyPort,
		p.InputProxyUsername, p.InputProxyPassword)
	return nowSettings
}

func (p *ProxySettings) GetInfos() (bool, []string, string) {

	inputInfo := []string{
		p.UseWhichProxyProtocol,
		p.InputProxyAddress,
		p.InputProxyPort,
	}
	if p.InputProxyUsername != "" && p.InputProxyPassword != "" {
		inputInfo = append(inputInfo, p.InputProxyUsername, p.InputProxyPassword)
	}

	return p.UseProxy, inputInfo, p.LocalHttpProxyServerPort
}