91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os/signal"
|
|
"strings"
|
|
"syscall"
|
|
|
|
"codeberg.org/arimelody/ari-stream-tools/config"
|
|
"codeberg.org/arimelody/ari-stream-tools/learning"
|
|
"codeberg.org/arimelody/ari-stream-tools/music"
|
|
"codeberg.org/arimelody/ari-stream-tools/static"
|
|
"codeberg.org/arimelody/ari-stream-tools/twitch"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
failed := make(chan error)
|
|
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
gin.SetMode(gin.ReleaseMode)
|
|
|
|
cfg, err := config.LoadConfig()
|
|
if err != nil { log.Fatalf("Failed to load config: %v", err) }
|
|
|
|
srv := gin.New()
|
|
// srv.Use(gin.Logger())
|
|
srv.SetTrustedProxies([]string{"127.0.0.1", "::1"})
|
|
|
|
learningService, err := learning.New(ctx, learning.ServiceConfig{
|
|
TitleFilePath: "learning-title.txt",
|
|
})
|
|
if err != nil {
|
|
log.Printf("Failed to create learning service: %v", err)
|
|
srv.Group("learning").Any("/*any", func(ctx *gin.Context) {
|
|
ctx.String(http.StatusServiceUnavailable, http.StatusText(http.StatusServiceUnavailable))
|
|
})
|
|
} else {
|
|
learningService.BindRoutes(srv.Group("learning"))
|
|
}
|
|
|
|
musicService, err := music.New(ctx)
|
|
if err != nil {
|
|
log.Printf("Failed to create music service: %v", err)
|
|
srv.Group("music").Any("/*any", func(ctx *gin.Context) {
|
|
ctx.String(http.StatusServiceUnavailable, http.StatusText(http.StatusServiceUnavailable))
|
|
})
|
|
} else {
|
|
musicService.BindRoutes(srv.Group("music"))
|
|
go musicService.Run(ctx)
|
|
}
|
|
|
|
twitchService, err := twitch.New(ctx, twitch.ServiceOptions{
|
|
Host: cfg.Host,
|
|
Port: cfg.Port,
|
|
ChannelName: cfg.Twitch.ChannelName,
|
|
ClientID: cfg.Twitch.ClientID,
|
|
ClientSecret: cfg.Twitch.ClientSecret,
|
|
})
|
|
if err != nil {
|
|
log.Printf("Failed to create Twitch service: %v", err)
|
|
srv.Group("twitch").Any("/*any", func(ctx *gin.Context) {
|
|
ctx.String(http.StatusServiceUnavailable, http.StatusText(http.StatusServiceUnavailable))
|
|
})
|
|
} else {
|
|
twitchService.BindRoutes(srv.Group("twitch"))
|
|
go twitchService.Run(ctx)
|
|
}
|
|
|
|
public := srv.Group("/public")
|
|
public.GET("/*path", func(ctx *gin.Context) {
|
|
filepath := strings.TrimPrefix(ctx.Request.URL.Path, "/public/")
|
|
http.ServeFileFS(ctx.Writer, ctx.Request, static.PublicFS, filepath)
|
|
})
|
|
|
|
go func() {
|
|
log.Printf("Now serving at http://%s:%d\n", cfg.Host, cfg.Port)
|
|
failed <- srv.Run(fmt.Sprintf("%s:%d", cfg.Host, cfg.Port))
|
|
}()
|
|
|
|
select {
|
|
case err := <-failed:
|
|
log.Fatal(err)
|
|
case <-ctx.Done():
|
|
log.Print("Shutting down...")
|
|
}
|
|
}
|