From c93835eb0cb68ae0a96f78d65af3aaa651d1b0cb Mon Sep 17 00:00:00 2001
From: Sheetal Umesh Kumar <sheetaluk@hypothes.is>
Date: Wed, 25 Jan 2017 14:59:46 -0500
Subject: [PATCH] Show annotation count on the page, so that users don't have
 to open to sidebar to know how many annotations are in the page.

Fixes: https://github.com/hypothesis/product-backlog/issues/129
---
 scripts/gulp/live-reload-server.js           |  6 +-
 src/annotator/annotation-counts.js           | 26 ++++++++
 src/annotator/sidebar.coffee                 |  6 +-
 src/annotator/test/annotation-counts-test.js | 67 ++++++++++++++++++++
 src/shared/bridge-events.js                  | 10 +++
 src/sidebar/annotation-metadata.js           | 19 ++++++
 src/sidebar/frame-sync.js                    | 12 ++++
 src/sidebar/test/annotation-fixtures.js      | 20 ++++++
 src/sidebar/test/annotation-metadata-test.js | 22 +++++++
 src/sidebar/test/frame-sync-test.js          |  7 ++
 10 files changed, 193 insertions(+), 2 deletions(-)
 create mode 100644 src/annotator/annotation-counts.js
 create mode 100644 src/annotator/test/annotation-counts-test.js
 create mode 100644 src/shared/bridge-events.js
