E5SubBotForSQLite/main.go at master · rainerosion/E5SubBotForSQLite · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
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
	logger   *log.Logger
)

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

var dbPath string

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)
	}

	dbPath = strings.Trim(viper.GetString("dbfile"), "")
	logger.Println("sqlite3 db file ->", dbPath)

	//fmt.Println(path)
	db, err := sql.Open(dbDriverName, dbPath)
	if err != nil {
		logger.Fatal(err)
	}
	logger.Println("Connect SQLite Success!")
	if ok, err := CreateTB(); !ok {
		logger.Fatal(err)
	}
	defer db.Close()
	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)
}