Add set-local-storage-item/trusted-set-local-storage-item scriptlets

Related discussion: - uBlockOrigin/uBlock-issues#2697
gorhill · Jun 20, 2023 · 9443ba8 · 9443ba8
1 parent db9662f
commit 9443ba8
Showing 1 changed file with 92 additions and 0 deletions.
diff --git a/assets/resources/scriptlets.js b/assets/resources/scriptlets.js
@@ -2765,6 +2765,61 @@ function setCookie(
  );
 }
 
+/*******************************************************************************
+ * 
+ * set-local-storage-item.js
+ * 
+ * Set a local storage entry to a specific, allowed value.
+ * 
+ * Reference:
+ * https://github.com/AdguardTeam/Scriptlets/blob/master/src/scriptlets/set-local-storage-item.js
+ * 
+ **/
+
+builtinScriptlets.push({
+ name: 'set-local-storage-item.js',
+ fn: setLocalStorageItem,
+ world: 'ISOLATED',
+});
+function setLocalStorageItem(
+ key = '',
+ value = ''
+) {
+ if ( key === '' ) { return; }
+ if ( value === '' ) { return; }
+
+ let actualValue;
+ if ( value === 'undefined' ) {
+ actualValue = undefined;
+ } else if ( value === 'false' ) {
+ actualValue = false;
+ } else if ( value === 'true' ) {
+ actualValue = true;
+ } else if ( value === 'null' ) {
+ actualValue = null;
+ } else if ( value === '{}' ) {
+ actualValue = '{}';
+ } else if ( value === '[]' ) {
+ actualValue = '[]';
+ } else if ( value === "''" ) {
+ actualValue = '';
+ } else if ( value === 'yes' ) {
+ actualValue = 'yes';
+ } else if ( value === 'no' ) {
+ actualValue = 'no';
+ } else if ( /^\d+$/.test(value) ) {
+ actualValue = parseInt(value, 10);
+ if ( actualValue > 32767 ) { return; }
+ } else {
+ return;
+ }
+
+ try {
+ self.localStorage.setItem(key, `${actualValue}`);
+ } catch(ex) {
+ }
+}
+
 /*******************************************************************************
  * 
  * Scriplets below this section are only available for filter lists from
@@ -2903,4 +2958,41 @@ function trustedSetCookie(
  );
 }
 
+/*******************************************************************************
+ * 
+ * trusted-set-local-storage-item.js
+ * 
+ * Set a local storage entry to an arbitrary value.
+ * 
+ * Reference:
+ * https://github.com/AdguardTeam/Scriptlets/blob/master/src/scriptlets/trusted-set-local-storage-item.js
+ * 
+ **/
+
+builtinScriptlets.push({
+ name: 'trusted-set-local-storage-item.js',
+ requiresTrust: true,
+ fn: trustedSetLocalStorageItem,
+ world: 'ISOLATED',
+});
+function trustedSetLocalStorageItem(
+ key = '',
+ value = ''
+) {
+ if ( key === '' ) { return; }
+ if ( value === '' ) { return; }
+
+ let actualValue = value;
+ if ( value === '$now$' ) {
+ actualValue = Date.now();
+ } else if ( value === '$currentDate$' ) {
+ actualValue = `${Date()}`;
+ }
+
+ try {
+ self.localStorage.setItem(key, `${actualValue}`);
+ } catch(ex) {
+ }
+}
+
 /******************************************************************************/