Improve code reuse

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

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