mirrors/rimgo
1
0
Fork 0
mirror of https://codeberg.org/video-prize-ranch/rimgo.git synced 2025-12-13 19:55:21 +00:00
Code Issues Projects Releases Wiki Activity

Improve code reuse

Browse Source
This commit is contained in:
video-prize-ranch
2022-05-23 11:10:50 -04:00
parent 5569feb2c8
commit 6eee86d682
7 changed files with 65 additions and 90 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
pages/gallery.go
View File

@@ -1,34 +0,0 @@
package pages
import (
"codeberg.org/video-prize-ranch/rimgo/api"
"codeberg.org/video-prize-ranch/rimgo/types"
"codeberg.org/video-prize-ranch/rimgo/utils"
"github.com/gofiber/fiber/v2"
)
func HandleGallery(c *fiber.Ctx) error {
utils.SetHeaders(c)
c.Set("Content-Security-Policy", "default-src 'none'; media-src 'self'; style-src 'unsafe-inline' 'self'; img-src 'self'; font-src 'self'; manifest-src 'self'; block-all-mixed-content")
album, err := api.FetchAlbum(c.Params("galleryID"))
if err != nil {
return err
}
comments := []types.Comment{}
if album.SharedWithCommunity {
c.Set("Cache-Control", "public,max-age=604800")
comments, err = api.FetchComments(c.Params("galleryID"))
if err != nil {
return err
}
} else {
c.Set("Cache-Control", "public,max-age=31557600")
}
return c.Render("post", fiber.Map{
"post": album,
"comments": comments,
})
}

20
pages/post.go
View File

@@ -2,6 +2,7 @@ package pages
import (
"fmt"
"strings"
"codeberg.org/video-prize-ranch/rimgo/api"
"codeberg.org/video-prize-ranch/rimgo/types"
@@ -13,15 +14,18 @@ func HandlePost(c *fiber.Ctx) error {
utils.SetHeaders(c)
c.Set("Content-Security-Policy", "default-src 'none'; media-src 'self'; style-src 'self'; img-src 'self'; font-src 'self'; manifest-src 'self'; block-all-mixed-content")
post, err := api.FetchPosts(c.Params("postID"))
if post.Id == "" {
post, err := types.Album{}, error(nil)
switch {
case strings.HasPrefix(c.Path(), "/a"):
post, err = api.FetchAlbum(c.Params("postID"))
println(post.Title)
case strings.HasPrefix(c.Path(), "/gallery"):
post, err = api.FetchPosts(c.Params("postID"))
if post.Id == "" {
post, err = api.FetchMedia(c.Params("postID"))
if post.Id == "" {
return fmt.Errorf("404 page not found")
}
}
default:
post, err = api.FetchMedia(c.Params("postID"))
}
if post.Id == "" || strings.Contains(err.Error(), "404") {
return fmt.Errorf("404 page not found")
}
if err != nil {
return err