package proxy

import (
	"fmt"
	"ghproxy/config"
	"net/http"
	"sync"
	"time"

	httpc "github.com/satomitouka/touka-httpc"
)

var BufferSize int = 32 * 1024 // 32KB

var (
	tr         *http.Transport
	BufferPool *sync.Pool
	client     *httpc.Client
)

func InitReq(cfg *config.Config) {
	initHTTPClient(cfg)

	// 初始化固定大小的缓存池
	BufferPool = &sync.Pool{
		New: func() interface{} {
			return make([]byte, BufferSize)
		},
	}
}

func initHTTPClient(cfg *config.Config) {
	/*
		ctr = &http.Transport{
			MaxIdleConns:          100,
			MaxConnsPerHost:       60,
			IdleConnTimeout:       20 * time.Second,
			TLSHandshakeTimeout:   10 * time.Second,
			ExpectContinueTimeout: 1 * time.Second,
			ResponseHeaderTimeout: 10 * time.Second,
			DialContext: (&net.Dialer{
				Timeout:   30 * time.Second,
				KeepAlive: 30 * time.Second,
			}).DialContext,
		}
	*/
	if cfg.Httpc.Mode == "auto" {
		tr = &http.Transport{
			//MaxIdleConns:    160,
			IdleConnTimeout: 30 * time.Second,
			WriteBufferSize: 32 * 1024, // 32KB
			ReadBufferSize:  32 * 1024, // 32KB
		}
	} else if cfg.Httpc.Mode == "advanced" {
		tr = &http.Transport{
			MaxIdleConns:        cfg.Httpc.MaxIdleConns,
			MaxConnsPerHost:     cfg.Httpc.MaxConnsPerHost,
			MaxIdleConnsPerHost: cfg.Httpc.MaxIdleConnsPerHost,
			WriteBufferSize:     32 * 1024, // 32KB
			ReadBufferSize:      32 * 1024, // 32KB
		}
	} else {
		// 错误的模式
		logError("unknown httpc mode: %s", cfg.Httpc.Mode)
		fmt.Println("unknown httpc mode: ", cfg.Httpc.Mode)
		logWarning("use Auto to Run HTTP Client")
		fmt.Println("use Auto to Run HTTP Client")
		tr = &http.Transport{
			//MaxIdleConns:    160,
			IdleConnTimeout: 30 * time.Second,
			WriteBufferSize: 32 * 1024, // 32KB
			ReadBufferSize:  32 * 1024, // 32KB
		}
	}
	if cfg.Outbound.Enabled {
		initTransport(cfg, tr)
	}
	if cfg.Server.Debug {
		client = httpc.New(
			httpc.WithTransport(tr),
			httpc.WithDumpLog(),
		)
	} else {
		client = httpc.New(
			httpc.WithTransport(tr),
		)
	}
}