chirpnest/bot.go

package main

import (
	"encoding/json"
	"fmt"
	tb "gopkg.in/tucnak/telebot.v2"
	"log"
	"regexp"
	"time"
)

func BotHandlers(b *tb.Bot) {
	b.Handle("/hello", func(m *tb.Message) {
		s, err := botHello(m)
		logOnError(err, "/hello")
		if err == nil {
			b.Send(m.Sender, s)
		}
	})

	b.Handle("/msg_rescan", func(m *tb.Message) {
		s, err := botMsgRescan(m)
		logOnError(err, "/msg_rescan")
		if err == nil {
			b.Send(m.Sender, s)
		}
	})
	b.Handle("/msg_rescan_all", botMsgRescanAll)

	b.Handle(tb.OnPhoto, botPhoto)
	b.Handle(tb.OnChannelPost, botChannelPost)
	b.Handle(tb.OnQuery, botQuery)
	b.Handle(tb.OnText, botText)

	b.Start()
}

func botPhoto(m *tb.Message) {
	fmt.Println("botPhoto :", m.Text)
	//	photos only
}

func botHello(m *tb.Message) (string, error) {
	fmt.Println("botHello :", m.Text)
	if !m.Private() {
		fmt.Println("botHello : !m.Private()")
		return ``, nil
	}
	//	fmt.Println("Hello payload :", m.Payload) // <PAYLOAD>
	PrintText(m)
	return `hello world`, nil
}

func botChannelPost(m *tb.Message) {
	fmt.Println("botChannelPost :", m.Text)
	PrintText(m)
	//	channel posts only
}

func botQuery(q *tb.Query) {
	fmt.Println("botQuery")
	//	incoming inline queries
}

func botText(m *tb.Message) {
	fmt.Println("botText :", m.Text)
	PrintText(m)
	// all the text messages that weren't
	// captured by existing handlers
}

func botMsgRescan(m *tb.Message) (string, error) {
	fmt.Println("botRescanMsg :", m.Text)
	if !m.Private() {
		fmt.Println("botRescanMsg : !m.Private()")
		return ``, nil
	}
	r := regexp.MustCompile("^[0-9]+$")
	if r.MatchString(m.Payload) {
		p := JobPayloadRescanMsg{
			Query:    fmt.Sprintf("SELECT o.id FROM obj o WHERE o.id = %s AND o.obj_type_id = %d AND o.obj_sub_type_id = %d;", m.Payload, objTypeMessage, objSubTypeMessageUnknown),
			MsgID64:  int64(m.ID),
			ChatID64: m.Chat.ID,
		}
		b, _ := json.Marshal(p)
		log.Printf("botMsgRescan : json : %s\n", string(b))
		err := createJob(objSubTypeJobRescanMsg, objJobPriorityRescanMsg, int64(m.Sender.ID), time.Now(), b)
		logOnError(err, "botMsgRescan : createJob(objSubTypeJobRescanMsg)")
		if err != nil {
			return "Error scheduling the rescan for msg #" + m.Payload, nil
		} else {
			return "Rescaning msg #" + m.Payload, nil
		}

	}
	r = regexp.MustCompile("^all$")
	if r.MatchString(m.Payload) {
		p := JobPayloadRescanMsg{
			Query:    fmt.Sprintf("SELECT o.id FROM obj o WHERE o.obj_type_id = %d AND o.obj_sub_type_id = %d;", objTypeMessage, objSubTypeMessageUnknown),
			MsgID64:  int64(m.ID),
			ChatID64: m.Chat.ID,
		}
		b, _ := json.Marshal(p)
		err := createJob(objSubTypeJobRescanMsg, objJobPriorityRescanAllMsg, int64(m.Sender.ID), time.Now(), b)
		logOnError(err, "botMsgRescan : createJob(objSubTypeJobRescanMsg)")
		if err != nil {
			return "Error scheduling the rescan for all msg", nil
		} else {
			return "Rescaning all msg scheduled", nil
		}
	}
	return "/msg_rescan msg_id or /msg_rescan all", nil
}

func botMsgRescanAll(m *tb.Message) {
	if !m.Private() {
		return
	}
	p := JobPayloadRescanMsg{
		Query:    fmt.Sprintf("SELECT o.id FROM obj o WHERE o.obj_type_id = %d AND o.obj_sub_type_id = %d;", objTypeMessage, objSubTypeMessageUnknown),
		MsgID64:  int64(m.ID),
		ChatID64: m.Chat.ID,
	}
	b, _ := json.Marshal(p)
	err := createJob(objSubTypeJobRescanMsg, objJobPriorityRescanAllMsg, int64(m.Sender.ID), time.Now(), b)
	logOnError(err, "botMsgRescan : createJob(objSubTypeJobRescanMsg)")

	if err != nil {
		c := TGCommand{
			Type:         commandReplyMsg,
			Text:         "Error scheduling the rescan for all msg.",
			FromMsgID64:  int64(m.ID),
			FromChatID64: m.Chat.ID,
		}
		TGCmdQueue <- c
	} else {
		c := TGCommand{
			Type:         commandReplyMsg,
			Text:         "Rescaning all msg scheduled.",
			FromMsgID64:  int64(m.ID),
			FromChatID64: m.Chat.ID,
		}
		TGCmdQueue <- c
	}
	return
}