From a66b676481d273508927e64a22e388dc302890ba Mon Sep 17 00:00:00 2001
From: FChannel <>
Date: Sun, 22 May 2022 14:08:36 -0700
Subject: route organization

---
 route/routes/api.go | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 route/routes/api.go

(limited to 'route/routes/api.go')

diff --git a/route/routes/api.go b/route/routes/api.go
new file mode 100644
index 0000000..080d88d
--- /dev/null
+++ b/route/routes/api.go
@@ -0,0 +1,55 @@
+package routes
+
+import (
+	"io/ioutil"
+	"net/http"
+	"time"
+
+	"github.com/FChannel0/FChannel-Server/config"
+	"github.com/FChannel0/FChannel-Server/util"
+	"github.com/gofiber/fiber/v2"
+)
+
+func Media(c *fiber.Ctx) error {
+	if c.Query("hash") != "" {
+		return RouteImages(c, c.Query("hash"))
+	}
+
+	return c.SendStatus(404)
+}
+
+func RouteImages(ctx *fiber.Ctx, media string) error {
+	req, err := http.NewRequest("GET", config.MediaHashs[media], nil)
+	if err != nil {
+		return util.MakeError(err, "RouteImages")
+	}
+
+	client := http.Client{
+		Timeout: 5 * time.Second,
+	}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return util.MakeError(err, "RouteImages")
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode != 200 {
+		fileBytes, err := ioutil.ReadFile("./static/notfound.png")
+		if err != nil {
+			return util.MakeError(err, "RouteImages")
+		}
+
+		_, err = ctx.Write(fileBytes)
+		return util.MakeError(err, "RouteImages")
+	}
+
+	body, _ := ioutil.ReadAll(resp.Body)
+	for name, values := range resp.Header {
+		for _, value := range values {
+			ctx.Append(name, value)
+		}
+	}
+
+	return ctx.Send(body)
+}
-- 
cgit v1.2.3