package transport

import (
	"crypto/tls"
	"io"
	"net"
	"strings"
	"time"

	"github.com/haxii/fastproxy/bytebufferpool"
)

var defaultDialer Dialer

//DialTLS dial tls without pool
func DialTLS(addr string, tlsConfig *tls.Config) (net.Conn, error) {
	return defaultDialer.Dial(addr, -1, true, tlsConfig)
}

//Dial dial without pool
func Dial(addr string) (net.Conn, error) {
	return defaultDialer.Dial(addr, -1, false, nil)
}

// Forward forward remote and local connection
// It returns the number of bytes write to dst
// and the first error encountered while writing, if any.
func Forward(dst io.Writer, src io.Reader, idle time.Duration) (int64, error) {
	buffer := bytebufferpool.Get()
	defer bytebufferpool.Put(buffer)
	var err, e error
	var wn int64
	if wn, e = buffer.CopyWithIdleDuration(dst, src, idle); e != nil {
		errStr := e.Error()
		if !(strings.Contains(errStr, "broken pipe") ||
			strings.Contains(errStr, "reset by peer") ||
			strings.Contains(errStr, "i/o timeout") ||
			strings.Contains(errStr, "idle time out")) {
			err = e
		}
	}
	return wn, err
}