helper.go

package main

import (
	"fmt"
	"os"

	"charm.land/bubbles/v2/list"
	tea "charm.land/bubbletea/v2"
	"github.com/yashranjan1/commit/internal/components"
	"github.com/yashranjan1/commit/internal/messages"
	"github.com/yashranjan1/commit/internal/requests"
)

func InitFeedbackArea() *components.InputForm {
	title := "What would you like to change about the commit? (press shift+enter to make a new line)"
	onAccept := func(msg string) tea.Cmd {
		return func() tea.Msg {
			return messages.Correction{
				Prompt: msg,
			}
		}
	}
	return components.NewInputForm(title, onAccept, 3)
}

func InitAPIKey() *components.InputForm {
	title := "Enter your API key"
	onAccept := func(msg string) tea.Cmd {
		return func() tea.Msg {
			return messages.APIEntered{
				APIKey: msg,
			}
		}
	}
	return components.NewInputForm(title, onAccept, 3)
}

func InitChangeArea() *components.InputForm {
	title := "Make your change"
	onAccept := func(msg string) tea.Cmd {
		return func() tea.Msg {
			return messages.ChangedCommit{
				Commit: msg,
			}
		}
	}
	return components.NewInputForm(title, onAccept, 7)
}

func InitOllamaOpts() *components.SelectForm {
	items := []list.Item{
		components.Option{
			Name:  "Start it",
			Value: 0,
		},
		components.Option{
			Name:  "No I wanna write my own commit (exit)",
			Value: 1,
		},
	}

	onSelect := func(item components.Option) tea.Cmd {
		return func() tea.Msg {
			return messages.StartOllama{
				Msg: item.Value,
			}
		}
	}

	config := components.SelectFormConfig{
		Title:    "Ollama server not running, would you like me to start it?",
		Items:    items,
		Width:    100,
		OnSelect: onSelect,
	}

	l := components.InitSelectForm(config)

	return l
}

func InitCommitConfirmOpts() *components.SelectForm {
	items := []list.Item{
		components.Option{
			Name:  "Yes, commit it",
			Value: 0,
		},
		components.Option{
			Name:  "No, I want to request a change",
			Value: 1,
		},
		components.Option{
			Name:  "No, I want to make some changes",
			Value: 2,
		},
	}

	onSelect := func(item components.Option) tea.Cmd {
		return func() tea.Msg {
			return messages.CommitConfirmation{
				Msg: item.Value,
			}
		}
	}

	config := components.SelectFormConfig{
		Title:    "Would you like to commit this? If yes, the following command will be run: git commit -m <COMMIT_MESSAGE>",
		Items:    items,
		Width:    100,
		OnSelect: onSelect,
	}

	l := components.InitSelectForm(config)

	return l
}

func InitPickProvider() *components.SelectForm {
	items := []list.Item{}

	providers := []string{
		"Ollama",
		"Gemini",
	}

	for i, provider := range providers {
		items = append(items,
			components.Option{
				Name:  provider,
				Value: i,
			},
		)
	}

	onSelect := func(item components.Option) tea.Cmd {
		return func() tea.Msg {
			return messages.ProviderPicked{
				Name: item.Name,
			}
		}
	}

	path, _ := os.UserConfigDir()

	title := fmt.Sprintf("You haven't picked a provider to use, which one would you like to use? (you can change your selection in the \"%s/commit/config.json\" file)", path)

	config := components.SelectFormConfig{
		Title:    title,
		Items:    items,
		Width:    100,
		OnSelect: onSelect,
	}

	l := components.InitSelectForm(config)

	return l
}

func InitModelPickOpts(models requests.GetModelsResponse) *components.SelectForm {
	items := []list.Item{}

	for i, model := range models.Models {
		items = append(items,
			components.Option{
				Name:  model.Name,
				Value: i,
			},
		)
	}

	onSelect := func(item components.Option) tea.Cmd {
		return func() tea.Msg {
			return messages.ModelPicked{
				Name: item.Name,
			}
		}
	}

	path, _ := os.UserConfigDir()

	title := fmt.Sprintf("You haven't picked a model to use, which one would you like to use? (you can change your selection in the \"%s/commit/config.json\" file)", path)

	config := components.SelectFormConfig{
		Title:    title,
		Items:    items,
		Width:    100,
		OnSelect: onSelect,
	}

	l := components.InitSelectForm(config)

	return l
}