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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
|
package main
import (
"encoding/json"
"fmt"
"log"
"net/http"
"os"
"os/signal"
"syscall"
"github.com/ketsuna-org/bot-creator-api/internal"
)
var botList = make(map[string]*internal.Bot)
func init() {
// Initialisation de l'application
}
func main() {
// Créer un ServeMux
mux := http.NewServeMux()
// Route principale
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("Hello, World!"))
})
// Route POST /create/{bot_token}
mux.HandleFunc("POST /create/{bot_token}", func(w http.ResponseWriter, r *http.Request) {
// Extraire le token du bot de l'URL
botToken := r.PathValue("bot_token")
// we need to retrieve the amount of bots running
bot := &internal.Bot{
BotToken: botToken,
ProcessID: fmt.Sprint(len(botList) + 5555), // or any unique identifier
}
// Let's check if this discord bot exists
if _, ok := botList[botToken]; ok {
log.Printf("[SERVER] Bot already running: %s", botToken)
http.Error(w, "Bot already running", http.StatusConflict)
return
}
// let's check if it exist on Discord
httpClient := &http.Client{}
req, err := http.NewRequest("GET", "https://discord.com/api/v10/users/@me", nil)
if err != nil {
log.Printf("[SERVER] Error creating request: %v", err)
http.Error(w, "Error creating request", http.StatusInternalServerError)
return
}
req.Header.Set("Authorization", "Bot "+botToken)
resp, err := httpClient.Do(req)
if err != nil {
log.Printf("[SERVER] Error sending request: %v", err)
http.Error(w, "Error sending request", http.StatusInternalServerError)
return
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
log.Printf("[SERVER] Bot not found: %s", botToken)
http.Error(w, "Bot not found", http.StatusNotFound)
return
}
// let's check if the bot is already running
bot, err = internal.Start(bot)
if err != nil {
log.Printf("[SERVER] Error starting bot: %v", err)
http.Error(w, "Error starting bot", http.StatusInternalServerError)
return
}
botList[botToken] = bot
log.Printf("[SERVER] Bot started successfully")
w.WriteHeader(http.StatusOK)
w.Write([]byte("Bot started successfully"))
})
// Route POST /stop/{bot_token}
mux.HandleFunc("POST /stop/{bot_token}", func(w http.ResponseWriter, r *http.Request) {
// Extraire le token du bot de l'URL
botToken := r.PathValue("bot_token")
bot, ok := botList[botToken]
if !ok {
http.Error(w, "Bot not found", http.StatusNotFound)
return
}
if err := bot.Stop(); err != nil {
log.Printf("[SERVER] Error stopping bot: %v", err)
http.Error(w, "Error stopping bot", http.StatusInternalServerError)
return
}
delete(botList, botToken)
log.Printf("[SERVER] Bot stopped successfully")
w.WriteHeader(http.StatusOK)
w.Write([]byte("Bot stopped successfully"))
})
// Route POST /update/{bot_token}
mux.HandleFunc("POST /update/{bot_token}", func(w http.ResponseWriter, r *http.Request) {
// Extraire le token du bot de l'URL
botToken := r.PathValue("bot_token")
bot, ok := botList[botToken]
if !ok {
http.Error(w, "Bot not found", http.StatusNotFound)
return
}
body := make(map[string]interface{})
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
log.Printf("[SERVER] Error decoding JSON: %v", err)
http.Error(w, "Invalid JSON", http.StatusBadRequest)
return
}
data, err := json.Marshal(body)
if err != nil {
log.Printf("[SERVER] Error marshaling JSON: %v", err)
http.Error(w, "Error marshaling JSON", http.StatusInternalServerError)
return
}
if err := bot.SendMessage(string(data)); err != nil {
log.Printf("[SERVER] Error sending message: %v", err)
http.Error(w, "Error sending message", http.StatusInternalServerError)
return
}
log.Printf("[SERVER] Bot updated successfully")
w.WriteHeader(http.StatusOK)
w.Write([]byte("Bot updated successfully"))
})
// Gestion des signaux pour l'arrêt propre
signals := make(chan os.Signal, 1)
signal.Notify(signals, os.Interrupt)
signal.Notify(signals, syscall.SIGTERM)
go func() {
sig := <-signals
log.Printf("Received signal: %s", sig)
// Arrêter tous les bots en cours
for _, bot := range botList {
if err := bot.Stop(); err != nil {
log.Printf("[SERVER] Error stopping bot: %v", err)
}
delete(botList, bot.BotToken)
}
// Quitter l'application
os.Exit(0)
}()
// Démarrer le serveur HTTP
log.Printf("[SERVER] Starting server on :2030")
log.Fatal(http.ListenAndServe(":2030", mux))
}
|
