tg/cmd/test/main.go

package main

import (
	"log"
	"os"
	"strings"
	"fmt"

	"github.com/mojosa-software/got/tg"
	"math/rand"
	"strconv"
)

type BotData struct {
	Name string
}

type SessionData struct {
	Counter int
}

type MutateMessageWidget struct {
	Mutate func(string) string
}

func NewMutateMessageWidget(fn func(string) string) *MutateMessageWidget {
	ret := &MutateMessageWidget{}
	ret.Mutate = fn
	return ret
}

func (w *MutateMessageWidget) Serve(c *tg.Context) {
	args, ok := c.Arg.(tg.ArgSlice)
	if ok {
		for _, arg := range args {
			c.Sendf("%v", arg)
		}
	}
	for u := range c.Input() {
		text := u.Message.Text
		c.Sendf("%s", w.Mutate(text))
	}
}

func (w *MutateMessageWidget) Filter(u *tg.Update, _ tg.MessageMap) bool {
	if u.Message == nil {
		return true
	}
	return false
}

func ExtractSessionData(c *tg.Context) *SessionData {
	return c.Session.Data.(*SessionData)
}

var (
	homeButton = tg.NewButton("Home").Go("/")
	backButton = tg.NewButton("Back").Go("..")
	backKeyboard = tg.NewKeyboard().Row(
		backButton,
	)

	incDecKeyboard = tg.NewKeyboard().Row(
		tg.NewButton("+").ActionFunc(func(c *tg.Context) {
			d := ExtractSessionData(c)
			d.Counter++
			c.Sendf("%d", d.Counter)
		}),
		tg.NewButton("-").ActionFunc(func(c *tg.Context) {
			d := ExtractSessionData(c)
			d.Counter--
			c.Sendf("%d", d.Counter)
		}),
	).Row(
		backButton,
	)

	navKeyboard = 

	sendLocationKeyboard = tg.NewKeyboard().Row(
		tg.NewButton("Send location").
			WithSendLocation(true).
			ActionFunc(func(c *tg.Context) {
				l := c.Message.Location
				c.Sendf(
					"Longitude: %f\n"+
					"Latitude: %f\n"+
					"Heading: %d"+
					"",
					l.Longitude,
					l.Latitude,
					l.Heading,
				)
			}),
	).Row(
		backButton,
	).Reply()
)

