mirror of
https://github.com/mgerb/go-discord-bot
synced 2026-01-10 09:02:49 +00:00
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package webserver
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"golang.org/x/crypto/acme/autocert"
|
|
|
|
"github.com/go-chi/chi"
|
|
"github.com/go-chi/chi/middleware"
|
|
"github.com/mgerb/go-discord-bot/server/config"
|
|
"github.com/mgerb/go-discord-bot/server/webserver/handlers"
|
|
"github.com/mgerb/go-discord-bot/server/webserver/pubg"
|
|
)
|
|
|
|
func getRouter() *chi.Mux {
|
|
|
|
r := chi.NewRouter()
|
|
|
|
r.Use(middleware.RequestID)
|
|
r.Use(middleware.DefaultCompress)
|
|
|
|
if !config.Flags.Prod {
|
|
r.Use(middleware.Logger)
|
|
}
|
|
|
|
workDir, _ := os.Getwd()
|
|
|
|
FileServer(r, "/static", http.Dir(filepath.Join(workDir, "./dist/static")))
|
|
FileServer(r, "/public/sounds", http.Dir(filepath.Join(workDir, "./sounds")))
|
|
FileServer(r, "/public/youtube", http.Dir(filepath.Join(workDir, "./youtube")))
|
|
|
|
r.NotFound(func(w http.ResponseWriter, r *http.Request) {
|
|
http.ServeFile(w, r, "./dist/index.html")
|
|
})
|
|
|
|
// configure end points
|
|
r.Get("/soundlist", handlers.SoundList)
|
|
r.Put("/upload", handlers.FileUpload)
|
|
r.Get("/ytdownloader", handlers.Downloader)
|
|
r.Get("/stats/pubg", pubg.Handler)
|
|
|
|
return r
|
|
}
|
|
|
|
// Start -
|
|
func Start() {
|
|
|
|
// start gathering pubg data from the api
|
|
if config.Config.Pubg.Enabled {
|
|
pubg.Start(config.Config.Pubg.APIKey, config.Config.Pubg.Players)
|
|
}
|
|
|
|
router := getRouter()
|
|
|
|
if config.Flags.TLS {
|
|
|
|
// start server on port 80 to redirect
|
|
go http.ListenAndServe(":80", http.HandlerFunc(redirect))
|
|
|
|
// start TLS server
|
|
log.Fatal(http.Serve(autocert.NewListener(), router))
|
|
|
|
} else {
|
|
|
|
// start basic server
|
|
http.ListenAndServe(config.Config.ServerAddr, router)
|
|
}
|
|
}
|
|
|
|
// FileServer conveniently sets up a http.FileServer handler to serve
|
|
// static files from a http.FileSystem.
|
|
func FileServer(r chi.Router, path string, root http.FileSystem) {
|
|
if strings.ContainsAny(path, "{}*") {
|
|
panic("FileServer does not permit URL parameters.")
|
|
}
|
|
|
|
fs := http.StripPrefix(path, http.FileServer(root))
|
|
|
|
if path != "/" && path[len(path)-1] != '/' {
|
|
r.Get(path, http.RedirectHandler(path+"/", 301).ServeHTTP)
|
|
path += "/"
|
|
}
|
|
path += "*"
|
|
|
|
r.Get(path, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
fs.ServeHTTP(w, r)
|
|
}))
|
|
}
|
|
|
|
// redirect to https
|
|
func redirect(w http.ResponseWriter, req *http.Request) {
|
|
// remove/add not default ports from req.Host
|
|
target := "https://" + req.Host + req.URL.Path
|
|
if len(req.URL.RawQuery) > 0 {
|
|
target += "?" + req.URL.RawQuery
|
|
}
|
|
|
|
http.Redirect(w, req, target, http.StatusTemporaryRedirect)
|
|
}
|