package main

import (
	"log"
	"os"
	"strings"

	"github.com/mojosa-software/got/src/tx"
)

var startScreenButton = tx.NewButton().
	WithText("🏠 To the start screen").
	ScreenChange("start")

var beh = tx.NewBehaviour().
	// The function will be called every time
	// the bot is started.
	OnStartFunc(func(c *tx.Context) {
		c.V["counter"] = new(int)
		c.ChangeScreen("start")
	}).WithKeyboards(
	// Increment/decrement keyboard.
	tx.NewKeyboard("inc/dec").Row(
		tx.NewButton().WithText("+").ActionFunc(func(c *tx.Context) {
			counter := c.V["counter"].(*int)
			*counter++
			c.Sendf("%d", *counter)
		}),
		tx.NewButton().WithText("-").ActionFunc(func(c *tx.Context) {
			counter := c.V["counter"].(*int)
			*counter--
			c.Sendf("%d", *counter)
		}),
	).Row(
		startScreenButton,
	),
	// The navigational keyboard.
	tx.NewKeyboard("nav").Row(
		tx.NewButton().WithText("Inc/Dec").ScreenChange("inc/dec"),
	).Row(
		tx.NewButton().WithText("Upper case").ScreenChange("upper-case"),
		tx.NewButton().WithText("Lower case").ScreenChange("lower-case"),
	),
	// The keyboard to return to the start screen.
	tx.NewKeyboard("nav-start").Row(
		startScreenButton,
	),
).WithScreens(
	tx.NewScreen("start").
		WithText("The bot started!").
		Keyboard("nav"),
	tx.NewScreen("inc/dec").
		WithText(
			"The screen shows how"+
				"user separated data works"+
				"by saving the counter for each of them",
		).
		Keyboard("inc/dec").
		// The function will be called when reaching the screen.
		ActionFunc(func(c *tx.Context) {
			counter := c.V["counter"].(*int)
			c.Sendf("Current counter value equals %d", *counter)
		}),
	tx.NewScreen("upper-case").
		WithText("Type text and the bot will send you the upper case version to you").
		Keyboard("nav-start").
		ActionFunc(func(c *tx.Context) {
			for {
				s, err := c.ReadTextMessage()
				if err == tx.NotAvailableErr {
					break
				}
				c.Sendf("%s", strings.ToUpper(s))
			}
		}),
)

func main() {
	token := os.Getenv("BOT_TOKEN")

	bot, err := tx.NewBot(token, beh, nil)
	if err != nil {
		log.Panic(err)
	}

	bot.Debug = true

	log.Printf("Authorized on account %s", bot.Self.UserName)
	bot.Run()
}