From 0c6b23cde0e749556514cc08f8028e44793d105c Mon Sep 17 00:00:00 2001
From: Odyssey <odyssey346@disroot.org>
Date: Wed, 30 Nov 2022 18:32:15 +0100
Subject: [PATCH] Update header, and fix raw from fileview.

Signed-off-by: Odyssey <odyssey346@disroot.org>
---
 main.go               | 10 ++++++++++
 public/css/global.css |  1 +
 views/file.html       |  2 +-
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 157a8d3..bcb598f 100644
--- a/main.go
+++ b/main.go
@@ -115,6 +115,7 @@ func main() {
 			"file":     file,
 			"fullname": c.Params("user") + "/" + c.Params("repo"),
 			"name":     c.Params("file"),
+			"branch":   c.Params("branch"),
 		})
 	})
 	app.Get("/avatar/:id", func(c *fiber.Ctx) error {
@@ -135,6 +136,15 @@ func main() {
 		c.Response().Header.Del(fiber.HeaderServer)
 		return nil
 	})
+	app.Get("/raw/:user/:repo/:branch/:file", ratelimiter, func(c *fiber.Ctx) error {
+		url := "https://raw.githubusercontent.com/" + c.Params("user") + "/" + c.Params("repo") + "/" + c.Params("branch") + "/" + c.Params("file")
+		if err := proxy.Do(c, url); err != nil {
+			return err
+		}
+		// Remove Server header from response
+		c.Response().Header.Del(fiber.HeaderServer)
+		return nil
+	})
 	val, ok := os.LookupEnv("GOTHUB_PORT")
 	if !ok {
 		val = "3000"
diff --git a/public/css/global.css b/public/css/global.css
index 5e19f90..d19e324 100644
--- a/public/css/global.css
+++ b/public/css/global.css
@@ -192,6 +192,7 @@ a:hover {
     color: var(--text);
     padding: 8px;
     border-radius: 8px;
+    white-space: pre-wrap;
 }
 
 
diff --git a/views/file.html b/views/file.html
index d953e8f..5f1ff7b 100644
--- a/views/file.html
+++ b/views/file.html
@@ -11,7 +11,7 @@
             <pre class="filePre">{{ .file}}</pre>
         </div>
         <div class="downloadParent">
-            <a href="/raw/{{.fullname}}/{{.name}}" class="downloadButton">Download</a>
+            <a href="/raw/{{.fullname}}/{{.branch}}/{{.name}}" class="downloadButton">Download</a>
         </div>
     {{end}}
 </main>
\ No newline at end of file