package chain

import (
	"bytes"
	"context"
	"fmt"
	"net"

	"github.com/go-gost/gost/pkg/connector"
	"github.com/go-gost/gost/pkg/hosts"
	"github.com/go-gost/gost/pkg/logger"
	"github.com/go-gost/gost/pkg/resolver"
)

type Router struct {
	Retries  int
	Chain    Chainer
	Hosts    hosts.HostMapper
	Resolver resolver.Resolver
	Logger   logger.Logger
}

func (r *Router) Dial(ctx context.Context, network, address string) (conn net.Conn, err error) {
	conn, err = r.dial(ctx, network, address)
	if err != nil {
		return
	}
	if network == "udp" || network == "udp4" || network == "udp6" {
		if _, ok := conn.(net.PacketConn); !ok {
			return &packetConn{conn}, nil
		}
	}
	return
}

func (r *Router) dial(ctx context.Context, network, address string) (conn net.Conn, err error) {
	count := r.Retries + 1
	if count <= 0 {
		count = 1
	}
	r.Logger.Debugf("dial %s/%s", address, network)

	for i := 0; i < count; i++ {
		var route *Route
		if r.Chain != nil {
			route = r.Chain.Route(network, address)
		}

		if r.Logger.IsLevelEnabled(logger.DebugLevel) {
			buf := bytes.Buffer{}
			for _, node := range route.Path() {
				fmt.Fprintf(&buf, "%s@%s > ", node.Name, node.Addr)
			}
			fmt.Fprintf(&buf, "%s", address)
			r.Logger.Debugf("route(retry=%d) %s", i, buf.String())
		}

		address, err = resolve(ctx, "ip", address, r.Resolver, r.Hosts, r.Logger)
		if err != nil {
			r.Logger.Error(err)
			break
		}

		if route != nil {
			route.logger = r.Logger
		}

		conn, err = route.Dial(ctx, network, address)
		if err == nil {
			break
		}
		r.Logger.Errorf("route(retry=%d) %s", i, err)
	}

	return
}

func (r *Router) Bind(ctx context.Context, network, address string, opts ...connector.BindOption) (ln net.Listener, err error) {
	count := r.Retries + 1
	if count <= 0 {
		count = 1
	}
	r.Logger.Debugf("bind on %s/%s", address, network)

	for i := 0; i < count; i++ {
		var route *Route
		if r.Chain != nil {
			route = r.Chain.Route(network, address)
		}

		if r.Logger.IsLevelEnabled(logger.DebugLevel) {
			buf := bytes.Buffer{}
			for _, node := range route.Path() {
				fmt.Fprintf(&buf, "%s@%s > ", node.Name, node.Addr)
			}
			fmt.Fprintf(&buf, "%s", address)
			r.Logger.Debugf("route(retry=%d) %s", i, buf.String())
		}

		ln, err = route.Bind(ctx, network, address, opts...)
		if err == nil {
			break
		}
		r.Logger.Errorf("route(retry=%d) %s", i, err)
	}

	return
}

type packetConn struct {
	net.Conn
}

func (c *packetConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
	n, err = c.Read(b)
	addr = c.Conn.RemoteAddr()
	return
}

func (c *packetConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
	return c.Write(b)
}