Fix no storage events

Signed-off-by: Simon Bennetts <[email protected]>

CHANGELOG.md

@@ -4,3 +4,6 @@ All notable changes to this add-on will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
 ## Unreleased
+
+### Fixed
+- Storage events not being reported.

source/Background/index.ts

@@ -196,7 +196,7 @@ async function handleMessage(
     case REPORT_OBJECT: {
       const repObj = JSON.parse(request.data);
       if (repObj.type === LOCAL_STORAGE || repObj.type === SESSION_STORAGE) {
-        const repStorage = new ReportedStorage('', '', '', '', '');
+        const repStorage = new ReportedStorage('', '', '', '', '', '');
         Object.assign(repStorage, repObj);
         const repStorStr = repStorage.toShortString();
         if (reportedStorage.has(repStorStr)) {

source/ContentScript/index.ts

@@ -47,8 +47,9 @@ function reportStorage(
   storage: Storage,
   fn: (re: ReportedStorage) => void
 ): void {
+  const url = window.location.href;
   for (const key of Object.keys(storage)) {
-    fn(new ReportedStorage(name, '', key, '', storage.getItem(key)));
+    fn(new ReportedStorage(name, '', key, '', storage.getItem(key), url));
   }
 }
 
@@ -131,19 +132,21 @@ function reportPageForms(
   doc: Document,
   fn: (re: ReportedObject) => void
 ): void {
+  const url = window.location.href;
   Array.prototype.forEach.call(doc.forms, (form: HTMLFormElement) => {
-    fn(new ReportedElement(form));
+    fn(new ReportedElement(form, url));
   });
 }
 
 function reportPageLinks(
   doc: Document,
   fn: (re: ReportedObject) => void
 ): void {
+  const url = window.location.href;
   Array.prototype.forEach.call(
     doc.links,
     (link: HTMLAnchorElement | HTMLAreaElement) => {
-      fn(new ReportedElement(link));
+      fn(new ReportedElement(link, url));
     }
   );
 }
@@ -152,8 +155,9 @@ function reportElements(
   collection: HTMLCollection,
   fn: (re: ReportedObject) => void
 ): void {
+  const url = window.location.href;
   Array.prototype.forEach.call(collection, (element: Element) => {
-    fn(new ReportedElement(element));
+    fn(new ReportedElement(element, url));
   });
 }
 

source/types/ReportedModel.ts

@@ -42,7 +42,7 @@ class ReportedObject {
     id: string,
     nodeName: string,
     text: string | null,
-    url: string = window.location.href
+    url: string
   ) {
     this.timestamp = Date.now();
     this.type = type;
@@ -96,13 +96,14 @@ class ReportedStorage extends ReportedObject {
 }
 
 class ReportedElement extends ReportedObject {
-  public constructor(element: Element) {
+  public constructor(element: Element, url: string) {
     super(
       'nodeAdded',
       element.tagName,
       element.id,
       element.nodeName,
-      element.textContent
+      element.textContent,
+      url
     );
     if (element.tagName === 'A') {
       // This gets the full URL rather than a relative one

test/ContentScript/unitTests.test.ts

@@ -47,7 +47,8 @@ test('ReportedObject toString as expected', () => {
     'b',
     'c',
     'd',
-    'e'
+    'e',
+    'http://localhost/'
   );
 
   // Then
@@ -59,7 +60,10 @@ test('ReportedObject toString as expected', () => {
 test('ReportedElement P toString as expected', () => {
   // Given / When
   const el: Element = document.createElement('p');
-  const ro: src.ReportedElement = new src.ReportedElement(el);
+  const ro: src.ReportedElement = new src.ReportedElement(
+    el,
+    'http://localhost/'
+  );
 
   // Then
   expect(ro.toNonTimestampString()).toBe(
@@ -73,7 +77,10 @@ test('ReportedElement A toString as expected', () => {
   const linkText = document.createTextNode('Title');
   a.appendChild(linkText);
   a.setAttribute('href', 'https://example.com');
-  const ro: src.ReportedElement = new src.ReportedElement(a);
+  const ro: src.ReportedElement = new src.ReportedElement(
+    a,
+    'http://localhost/'
+  );
 
   // Then
   expect(ro.toNonTimestampString()).toBe(