97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"ladder/handlers"
|
|
|
|
"github.com/akamensky/argparse"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/gofiber/fiber/v2/middleware/basicauth"
|
|
"github.com/gofiber/fiber/v2/middleware/favicon"
|
|
)
|
|
|
|
//go:embed favicon.ico
|
|
var faviconData string
|
|
//go:embed styles.css
|
|
var cssData embed.FS
|
|
|
|
func main() {
|
|
parser := argparse.NewParser("ladder", "Every Wall needs a Ladder")
|
|
|
|
portEnv := os.Getenv("PORT")
|
|
if os.Getenv("PORT") == "" {
|
|
portEnv = "8080"
|
|
}
|
|
port := parser.String("p", "port", &argparse.Options{
|
|
Required: false,
|
|
Default: portEnv,
|
|
Help: "Port the webserver will listen on",
|
|
})
|
|
|
|
prefork := parser.Flag("P", "prefork", &argparse.Options{
|
|
Required: false,
|
|
Help: "This will spawn multiple processes listening",
|
|
})
|
|
|
|
err := parser.Parse(os.Args)
|
|
if err != nil {
|
|
fmt.Print(parser.Usage(err))
|
|
}
|
|
|
|
if os.Getenv("PREFORK") == "true" {
|
|
*prefork = true
|
|
}
|
|
|
|
app := fiber.New(
|
|
fiber.Config{
|
|
Prefork: *prefork,
|
|
GETOnly: true,
|
|
},
|
|
)
|
|
|
|
userpass := os.Getenv("USERPASS")
|
|
if userpass != "" {
|
|
userpass := strings.Split(userpass, ":")
|
|
app.Use(basicauth.New(basicauth.Config{
|
|
Users: map[string]string{
|
|
userpass[0]: userpass[1],
|
|
},
|
|
}))
|
|
}
|
|
|
|
app.Use(favicon.New(favicon.Config{
|
|
Data: []byte(faviconData),
|
|
URL: "/favicon.ico",
|
|
}))
|
|
|
|
if os.Getenv("NOLOGS") != "true" {
|
|
app.Use(func(c *fiber.Ctx) error {
|
|
log.Println(c.Method(), c.Path())
|
|
return c.Next()
|
|
})
|
|
}
|
|
|
|
app.Get("/", handlers.Form)
|
|
app.Get("/styles.css", func(c *fiber.Ctx) error {
|
|
cssData, err := cssData.ReadFile("styles.css")
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).SendString("Internal Server Error")
|
|
}
|
|
c.Set("Content-Type", "text/css")
|
|
return c.Send(cssData)
|
|
})
|
|
app.Get("ruleset", handlers.Ruleset)
|
|
|
|
app.Get("raw/*", handlers.Raw)
|
|
app.Get("api/*", handlers.Api)
|
|
app.Get("ruleset", handlers.Raw)
|
|
app.Get("/*", handlers.ProxySite)
|
|
|
|
log.Fatal(app.Listen(":" + *port))
|
|
}
|