package ui

import (
	"github.com/charmbracelet/bubbles/key"
	"github.com/charmbracelet/bubbles/list"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"

	"github.com/taigrr/gico/commits"
)

const (
	authors SettingsCursor = iota
	repos
)

type (
	SettingsCursor int
	Settings       struct {
		AllAuthors       selectablelist
		SelectedAuthors  []string
		AllRepos         selectablelist
		SelectedRepos    []string
		cursor           SettingsCursor
		highlightedEntry int
		AuthorList       list.Model
		RepoList         list.Model
	}
)

type selectablelist []selectable

type selectable struct {
	text     string
	selected bool
}

func (i selectable) Title() string       { return i.text }
func (i selectable) FilterValue() string { return i.text }
func (i selectablelist) GetSelected() []string {
	selected := []string{}
	for _, v := range i {
		if v.selected {
			selected = append(selected, v.text)
		}
	}
	return selected
}

var settingsKey = key.NewBinding(
	key.WithKeys("ctrl+g"),
	key.WithHelp("ctrl+g", "press ctrl+g to open settings"),
)

func (m Settings) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	if msg, ok := msg.(tea.KeyMsg); ok {
		if msg.String() == "tab" {
			if m.cursor == authors {
				m.cursor = repos
			} else {
				m.cursor = authors
			}
		}
	}
	if msg, ok := msg.(tea.WindowSizeMsg); ok {
		m.AuthorList.SetHeight(msg.Height - 8)
		m.RepoList.SetHeight(msg.Height - 8)
	}
	switch m.cursor {
	case authors:
		var cmd tea.Cmd
		batch := []tea.Cmd{}
		switch msg := msg.(type) {
		case tea.KeyMsg:
			switch msg.String() {
			case " ", "enter":
				selected := 0
				if !m.AuthorList.IsFiltered() {
					selected = m.AuthorList.Index()
					m.AllAuthors[selected].selected = !m.AllAuthors[selected].selected
					cmd = m.AuthorList.SetItem(selected, m.AllAuthors[selected])
					batch = append(batch, cmd)
				} else {
					val := m.AuthorList.SelectedItem()
					selectedItem, ok := val.(selectable)
					if ok {
						for i, v := range m.AllAuthors {
							if v.text == selectedItem.text {
								selected = i
								break
							}
						}
						m.AllAuthors[selected].selected = !m.AllAuthors[selected].selected
						cmd = m.AuthorList.SetItem(selected, m.AllAuthors[selected])
						batch = append(batch, cmd)
					}
				}
			}
		}
		m.SelectedAuthors = m.AllAuthors.GetSelected()
		m.AuthorList, cmd = m.AuthorList.Update(msg)
		batch = append(batch, cmd)
		return m, tea.Batch(batch...)
	case repos:
		var cmd tea.Cmd
		batch := []tea.Cmd{}
		switch msg := msg.(type) {
		case tea.KeyMsg:
			switch msg.String() {
			case " ", "enter":
				selected := 0
				if !m.RepoList.IsFiltered() {
					selected = m.RepoList.Index()
					m.AllRepos[selected].selected = !m.AllRepos[selected].selected
					cmd = m.RepoList.SetItem(selected, m.AllRepos[selected])
					batch = append(batch, cmd)
				} else {
					val := m.RepoList.SelectedItem()
					selectedItem, ok := val.(selectable)
					if ok {
						for i, v := range m.AllRepos {
							if v.text == selectedItem.text {
								selected = i
								break
							}
						}
						m.AllRepos[selected].selected = !m.AllRepos[selected].selected
						cmd = m.RepoList.SetItem(selected, m.AllRepos[selected])
						batch = append(batch, cmd)
					}
				}
			}
		}
		m.SelectedRepos = m.AllRepos.GetSelected()
		m.RepoList, cmd = m.RepoList.Update(msg)
		batch = append(batch, cmd)
		return m, tea.Batch(batch...)
	}

	return m, nil
}

func (m Settings) Init() tea.Cmd {
	return nil
}

func (m Settings) View() string {
	return lipgloss.JoinHorizontal(lipgloss.Top, m.AuthorList.View(), m.RepoList.View())
	// return fmt.Sprintf("This is the settings view %s", "fmt")
}

func NewSettings() (Settings, error) {
	var m Settings
	var err error
	m.cursor = authors
	allRepos, err := commits.GetMRRepos()
	if err != nil {
		return m, err
	}
	allAuthors, err := commits.RepoSet(allRepos).GetRepoAuthors()
	if err != nil {
		return m, err
	}

	m.AllRepos = []selectable{}
	for _, v := range allRepos {
		m.AllRepos = append(m.AllRepos, selectable{text: v, selected: true})
	}

	m.AllAuthors = []selectable{}
	for _, v := range allAuthors {
		m.AllAuthors = append(m.AllAuthors, selectable{text: v, selected: false})
	}
	m.SelectedRepos = m.AllRepos.GetSelected()
	email, _ := commits.GetAuthorEmail()
	if email != "" {
		m.SelectedAuthors = append(m.SelectedAuthors, email)
	}
	name, _ := commits.GetAuthorName()
	if name != "" {
		m.SelectedAuthors = append(m.SelectedAuthors, name)
	}
	for _, v := range m.SelectedAuthors {
	inner:
		for i, s := range m.AllAuthors {
			if s.text == v {
				m.AllAuthors[i].selected = true
				break inner
			}
		}
	}
	repoItems := []list.Item{}
	for _, v := range m.AllRepos {
		repoItems = append(repoItems, v)
	}
	m.RepoList = list.New(repoItems, selectableDelegate{}, 0, 4)

	authItems := []list.Item{}
	for _, v := range m.AllAuthors {
		authItems = append(authItems, v)
	}
	m.AuthorList = list.New(authItems, selectableDelegate{}, 0, 4)
	return m, nil
}