From 43317bb6470f1f7f40051ef4f3e47c0e9bfc1e05 Mon Sep 17 00:00:00 2001
From: c9s <yoan@maicoin.com>
Date: Wed, 12 Jan 2022 21:45:12 +0800
Subject: [PATCH] add state and telegram example

---
 examples/telebot/main.go                | 99 +++++++++++++++++++++++++
 pkg/{bbgo => interact}/interact.go      | 18 ++++-
 pkg/{bbgo => interact}/interact_test.go | 34 ++++++++-
 3 files changed, 147 insertions(+), 4 deletions(-)
 create mode 100644 examples/telebot/main.go
 rename pkg/{bbgo => interact}/interact.go (83%)
 rename pkg/{bbgo => interact}/interact_test.go (65%)

diff --git a/examples/telebot/main.go b/examples/telebot/main.go
new file mode 100644
index 000000000..b24774c5c
--- /dev/null
+++ b/examples/telebot/main.go
@@ -0,0 +1,99 @@
+package main
+
+import (
+	"fmt"
+	"log"
+	"os"
+	"time"
+
+	"github.com/google/uuid"
+	tb "gopkg.in/tucnak/telebot.v2"
+)
+
+func main() {
+	b, err := tb.NewBot(tb.Settings{
+		// You can also set custom API URL.
+		// If field is empty it equals to "https://api.telegram.org".
+		// URL: "http://195.129.111.17:8012",
+
+		Token:  os.Getenv("TELEGRAM_BOT_TOKEN"),
+		Poller: &tb.LongPoller{Timeout: 10 * time.Second},
+		// Synchronous: false,
+		Verbose: true,
+		// ParseMode:   "",
+		// Reporter:    nil,
+		// Client:      nil,
+		// Offline:     false,
+	})
+
+	if err != nil {
+		log.Fatal(err)
+		return
+	}
+
+	b.Handle("/hello", func(m *tb.Message) {
+		fmt.Printf("message: %#v\n", m)
+		// b.Send(m.Sender, "Hello World!")
+	})
+
+	b.Handle(tb.OnText, func(m *tb.Message) {
+		fmt.Printf("text: %#v\n", m)
+		// all the text messages that weren't
+		// captured by existing handlers
+	})
+
+	b.Handle(tb.OnQuery, func(q *tb.Query) {
+		fmt.Printf("query: %#v\n", q)
+
+		// r := &tb.ReplyMarkup{}
+		// r.URL("test", "https://media.tenor.com/images/f176705ae1bb3c457e19d8cd71718ac0/tenor.gif")
+		urls := []string{
+			// "https://media.tenor.com/images/aae0cdf3c5a291cd7b96432180f6eee3/tenor.png",
+			// "https://media.tenor.com/images/905c1a9b1f56ae3c458b1ef58fd46357/tenor.png",
+
+			"https://media.tenor.com/images/2e69768f9537957ed3015a80ebc3f0f1/tenor.gif",
+			"https://media.tenor.com/images/6fcd72b29127a55e5c35db86d06d665c/tenor.gif",
+			"https://media.tenor.com/images/05dbf5bf3a3b88275bb045691541dc53/tenor.gif",
+			"https://media.tenor.com/images/0e1a52cfe5616c1509090d6ec2312db0/tenor.gif",
+			"https://media.tenor.com/images/1ca04a449b26e1f7d45682a79d2c8697/tenor.gif",
+			"https://media.tenor.com/images/a2844b186fb71c376226b56c4ea7730a/tenor.gif",
+			"https://media.tenor.com/images/ec636a1ebce1a3fc1c795b851c125b31/tenor.gif",
+			"https://media.tenor.com/images/ae103819cb05a0cf7497900b77b87d80/tenor.gif",
+		}
+
+		results := make(tb.Results, len(urls)) // []tb.Result
+		for i, url := range urls {
+			// result := &tb.PhotoResult{
+			result := &tb.GifResult{
+				ResultBase: tb.ResultBase{
+					// Type:        "photo",
+					// Content:     nil,
+					// ReplyMarkup: nil,
+					ID:          uuid.New().String(),
+				},
+
+				URL: url,
+
+				// required for photos
+				ThumbURL: url,
+			}
+
+			results[i] = result
+			// needed to set a unique string ID for each result
+			// results[i].SetResultID(strconv.Itoa(i))
+		}
+
+		err := b.Answer(q, &tb.QueryResponse{
+			QueryID: q.ID,
+			Results: results,
+			// CacheTime:  60, // a minute
+			IsPersonal: true,
+		})
+
+		if err != nil {
+			log.Println(err)
+		}
+	})
+
+	b.Start()
+}
diff --git a/pkg/bbgo/interact.go b/pkg/interact/interact.go
similarity index 83%
rename from pkg/bbgo/interact.go
rename to pkg/interact/interact.go
index a7a479bbe..e7833108c 100644
--- a/pkg/bbgo/interact.go
+++ b/pkg/interact/interact.go
@@ -1,4 +1,4 @@
-package bbgo
+package interact
 
 import (
 	"reflect"
@@ -11,11 +11,23 @@ import (
 
 // Interact implements the interaction between bot and message software.
 type Interact struct {
-	Commands map[string]func()
+	Commands map[string]interface{}
+}
+
+func New()  *Interact {
+	return &Interact{}
+}
+
+func (i *Interact) Command(command string, f interface{}) {
+	i.Commands[command] = f
 }
 
 func (i *Interact) HandleTelegramMessage(msg *telebot.Message) {
-
+	// For registered commands, will contain the string payload
+	// msg.Payload
+	// msg.Text
+	args := parseCommand(msg.Text)
+	_ = args
 }
 
 func parseCommand(src string) (args []string) {
diff --git a/pkg/bbgo/interact_test.go b/pkg/interact/interact_test.go
similarity index 65%
rename from pkg/bbgo/interact_test.go
rename to pkg/interact/interact_test.go
index 54c74a888..e4c5b0055 100644
--- a/pkg/bbgo/interact_test.go
+++ b/pkg/interact/interact_test.go
@@ -1,4 +1,4 @@
-package bbgo
+package interact
 
 import (
 	"errors"
@@ -42,3 +42,35 @@ func Test_parseCommand(t *testing.T) {
 	assert.Equal(t, "3.1415926", args[2])
 	assert.Equal(t, "market", args[3])
 }
+
+type TestState int
+
+const (
+	TestStateStart = 0
+
+	TestStateShowNetAssetValue = 101
+
+	TestStateShowPositionChoosingSymbol = 201
+
+	TestStateClosePositionChoosingSymbol = 301
+	TestStateClosePositionEnteringPercentage = 302
+	TestStateClosePositionConfirming = 303
+)
+
+type TestInteraction struct {
+	State TestState
+}
+
+func (m *TestInteraction) HandleMessage() {
+
+}
+
+func (m *TestInteraction) Commands(interact *Interact) {
+	interact.Command("closePosition", func() {
+
+	})
+}
+
+func TestCustomInteraction(t *testing.T) {
+
+}