1 files changed, 212 insertions, 0 deletions
diff --git a/go/activities/draft.go b/go/activities/draft.go
new file mode 100644
index 00000000..10ce9671
--- /dev/null
+++ b/go/activities/draft.go
@@ -0,0 +1,212 @@
+package activities
+
+import (
+	"log"
+
+	"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.ChanPlayerControl
+	draft  *game.Draft
+
+	setInput  *ui.TextInput
+	descInput *ui.TextInput
+	deckList  *ui.PocList
+}
+
+func NewDraft(width, height int, playerName string) *Draft {
+	d := &Draft{
+		Collection: ui.Collection{Width: width, Height: height},
+	}
+
+	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,
+		DRAFT_BUTTON_WIDTH,
+		"?",
+		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())
+		}))
+
+	d.player = game.NewDraftPlayer(playerName)
+	d.ctrl = game.NewChanPlayerControl(d.player)
+	d.player.Ctrl = d.ctrl
+
+	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.TouchManager.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.RemoveWidget(grid)
+		d.ctrl.SendAction(game.NewDraftPick(d.player, pack, card))
+	}
+	grid.RegisterHandler("click", selectCard)
+	d.AddWidget(grid)
+}
+
+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 = game.NewDraft([]*game.Player{d.player}, desc, sets)
+
+	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()
+}