publapi/main.go
Odyssey 6d0d1a3cb0 bruhhh
Signed-off-by: Odyssey <odyssey346@disroot.org>
2023-01-07 18:57:57 +01:00

62 lines
1.2 KiB
Go

package main
import (
"os"
"os/exec"
"github.com/ProjectSegfault/publapi/pages"
"github.com/gofiber/fiber/v2"
log "github.com/sirupsen/logrus"
"runtime"
)
// publapi is a simple API for Project Segfault's public shared server (pubnix).
func main() {
app := fiber.New()
app.Get("/", func(c *fiber.Ctx) error {
return c.JSON(fiber.Map{
"message": "welcome to publapi",
"status": c.Response().StatusCode(),
})
})
app.Get("/online", func(c *fiber.Ctx) error {
if runtime.GOOS == "windows" {
return c.JSON(fiber.Map{
"message": "/online is not supported on Windows",
"status": c.Response().StatusCode(),
})
}
// Get the number of users online
out, err := exec.Command("bash", "-c", "/usr/bin/users | /usr/bin/wc -l").Output()
log.Info(string(out))
if err != nil {
log.Error(err)
return c.SendStatus(fiber.StatusInternalServerError)
}
output := string(out)
return c.JSON(fiber.Map{
"users": output,
"status": c.Response().StatusCode(),
})
})
app.Post("/signup", pages.SignupPage)
app.Listen(GetPort())
}
// GetPort returns the port to listen on
func GetPort() string {
port := os.Getenv("PUBLAPI_PORT")
if port == "" {
port = "3000"
}
return ":" + port
}