diff --git a/scripts/gulp/live-reload-server.js b/scripts/gulp/live-reload-server.js
index 7ae6c0a4264..03e41923ab2 100644
--- a/scripts/gulp/live-reload-server.js
+++ b/scripts/gulp/live-reload-server.js
@@ -36,7 +36,11 @@ function LiveReloadServer(port, appServer) {
       <title>Hypothesis Client Test</title>
     </head>
     <body>
-      <pre style="margin: 75px;">${changelogText()}</pre>
+      <div data-hypothesis-trigger style="margin: 75px 0 0 75px;">
+        Number of annotations:
+        <span data-hypothesis-annotation-count>...</span>
+      </div>
+      <pre style="margin: 20px 75px 75px 75px;">${changelogText()}</pre>
       <script>
       var appHost = document.location.hostname;
 
diff --git a/src/annotator/annotation-counts.js b/src/annotator/annotation-counts.js
new file mode 100644
index 00000000000..2103a3b4163
--- /dev/null
+++ b/src/annotator/annotation-counts.js
@@ -0,0 +1,26 @@
+'use strict';
+
+var events = require('../shared/bridge-events');
+
+var ANNOTATION_COUNT_ATTR = 'data-hypothesis-annotation-count';
+
+/**
+ * Update the elements in the container element with the count data attribute
+ * with the new annotation count.
+ *
+ * @param {Element} rootEl - The DOM element which contains the elements that
+ * display annotation count.
+ */
+
+function annotationCounts(rootEl, crossframe) {
+  crossframe.on(events.PUBLIC_ANNOTATION_COUNT_CHANGED, updateAnnotationCountElems);
+
+  function updateAnnotationCountElems(newCount) {
+    var elems = rootEl.querySelectorAll('['+ANNOTATION_COUNT_ATTR+']');
+    Array.from(elems).forEach(function(elem) {
+      elem.textContent = newCount;
+    });
+  }
+}
+
+module.exports = annotationCounts;
diff --git a/src/annotator/sidebar.coffee b/src/annotator/sidebar.coffee
index ab74b221e17..623dcefda89 100644
--- a/src/annotator/sidebar.coffee
+++ b/src/annotator/sidebar.coffee
@@ -3,6 +3,7 @@ raf = require('raf')
 Hammer = require('hammerjs')
 
 Host = require('./host')
+annotationCounts = require('./annotation-counts')
 sidebarTrigger = require('./sidebar-trigger')
 
 # Minimum width to which the frame can be resized.
@@ -37,13 +38,16 @@ module.exports = class Sidebar extends Host
     this._setupSidebarEvents()
 
   _setupDocumentEvents: ->
-    sidebarTrigger(document, @show.bind(this))
+    sidebarTrigger(document.body, @show.bind(this))
+
     @element.on 'click', (event) =>
       if !@selectedTargets?.length
         this.hide()
     return this
 
   _setupSidebarEvents: ->
+    annotationCounts(document.body, @crossframe)
+
     @crossframe.on('show', this.show.bind(this))
     @crossframe.on('hide', this.hide.bind(this))
 
diff --git a/src/annotator/test/annotation-counts-test.js b/src/annotator/test/annotation-counts-test.js
new file mode 100644
index 00000000000..0102ab3b89d
--- /dev/null
+++ b/src/annotator/test/annotation-counts-test.js
@@ -0,0 +1,67 @@
+'use strict';
+
+var annotationCounts = require('../annotation-counts');
+
+describe('annotationCounts', function () {
+  var countEl1;
+  var countEl2;
+  var CrossFrame;
+  var fakeCrossFrame;
+  var sandbox;
+
+  beforeEach(function () {
+    CrossFrame = null;
+    fakeCrossFrame = {};
+    sandbox = sinon.sandbox.create();
+
+    countEl1 = document.createElement('button');
+    countEl1.setAttribute('data-hypothesis-annotation-count');
+    document.body.appendChild(countEl1);
+
+    countEl2 = document.createElement('button');
+    countEl2.setAttribute('data-hypothesis-annotation-count');
+    document.body.appendChild(countEl2);
+
+    fakeCrossFrame.on = sandbox.stub().returns(fakeCrossFrame);
+    
+    CrossFrame = sandbox.stub();
+    CrossFrame.returns(fakeCrossFrame);
+  });
+
+  afterEach(function () {
+    sandbox.restore();
+    countEl1.remove();
+    countEl2.remove();
+  });
+
+  describe('listen for "publicAnnotationCountChanged" event', function () {
+    var emitEvent = function () {
+      var crossFrameArgs;
+      var evt;
+      var fn;
+
+      var event = arguments[0];
+      var args = 2 <= arguments.length ? Array.prototype.slice.call(arguments, 1) : [];
+
+      crossFrameArgs = fakeCrossFrame.on.args;
+      for (var i = 0, len = crossFrameArgs.length; i < len; i++) {
+        evt = crossFrameArgs[i][0];
+        fn = crossFrameArgs[i][1];
+
+        if (event === evt) {
+          fn.apply(null, args);
+        }
+      }
+    };
+
+    it('displays the updated annotation count on the appropriate elements', function () {
+      var newCount = 10;
+      annotationCounts(document.body, fakeCrossFrame);
+
+      emitEvent('publicAnnotationCountChanged', newCount);
+
+      assert.equal(countEl1.textContent, newCount);
+      assert.equal(countEl2.textContent, newCount);
+    });
+  });
+});
diff --git a/src/shared/bridge-events.js b/src/shared/bridge-events.js
new file mode 100644
index 00000000000..3cca12373d9
--- /dev/null
+++ b/src/shared/bridge-events.js
@@ -0,0 +1,10 @@
+'use strict';
+
+/**
+ * This module defines the set of global events that are dispatched
+ * across the bridge between the sidebar and annotator
+ */
+module.exports = {
+  /** The set of annotations was updated. */
+  PUBLIC_ANNOTATION_COUNT_CHANGED: 'publicAnnotationCountChanged',
+};
diff --git a/src/sidebar/annotation-metadata.js b/src/sidebar/annotation-metadata.js
index 58c92af1a61..a6b7b1f0346 100644
--- a/src/sidebar/annotation-metadata.js
+++ b/src/sidebar/annotation-metadata.js
@@ -101,6 +101,24 @@ function isNew(annotation) {
   return !annotation.id;
 }
 
+/** Return `true` if the given annotation is public, `false` otherwise. */
+function isPublic(annotation) {
+  var isPublic = false;
+
+  if (!annotation.permissions) {
+    return isPublic;
+  }
+
+  annotation.permissions.read.forEach(function(perm) {
+    var readPermArr = perm.split(':');
+    if (readPermArr.length === 2 && readPermArr[0] === 'group') {
+      isPublic = true;
+    }
+  });
+
+  return isPublic;
+}
+
 /**
  * Return `true` if `annotation` has a selector.
  *
@@ -169,6 +187,7 @@ module.exports = {
   isNew: isNew,
   isOrphan: isOrphan,
   isPageNote: isPageNote,
+  isPublic: isPublic,
   isReply: isReply,
   isWaitingToAnchor: isWaitingToAnchor,
   location: location,
diff --git a/src/sidebar/frame-sync.js b/src/sidebar/frame-sync.js
index 15e570e55d4..4f9a29c8bcd 100644
--- a/src/sidebar/frame-sync.js
+++ b/src/sidebar/frame-sync.js
@@ -1,6 +1,7 @@
 'use strict';
 
 var events = require('./events');
+var bridgeEvents = require('../shared/bridge-events');
 var metadata = require('./annotation-metadata');
 var uiConstants = require('./ui-constants');
 
@@ -50,6 +51,7 @@ function FrameSync($rootScope, $window, Discovery, annotationUI, bridge) {
   function setupSyncToFrame() {
     // List of loaded annotations in previous state
     var prevAnnotations = [];
+    var prevPublicAnns = 0;
 
     annotationUI.subscribe(function () {
       var state = annotationUI.getState();
@@ -57,6 +59,7 @@ function FrameSync($rootScope, $window, Discovery, annotationUI, bridge) {
         return;
       }
 
+      var publicAnns = 0;
       var inSidebar = new Set();
       var added = [];
 
@@ -66,6 +69,10 @@ function FrameSync($rootScope, $window, Discovery, annotationUI, bridge) {
           return;
         }
 
+        if (metadata.isPublic(annot)) {
+          ++publicAnns;
+        }
+
         inSidebar.add(annot.$tag);
         if (!inFrame.has(annot.$tag)) {
           added.push(annot);
@@ -89,6 +96,11 @@ function FrameSync($rootScope, $window, Discovery, annotationUI, bridge) {
         bridge.call('deleteAnnotation', formatAnnot(annot));
         inFrame.delete(annot.$tag);
       });
+
+      if (publicAnns !== prevPublicAnns) {
+        bridge.call(bridgeEvents.PUBLIC_ANNOTATION_COUNT_CHANGED, publicAnns);
+        prevPublicAnns = publicAnns;
+      }
     });
   }
 
diff --git a/src/sidebar/test/annotation-fixtures.js b/src/sidebar/test/annotation-fixtures.js
index 922a15de74b..162165c359c 100644
--- a/src/sidebar/test/annotation-fixtures.js
+++ b/src/sidebar/test/annotation-fixtures.js
@@ -16,6 +16,25 @@ function defaultAnnotation() {
   };
 }
 
+/**
+ * Return a fake public annotation with the basic properties filled in.
+ */
+function publicAnnotation() {
+  return {
+    id: 'pubann',
+    document: {
+      title: 'A special document',
+    },
+    permissions: {
+      read:['group:__world__'],
+    },
+    target: [{source: 'source', 'selector': []}],
+    uri: 'http://example.com',
+    user: 'acct:bill@localhost',
+    updated: '2015-05-10T20:18:56.613388+00:00',
+  };
+}
+
 /** Return an annotation domain model object for a new annotation
  * (newly-created client-side, not yet saved to the server).
  */
@@ -120,6 +139,7 @@ function oldReply() {
 
 module.exports = {
   defaultAnnotation: defaultAnnotation,
+  publicAnnotation: publicAnnotation,
   newAnnotation: newAnnotation,
   newEmptyAnnotation: newEmptyAnnotation,
   newHighlight: newHighlight,
diff --git a/src/sidebar/test/annotation-metadata-test.js b/src/sidebar/test/annotation-metadata-test.js
index 04ba3493b81..9d6a92e6315 100644
--- a/src/sidebar/test/annotation-metadata-test.js
+++ b/src/sidebar/test/annotation-metadata-test.js
@@ -3,6 +3,8 @@
 var annotationMetadata = require('../annotation-metadata');
 var fixtures = require('./annotation-fixtures');
 
+var unroll = require('../../shared/test/util').unroll;
+
 var documentMetadata = annotationMetadata.documentMetadata;
 var domainAndTitle = annotationMetadata.domainAndTitle;
 
@@ -247,6 +249,26 @@ describe('annotation-metadata', function () {
     });
   });
 
+  describe('.isPublic', function () {
+    it('returns true if an annotation is shared within a group', function () {
+      assert.isTrue(annotationMetadata.isPublic(fixtures.publicAnnotation()));
+    });
+
+    unroll('returns false if an annotation is not publicly readable', function (testCase) {
+      var annotation = Object.assign(fixtures.defaultAnnotation(), {permissions: testCase});
+      assert.isFalse(annotationMetadata.isPublic(annotation));
+    }, [{
+      read:['acct:someemail@localhost'],
+    }, {
+      read:['something invalid'],
+    }]);
+
+    it('returns false if an annotation is missing permissions', function () {
+      var annotation = Object.assign(fixtures.defaultAnnotation());
+      assert.isFalse(annotationMetadata.isPublic(annotation));
+    });
+  });
+
   describe('.isOrphan', function () {
     it('returns true if an annotation failed to anchor', function () {
       var annotation = Object.assign(fixtures.defaultAnnotation(), {$orphan: true});
diff --git a/src/sidebar/test/frame-sync-test.js b/src/sidebar/test/frame-sync-test.js
index afaec176b53..d9b61e26a98 100644
--- a/src/sidebar/test/frame-sync-test.js
+++ b/src/sidebar/test/frame-sync-test.js
@@ -121,6 +121,13 @@ describe('FrameSync', function () {
     });
   });
 
+  context('when annotation count has changed', function () {
+    it('sends a "publicAnnotationCountChanged" message to the frame', function () {
+      fakeAnnotationUI.setState({annotations: [annotationFixtures.publicAnnotation()]});
+      assert.calledWithMatch(fakeBridge.call, 'publicAnnotationCountChanged', sinon.match(1));
+    });
+  });
+
   context('when annotations are removed from the sidebar', function () {
     it('sends a "deleteAnnotation" message to the frame', function () {
       fakeAnnotationUI.setState({annotations: [fixtures.ann]});
