/*
 * MIT License
 *
 * Copyright (c) 2020 Nicolas JUHEL
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 *
 *
 */

package httpcli

import (
	"context"
	"net"
	"net/http"
	"net/url"
	"time"

	libtls "github.com/nabbar/golib/certificates"
	liberr "github.com/nabbar/golib/errors"
	libptc "github.com/nabbar/golib/network/protocol"
	"golang.org/x/net/http2"
)

const (
	ClientTimeout30Sec = 30 * time.Second
	ClientTimeout10Sec = 10 * time.Second
	ClientTimeout5Sec  = 5 * time.Second
	ClientTimeout1Sec  = 1 * time.Second

	ClientNetworkTCP = "tcp"
	ClientNetworkUDP = "udp"
)

type FctHttpClient func() *http.Client

type HttpClient interface {
	Do(req *http.Request) (*http.Response, error)
}

func GetTransport(DisableKeepAlive, DisableCompression, ForceHTTP2 bool) *http.Transport {
	return &http.Transport{
		Proxy:              http.ProxyFromEnvironment,
		DialContext:        nil,
		DialTLSContext:     nil,
		TLSClientConfig:    nil,
		DisableKeepAlives:  DisableKeepAlive,
		DisableCompression: DisableCompression,
		ForceAttemptHTTP2:  ForceHTTP2,
	}
}

func SetTransportTLS(tr *http.Transport, tls libtls.TLSConfig, servername string) {
	tr.TLSClientConfig = tls.TlsConfig(servername)
}

func SetTransportProxy(tr *http.Transport, proxyUrl *url.URL) {
	tr.Proxy = http.ProxyURL(proxyUrl)
}

func SetTransportDial(tr *http.Transport, forceIp bool, netw libptc.NetworkProtocol, ip, local string) {
	var (
		fctDial func(ctx context.Context, network, address string) (net.Conn, error)
	)

	if forceIp && len(local) > 0 {
		u := &url.URL{
			Host: local,
		}
		fctDial = func(ctx context.Context, network, address string) (net.Conn, error) {
			dl := &net.Dialer{
				LocalAddr: &net.TCPAddr{
					IP: net.ParseIP(u.Hostname()),
				},
			}
			return dl.DialContext(ctx, netw.Code(), ip)
		}
	} else if forceIp {
		fctDial = func(ctx context.Context, network, address string) (net.Conn, error) {
			dl := &net.Dialer{}
			return dl.DialContext(ctx, netw.Code(), ip)
		}
	} else {
		dl := &net.Dialer{}
		fctDial = dl.DialContext
	}

	tr.DialContext = fctDial
}

func GetClient(tr *http.Transport, http2Tr bool, GlobalTimeout time.Duration) (*http.Client, liberr.Error) {
	if http2Tr {
		if e := http2.ConfigureTransport(tr); e != nil {
			return nil, ErrorClientTransportHttp2.Error(e)
		}
	}

	c := &http.Client{
		Transport: tr,
	}

	if GlobalTimeout != 0 {
		c.Timeout = GlobalTimeout
	}

	return c, nil
}