mirror of
https://github.com/nxshock/simplefileshare.git
synced 2025-07-02 00:13:36 +05:00
68 lines
1.2 KiB
Go
68 lines
1.2 KiB
Go
package main
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func init() {
|
|
log.SetOutput(os.Stderr)
|
|
log.SetFormatter(&log.TextFormatter{ForceColors: true, DisableTimestamp: true})
|
|
log.SetLevel(log.ErrorLevel)
|
|
|
|
err := initConfig()
|
|
if err != nil {
|
|
log.Fatalln("initConfig:", err)
|
|
}
|
|
|
|
log.SetLevel(config.LogLevel)
|
|
|
|
err = initTemplates()
|
|
if err != nil {
|
|
log.Fatalln("initTemplates:", err)
|
|
}
|
|
|
|
if config.RemoveFilePeriod > 0 {
|
|
go removeOldFilesThread(time.Duration(config.RemoveFilePeriod) * time.Hour)
|
|
}
|
|
|
|
http.HandleFunc("/", HandleRoot)
|
|
http.HandleFunc("/icon", HandleIcon)
|
|
http.HandleFunc("/upload", HandleUpload)
|
|
http.HandleFunc("/download", HandleDownload)
|
|
http.HandleFunc("/stream", HandleStream)
|
|
}
|
|
|
|
func main() {
|
|
go func() {
|
|
u, err := url.Parse(config.ListenAddress)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
syscall.Umask(0007)
|
|
|
|
listener, err := net.Listen(u.Scheme, u.Host+u.Path)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
|
|
err = http.Serve(listener, http.DefaultServeMux)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
}()
|
|
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
|
|
|
<-c
|
|
log.Debugln("Stop signal received.")
|
|
}
|