From 8e65053cc1db732577a4245e9bad09696aade591 Mon Sep 17 00:00:00 2001
From: sad <sad@poptart.cat>
Date: Wed, 28 Feb 2024 06:17:37 -0700
Subject: [PATCH] implemented compression

---
 main.go        | 82 +++++++++++++++++++++++++++++++++++++++++---------
 www/index.html | 40 ++++++++++++++++++------
 2 files changed, 98 insertions(+), 24 deletions(-)

diff --git a/main.go b/main.go
index 7574188..608b122 100644
--- a/main.go
+++ b/main.go
@@ -7,6 +7,9 @@ import (
 	"os"
 	"strconv"
 	"time"
+  "compress/gzip"
+  "strings"
+  "bytes"
 
 	"github.com/BurntSushi/toml"
 	"github.com/gabriel-vasile/mimetype"
@@ -109,31 +112,76 @@ func Exists(path string) bool {
 	return true
 }
 
+func isValidGzipHeader(data []byte) bool {
+    return len(data) >= 2 && data[0] == 0x1f && data[1] == 0x8b
+}
+
 func UploadHandler(w http.ResponseWriter, r *http.Request) {
 	// expiry time
 	var ttl int64
 
 	ttl = 0
 
-	file, _, err := r.FormFile("file")
+	var file, header, err = r.FormFile("file")
 	if err != nil {
 		w.WriteHeader(http.StatusBadRequest)
 		return
 	}
-	defer file.Close()
+  defer file.Close()
 
-	mtype, err := mimetype.DetectReader(file)
-	if err != nil {
-		w.Write([]byte("error detecting the mime type of your file\n"))
-		return
-	}
-	file.Seek(0, 0)
+  var buffer bytes.Buffer
+  _, err = io.Copy(&buffer, file)
+  if err != nil {
+    log.Error().Err(err).Msg("error reading file data")
+    w.WriteHeader(http.StatusInternalServerError)
+    return
+  }
+
+  var data bytes.Buffer
+  var name = header.Filename
+  println("file: " + name)
+  if strings.HasSuffix(name, ".5000") {
+    buffer.WriteTo(&data)
+    if isValidGzipHeader(data.Bytes()[:2]) {
+      gz, err := gzip.NewReader(&data)
+        if err != nil {
+          log.Error().Err(err).Msg("error creating gzip reader")
+          w.WriteHeader(http.StatusInternalServerError)
+          return
+        }
+        defer gz.Close()
+
+        _, err = io.Copy(&buffer, gz)
+        if err != nil {
+          log.Error().Err(err).Msg("error decompressing gzip file")
+          w.WriteHeader(http.StatusInternalServerError)
+          return
+        }
+
+        name = strings.TrimSuffix(name, ".5000")
+        data.Write(buffer.Bytes())
+    } else {
+      log.Error().Msg("Invalid gzip file")
+      w.WriteHeader(http.StatusBadRequest)
+      return
+      }
+  } else {
+    data.Write(buffer.Bytes())
+  }
+
+  mtype, err := mimetype.DetectReader(&data)
+  if err != nil {
+    log.Error().Err(err).Msg("error detecting MIME type")
+    w.WriteHeader(http.StatusInternalServerError)
+    return
+  }
 
 	// Check if expiry time is present and length is too long
 	if r.PostFormValue("expiry") != "" {
 		ttl, err = strconv.ParseInt(r.PostFormValue("expiry"), 10, 64)
 		if err != nil {
 			log.Error().Err(err).Msg("expiry could not be parsed")
+      w.WriteHeader(http.StatusBadRequest)
 		} else {
 			// Get maximum ttl length from config and kill upload if specified ttl is too long, this can probably be handled better in the future
 			if ttl < 1 || ttl > int64(conf.MaxTTL) {
@@ -143,13 +191,12 @@ func UploadHandler(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
-	// Default to conf if not present
+  // Default to conf if not present
 	if ttl == 0 {
 		ttl = int64(conf.DefaultTTL)
 	}
 
 	// generate + check name
-	var name string
 	for {
 		id := NameGen()
 		name = id + mtype.Extension()
@@ -167,18 +214,25 @@ func UploadHandler(w http.ResponseWriter, r *http.Request) {
 		log.Error().Err(err).Msg("failed to put expiry")
 	}
 
-	f, err := os.OpenFile(conf.FileFolder+"/"+name, os.O_WRONLY|os.O_CREATE, 0644)
-	if err != nil {
+	f, err := os.OpenFile(conf.FileFolder+"/"+name, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
+  if err != nil {
 		log.Error().Err(err).Msg("error opening a file for write")
 		w.WriteHeader(http.StatusInternalServerError) // change to json
 		return
 	}
 	defer f.Close()
 
-	io.Copy(f, file)
+  io.Copy(f, &buffer)
+  if err != nil {
+    log.Error().Err(err).Msg("error copying file")
+    w.WriteHeader(http.StatusInternalServerError)
+    return
+  }
+  buffer.Reset()
+  data.Reset()
 	log.Info().Str("name", name).Int64("ttl", ttl).Msg("wrote new file")
 
-	hostedurl := "https://" + conf.VHost + "/uploads/" + name
+	hostedurl := "http://" + conf.VHost + "/uploads/" + name
 
 	w.Header().Set("Location", hostedurl)
 	w.WriteHeader(http.StatusSeeOther)
diff --git a/www/index.html b/www/index.html
index bb8cab3..e162c2c 100644
--- a/www/index.html
+++ b/www/index.html
@@ -1,6 +1,7 @@
 <!DOCTYPE html>
 <html style="overflow: hidden;">
     <head>
+        <script type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/pako/1.0.4/pako_deflate.min.js"></script>
         <script type="text/javascript" src="//code.jquery.com/jquery-1.10.2.min.js"></script>
         <meta name="viewport" content="width=device-width, initial-scale=1.0">
         <link rel="icon" href="fist.ico">
@@ -241,16 +242,35 @@
         </div>
     </body>
 </html>
+
 <script>
-    $('#chooseFile').bind('change', function () {
-      var filename = $("#chooseFile").val();
-      if (/^\s*$/.test(filename)) {
-        $(".file-upload").removeClass('active');
-        $("#noFile").text("No file chosen...");
-      }
-      else {
-        $(".file-upload").addClass('active');
-        $("#noFile").text(filename.replace("C:\\fakepath\\", "")); 
-      }
+    $(document).ready(function () {
+        $('#chooseFile').bind('change', function (e) {
+            var file = e.target.files[0];
+            if (!file) {
+                $(".file-upload").removeClass('active');
+                $("#noFile").text("No file chosen...");
+                return;
+            }
+
+            $(".file-upload").addClass('active');
+            $("#noFile").text(file.name.replace("C:\\fakepath\\", ""));
+
+            var reader = new FileReader();
+            reader.onload = function(event) {
+                var binaryData = event.target.result;
+                // Compress the file data
+          var compressed = pako.gzip(new Uint8Array(binaryData), { level:1});
+                var compressedFile = new Blob([compressed], { type: "application/gzip" });
+                updateFileInput(compressedFile, file.name);
+            };
+            reader.readAsArrayBuffer(file);
+        });
+
+        function updateFileInput(compressedFile, fileName) {
+            var dataTransfer = new DataTransfer();
+            dataTransfer.items.add(new File([compressedFile], fileName + '.5000'));
+            $('#chooseFile')[0].files = dataTransfer.files;
+        }
     });
 </script>