package main

import (
	"context"
	"fmt"
	"log"
	"strings"

	"git.milar.in/hyprland-tools/hypr"
	"git.milar.in/milarin/slices"
)

func main() {
	client, err := hypr.GetDefaultClient()
	if err != nil {
		panic(err)
	}

	events, err := client.Subscribe(context.Background(), hypr.EventTypeMonitorAdded)
	if err != nil {
		panic(err)
	}

	for event := range events {
		monitorName := event.Data[0]
		log.Printf("monitor %s added", monitorName)

		activeWorkspace, err := client.GetActiveWorkspace()
		if err != nil {
			log.Println(fmt.Errorf("active workspace: %w", err))
		}

		if err := client.DispatchExpectOK(fmt.Sprintf("focusmonitor %s", monitorName)); err != nil {
			log.Println(fmt.Errorf("focus monitor '%s': %w", monitorName, err))
		}

		if err := FixAllSpecialWorkspacesOnMonitor(client, monitorName); err != nil {
			log.Println(err)
			continue
		}

		if err := client.DispatchExpectOK(fmt.Sprintf("focusmonitor %s", activeWorkspace.Monitor)); err != nil {
			log.Println(fmt.Errorf("focus monitor '%s': %w", monitorName, err))
		}
	}
}

func FixAllSpecialWorkspacesOnMonitor(client *hypr.Client, monitorName string) error {
	workspaces, err := client.GetWorkspaces()
	if err != nil {
		return fmt.Errorf("retrieve workspaces from Hyprland: %w", err)
	}

	workspacesOnMonitor := slices.Filter(workspaces, WorkspaceIsOnMonitor(monitorName))
	specialWorkspacesOnMonitor := slices.Filter(workspacesOnMonitor, WorkspaceIsSpecial)

	for _, workspace := range specialWorkspacesOnMonitor {
		if err := ToggleSpecialWorkspace(client, workspace.Name); err != nil {
			return fmt.Errorf("toggle special workspace '%s': %w", workspace.Name, err)
		}

		if err := ToggleSpecialWorkspace(client, workspace.Name); err != nil {
			return fmt.Errorf("toggle special workspace '%s': %w", workspace.Name, err)
		}

		log.Printf("special workspace '%s' fixed", workspace.Name)
	}

	return nil
}

func WorkspaceIsSpecial(w *hypr.Workspace) bool {
	return w.ID < 0
}

func WorkspaceIsOnMonitor(monitorName string) func(w *hypr.Workspace) bool {
	return func(w *hypr.Workspace) bool {
		return w.Monitor == monitorName
	}
}

func ToggleSpecialWorkspace(client *hypr.Client, workspaceName string) error {
	if workspaceName == "special" {
		return client.DispatchExpectOK("togglespecialworkspace")
	}

	specialWorkspaceName := strings.TrimPrefix(workspaceName, "special:")
	return client.DispatchExpectOK(fmt.Sprintf("togglespecialworkspace %s", specialWorkspaceName))
}