package activities

import (
	"log"
	"time"

	"github.com/hajimehoshi/ebiten/v2"
	"github.com/hajimehoshi/ebiten/v2/inpututil"

	"muhq.space/muhqs-game/go/game"
	"muhq.space/muhqs-game/go/ui"
)

const (
	DRAFT_BUTTON_WIDTH  = 150
	DRAFT_BUTTON_HEIGHT = 40
)

type Draft struct {
	ui.Collection
	player *game.Player
	ctrl   game.PlayerControl
	draft  *game.Draft

	setInput  *ui.TextInput
	descInput *ui.TextInput
	deckList  *ui.PocList
	timer     *ui.TimerBar
}

func NewLocalDraft(width, height int, playerName string) *Draft {
	d := &Draft{
		Collection: ui.Collection{Width: width, Height: height},
	}

	d.player = game.NewDraftPlayer(playerName)
	d.ctrl = game.NewChanPlayerControl(d.player)
	d.player.Ctrl = d.ctrl

	d.AddWidget(ui.NewFixedTextBox(
		(d.Width-DRAFT_BUTTON_WIDTH)/2-DRAFT_BUTTON_WIDTH,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-3*DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"sets:",
	).Centering(true))

	d.setInput = ui.NewTextInput(
		(d.Width-DRAFT_BUTTON_WIDTH)/2,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-3*DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"base,magic,equipments",
	)
	d.AddWidget(d.setInput)

	d.AddWidget(ui.NewFixedTextBox(
		(d.Width-DRAFT_BUTTON_WIDTH)/2-DRAFT_BUTTON_WIDTH,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-2*DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"draft:",
	).Centering(true))

	d.descInput = ui.NewTextInput(
		(d.Width-DRAFT_BUTTON_WIDTH)/2,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-2*DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"3x[2;8]",
	)
	d.AddWidget(d.descInput)

	d.AddWidget(ui.NewRoundSimpleButton(
		(d.Width-DRAFT_BUTTON_WIDTH)/2+DRAFT_BUTTON_WIDTH+20,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-2*DRAFT_BUTTON_HEIGHT,
		30,
		"?",
		func(*ui.SimpleButton) {
			ui.OpenUrl("https://muhq.space/muhqs-game/build/en/rules.html#draft-notation")
		}))

	d.AddWidget(ui.NewFixedTextBox(
		(d.Width-DRAFT_BUTTON_WIDTH)/2-DRAFT_BUTTON_WIDTH,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"AI:",
	).Centering(true))

	aiNInput := ui.NewNumberChoice(
		(d.Width-DRAFT_BUTTON_WIDTH)/2,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2-DRAFT_BUTTON_HEIGHT,
		1,
		nil,
	)
	d.AddWidget(aiNInput)

	d.AddWidget(ui.NewSimpleButton(
		(d.Width-DRAFT_BUTTON_WIDTH)/2,
		(d.Height-DRAFT_BUTTON_HEIGHT)/2,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"Start",
		func(*ui.SimpleButton) {
			d.startDraft(d.descInput.TextOrLabel(), d.setInput.TextOrLabel(), aiNInput.GetChoosen())
		}))

	return d
}

func StartNewRemoteDraft(width, height int, ctrl game.PlayerControl, firstPack *game.PlayerNotification) *Draft {
	d := &Draft{
		Collection: ui.Collection{Width: width, Height: height},
		player:     ctrl.Player(),
		ctrl:       ctrl,
	}

	d.presentPack(firstPack)
	return d
}

func (d *Draft) Update() error {
	if n := d.ctrl.RecvNotification(); n != nil {
		log.Println("Received draft pick: %v\n", n)
		d.presentPack(n)
	}

	if err := ui.Update(); err != nil {
		return err
	}

	if err := d.Collection.Update(); err != nil {
		return err
	}

	if inpututil.IsKeyJustPressed(ebiten.KeyEscape) {
		PopActivity()
	}

	return nil
}

func (d *Draft) presentPack(n *game.PlayerNotification) {
	pack := n.Context.(game.PileOfCards)
	grid := ui.NewCardGrid(0, 0, d.Width-SEALED_DECK_STRIP_WIDTH, d.Height, 0.5, pack).Columns(3)
	selectCard := func(x, y int) {
		_card := grid.FindObjectAt(x, y)
		if _card == nil {
			return
		}

		card := _card.(*game.Card)
		d.ctrl.SendAction(game.NewDraftPick(d.player, pack, card))
		d.RemoveWidget(grid)
		if d.timer != nil {
			d.timer = nil
			d.RemoveWidget(d.timer)
		}
	}
	grid.RegisterHandler("click", selectCard)
	d.AddWidget(grid)

	if d.draft == nil {
		t := time.Duration(pack.Size()*5) * time.Second
		timer := ui.NewTimerBar(0, 10, d.Width, 5, t)
		d.AddWidget(timer)
	}
}

func (d *Draft) Layout(width, height int) (int, int) {
	return d.Width, d.Height
}

func (d *Draft) startDraft(desc string, setList string, aiN int) {

	log.Printf("starting %s draft from '%s'\n", desc, setList)

	var sets []game.SetIdentifier
	var err error
	if sets, err = game.SetListToSets(setList); err != nil {
		d.setInput.Bg(ui.WarningBg)
		d.setInput.SetInput("")
		return
	}

	d.draft, err = game.NewDraft([]*game.Player{d.player}, desc, sets)
	if err != nil {
		d.descInput.Bg(ui.WarningBg)
		d.descInput.SetInput("")
		return
	}

	log.Println(aiN)
	for _ = range aiN {
		log.Println("add ai")
		d.draft.AddRandomAi()
	}

	if !d.draft.Valid() {
		d.descInput.Bg(ui.WarningBg)
		d.descInput.SetInput("")
		return
	}
	d.Clear()

	d.deckList = ui.NewPocList(d.Width-SEALED_DECK_STRIP_WIDTH/2, d.Height-DRAFT_BUTTON_HEIGHT-20, d.player.Deck)
	d.deckList.Bg(ui.Gray)
	{
		cardsInDeck := 0
		d.deckList.RegisterHandler("update", func() error {
			if n := d.player.Deck.Size(); n > cardsInDeck {
				d.deckList.ForceRedraw()
				cardsInDeck = n
			}
			return nil
		})
	}
	d.AddWidget(d.deckList)

	d.AddWidget(ui.NewSimpleButton(
		d.Width-SEALED_DECK_STRIP_WIDTH/2-DRAFT_BUTTON_WIDTH/2,
		d.Height-DRAFT_BUTTON_HEIGHT,
		DRAFT_BUTTON_WIDTH,
		DRAFT_BUTTON_HEIGHT,
		"Done",
		func(*ui.SimpleButton) {
			d.done()
		}))

	go d.draft.Run()
}

func (d *Draft) done() {
	PopActivity()
}

func (d *Draft) GetDeckList() string {
	return d.player.Deck.ToList()
}