package proxy

import (
	"net/http/httputil"
	"net/url"
)

var _default *httputil.ReverseProxy

var prefixs = make([]string, 0, 5)
var proxys = make([]*httputil.ReverseProxy, 0, 5)

// SetDefaultProxy 设置默认代理
func SetDefaultProxy(addr string) {
	url, _ := url.Parse(addr)
	_default = httputil.NewSingleHostReverseProxy(url)
}

// AddProxy 添加代理
func AddProxy(prefix, addr string) {
	prefixs = append(prefixs, prefix)

	url, _ := url.Parse(addr)
	proxys = append(proxys, httputil.NewSingleHostReverseProxy(url))
}

// GetDefaultProxy 获取默认代理
func GetDefaultProxy() *httputil.ReverseProxy {
	return _default
}

// GetProxy 获取代理
func GetProxy(prefix string) *httputil.ReverseProxy {
	for i := 0; i < len(prefixs); i++ {
		if prefix == prefixs[i] {
			return proxys[i]
		}
	}
	return nil
}