Files
soundcloak/main.go
2024-08-24 11:04:20 +03:00

100 lines
2.6 KiB
Go

package main
import (
"context"
_ "embed"
"fmt"
"log"
"net/url"
"github.com/gofiber/fiber/v2"
"github.com/gofiber/fiber/v2/middleware/compress"
"github.com/gofiber/fiber/v2/middleware/earlydata"
"github.com/gofiber/fiber/v2/middleware/recover"
"github.com/maid-zone/soundcloak/lib/cfg"
"github.com/maid-zone/soundcloak/lib/sc"
"github.com/maid-zone/soundcloak/templates"
)
func main() {
app := fiber.New()
app.Use(compress.New())
app.Use(recover.New())
app.Use(earlydata.New())
app.Static("/", "assets", fiber.Static{Compress: true, MaxAge: 3600})
app.Static("/js/hls.js/", "node_modules/hls.js/dist", fiber.Static{Compress: true, MaxAge: 3600})
app.Get("/search", func(c *fiber.Ctx) error {
q := c.Query("q")
if q == "" {
return c.SendStatus(404)
}
t := c.Query("type")
switch t {
case "tracks":
p, err := sc.SearchTracks("?q=" + url.QueryEscape(q))
if err != nil {
fmt.Printf("error getting tracks for %s: %s\n", q, err)
return err
}
c.Set("Content-Type", "text/html")
return templates.Base("tracks: "+q, templates.SearchTracks(p)).Render(context.Background(), c)
case "users":
p, err := sc.SearchUsers("?q=" + url.QueryEscape(q))
if err != nil {
fmt.Printf("error getting users for %s: %s\n", q, err)
return err
}
c.Set("Content-Type", "text/html")
return templates.Base("users: "+q, templates.SearchUsers(p)).Render(context.Background(), c)
}
return c.SendStatus(404)
})
app.Get("/:user/:track", func(c *fiber.Ctx) error {
track, err := sc.GetTrack(c.Params("user") + "/" + c.Params("track"))
if err != nil {
fmt.Printf("error getting %s from %s: %s\n", c.Params("track"), c.Params("user"), err)
return err
}
stream, err := track.GetStream()
if err != nil {
fmt.Printf("error getting %s stream from %s: %s\n", c.Params("track"), c.Params("user"), err)
}
c.Set("Content-Type", "text/html")
return templates.Base(track.Title+" by "+track.Author.Username, templates.Track(track, stream)).Render(context.Background(), c)
})
app.Get("/:user", func(c *fiber.Ctx) error {
//h := time.Now()
usr, err := sc.GetUser(c.Params("user"))
if err != nil {
fmt.Printf("error getting %s: %s\n", c.Params("user"), err)
return err
}
//fmt.Println("getuser", time.Since(h))
//h = time.Now()
p, err := usr.GetTracks(c.Query("pagination", "?limit=20"))
if err != nil {
fmt.Printf("error getting %s tracks: %s\n", c.Params("user"), err)
return err
}
//fmt.Println("gettracks", time.Since(h))
c.Set("Content-Type", "text/html")
return templates.Base(usr.Username, templates.User(usr, p)).Render(context.Background(), c)
})
log.Fatal(app.Listen(cfg.Addr))
}