package controllers

import (
	"github.com/jesseduffield/lazygit/pkg/gocui"
	"github.com/jesseduffield/lazygit/pkg/gui/types"
)

type SearchPromptController struct {
	baseController
	c *ControllerCommon
}

var _ types.IController = &SearchPromptController{}

func NewSearchPromptController(
	c *ControllerCommon,
) *SearchPromptController {
	return &SearchPromptController{
		baseController: baseController{},
		c:              c,
	}
}

func (self *SearchPromptController) GetKeybindings(opts types.KeybindingsOpts) []*types.Binding {
	return []*types.Binding{
		{
			Keys:    []gocui.Key{gocui.NewKeyName(gocui.KeyEnter)},
			Handler: self.confirm,
		},
		{
			Keys:    opts.GetKeys(opts.Config.Universal.Return),
			Handler: self.cancel,
		},
		{
			Keys:    opts.GetKeys(opts.Config.Universal.PrevItem),
			Handler: self.prevHistory,
		},
		{
			Keys:    opts.GetKeys(opts.Config.Universal.NextItem),
			Handler: self.nextHistory,
		},
	}
}

func (self *SearchPromptController) Context() types.Context {
	return self.context()
}

func (self *SearchPromptController) context() types.Context {
	return self.c.Contexts().Search
}

func (self *SearchPromptController) confirm() error {
	self.c.Helpers().Search.Confirm()
	return nil
}

func (self *SearchPromptController) cancel() error {
	self.c.Helpers().Search.CancelPrompt()
	return nil
}

func (self *SearchPromptController) prevHistory() error {
	self.c.Helpers().Search.ScrollHistory(1)
	return nil
}

func (self *SearchPromptController) nextHistory() error {
	self.c.Helpers().Search.ScrollHistory(-1)
	return nil
}