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
|
package main
import (
"encoding/json"
"log"
"net/http"
"os"
"os/signal"
"syscall"
"github.com/ketsuna-org/bot-creator-api/internal"
zmq "github.com/pebbe/zmq4"
)
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!"))
})
// Contexte ZeroMQ
ctx, err := zmq.NewContext()
if err != nil {
log.Fatalf("[SERVER] Failed to create context: %v", err)
}
defer ctx.Term()
// Socket dealer ZeroMQ
dealer, err := ctx.NewSocket(zmq.REP)
if err != nil {
log.Fatalf("[SERVER] Failed to create dealer: %v", err)
}
defer dealer.Close()
err = dealer.Bind("tcp://*:5555")
if err != nil {
log.Fatalf("[SERVER] Failed to bind dealer: %v", err)
}
// 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.URL.Query().Get("bot_token")
bot := &internal.Bot{
BotToken: botToken,
}
bot, err := internal.Start(bot, dealer)
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.URL.Query().Get("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.URL.Query().Get("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))
}
|
