RajaBot/main.go at main · FDeghy/RajaBot · 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
package main

import (
	"RajaBot/bot"
	"RajaBot/bot/adminHandler"
	"RajaBot/config"
	"RajaBot/core"
	"RajaBot/database"
	"RajaBot/payment"
	"RajaBot/prometheus"
	"RajaBot/tools"
	"flag"
	"fmt"
	"log"
)

func main() {
	fl := flag.Bool("addall", false, "add free sub to all users that have sub from 1 farvardin")
	flag.Parse()

	if *fl {
		addall()
		return
	}

	err := config.ParseConfig("config.toml")
	if err != nil {
		log.Fatalln("failed to parse config.")
	}

	err = bot.CreateBot()
	if err != nil {
		log.Fatalf("failed to create bot.\nerror: %v\n", err)
	}

	err = database.StartDatabase()
	if err != nil {
		log.Fatalln("failed to start database.")
	}

	err = core.StartCore()
	if err != nil {
		log.Fatalln("failed to start core.")
	}

	err = prometheus.StartProm()
	if err != nil {
		log.Fatalf("failed to start prometheus.\nerror: %v\n", err)
	}

	err = payment.StartPaymentServer()
	if err != nil {
		log.Fatalf("failed to start payment server.\nerror: %v\n", err)
	}

	err = bot.StartBot()
	if err != nil {
		log.Fatalf("failed to start bot.\nerror: %v\n", err)
	}
}

func addall() {
	days := 30

	err := config.ParseConfig("config.toml")
	if err != nil {
		log.Fatalln("failed to parse config.")
	}

	err = bot.CreateBot()
	if err != nil {
		log.Fatalf("failed to create bot.\nerror: %v\n", err)
	}

	err = database.StartDatabase()
	if err != nil {
		log.Fatalln("failed to start database.")
	}

	for _, user := range database.GetAllSubscription() {
		if (user.ExpirationDate-2592000) >= 1710361800 && database.IsHavePayment(user.UserID) {
			tools.AddDaysSub(user.UserID, days)
			_, err := tools.Bot.SendMessage(user.UserID, fmt.Sprintf(adminHandler.AddSubMsg, days), nil)
			if err != nil {
				log.Printf("Error: %v, User: %v", err, user.UserID)
				continue
			}
			log.Printf("add %v days to %v", days, user.UserID)
		}
	}
}