From 102078744e2b9be3f66adf6efc8964beb734eb8a Mon Sep 17 00:00:00 2001
From: David Ralph <me@davidjcralph.co.uk>
Date: Wed, 8 Sep 2021 20:58:07 +0100
Subject: [PATCH] feat: multiple custom backgrounds (WIP)

---
 .../main/marketplace/sections/Create.jsx      |  2 +-
 .../modals/main/scss/settings/_buttons.scss   |  5 ++
 .../sections/background/Background.jsx        | 72 ++++++++++++++++---
 .../widgets/background/Background.jsx         | 14 +++-
 4 files changed, 79 insertions(+), 14 deletions(-)

diff --git a/src/components/modals/main/marketplace/sections/Create.jsx b/src/components/modals/main/marketplace/sections/Create.jsx
index 54a7fd75..53228f02 100644
--- a/src/components/modals/main/marketplace/sections/Create.jsx
+++ b/src/components/modals/main/marketplace/sections/Create.jsx
@@ -149,7 +149,7 @@ export default class Create extends PureComponent {
         <p>{addons.create.metadata.screenshot_url}</p>
         <input type='text' value={this.state.addonMetadata.screenshot_url} onInput={(e) => setMetadata(e.target.value, 'screenshot_url')}/>
         <p>{addons.create.metadata.description}</p>
-        <textarea className='settingsTextarea' value={this.state.addonMetadata.description} onInput={(e) => setMetadata(e.target.value, 'description')}></textarea>
+        <textarea className='settingsTextarea' value={this.state.addonMetadata.description} onInput={(e) => setMetadata(e.target.value, 'description')}/>
         <br/>
         <button onClick={() => this.changeTab(1)} className='uploadbg' style={{ marginRight: '10px' }}>{welcome.buttons.previous}</button>
         <button onClick={() => this.changeTab(this.state.addonMetadata.type)} className='uploadbg' disabled={nextDescriptionDisabled}>{welcome.buttons.next}</button>
diff --git a/src/components/modals/main/scss/settings/_buttons.scss b/src/components/modals/main/scss/settings/_buttons.scss
index 65d54c53..c444d533 100644
--- a/src/components/modals/main/scss/settings/_buttons.scss
+++ b/src/components/modals/main/scss/settings/_buttons.scss
@@ -59,3 +59,8 @@
 .import {
   margin-left: 20px;
 }
+
+.round-small {
+  height: 10px !important;
+  width: 10px !important;
+}
diff --git a/src/components/modals/main/settings/sections/background/Background.jsx b/src/components/modals/main/settings/sections/background/Background.jsx
index 8a3bd192..3b9c62f5 100644
--- a/src/components/modals/main/settings/sections/background/Background.jsx
+++ b/src/components/modals/main/settings/sections/background/Background.jsx
@@ -1,4 +1,4 @@
-import { PureComponent } from 'react';
+import { PureComponent, Fragment } from 'react';
 import { toast } from 'react-toastify';
 
 import Checkbox from '../../Checkbox';
@@ -16,7 +16,7 @@ export default class BackgroundSettings extends PureComponent {
   constructor() {
     super();
     this.state = {
-      customBackground: localStorage.getItem('customBackground') || '',
+      customBackground: [''],
       backgroundType: localStorage.getItem('backgroundType') || 'api',
       backgroundCategories: [window.language.modals.main.loading]
     };
@@ -25,26 +25,54 @@ export default class BackgroundSettings extends PureComponent {
   }
 
   resetCustom = () => {
-    localStorage.setItem('customBackground', '');
+    localStorage.setItem('customBackground', '[""]');
     this.setState({
-      customBackground: ''
+      customBackground: ['']
     });
     toast(window.language.toasts.reset);
     EventBus.dispatch('refresh', 'background');
   }
 
-  customBackground(e, text) {
+  customBackground(e, text, index) {
     const result = (text === true) ? e.target.value : e.target.result;
-    localStorage.setItem('customBackground', result);
+
+    const customBackground = this.state.customBackground;
+    customBackground[index] = result;
     this.setState({
-      customBackground: result
+      customBackground
     });
+    this.forceUpdate();
+
+    localStorage.setItem('customBackground', JSON.stringify(customBackground));
     EventBus.dispatch('refresh', 'background');
   }
 
-  videoCustomSettings = () => {
+  addCustomBackground() {
     const customBackground = this.state.customBackground;
+    customBackground.push('');
 
+    this.setState({
+      customBackground
+    });
+    this.forceUpdate();
+
+    localStorage.setItem('customBackground', JSON.stringify(customBackground));
+  }
+
+  removeCustomBackground(index) { 
+    const customBackground = this.state.customBackground;
+    customBackground.splice(index, 1);
+
+    this.setState({
+      customBackground
+    });
+    this.forceUpdate();
+
+    localStorage.setItem('customBackground', JSON.stringify(customBackground));
+  }
+
+  videoCustomSettings = (index) => {
+    const customBackground = this.state.customBackground[index];
     if (customBackground.startsWith('data:video/') || customBackground.endsWith('.mp4') || customBackground.endsWith('.webm') || customBackground.endsWith('.ogg')) { 
       return (
         <>
@@ -63,6 +91,17 @@ export default class BackgroundSettings extends PureComponent {
     }
   }
 
+  getCustom() {
+    let data;
+    try {
+      data = JSON.parse(localStorage.getItem('customBackground'));
+    } catch (e) {
+      data = [localStorage.getItem('customBackground')];
+    }
+
+    return data;
+  }
+
   async getBackgroundCategories() {
     const data = await (await fetch(window.constants.API_URL + '/images/categories', { signal: this.controller.signal })).json();
 
@@ -76,6 +115,10 @@ export default class BackgroundSettings extends PureComponent {
   }
 
   componentDidMount() {
+    this.setState({
+      customBackground: this.getCustom()
+    });
+
     if (navigator.onLine === false || localStorage.getItem('offlineMode') === 'true') {
       return this.setState({
         backgroundCategories: [window.language.modals.update.offline.title]
@@ -144,11 +187,18 @@ export default class BackgroundSettings extends PureComponent {
       <>
         <ul>
           <p>{background.source.custom_background} <span className='modalLink' onClick={this.resetCustom}>{this.language.buttons.reset}</span></p>
-          <input type='text' value={this.state.customBackground} onChange={(e) => this.customBackground(e, true)}></input>
-          <span className='modalLink' onClick={() => document.getElementById('bg-input').click()}>{background.source.upload}</span>
+          {this.state.customBackground.map((_url, index) => (
+            <Fragment key={index}>
+              {this.state.customBackground.length > 1 ? <button className='reset round round-small' onClick={() => this.removeCustomBackground(index)}>x</button> : null}
+              <input type='text' value={this.state.customBackground[index]} onChange={(e) => this.customBackground(e, true, index)}></input>
+              <span className='modalLink' onClick={() => document.getElementById('bg-input').click(index)}>{background.source.upload}</span>
+              {this.videoCustomSettings(index)}
+              <br/><br/>  
+            </Fragment>
+          ))}
+          <button className='uploadbg' onClick={() => this.addCustomBackground()}>Add</button>
           <FileUpload id='bg-input' accept='image/jpeg, image/png, image/webp, image/webm, image/gif, video/mp4, video/webm, video/ogg' loadFunction={(e) => this.customBackground(e)} />
         </ul>
-        {this.videoCustomSettings()}
       </>
     );
 
diff --git a/src/components/widgets/background/Background.jsx b/src/components/widgets/background/Background.jsx
index 428ee827..34f054ff 100644
--- a/src/components/widgets/background/Background.jsx
+++ b/src/components/widgets/background/Background.jsx
@@ -189,14 +189,24 @@ export default class Background extends PureComponent {
       break;
 
       case 'custom':
-        const customBackground = localStorage.getItem('customBackground');
+        let customBackground;
+        try {
+          customBackground = JSON.parse(localStorage.getItem('customBackground'));
+        } catch (e) {
+          // move to new format
+          customBackground = [localStorage.getItem('customBackground')];
+          localStorage.setItem('customBackground', JSON.stringify(customBackground));
+        }
+
+        // pick random
+        customBackground = customBackground[Math.floor(Math.random() * customBackground.length)];
 
         // allow users to use offline images
         if (offline && !customBackground.startsWith('data:')) {
           return this.setState(offlineBackground());
         }
 
-        if (customBackground !== '' && customBackground !== 'undefined') {
+        if (customBackground !== '' && customBackground !== 'undefined' && customBackground !== ['']) {
           this.setState({
             url: customBackground,
             type: 'custom',