package mirror

import (
	"log"
	"net"
	"strings"

	"github.com/go-i2p/go-meta-listener"
	"github.com/go-i2p/onramp"

	wileedot "github.com/opd-ai/wileedot"
)

type Mirror struct {
	*meta.MetaListener
	Onion  *onramp.Onion
	Garlic *onramp.Garlic
}

var _ net.Listener = &Mirror{}

func (m *Mirror) Close() error {
	if err := m.MetaListener.Close(); err != nil {
		log.Println("Error closing MetaListener:", err)
	}
	if err := m.Onion.Close(); err != nil {
		log.Println("Error closing Onion:", err)
	}
	if err := m.Garlic.Close(); err != nil {
		log.Println("Error closing Garlic:", err)
	}
	return nil
}

func NewMirror(name string) (*Mirror, error) {
	inner := meta.NewMetaListener()
	name = strings.TrimSpace(name)
	name = strings.ReplaceAll(name, " ", "")
	if name == "" {
		name = "mirror"
	}
	onion, err := onramp.NewOnion("metalistener-" + name)
	if err != nil {
		return nil, err
	}
	garlic, err := onramp.NewGarlic("metalistener-"+name, "127.0.0.1:7656", onramp.OPT_WIDE)
	if err != nil {
		return nil, err
	}
	ml := &Mirror{
		MetaListener: inner,
		Onion:        onion,
		Garlic:       garlic,
	}
	return ml, nil
}

func (ml Mirror) Listen(name, addr, certdir string, hiddenTls bool) (net.Listener, error) {
	log.Println("Starting Mirror Listener")
	log.Printf("Actual args: name: '%s' addr: '%s' certDir: '%s' hiddenTls: '%t'\n", name, addr, certdir, hiddenTls)
	// determine if addr is an IP address and parse it
	//host, port, err := net.SplitHostPort(addr)
	//if err != nil {
	//host = addr
	//port = "3000"
	//}
	//addr = net.JoinHostPort(host, port)
	// Listen on plain HTTP
	tcpListener, err := net.Listen("tcp", "localhost:3000")
	if err != nil {
		return nil, err
	}
	if err := ml.AddListener("http", tcpListener); err != nil {
		return nil, err
	}
	log.Println("HTTP Local listener added http://", tcpListener.Addr())
	if hiddenTls {
		onionListener, err := ml.Onion.ListenTLS()
		if err != nil {
			return nil, err
		}
		if err := ml.AddListener("onion", onionListener); err != nil {
			return nil, err
		}
		log.Println("OnionTLS listener added https://", onionListener.Addr())
		garlicListener, err := ml.Garlic.ListenTLS()
		if err != nil {
			return nil, err
		}
		if err := ml.AddListener("garlic", garlicListener); err != nil {
			return nil, err
		}
		log.Println("GarlicTLS listener added https://", garlicListener.Addr())
	} else {
		onionListener, err := ml.Onion.Listen()
		if err != nil {
			return nil, err
		}
		if err := ml.AddListener("onion", onionListener); err != nil {
			return nil, err
		}
		log.Println("Onion listener added http://", onionListener.Addr())
		garlicListener, err := ml.Garlic.Listen()
		if err != nil {
			return nil, err
		}
		if err := ml.AddListener("garlic", garlicListener); err != nil {
			return nil, err
		}
		log.Println("Garlic listener added http://", garlicListener.Addr())
	}
	if addr != "" {
		cfg := wileedot.Config{
			Domain:         name,
			AllowedDomains: []string{name},
			CertDir:        certdir,
			Email:          addr,
		}
		tlsListener, err := wileedot.New(cfg)
		if err != nil {
			return nil, err
		}
		if err := ml.AddListener("tls", tlsListener); err != nil {
			return nil, err
		}
		log.Println("TLS listener added https://", tlsListener.Addr())
	}
	return &ml, nil
}

// Listen creates a new Mirror instance and sets up listeners for TLS, Onion, and Garlic.
// It returns the Mirror instance and any error encountered during setup.
// name is the domain name used for the TLS listener, required for Let's Encrypt.
// addr is the email address used for Let's Encrypt registration.
// It is recommended to use a valid email address for production use.
func Listen(name, addr, certdir string, hiddenTls bool) (net.Listener, error) {
	ml, err := NewMirror(name)
	if err != nil {
		return nil, err
	}
	return ml.Listen(name, addr, certdir, hiddenTls)
}