2022-02-06 02:22:23 +04:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"dwelling-upload/internal/configuration"
|
2022-02-06 15:25:27 +04:00
|
|
|
"dwelling-upload/internal/handlers"
|
2022-02-07 19:35:48 +04:00
|
|
|
"dwelling-upload/pkg/logging"
|
2022-02-06 02:22:23 +04:00
|
|
|
"dwelling-upload/pkg/server"
|
2022-02-08 00:26:59 +04:00
|
|
|
"dwelling-upload/pkg/utils"
|
|
|
|
"dwelling-upload/pkg/watcher"
|
2022-02-06 02:22:23 +04:00
|
|
|
"flag"
|
2022-03-31 16:10:56 +04:00
|
|
|
"fmt"
|
2022-02-06 02:22:23 +04:00
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
"os/signal"
|
|
|
|
"syscall"
|
|
|
|
)
|
|
|
|
|
2022-02-06 15:25:27 +04:00
|
|
|
var configPath *string = flag.String("conf", "config.yaml", "path to configuration file")
|
2022-02-07 21:12:30 +04:00
|
|
|
var logToStdout *bool = flag.Bool("log-stdout", false, "write logs to stdout")
|
2022-03-31 16:10:56 +04:00
|
|
|
var showVersion *bool = flag.Bool("v", false, "show version")
|
2022-02-06 02:22:23 +04:00
|
|
|
|
2022-02-07 22:04:00 +04:00
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
|
2022-03-31 16:10:56 +04:00
|
|
|
if *showVersion {
|
2022-06-13 22:34:15 +04:00
|
|
|
fmt.Println("dwelling-upload Ver. 22.24.1\nCopyright (c) 2022 Alexander \"Arav\" Andreev <me@arav.top>")
|
2022-03-31 16:10:56 +04:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2022-02-07 22:04:00 +04:00
|
|
|
config, err := configuration.LoadConfiguration(*configPath)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalln(err)
|
|
|
|
}
|
|
|
|
|
2022-02-14 20:21:19 +04:00
|
|
|
defer func() {
|
2022-03-06 21:52:07 +04:00
|
|
|
if typ, addr := config.SplitNetworkAddress(); typ == "unix" {
|
2022-02-14 20:21:19 +04:00
|
|
|
os.Remove(addr)
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2022-02-07 21:12:30 +04:00
|
|
|
if *logToStdout {
|
|
|
|
config.Log.ToStdout = true
|
|
|
|
}
|
|
|
|
|
|
|
|
logErr, err := logging.NewLogger(config.Log.Error, config.Log.ToStdout)
|
2022-02-07 19:35:48 +04:00
|
|
|
if err != nil {
|
2022-02-08 00:26:59 +04:00
|
|
|
log.Fatalln("error logger:", err)
|
2022-02-07 19:35:48 +04:00
|
|
|
}
|
|
|
|
defer logErr.Close()
|
|
|
|
|
2022-02-07 21:12:30 +04:00
|
|
|
logUpload, err := logging.NewLogger(config.Log.Upload, config.Log.ToStdout)
|
2022-02-07 19:35:48 +04:00
|
|
|
if err != nil {
|
2022-02-08 00:26:59 +04:00
|
|
|
log.Fatalln("upload logger:", err)
|
2022-02-07 19:35:48 +04:00
|
|
|
}
|
|
|
|
defer logUpload.Close()
|
2022-02-06 15:25:27 +04:00
|
|
|
|
2022-02-07 21:12:30 +04:00
|
|
|
logDownload, err := logging.NewLogger(config.Log.Download, config.Log.ToStdout)
|
2022-02-07 19:35:48 +04:00
|
|
|
if err != nil {
|
2022-02-08 00:26:59 +04:00
|
|
|
log.Fatalln("download logger:", err)
|
2022-02-07 19:35:48 +04:00
|
|
|
}
|
|
|
|
defer logDownload.Close()
|
|
|
|
|
2022-07-01 03:32:12 +04:00
|
|
|
logDelete, err := logging.NewLogger(config.Log.Delete, config.Log.ToStdout)
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalln("delete logger:", err)
|
|
|
|
}
|
|
|
|
defer logDelete.Close()
|
|
|
|
|
2022-03-07 23:12:02 +04:00
|
|
|
watcha, err := watcher.NewInotifyWatcher()
|
2022-02-08 00:26:59 +04:00
|
|
|
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 {
|
2022-03-06 21:52:50 +04:00
|
|
|
logErr.Fatalf("failed to get initial size of %s: %s", config.Uploads.Directory, err)
|
2022-02-08 00:26:59 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
watcha.WatchForMask(uploadDirNotify, watcher.CrDelMask)
|
|
|
|
|
2022-07-01 03:32:12 +04:00
|
|
|
hand := handlers.NewUploadHandlers(config, logErr, logUpload, logDownload, logDelete, &uploadDirSize)
|
2022-02-06 02:22:23 +04:00
|
|
|
srv := server.NewHttpServer()
|
|
|
|
|
2022-02-07 22:52:01 +04:00
|
|
|
srv.SetNotFoundHandler(hand.NotFound)
|
2022-02-08 17:46:16 +04:00
|
|
|
srv.ServeStatic("/assets/*filepath", hand.AssetsFS())
|
2022-02-06 15:25:27 +04:00
|
|
|
srv.GET("/", hand.Index)
|
|
|
|
srv.POST("/", hand.Upload)
|
2022-07-01 03:32:53 +04:00
|
|
|
srv.POST("/:hash/delete", hand.Delete)
|
2022-02-07 04:46:46 +04:00
|
|
|
srv.GET("/f/:hash/:name", hand.Download)
|
2022-07-01 03:32:53 +04:00
|
|
|
srv.DELETE("/:hash", hand.Delete)
|
2022-02-06 15:25:27 +04:00
|
|
|
|
2022-02-06 02:22:23 +04:00
|
|
|
if err := srv.Start(config.SplitNetworkAddress()); err != nil {
|
2022-03-06 21:52:50 +04:00
|
|
|
logErr.Fatalln("failed to start a server:", err)
|
2022-02-06 02:22:23 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
doneSignal := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(doneSignal, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
|
2022-05-25 00:37:33 +04:00
|
|
|
logReopenSignal := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(logReopenSignal, syscall.SIGHUP)
|
2022-02-08 02:05:38 +04:00
|
|
|
|
|
|
|
go func() {
|
|
|
|
for {
|
|
|
|
select {
|
2022-05-25 00:37:33 +04:00
|
|
|
case <-logReopenSignal:
|
|
|
|
logErr.Reopen(config.Log.Error)
|
|
|
|
logUpload.Reopen(config.Log.Upload)
|
|
|
|
logDownload.Reopen(config.Log.Download)
|
2022-02-08 02:05:38 +04:00
|
|
|
case <-uploadDirNotify:
|
|
|
|
sz, err := utils.DirectorySize(config.Uploads.Directory)
|
|
|
|
if err != nil {
|
2022-03-06 21:52:50 +04:00
|
|
|
logErr.Println("failed to get uploads directory size:", err)
|
2022-02-08 02:05:38 +04:00
|
|
|
}
|
|
|
|
|
|
|
|
if sz > 0 {
|
|
|
|
uploadDirSize = sz
|
|
|
|
}
|
2022-02-08 00:26:59 +04:00
|
|
|
}
|
|
|
|
}
|
2022-02-08 02:05:38 +04:00
|
|
|
}()
|
|
|
|
|
2022-05-25 00:37:33 +04:00
|
|
|
<-doneSignal
|
|
|
|
|
|
|
|
if err := srv.Stop(); err != nil {
|
|
|
|
logErr.Fatalln("failed to properly shutdown a server:", err)
|
|
|
|
}
|
2022-02-06 02:22:23 +04:00
|
|
|
}
|