package xnet

import (
	"crypto/tls"
	"crypto/x509"
	"fmt"
	"io/ioutil"
	"net"
	"os"
)

type TcpListenerEx struct {
	*net.TCPListener
	listen net.Listener
}

func NewTcpServer(host string, fd uintptr) (*TcpListenerEx, error) {
	if fd > 0 {
		file := os.NewFile(fd, "")
		ln, err := net.FileListener(file)
		if err != nil {
			fmt.Println("Clone fd failed:", fd)
			return nil, err
		}
		tcpLn := ln.(*net.TCPListener)
		l := &TcpListenerEx{}
		l.TCPListener = tcpLn
		fmt.Println("Clone fd ok:", fd)
		return l, nil
	}
	tcpAddr, err := net.ResolveTCPAddr("tcp4", host)
	if err != nil {
		return nil, err
	}

	listener, err := net.ListenTCP("tcp", tcpAddr)
	if listener == nil || err != nil {
		fmt.Printf("Listen Host %s Failed\n", host)
		return nil, err
	}

	l := &TcpListenerEx{}
	l.TCPListener = listener

	return l, nil
}

func NewTLSTcpServer(host string, fd uintptr, config *TLSConfig) (*TcpListenerEx, error) {
	cert, err := tls.LoadX509KeyPair(config.Cert, config.Key)
	if err != nil {
		fmt.Println(err)
		return nil, err
	}
	pool := x509.NewCertPool()
	caCrt, err := ioutil.ReadFile(config.CA)
	if err != nil {
		fmt.Println("ReadFile err:", err)
		return nil, err
	}
	pool.AppendCertsFromPEM(caCrt)

	tlsConfig := &tls.Config{Certificates: []tls.Certificate{cert}, ClientCAs: pool}
	if config.AuthClient == false {
		tlsConfig.ClientAuth = tls.NoClientCert
	} else {
		tlsConfig.ClientAuth = tls.RequireAndVerifyClientCert
	}

	if fd > 0 {
		file := os.NewFile(fd, "")
		ln, err := net.FileListener(file)
		if err != nil {
			fmt.Println("Clone fd failed:", fd)
			return nil, err
		}
		tcpLn := ln.(*net.TCPListener)
		tmpLn := tls.NewListener(tcpLn, tlsConfig)
		l := &TcpListenerEx{}
		l.listen = tmpLn
		fmt.Println("Clone fd ok:", fd)
		return l, nil
	}
	fmt.Println("start tls listen:", host)
	listener, err := tls.Listen("tcp", host, tlsConfig)
	if listener == nil || err != nil {
		fmt.Printf("Listen Host %s Failed\n", host)
		return nil, err
	}
	l := &TcpListenerEx{}
	l.listen = listener

	return l, nil
}

func (me *TcpListenerEx) Accept() (*ConnEx, error) {
	if me.TCPListener != nil {
		connEx := &ConnEx{}
		conn, err := me.TCPListener.AcceptTCP()
		if err != nil {
			return connEx, err
		}
		conn.SetWriteBuffer(32 * 1024)
		conn.SetReadBuffer(32 * 1024)
		connEx.Conn = conn
		connEx.connIf = nil
		connEx.bConnected.Set(true)
		connEx.Init()
		return connEx, nil
	} else if me.listen != nil {
		connEx := &ConnEx{}
		conn, err := me.listen.Accept()
		if err != nil {
			return connEx, err
		}
		tcpConn, ok := conn.(*net.TCPConn)
		if ok {
			tcpConn.SetWriteBuffer(32 * 1024)
			tcpConn.SetReadBuffer(32 * 1024)
		}
		connEx.Conn = conn
		connEx.connIf = nil
		connEx.bConnected.Set(true)
		connEx.Init()
		return connEx, nil
	} else {
		return nil, nil
	}
}