upload to github
This commit is contained in:
commit
ae9ca74ac6
23 changed files with 2133 additions and 0 deletions
85
main.go
Normal file
85
main.go
Normal file
|
|
@ -0,0 +1,85 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"embed"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
"beat-harvester/src/config"
|
||||
"beat-harvester/src/handlers"
|
||||
"beat-harvester/src/utils"
|
||||
)
|
||||
|
||||
//go:embed src/static/*
|
||||
var staticFS embed.FS
|
||||
|
||||
func main() {
|
||||
// Initialize configuration
|
||||
cfg := config.Load()
|
||||
|
||||
// Check dependencies
|
||||
if err := utils.CheckDependencies(); err != nil {
|
||||
log.Fatalf("Missing dependencies: %v", err)
|
||||
}
|
||||
|
||||
// Setup router
|
||||
r := setupRoutes(cfg)
|
||||
|
||||
// Create server
|
||||
srv := &http.Server{
|
||||
Addr: ":" + cfg.Port,
|
||||
Handler: r,
|
||||
}
|
||||
|
||||
// Start graceful shutdown handler
|
||||
go handleGracefulShutdown(srv)
|
||||
|
||||
log.Printf("Music Downloader Server starting on port %s", cfg.Port)
|
||||
log.Printf("Default output path: %s", cfg.DefaultOutputPath)
|
||||
log.Printf("Web interface: http://localhost:%s", cfg.Port)
|
||||
|
||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
log.Fatalf("Server failed to start: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func setupRoutes(cfg *config.Config) *mux.Router {
|
||||
r := mux.NewRouter()
|
||||
|
||||
// Initialize handlers with config
|
||||
h := handlers.New(cfg, staticFS)
|
||||
|
||||
// API routes
|
||||
api := r.PathPrefix("/api").Subrouter()
|
||||
api.HandleFunc("/download", h.Download).Methods("POST")
|
||||
api.HandleFunc("/health", h.Health).Methods("GET")
|
||||
|
||||
// WebSocket route
|
||||
r.HandleFunc("/ws", h.WebSocket)
|
||||
|
||||
// Serve static files
|
||||
r.PathPrefix("/").Handler(h.Static())
|
||||
|
||||
return r
|
||||
}
|
||||
|
||||
func handleGracefulShutdown(srv *http.Server) {
|
||||
sigChan := make(chan os.Signal, 1)
|
||||
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-sigChan
|
||||
|
||||
log.Println("Shutting down server...")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
|
||||
if err := srv.Shutdown(ctx); err != nil {
|
||||
log.Fatalf("Server shutdown failed: %v", err)
|
||||
}
|
||||
log.Println("Server stopped")
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue