E5BotForSQLite/main.go

package main

import (
	"database/sql"
	"fmt"
	"github.com/robfig/cron/v3"
	"github.com/spf13/viper"
	"golang.org/x/net/proxy"
	tb "gopkg.in/tucnak/telebot.v2"
	"log"
	"net/http"
	"strconv"
	"strings"
	"time"
)

var (
	BotToken string
	Socks5   string
	bot      *tb.Bot
	db       *sql.DB
	logger   *log.Logger
)

const (
	dbDriverName = "mysql"
	logo         = `
  ______ _____ _____       _     ____        _   
 |  ____| ____/ ____|     | |   |  _ \      | |  
 | |__  | |__| (___  _   _| |__ | |_) | ___ | |_ 
 |  __| |___ \\___ \| | | | '_ \|  _ < / _ \| __|
 | |____ ___) |___) | |_| | |_) | |_) | (_) | |_ 
 |______|____/_____/ \__,_|_.__/|____/ \___/ \__|
`
)

func main() {
	BotStart()
}
func BotStart() {
	MakeHandle()
	TaskLaunch()
	logger.Println("Bot Start")
	fmt.Println("------------")
	bot.Start()
}
func MakeHandle() {
	logger.Println("Make Handle……")
	//所有用户
	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() {
	task := cron.New()
	//每三小时执行一次
	task.AddFunc(viper.GetString("cron"), SignTask)
	//log分为每天
	task.AddFunc(" 0 0 * * *", InitLogger)
	//  */1 * * * *    1 */3 * * *
	logger.Println("Cron Task Start……")
	task.Start()
}
func init() {
	fmt.Println(logo)
	InitLogger()
	//read config
	logger.Println("Read Config……")
	viper.SetConfigName("config")
	viper.AddConfigPath(".")
	err := viper.ReadInConfig()
	if err != nil {
		logger.Fatal(err)
	}
	host := viper.GetString("mysql.host")
	user := viper.GetString("mysql.user")
	port := viper.GetString("mysql.port")
	pwd := viper.GetString("mysql.password")
	database := viper.GetString("mysql.database")
	path := strings.Join([]string{user, ":", pwd, "@tcp(", host, ":", port, ")/", database, "?charset=utf8"}, "")
	//fmt.Println(path)
	db, err = sql.Open(dbDriverName, path)
	if err != nil {
		logger.Fatal(err)
	}
	logger.Println("Connect MySQL Success!")
	if ok, err := CreateTB(db); !ok {
		logger.Fatal(err)
	}

	BotToken = viper.GetString("bot_token")
	Socks5 = viper.GetString("socks5")
	//set bot
	logger.Println("Bot Settings……")
	Poller := &tb.LongPoller{Timeout: 15 * time.Second}
	spamProtected := tb.NewMiddlewarePoller(Poller, func(upd *tb.Update) bool {
		if upd.Message == nil {
			return true
		}
		if !upd.Message.Private() {
			return false
		}
		return true
	})
	botsettings := tb.Settings{
		Token:  BotToken,
		Poller: spamProtected,
	}
	//set socks5
	if Socks5 != "" {
		logger.Println("Proxy:" + Socks5)
		dialer, err := proxy.SOCKS5("tcp", Socks5, nil, proxy.Direct)
		if err != nil {
			logger.Println(err)
		}
		httpTransport := &http.Transport{}
		httpClient := &http.Client{Transport: httpTransport}
		httpTransport.Dial = dialer.Dial
		botsettings.Client = httpClient
	}
	//create bot
	bot, err = tb.NewBot(botsettings)
	if err != nil {
		logger.Fatal(err)
	}
	logger.Println("Bot: " + strconv.Itoa(bot.Me.ID) + " " + bot.Me.Username)
}