marcobaobao
e1510d28d2
Session serialization will use gob encoding instead of json. Binary size will likely be reduced. General backend code refactoring.
119 lines
2.3 KiB
Go
119 lines
2.3 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/fs"
|
|
"log"
|
|
"net/http"
|
|
"net/rpc"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server/internal"
|
|
middlewares "github.com/marcopeocchi/yt-dlp-web-ui/server/middleware"
|
|
"github.com/marcopeocchi/yt-dlp-web-ui/server/rest"
|
|
ytdlpRPC "github.com/marcopeocchi/yt-dlp-web-ui/server/rpc"
|
|
)
|
|
|
|
type serverConfig struct {
|
|
frontend fs.FS
|
|
port int
|
|
db *internal.MemoryDB
|
|
mq *internal.MessageQueue
|
|
}
|
|
|
|
func RunBlocking(port int, frontend fs.FS) {
|
|
var db internal.MemoryDB
|
|
db.Restore()
|
|
|
|
mq := internal.NewMessageQueue()
|
|
go mq.Subscriber()
|
|
|
|
srv := newServer(serverConfig{
|
|
frontend: frontend,
|
|
port: port,
|
|
db: &db,
|
|
mq: mq,
|
|
})
|
|
|
|
// http-post
|
|
go gracefulShutdown(srv, &db)
|
|
go autoPersist(time.Minute*5, &db)
|
|
|
|
log.Fatal(srv.ListenAndServe())
|
|
}
|
|
|
|
func newServer(c serverConfig) *http.Server {
|
|
service := ytdlpRPC.Container(c.db, c.mq)
|
|
rpc.Register(service)
|
|
|
|
r := chi.NewRouter()
|
|
|
|
r.Use(middlewares.CORS)
|
|
r.Use(middleware.Logger)
|
|
|
|
sh := middlewares.NewSpaHandler("index.html", c.frontend)
|
|
sh.AddClientRoute("/settings")
|
|
sh.AddClientRoute("/archive")
|
|
sh.AddClientRoute("/login")
|
|
|
|
r.Get("/*", sh.Handler())
|
|
|
|
// Archive routes
|
|
r.Route("/archive", func(r chi.Router) {
|
|
r.Use(middlewares.Authenticated)
|
|
r.Post("/downloaded", rest.ListDownloaded)
|
|
r.Post("/delete", rest.DeleteFile)
|
|
r.Get("/d/{id}", rest.SendFile)
|
|
})
|
|
|
|
// Authentication routes
|
|
r.Route("/auth", func(r chi.Router) {
|
|
r.Post("/login", rest.Login)
|
|
r.Get("/logout", rest.Logout)
|
|
})
|
|
|
|
// RPC handlers
|
|
r.Route("/rpc", func(r chi.Router) {
|
|
r.Use(middlewares.Authenticated)
|
|
r.Get("/ws", ytdlpRPC.WebSocket)
|
|
r.Post("/http", ytdlpRPC.Post)
|
|
})
|
|
|
|
return &http.Server{
|
|
Addr: fmt.Sprintf(":%d", c.port),
|
|
Handler: r,
|
|
}
|
|
}
|
|
|
|
func gracefulShutdown(srv *http.Server, db *internal.MemoryDB) {
|
|
ctx, stop := signal.NotifyContext(context.Background(),
|
|
os.Interrupt,
|
|
syscall.SIGTERM,
|
|
syscall.SIGQUIT,
|
|
)
|
|
|
|
go func() {
|
|
<-ctx.Done()
|
|
log.Println("shutdown signal received")
|
|
|
|
defer func() {
|
|
db.Persist()
|
|
stop()
|
|
srv.Shutdown(context.TODO())
|
|
}()
|
|
}()
|
|
}
|
|
|
|
func autoPersist(d time.Duration, db *internal.MemoryDB) {
|
|
for {
|
|
db.Persist()
|
|
time.Sleep(d)
|
|
}
|
|
}
|