publapi/main.go
Odyssey 41ecec808c Switch logger
Signed-off-by: Odyssey <odyssey346@disroot.org>
2023-01-07 17:54:04 +01:00

51 lines
976 B
Go

package main
import (
"os"
"os/exec"
"github.com/ProjectSegfault/publapi/pages"
"github.com/gofiber/fiber/v2"
log "github.com/sirupsen/logrus"
)
// 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 {
// Get the number of users online
out, err := exec.Command("users | wc -l").Output()
if err != nil {
log.Error(err)
return c.SendStatus(fiber.StatusInternalServerError)
}
return c.JSON(fiber.Map{
"users": out,
"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
}