88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/debridmediamanager.com/zurg/internal/config"
|
|
"github.com/debridmediamanager.com/zurg/internal/net"
|
|
"github.com/debridmediamanager.com/zurg/internal/torrent"
|
|
"github.com/debridmediamanager.com/zurg/internal/zfs"
|
|
"github.com/debridmediamanager.com/zurg/pkg/logutil"
|
|
"github.com/hashicorp/golang-lru/v2/expirable"
|
|
)
|
|
|
|
func main() {
|
|
rlog := logutil.NewLogger()
|
|
log := rlog.Named("zurg")
|
|
|
|
config, configErr := config.LoadZurgConfig("./config.yml")
|
|
if configErr != nil {
|
|
log.Panicf("Config failed to load: %v", configErr)
|
|
}
|
|
|
|
cache := expirable.NewLRU[string, string](1e4, nil, time.Hour)
|
|
|
|
t := torrent.NewTorrentManager(config, cache)
|
|
|
|
mux := http.NewServeMux()
|
|
net.Router(mux, config, t, cache)
|
|
|
|
addr := fmt.Sprintf(":%s", config.GetPort())
|
|
server := &http.Server{Addr: addr, Handler: mux}
|
|
|
|
mountPoint := config.GetMountPoint()
|
|
if _, err := os.Stat(mountPoint); os.IsNotExist(err) {
|
|
if err := os.Mkdir(mountPoint, 0755); err != nil {
|
|
log.Panicf("Failed to create mount point: %v", err)
|
|
}
|
|
}
|
|
|
|
shutdown := make(chan os.Signal, 1)
|
|
signal.Notify(shutdown, os.Interrupt, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
log.Errorf("Server panic: %v\n", r)
|
|
}
|
|
}()
|
|
log.Infof("Starting server on %s", addr)
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
log.Panicf("Failed to start server: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Start the mount in a goroutine with panic recovery.
|
|
go func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
log.Errorf("Mount panic: %v\n", r)
|
|
}
|
|
}()
|
|
log.Infof("Mounting on %s", mountPoint)
|
|
if err := zfs.Mount(mountPoint, config, t); err != nil {
|
|
log.Panicf("Failed to mount: %v", err)
|
|
}
|
|
}()
|
|
|
|
<-shutdown
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
if err := server.Shutdown(ctx); err != nil {
|
|
log.Errorf("Server shutdown error: %v\n", err)
|
|
}
|
|
if err := zfs.Unmount(mountPoint); err != nil {
|
|
log.Errorf("Unmount error: %v\n", err)
|
|
}
|
|
|
|
log.Info("BYE")
|
|
}
|