Add Checkerr And SQL Control

main.go

@@ -1,34 +1,81 @@
 package main
 
 import (
+	"database/sql"
 	"fmt"
 	"github.com/spf13/viper"
+	"golang.org/x/net/proxy"
 	tb "gopkg.in/tucnak/telebot.v2"
+	"log"
+	"net/http"
+	"os"
 	"time"
 )
 
 var (
 	BotToken string
+	Socks5   string
 )
 
-func main() {
+const (
+	dbDriverName = "sqlite3"
+	dbName       = "./data.db"
+)
+
+func init() {
 	viper.SetConfigName("config")
 	viper.AddConfigPath(".")
 	err := viper.ReadInConfig()
-	if err != nil {
-		panic(fmt.Errorf("Fatal error config file: %s \n", err))
-	}
+	CheckErr(err)
 	BotToken = viper.GetString("bot_token")
-	b, err := tb.NewBot(tb.Settings{
+	Socks5 = viper.GetString("socks5")
+
+}
+func main() {
+	botsettings := tb.Settings{
 		Token:  BotToken,
 		Poller: &tb.LongPoller{Timeout: 10 * time.Second},
-	})
-	if err != nil {
-		return
 	}
+	if Socks5 != "" {
+		fmt.Println("Proxy:" + Socks5)
+		dialer, err := proxy.SOCKS5("tcp", Socks5, nil, proxy.Direct)
+		if err != nil {
+			log.Fatal("Error creating dialer, aborting.")
+		}
+		httpTransport := &http.Transport{}
+		httpClient := &http.Client{Transport: httpTransport}
+		httpTransport.Dial = dialer.Dial
+		botsettings.Client = httpClient
+	}
+	db, err := sql.Open(dbDriverName, dbName)
+	CheckErr(err)
+	if !FileExist(dbName) {
+		CreateTB(db)
+	}
+	b, err := tb.NewBot(botsettings)
+	CheckErr(err)
 	//b.Handle(tb.OnText, func(m *tb.Message) {
 	//	b.Send(m.Sender, "hello world")
 	//})
 
 	b.Start()
 }
+func CheckErr(err error) bool {
+	if err != nil {
+		log.Println(err)
+		fmt.Println("error: ", err.Error())
+		panic(err)
+		return false
+	}
+	return true
+}
+func FileExist(Path string) bool {
+	if _, err := os.Stat(Path); err != nil {
+		if os.IsNotExist(err) {
+			return false
+		} else {
+			CheckErr(err)
+		}
+	}
+	return true
+}