var beh = tg.NewBehaviour().
WithInitFunc(func(c *tg.Context) {
	// The session initialization.
	c.Session.Data = &SessionData{}
}).WithRootNode(tg.NewRootNode(
	// The "/" widget.
	tg.WidgetFunc(func(c *tg.Context) tg.UIs {
		return tg.UIs{

			tg.NewKeyboard().Row(
					tg.NewButton("GoT Github page").
						WithUrl("https://github.com/mojosa-software/got"),
				).Inline().Widget(
					fmt.Sprintf(
						"Hello, %s"
						"The testing bot started!\n",
						"You can see the basics of usage in the ",
						"cmd/test/main.go file!",
						c.SentFrom().UserName,
					),
				),

			tg.NewKeyboard().Row(
					tg.NewButton("Inc/Dec").Go("/inc-dec"),
				).Row(
					tg.NewButton("Mutate messages").Go("/mutate-messages"),
				).Row(
					tg.NewButton("Send location").Go("/send-location"),
				).Reply().Widget(
					"Choose the point of your interest",
				),

		}
	)

	tg.NewNode(
		"mutate-messages", tg.NewPage().WithReply(
			tg.NewKeyboard().Row(
				tg.NewButton("Upper case").Go("upper-case"),
				tg.NewButton("Lower case").Go("lower-case"),
			).Row(
				backButton,
			).Reply().Widget(
				"Choose the function to mutate string",
			),
		),
		tg.NewNode(
			"upper-case", tg.NewPage().WithReply(
				backKeyboard.Reply().
					Widget(
						"Type a string and the bot will convert it to upper case",
					),
			).WithSub(
				NewMutateMessageWidget(strings.ToUpper),
			),
		),
		tg.NewNode(
			"lower-case", tg.NewPage().WithReply(
				backKeyboard.Reply().
					Widget(
						"Type a string and the bot will convert it to lower case",
					),
			).WithSub(
				NewMutateMessageWidget(strings.ToLower),
			),
		),
	),

	tg.NewNode(
		"inc-dec", tg.NewPage().WithReply(
				incDecKeyboard.Reply().Widget("Press the buttons to increment and decrement"),
			).ActionFunc(func(c *tg.Context) {
				// The function will be calleb before serving page.
				d := ExtractSessionData(c)
				c.Sendf("Current counter value = %d", d.Counter)
			}),
	),

	tg.NewNode(
		"send-location", tg.NewPage().WithReply(
			sendLocationKeyboard.Widget("Press the button to send your location!"),
		).WithInline(
			tg.NewKeyboard().Row(
				tg.NewButton(
					"Check",
				).WithData(
					"check",
				).ActionFunc(func(c *tg.Context) {
					d := ExtractSessionData(c)
					c.Sendf("Counter = %d", d.Counter)
				}),
			).Inline().Widget("Press the button to display your counter"),
		),
	),
)).WithCommands(
	tg.NewCommand("info").
		ActionFunc(func(c *tg.Context){
			c.SendfHTML(`<a href="https://res.cloudinary.com/demo/image/upload/v1312461204/sample.jpg">cock</a><strong>cock</strong> die`)
		}),
	tg.NewCommand("start").
		Desc(
			"start or restart the bot or move to the start screen",
		).Go("/"),
	tg.NewCommand("hello").
		Desc("sends the 'Hello, World!' message back").
		ActionFunc(func(c *tg.Context) {
			c.Sendf("Hello, World!")
		}),
	tg.NewCommand("read").
		Desc("reads a string and sends it back").
		WithWidget(
			tg.Func(func(c *tg.Context){
				str := c.ReadString("Type a string and I will send it back")
				c.Sendf2("You typed `%s`", str)
			}),
		),
	tg.NewCommand("image").
		Desc("sends a sample image").
		ActionFunc(func(c *tg.Context) {
			img := tg.NewFile("media/cat.jpg").Image().Caption("A cat!")
			c.Send(img)
		}),
	tg.NewCommand("botname").
		Desc("get the bot name").
		ActionFunc(func(c *tg.Context) {
			bd := c.Bot.Data.(*BotData)
			c.Sendf("My name is %q", bd.Name)
		}),
	tg.NewCommand("dynamic").
		Desc("check of the dynamic work").
		WidgetFunc(func(c *tg.Context){
			nRow, nBtn := rand.Int()%10, rand.Int()%5
			rows := []tg.ButtonRow{}
			for i:=0 ; i<nRow ; i++ {
				row := []*tg.Button{}
				for j:=0 ; j<nBtn ; j++ {
					row = append(row, tg.NewButton(
						strconv.Itoa(i) + " " + strconv.Itoa(j),
					))
				}
				rows = append(rows, row)
			}
			kbd := tg.NewKeyboard(rows...).ActionFunc(func(c *tg.Context){
				c.Sendf(u.)
			})Inline().Widget("sample text")
			c.Send(kbd)
			kbdChn := c.RunWidget(kbd)
			for u := range c.Input() {
				if kbd.Filter(u, nil) {
					continue
				}
				kbdChn.Send(u)
			}
		}),
)

var gBeh = tg.NewGroupBehaviour().
	InitFunc(func(c *tg.GC) {
	}).
	WithCommands(
		tg.NewGroupCommand("hello").ActionFunc(func(c *tg.GC) {
			c.Sendf("Hello, World!")
		}),
		tg.NewGroupCommand("mycounter").ActionFunc(func(c *tg.GC) {
			d := c.Session().Data.(*SessionData)
			c.Sendf("Your counter value is %d", d.Counter)
		}),
	)

func main() {
	log.Println(beh.Screens)
	token := os.Getenv("BOT_TOKEN")

	bot, err := tg.NewBot(token)
	if err != nil {
		log.Panic(err)
	}
	bot = bot.
		WithBehaviour(beh).
		Debug(true)

	bot.Data = &BotData{
		Name: "Jay",
	}

	log.Printf("Authorized on account %s", bot.Api.Self.UserName)
	err = bot.Run()
	if err != nil {
		panic(err)
	}
}