106 lines
2.4 KiB
Go
106 lines
2.4 KiB
Go
package bots
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/iyear/E5SubBot/task"
|
|
"github.com/robfig/cron/v3"
|
|
"github.com/spf13/viper"
|
|
"go.uber.org/zap"
|
|
"golang.org/x/net/proxy"
|
|
tb "gopkg.in/tucnak/telebot.v2"
|
|
"net/http"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
BotToken string
|
|
Socks5 string
|
|
bot *tb.Bot
|
|
)
|
|
|
|
const (
|
|
logo = `
|
|
______ _____ _____ _ ____ _
|
|
| ____| ____/ ____| | | | _ \ | |
|
|
| |__ | |__| (___ _ _| |__ | |_) | ___ | |_
|
|
| __| |___ \\___ \| | | | '_ \| _ < / _ \| __|
|
|
| |____ ___) |___) | |_| | |_) | |_) | (_) | |_
|
|
|______|____/_____/ \__,_|_.__/|____/ \___/ \__|
|
|
`
|
|
)
|
|
|
|
func BotStart() {
|
|
MakeHandle()
|
|
TaskLaunch()
|
|
fmt.Println("Bot Start")
|
|
fmt.Println("------------")
|
|
bot.Start()
|
|
}
|
|
func MakeHandle() {
|
|
fmt.Println("Make Handlers……")
|
|
//所有用户
|
|
bot.Handle("/start", bStart)
|
|
bot.Handle("/my", bMy)
|
|
bot.Handle("/bind", bBind1)
|
|
bot.Handle("/unbind", bUnBind)
|
|
bot.Handle("/export", bExport)
|
|
bot.Handle("/help", bHelp)
|
|
bot.Handle(tb.OnText, bOnText)
|
|
//管理员
|
|
bot.Handle("/task", bTask)
|
|
bot.Handle("/log", bLog)
|
|
}
|
|
func TaskLaunch() {
|
|
c := cron.New()
|
|
c.AddFunc(viper.GetString("cron"), task.SignTask)
|
|
fmt.Println("Cron Task Start……")
|
|
c.Start()
|
|
}
|
|
func init() {
|
|
fmt.Println(logo)
|
|
|
|
//read config
|
|
fmt.Println("Read Config……")
|
|
viper.SetConfigName("config")
|
|
viper.AddConfigPath(".")
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
zap.S().Errorw("failed to read config", "error", err)
|
|
}
|
|
BotToken = viper.GetString("bot_token")
|
|
Socks5 = viper.GetString("socks5")
|
|
Poller := &tb.LongPoller{Timeout: 15 * time.Second}
|
|
spamPoller := tb.NewMiddlewarePoller(Poller, func(upd *tb.Update) bool {
|
|
if upd.Message == nil {
|
|
return true
|
|
}
|
|
if !upd.Message.Private() {
|
|
return false
|
|
}
|
|
return true
|
|
})
|
|
botSetting := tb.Settings{
|
|
Token: BotToken,
|
|
Poller: spamPoller,
|
|
}
|
|
//set socks5
|
|
if Socks5 != "" {
|
|
fmt.Println("Proxy:" + Socks5)
|
|
dialer, err := proxy.SOCKS5("tcp", Socks5, nil, proxy.Direct)
|
|
if err != nil {
|
|
zap.S().Errorw("failed to make dialer", "error", err, "socks5", Socks5)
|
|
}
|
|
httpTransport := &http.Transport{}
|
|
httpClient := &http.Client{Transport: httpTransport}
|
|
httpTransport.Dial = dialer.Dial
|
|
botSetting.Client = httpClient
|
|
}
|
|
//create bot
|
|
bot, err = tb.NewBot(botSetting)
|
|
if err != nil {
|
|
zap.S().Errorw("failed to create bot", "error", err)
|
|
}
|
|
fmt.Println("Bot: " + strconv.Itoa(bot.Me.ID) + " " + bot.Me.Username)
|
|
}
|