From ac1bd586c9ed0ce72ed3cbacf53259a9e9b2e1d8 Mon Sep 17 00:00:00 2001
From: Xodrium <118943715+malloc62@users.noreply.github.com>
Date: Fri, 10 Feb 2023 23:44:25 -0500
Subject: [PATCH] File uploading stuff

---
 src/lib/components/FileUpload.svelte | 44 ++++++++++++++++++++++------
 src/lib/db/db.js                     | 37 +++++++++++++++++------
 2 files changed, 63 insertions(+), 18 deletions(-)

diff --git a/src/lib/components/FileUpload.svelte b/src/lib/components/FileUpload.svelte
index 568613b..9a8d5e8 100644
--- a/src/lib/components/FileUpload.svelte
+++ b/src/lib/components/FileUpload.svelte
@@ -7,6 +7,8 @@
     let files;
     let preview;
 
+    let progress = 0;
+
     function getBase64(image) {
         const reader = new FileReader();
         reader.readAsDataURL(image);
@@ -15,19 +17,40 @@
             uploadFunction(e.target.result);
         };
     };
+    async function uploadFunction(imageD) {
+        const imgData = imageD.split(',').pop();
 
-    async function uploadFunction(imgBase64) {
-        const imgData = imgBase64.split(',').pop();
+        var images = [];
 
-        var fData = (new FormData());
+        var fSize = 1024 * 256;
 
-        fData.append('img',imgData);
-        fData.append('extension',fileInput.value.split('.').pop());
+        for (var i = 0; i * fSize < imgData.length + 1; i++) {
+            images.push(imgData.substring(i * fSize, i * fSize + fSize));
+        }
+
+        var rId = (Math.random() + 1).toString(36).substring(7);;
+        var extension = fileInput.value.split('.').pop();
+
+        for (var i = 0; i < images.length; i++) {
+            var fData = (new FormData());
+
+            var image = images[i];
+            
+            fData.append('img',image);
+            fData.append('extension',extension);
+            fData.append('last',i == (images.length - 1));
+            fData.append('id',rId);
 
-        form = await fetch(`/api/fileCreate`, {
             method: 'POST',
-            body: fData
-        }).then(x => x.json());
+            form = await fetch(`/api/fileCreate`, {
+                method: 'POST',
+                body: fData,
+            }).then(x => x.json());
+
+            progress = (i+1) / (images.length);
+
+            await setTimeout(() => new Promise.resolve(),100);
+        }
     };
 </script>
 
@@ -42,13 +65,16 @@
     }
 </style>
 
-<form action='/api/postCreate' method='POST' >
+<form action='#' method='GET' >
     {#if preview}
         <img src={preview} alt="Image preview"/>
     {:else}
         <img src='/icon_sanifae.svg' alt="Image preview"/>
     {/if}
     <input class="hidden" id="file-to-upload" type="file" bind:files bind:this={fileInput} on:change={() => getBase64(files[0])}/>
+    <p>
+        Progress: {progress * 100}%
+    </p>
     <p> 
         <Button class="upload-btn" clickFunc={ () => fileInput.click() }>Upload</Button>
     </p>
diff --git a/src/lib/db/db.js b/src/lib/db/db.js
index 407a757..43a1708 100644
--- a/src/lib/db/db.js
+++ b/src/lib/db/db.js
@@ -6,7 +6,9 @@ const AUTH_ACTIONS = [
     'vote'
 ];
 
-const fileSizeLimit = 1024*1024*5;
+const fileSizeLimit = 1024*1024*16;
+
+var ridArray = {};
 
 import sqlite3 from 'sqlite3'
 import { open } from 'sqlite'
@@ -132,6 +134,8 @@ backend.login = async ({user, pass, cookies}) => {
 }
 
 backend.postCreate = async ({content, user}) => {
+    if (!content) return {'success': 'No post provided.'}
+
     var lengthCheck = checkLength(content,'Post content',1,10240);
 
     if (lengthCheck)
@@ -247,20 +251,35 @@ backend.token = async ({cookies}) => {
     return {data: existingAccounts[0].username};
 }
 
-backend.fileCreate = async({img, extension}) => {
-    const imgHash = createHash('md5').update(img).digest('hex');
+backend.fileCreate = async({img, extension,id, last }) => {
+    if (ridArray[id] !== '' && !(ridArray[id])) {
+        ridArray[id] = img;
+    } else {
+        ridArray[id] += img;
+    }
 
-    let lengthCheck = checkLength(img,'Image',fileSizeLimit);
+    const imgData = ridArray[id];
 
-    if (lengthCheck)
-        return lengthCheck;
+    if (last != 'true') {
+        return {'success': 'Image still proccessing...'}
+    } else {
+        ridArray[id] = false;
+    }
 
-    const extensionSafe = extension.replace(/(\s+)/g, '\\$1');
+    const imgHash = createHash('md5').update(imgData).digest('hex');
 
-    if (extensionSafe != 'png' && extensionSafe != 'jpg' && extensionSafe != 'svg' || extensionSafe != 'gif')
+    if (!imgHash)
+        return {'success': 'Image not provided.'}
+
+    if (imgHash.length > fileSizeLimit)
+        return {'success': 'Image too big.'}
+
+    const extensionSafe = extension.replace(/[^a-zA-Z]+/g, '\\$1');
+
+    if (extensionSafe != 'png' && extensionSafe != 'jpg' && extensionSafe != 'svg' && extensionSafe != 'gif')
         return { success: 'Illegal file extension.' };
 
-    writeFile(`${process.cwd()}/db/post-${imgHash}.${extensionSafe}`,img,{encoding: 'base64'});
+    writeFile(`${process.cwd()}/db/post-${imgHash}.${extensionSafe}`,imgData,{encoding: 'base64'});
 
     return { success: 'Successfully uploaded file.', 'href': `/img/${imgHash}.${extensionSafe}`};
 }