package ip

import (
	"context"
	"errors"
	"fmt"
	"io"
	"net"
	"net/http"
	"strings"
	"time"

	"github.com/olexsmir/viye/internal/viye"
)

type Tool struct{}

func (Tool) Name() string                 { return "ip" }
func (Tool) Match(ctx *viye.Context) bool { return len(ctx.Path) == 1 && ctx.Path[0] == "ip" }
func (Tool) Execute(ctx *viye.Context) (string, error) {
	pip, perr := getPublicIP()
	lip, lerr := getLocalIP()
	if perr != nil && lerr != nil {
		return "", fmt.Errorf("couldn't resolve public and local ips")
	}

	var buf strings.Builder
	_, _ = buf.WriteString("| ")

	if perr == nil && lerr == nil {
		_, _ = buf.WriteString(pip)
		_, _ = buf.WriteString(" / ")
		_, _ = buf.WriteString(lip)
	}

	if perr == nil && lerr != nil {
		_, _ = buf.WriteString(pip)
	}

	if lerr == nil && perr != nil {
		_, _ = buf.WriteString(pip)
	}

	_ = buf.WriteByte('\n')
	return buf.String(), nil
}

func getLocalIP() (string, error) {
	addrs, err := net.InterfaceAddrs()
	if err != nil {
		return "", err
	}
	for _, addr := range addrs {
		if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
			if ipnet.IP.To4() != nil {
				return ipnet.IP.String(), nil
			}
		}
	}
	return "", errors.New("couldn't get local error")
}

func getPublicIP() (string, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	req, err := http.NewRequestWithContext(ctx, http.MethodGet, "https://ifconfig.me/ip", nil)
	if err != nil {
		return "", err
	}

	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	ip, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}
	return strings.TrimSpace(string(ip)), nil
}