115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"dwelling-upload/internal/configuration"
|
|
"dwelling-upload/internal/handlers"
|
|
"dwelling-upload/pkg/logging"
|
|
"dwelling-upload/pkg/server"
|
|
"dwelling-upload/pkg/utils"
|
|
"dwelling-upload/pkg/watcher"
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
var configPath *string = flag.String("conf", "config.yaml", "path to configuration file")
|
|
var logToStdout *bool = flag.Bool("log-stdout", false, "write logs to stdout")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
config, err := configuration.LoadConfiguration(*configPath)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
defer func() {
|
|
if typ, addr := config.SplitNetworkAddress(); typ == "unix" {
|
|
os.Remove(addr)
|
|
}
|
|
}()
|
|
|
|
if *logToStdout {
|
|
config.Log.ToStdout = true
|
|
}
|
|
|
|
logErr, err := logging.NewLogger(config.Log.Error, config.Log.ToStdout)
|
|
if err != nil {
|
|
log.Fatalln("error logger:", err)
|
|
}
|
|
defer logErr.Close()
|
|
|
|
logUpload, err := logging.NewLogger(config.Log.Upload, config.Log.ToStdout)
|
|
if err != nil {
|
|
log.Fatalln("upload logger:", err)
|
|
}
|
|
defer logUpload.Close()
|
|
|
|
logDownload, err := logging.NewLogger(config.Log.Download, config.Log.ToStdout)
|
|
if err != nil {
|
|
log.Fatalln("download logger:", err)
|
|
}
|
|
defer logDownload.Close()
|
|
|
|
watcha, err := watcher.NewInotifyWatcher()
|
|
if err != nil {
|
|
logErr.Fatalln(err)
|
|
}
|
|
defer watcha.Close()
|
|
|
|
if err := watcha.AddWatch(config.Uploads.Directory, watcher.CrDelMask); err != nil {
|
|
logErr.Fatalln(err)
|
|
}
|
|
|
|
uploadDirNotify := make(chan uint32)
|
|
uploadDirSize, err := utils.DirectorySize(config.Uploads.Directory)
|
|
if err != nil {
|
|
logErr.Fatalf("failed to get initial size of %s: %s", config.Uploads.Directory, err)
|
|
}
|
|
|
|
watcha.WatchForMask(uploadDirNotify, watcher.CrDelMask)
|
|
|
|
hand := handlers.NewUploadHandlers(config, logErr, logUpload, logDownload, &uploadDirSize)
|
|
srv := server.NewHttpServer()
|
|
|
|
srv.SetNotFoundHandler(hand.NotFound)
|
|
srv.ServeStatic("/assets/*filepath", hand.AssetsFS())
|
|
srv.GET("/", hand.Index)
|
|
srv.POST("/", hand.Upload)
|
|
srv.GET("/f/:hash/:name", hand.Download)
|
|
|
|
if err := srv.Start(config.SplitNetworkAddress()); err != nil {
|
|
logErr.Fatalln("failed to start a server:", err)
|
|
}
|
|
|
|
doneSignal := make(chan os.Signal, 1)
|
|
signal.Notify(doneSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
|
|
closeSignal := make(chan bool, 1)
|
|
|
|
go func() {
|
|
for {
|
|
select {
|
|
case <-doneSignal:
|
|
if err := srv.Stop(); err != nil {
|
|
logErr.Fatalln("failed to properly shutdown a server:", err)
|
|
}
|
|
|
|
closeSignal <- true
|
|
case <-uploadDirNotify:
|
|
sz, err := utils.DirectorySize(config.Uploads.Directory)
|
|
if err != nil {
|
|
logErr.Println("failed to get uploads directory size:", err)
|
|
}
|
|
|
|
if sz > 0 {
|
|
uploadDirSize = sz
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
<-closeSignal
|
|
}
|