2024-07-19 22:07:04 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2024-10-30 18:15:55 +00:00
|
|
|
blossom "grain/app/src"
|
2024-08-06 19:35:52 +00:00
|
|
|
"grain/app/src/api"
|
|
|
|
"grain/app/src/routes"
|
2024-07-30 15:27:38 +00:00
|
|
|
"grain/config"
|
2024-07-31 15:56:55 +00:00
|
|
|
configTypes "grain/config/types"
|
2024-07-31 18:12:33 +00:00
|
|
|
relay "grain/server"
|
2024-09-17 14:35:31 +00:00
|
|
|
"grain/server/db/mongo"
|
2024-07-31 18:12:33 +00:00
|
|
|
"grain/server/utils"
|
2024-07-31 15:28:41 +00:00
|
|
|
"log"
|
|
|
|
"net/http"
|
2024-08-09 20:22:27 +00:00
|
|
|
"os"
|
|
|
|
"os/signal"
|
|
|
|
"sync"
|
|
|
|
"syscall"
|
2024-08-01 11:54:59 +00:00
|
|
|
"time"
|
2024-07-20 01:09:42 +00:00
|
|
|
|
2024-07-19 22:07:04 +00:00
|
|
|
"golang.org/x/net/websocket"
|
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2024-10-30 16:29:50 +00:00
|
|
|
utils.EnsureFileExists("config.yml", "app/static/examples/config.example.yml")
|
|
|
|
utils.EnsureFileExists("whitelist.yml", "app/static/examples/whitelist.example.yml")
|
|
|
|
utils.EnsureFileExists("blacklist.yml", "app/static/examples/blacklist.example.yml")
|
|
|
|
utils.EnsureFileExists("relay_metadata.json", "app/static/examples/relay_metadata.example.json")
|
|
|
|
|
|
|
|
restartChan := make(chan struct{})
|
|
|
|
go config.WatchConfigFile("config.yml", restartChan)
|
|
|
|
go config.WatchConfigFile("whitelist.yml", restartChan)
|
|
|
|
go config.WatchConfigFile("blacklist.yml", restartChan)
|
|
|
|
go config.WatchConfigFile("relay_metadata.json", restartChan)
|
|
|
|
|
|
|
|
signalChan := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(signalChan, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
for {
|
|
|
|
wg.Add(1)
|
|
|
|
|
|
|
|
cfg, err := config.LoadConfig("config.yml")
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error loading config: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = config.LoadWhitelistConfig("whitelist.yml")
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error loading whitelist config: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = config.LoadBlacklistConfig("blacklist.yml")
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error loading blacklist config: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Start event purging in the background.
|
|
|
|
go mongo.ScheduleEventPurging(cfg)
|
|
|
|
|
|
|
|
config.SetResourceLimit(&cfg.ResourceLimits)
|
|
|
|
client, err := mongo.InitDB(cfg)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error initializing database: ", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
config.SetRateLimit(cfg)
|
|
|
|
config.SetSizeLimit(cfg)
|
|
|
|
config.ClearTemporaryBans()
|
|
|
|
|
|
|
|
err = utils.LoadRelayMetadataJSON()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Failed to load relay metadata: ", err)
|
|
|
|
}
|
|
|
|
|
2024-10-30 18:15:55 +00:00
|
|
|
err = blossom.InitBlossom(cfg)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatal("Error initializing Blossom:", err)
|
|
|
|
}
|
2024-10-30 16:29:50 +00:00
|
|
|
mux := setupRoutes()
|
|
|
|
server := startServer(cfg, mux, &wg)
|
|
|
|
|
|
|
|
// Monitor for server restart or shutdown signals.
|
|
|
|
select {
|
|
|
|
case <-restartChan:
|
|
|
|
log.Println("Restarting server...")
|
|
|
|
server.Close()
|
|
|
|
wg.Wait()
|
|
|
|
time.Sleep(3 * time.Second)
|
|
|
|
case <-signalChan:
|
|
|
|
log.Println("Shutting down server...")
|
|
|
|
server.Close()
|
|
|
|
mongo.DisconnectDB(client)
|
|
|
|
wg.Wait()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
2024-07-30 13:58:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func setupRoutes() *http.ServeMux {
|
2024-07-23 20:40:39 +00:00
|
|
|
mux := http.NewServeMux()
|
2024-08-01 14:48:24 +00:00
|
|
|
mux.HandleFunc("/", ListenAndServe)
|
2024-08-06 19:35:52 +00:00
|
|
|
mux.HandleFunc("/import-results", api.ImportEvents)
|
|
|
|
mux.HandleFunc("/import-events", routes.ImportEvents)
|
2024-08-01 14:48:24 +00:00
|
|
|
mux.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("app/static"))))
|
|
|
|
mux.HandleFunc("/favicon.ico", func(w http.ResponseWriter, r *http.Request) {
|
2024-07-31 19:54:42 +00:00
|
|
|
http.ServeFile(w, r, "app/static/img/favicon.ico")
|
2024-08-01 14:48:24 +00:00
|
|
|
})
|
2024-07-30 13:58:20 +00:00
|
|
|
return mux
|
|
|
|
}
|
2024-07-25 02:09:54 +00:00
|
|
|
|
2024-08-09 20:22:27 +00:00
|
|
|
func startServer(config *configTypes.ServerConfig, mux *http.ServeMux, wg *sync.WaitGroup) *http.Server {
|
2024-08-01 11:54:59 +00:00
|
|
|
server := &http.Server{
|
|
|
|
Addr: config.Server.Port,
|
|
|
|
Handler: mux,
|
2024-08-08 12:40:23 +00:00
|
|
|
ReadTimeout: time.Duration(config.Server.ReadTimeout) * time.Second,
|
|
|
|
WriteTimeout: time.Duration(config.Server.WriteTimeout) * time.Second,
|
|
|
|
IdleTimeout: time.Duration(config.Server.IdleTimeout) * time.Second,
|
2024-08-01 11:54:59 +00:00
|
|
|
}
|
2024-08-09 20:22:27 +00:00
|
|
|
|
|
|
|
go func() {
|
2024-08-19 17:47:23 +00:00
|
|
|
defer wg.Done() // Notify that the server goroutine is done
|
2024-08-09 20:22:27 +00:00
|
|
|
fmt.Printf("Server is running on http://localhost%s\n", config.Server.Port)
|
|
|
|
err := server.ListenAndServe()
|
|
|
|
if err != nil && err != http.ErrServerClosed {
|
|
|
|
fmt.Println("Error starting server:", err)
|
|
|
|
}
|
|
|
|
}()
|
2024-08-19 17:47:23 +00:00
|
|
|
|
2024-08-09 20:22:27 +00:00
|
|
|
return server
|
2024-07-25 02:09:54 +00:00
|
|
|
}
|
2024-08-08 12:40:23 +00:00
|
|
|
|
2024-08-01 14:43:17 +00:00
|
|
|
var wsServer = &websocket.Server{
|
2024-08-08 12:40:23 +00:00
|
|
|
Handshake: func(config *websocket.Config, r *http.Request) error {
|
|
|
|
// Skip origin check
|
|
|
|
return nil
|
|
|
|
},
|
|
|
|
Handler: websocket.Handler(relay.WebSocketHandler),
|
2024-08-01 14:43:17 +00:00
|
|
|
}
|
2024-07-25 13:57:24 +00:00
|
|
|
|
|
|
|
func ListenAndServe(w http.ResponseWriter, r *http.Request) {
|
2024-08-08 12:40:23 +00:00
|
|
|
if r.Header.Get("Upgrade") == "websocket" {
|
|
|
|
wsServer.ServeHTTP(w, r)
|
|
|
|
} else if r.Header.Get("Accept") == "application/nostr+json" {
|
2024-08-09 12:55:30 +00:00
|
|
|
utils.RelayInfoHandler(w, r)
|
2024-08-08 12:40:23 +00:00
|
|
|
} else {
|
2024-08-09 13:24:32 +00:00
|
|
|
routes.IndexHandler(w, r)
|
2024-08-08 12:40:23 +00:00
|
|
|
}
|
2024-08-18 02:00:03 +00:00
|
|
|
}
|