package config

import (
	"gopkg.in/yaml.v3"
)

// GlobalConfig holds global settings from the .config section
type GlobalConfig struct {
	Shell        string `yaml:"shell,omitempty"`
	ProjectsPath string `yaml:"projects_path,omitempty"`
}

// ConfigFile represents the top-level config file: map of session name -> config
type ConfigFile map[string]TmuxConfigItemInput

// TmuxConfigItemInput represents a single tmux session configuration
type TmuxConfigItemInput struct {
	Root        string            `yaml:"root"`
	Name        string            `yaml:"name,omitempty"`
	BlankWindow bool              `yaml:"blank_window,omitempty"`
	Windows     []TmuxWindowInput `yaml:"windows,omitempty"`
}

// TmuxWindowInput can be either a string (directory path) or a TmuxWindow struct
type TmuxWindowInput struct {
	IsString bool
	String   string
	Window   *TmuxWindow
}

// UnmarshalYAML implements custom unmarshaling for TmuxWindowInput
func (w *TmuxWindowInput) UnmarshalYAML(value *yaml.Node) error {
	// Try string first
	var s string
	if err := value.Decode(&s); err == nil {
		w.IsString = true
		w.String = s
		return nil
	}

	// Try TmuxWindow struct
	var window TmuxWindow
	if err := value.Decode(&window); err == nil {
		w.IsString = false
		w.Window = &window
		return nil
	}

	return nil
}

// TmuxWindow represents a window configuration with name, cwd, and optional layout
type TmuxWindow struct {
	Name   string           `yaml:"name,omitempty"`
	Cwd    string           `yaml:"cwd"`
	Layout *TmuxLayoutInput `yaml:"layout,omitempty"`
}

// TmuxLayoutInput can be a string, array of strings, or TmuxPaneLayout
type TmuxLayoutInput struct {
	IsString   bool
	String     string
	IsArray    bool
	Array      []string
	PaneLayout *TmuxPaneLayout
}

// UnmarshalYAML implements custom unmarshaling for TmuxLayoutInput
func (l *TmuxLayoutInput) UnmarshalYAML(value *yaml.Node) error {
	// Try string first
	var s string
	if err := value.Decode(&s); err == nil {
		l.IsString = true
		l.String = s
		return nil
	}

	// Try array of strings
	var arr []string
	if err := value.Decode(&arr); err == nil {
		l.IsArray = true
		l.Array = arr
		return nil
	}

	// Try TmuxPaneLayout struct
	var pane TmuxPaneLayout
	if err := value.Decode(&pane); err == nil {
		l.IsString = false
		l.IsArray = false
		l.PaneLayout = &pane
		return nil
	}

	return nil
}

// TmuxPaneLayout represents a pane configuration
type TmuxPaneLayout struct {
	Cwd   string           `yaml:"cwd"`
	Cmd   string           `yaml:"cmd,omitempty"`
	Zoom  bool             `yaml:"zoom,omitempty"`
	Split *TmuxSplitLayout `yaml:"split,omitempty"`
}

// TmuxSplitLayout represents a split configuration
type TmuxSplitLayout struct {
	Direction string          `yaml:"direction"` // "h" or "v"
	Child     *TmuxPaneLayout `yaml:"child"`
}

// ParsedTmuxConfigItem is the resolved/parsed version of TmuxConfigItemInput
type ParsedTmuxConfigItem struct {
	Name    string
	Root    string
	Windows []ParsedTmuxWindow
}

// ParsedTmuxWindow is the resolved/parsed version of a window
type ParsedTmuxWindow struct {
	Name   string
	Cwd    string
	Layout TmuxPaneLayout
}

// DefaultEmptyPane is the default empty pane configuration
var DefaultEmptyPane = TmuxPaneLayout{
	Cwd: ".",
	Cmd: "",
}

// DefaultEmptyLayout is the default layout with horizontal and vertical splits
var DefaultEmptyLayout = TmuxPaneLayout{
	Cwd:  ".",
	Cmd:  "",
	Zoom: false,
	Split: &TmuxSplitLayout{
		Direction: "h",
		Child: &TmuxPaneLayout{
			Cwd: ".",
			Split: &TmuxSplitLayout{
				Direction: "v",
				Child: &TmuxPaneLayout{
					Cwd: ".",
				},
			},
		},
	},
}