From d5003b51a4ca7ed2c13ffd23b051ea31c8836ba0 Mon Sep 17 00:00:00 2001
From: KushBlazingJudah <59340248+KushBlazingJudah@users.noreply.github.com>
Date: Thu, 18 Nov 2021 23:25:50 -0400
Subject: board/thread view; images don't work

---
 main.go | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

(limited to 'main.go')

diff --git a/main.go b/main.go
index 02beaf4..2f8131d 100644
--- a/main.go
+++ b/main.go
@@ -116,9 +116,6 @@ func main() {
 		Views:   template,
 	})
 
-	app.Static("/public", "./public")
-	app.Static("/static", "./views")
-
 	/*
 	 Main actor
 	*/
@@ -135,7 +132,7 @@ func main() {
 	 Board actor
 	*/
 
-	app.Get("/:actor", routes.ActorIndex) //OutboxGet)
+	app.Get("/:actor", routes.OutboxGet)
 
 	app.Get("/:actor/:post", routes.ActorPostGet)
 	app.Get("/post", routes.ActorPost)
@@ -234,12 +231,19 @@ func main() {
 	})
 
 	app.Get("/api/media", func(c *fiber.Ctx) error {
-		return c.SendString("api media")
+		if c.Query("hash") != "" {
+			return RouteImages(c, c.Query("hash"))
+		}
+
+		return c.SendStatus(404)
 	})
 
 	// 404 handler
 	app.Use(routes.NotFound)
 
+	app.Static("/public", "./public")
+	app.Static("/static", "./views")
+
 	fmt.Println("Mod key: " + config.Key)
 	PrintAdminAuth()
 
@@ -979,7 +983,7 @@ func AddInstanceToIndexDB(actor string) error {
 	return nil
 }
 
-func RouteImages(w http.ResponseWriter, media string) error {
+func RouteImages(ctx *fiber.Ctx, media string) error {
 	req, err := http.NewRequest("GET", MediaHashs[media], nil)
 	if err != nil {
 		return err
@@ -1001,19 +1005,17 @@ func RouteImages(w http.ResponseWriter, media string) error {
 			return err
 		}
 
-		_, err = w.Write(fileBytes)
-		return err
+		return ctx.Send(fileBytes)
 	}
 
 	body, _ := ioutil.ReadAll(resp.Body)
 	for name, values := range resp.Header {
 		for _, value := range values {
-			w.Header().Set(name, value)
+			ctx.Append(name, value)
 		}
 	}
 
-	_, err = w.Write(body)
-	return err
+	return ctx.Send(body)
 }
 
 func TemplateFunctions(engine *html.Engine) {
-- 
cgit v1.2.3