package main

import (
	"fmt"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"sync"
	"syscall"
	"time"

	"github.com/fsnotify/fsnotify"
)

func getFreePort(port int) {
	// macOS-compatible: use lsof to find and kill process on port
	cmd := exec.Command("sh", "-c", fmt.Sprintf("lsof -ti :%d | xargs kill -9 2>/dev/null", port))
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	if err := cmd.Run(); err != nil {
		log.Printf("Failed to free port %d: %v", port, err)
	}
}

func main() {
	watcher, err := fsnotify.NewWatcher()
	if err != nil {
		log.Fatal(err)
	}
	defer watcher.Close()

	done := make(chan bool)
	var cmd *exec.Cmd
	var mu sync.Mutex
	var debounceTimer *time.Timer
	buildAndRestart := func() {
		mu.Lock()
		defer mu.Unlock()

		// Build the application
		buildCmd := exec.Command("go", "build", "-o", "./tmp/main", "./cmd/server/main.go")
		buildCmd.Stdout = os.Stdout
		buildCmd.Stderr = os.Stderr
		if err := buildCmd.Run(); err != nil {
			log.Println("Build failed:", err)
			return
		}

		// Free the port
		getFreePort(3000)

		// Restart the application
		if cmd != nil && cmd.Process != nil {
			if err := cmd.Process.Signal(syscall.SIGTERM); err != nil {
				log.Printf("Failed to terminate process: %v", err)
			}
			_ = cmd.Wait()
		}
		cmd = exec.Command("./tmp/main")
		cmd.Stdout = os.Stdout
		cmd.Stderr = os.Stderr
		if err := cmd.Start(); err != nil {
			log.Fatal(err)
		}
		log.Println("Application restarted")
	}
	go func() {
		buildAndRestart()
		for {
			select {
			case event, ok := <-watcher.Events:
				if !ok {
					return
				}
				if (event.Op&fsnotify.Write == fsnotify.Write || event.Op&fsnotify.Create == fsnotify.Create) && filepath.Ext(event.Name) == ".go" {
					log.Println("Modified file:", event.Name)
					if debounceTimer != nil {
						debounceTimer.Stop()
					}
					debounceTimer = time.AfterFunc(500*time.Millisecond, buildAndRestart)
				}
			case err := <-watcher.Errors:
				log.Println("error:", err)
			}
		}
	}()

	err = filepath.Walk(".", func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if info.IsDir() {
			switch info.Name() {
			case "vendor", "tmp", "node_modules", ".git", ".vscode", "docs", "tests", "scripts", "github":
				return filepath.SkipDir
			default:
				return watcher.Add(path)
			}
		}
		return nil
	})
	if err != nil {
		log.Fatal(err)
	}
	<-done
}