package common

import (
	"encoding/base64"
	"errors"
	"fmt"
	"regexp"

	"net"
	"strconv"
)

type Addr struct {
	scheam                  string
	user, passwd            string
	ip                      string
	port                    int
	httpAuthorizationHeader string
}

func ParseAddr(str string) (cfg *Addr, err error) {
	defer func() {
		if cfg != nil && cfg.user != "" && cfg.passwd != "" {
			cfg.httpAuthorizationHeader = fmt.Sprintf("Proxy-Authorization: Basic %s", base64.URLEncoding.EncodeToString([]byte(cfg.user+":"+cfg.passwd)))

		}

	}()
	r, _ := regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)@(\S+):(\d+)`)
	m := r.FindAllStringSubmatch(str, 1)

	if m != nil {

		addr, err := net.ResolveTCPAddr("tcp", m[0][4]+":"+m[0][5])
		if err != nil {
			return nil, errors.New("配置解析错误 " + m[0][4] + ":" + m[0][5] + " 不是有效的 地址:端口")
		}
		return &Addr{
			scheam: m[0][1],
			user:   m[0][2],
			passwd: m[0][3],
			ip:     m[0][4],
			port:   addr.Port,
		}, nil
	}
	r, _ = regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)@(\d+)`)
	m = r.FindAllStringSubmatch(str, 1)

	if m != nil {

		port, _ := strconv.Atoi(m[0][4])
		return &Addr{
			scheam: m[0][1],
			user:   m[0][2],
			passwd: m[0][3],
			ip:     "",
			port:   port,
		}, nil
	}
	r, _ = regexp.Compile(`^(http://|socks5://)?(\S+):(\S+)$`)
	m = r.FindAllStringSubmatch(str, 1)

	if m != nil {

		addr, err := net.ResolveTCPAddr("tcp", m[0][2]+":"+m[0][3])
		if err != nil {
			return nil, errors.New("配置解析错误 " + m[0][1] + ":" + m[0][2] + " 不是有效的 地址:端口")
		}

		return &Addr{
			scheam: m[0][1],
			user:   "",
			passwd: "",
			ip:     m[0][2],
			port:   addr.Port,
		}, nil
	}

	port, err := strconv.Atoi(str)
	if err != nil {
		return nil, errors.New("配置解析错误，请按照 用户名:密码@地址:端口 的方式填写，或者 用户名:密码@端口 或者 ip:端口 或者 只有端口")
	}
	return &Addr{port: port}, nil
}
func (c *Addr) IP() string {
	return c.ip
}
func (c *Addr) Addr() string {
	return fmt.Sprintf("%s:%d", c.ip, c.port)
}
func (c *Addr) Port() string {
	return fmt.Sprintf("%d", c.port)
}
func (c *Addr) String() string {
	if c.user == "" && c.passwd == "" {
		if c.ip == "" {
			return fmt.Sprintf("%d", c.port)
		}

		return fmt.Sprintf("%s%s:%d", c.scheam, c.ip, c.port)
	}

	return fmt.Sprintf("%s%s:%s@%s:%d", c.scheam, c.user, c.passwd, c.ip, c.port)
}
func (c *Addr) GetHttpAuthorizationHeader() string {
	return c.httpAuthorizationHeader
}
func (c *Addr) User() string {
	return c.user
}
func (c *Addr) Password() string {
	return c.passwd
}
func (c *Addr) Scheam() string {
	return c.scheam
}
func (c *Addr) HttpUrl() string {
	return "http://" + c.Addr()
}