// waste-daemon: the local peer process.
// Run one of these on each friend's machine.
package main

import (
	"flag"
	"fmt"
	"log"
	"net"
	"os"

	"github.com/waste-go/internal/crypto"
	"github.com/waste-go/internal/ipc"
	"github.com/waste-go/internal/mesh"
	"github.com/waste-go/internal/nat"
)

func main() {
	dataDir  := flag.String("data-dir", "~/.waste", "path to identity/config directory")
	alias    := flag.String("alias", "anon", "display name shown to peers (advisory only)")
	peerPort := flag.Int("peer-port", 17338, "port to accept incoming peer connections")
	ipcPort  := flag.Int("ipc-port", 17337, "port for local IPC (UI connects here)")
	relay    := flag.String("relay", "", "optional relay server address (host:port)")
	flag.Parse()

	// Load or generate identity keypair
	id, err := crypto.LoadOrCreate(expandHome(*dataDir), *alias)
	if err != nil {
		log.Fatalf("identity: %v", err)
	}
	log.Printf("daemon: local peer id: %s  alias: %s", id.PeerID().Short(), id.Alias)

	// Shared mesh state
	m := mesh.New(id)

	// Peer listener (inbound connections from other nodes)
	go func() {
		addr := fmt.Sprintf("0.0.0.0:%d", *peerPort)
		ln, err := net.Listen("tcp", addr)
		if err != nil {
			log.Fatalf("peer listener: %v", err)
		}
		log.Printf("daemon: listening for peers on %s", addr)
		for {
			conn, err := ln.Accept()
			if err != nil {
				log.Printf("peer accept error: %v", err)
				continue
			}
			go mesh.HandleConn(conn, m, false)
		}
	}()

	// NAT traversal / relay client
	go func() {
		if err := nat.Run(m, *relay); err != nil {
			log.Printf("nat: %v", err)
		}
	}()

	// IPC server — blocks until error
	if err := ipc.Run(m, *ipcPort); err != nil {
		log.Fatalf("ipc: %v", err)
	}
}

func expandHome(path string) string {
	if len(path) >= 2 && path[:2] == "~/" {
		if home, err := os.UserHomeDir(); err == nil {
			return home + path[1:]
		}
	}
	return path
}