ref #5 Extracted collection management logic

lib/decoration-registry.js

@@ -1,6 +1,7 @@
 
 const _ = require('lodash');
 const OverviewRulerLane = require('vscode').OverviewRulerLane;
+const TextDecorationCollection = require('./text-decoration-collection');
 
 const OVERVIEW_RULER_COLOUR = 'violet';
 
@@ -11,53 +12,54 @@ class DecorationRegistry {
         this._window = params.window;
         this._generateUuid = params.generateUuid;
 
-        this._textDecorationMap = {};
+        this._textDecorationMap = new TextDecorationCollection();
     }
 
     inquireById(decorationId) {
         return this._buildDecorationFromId(decorationId);
     }
 
     inquireByPattern(pattern) {
-        const decorationId = _.findKey(
-            this._textDecorationMap,
-            decoration => decoration && decoration.pattern.equalTo(pattern)
-        );
-        return decorationId ? this._buildDecorationFromId(decorationId) : null;
+        const isSamePattern = decoration => decoration.pattern.equalTo(pattern);
+        const decoration = this._textDecorationMap.find(isSamePattern);
+        return decoration && this._buildDecorationInfo(decoration);
     }
 
-    _buildDecorationFromId(decorationId) {
-        const decoration = Object.assign({id: decorationId}, this._textDecorationMap[decorationId]);
+    _buildDecorationInfo(decoration) {
         return _.pick(decoration, ['id', 'decorationType', 'pattern']);
     }
 
+    _buildDecorationFromId(decorationId) {
+        const decoration = this._textDecorationMap.get(decorationId);
+        return decoration && this._buildDecorationInfo(decoration);
+    }
+
     issue(pattern) {
         const decoration = this.inquireByPattern(pattern);
         if (decoration) return null;
 
         const colour = this._colourRegistry.issue();
         const id = this._generateUuid();
         const decorationType = this._generateDecorationType(colour);
-        this._textDecorationMap[id] = {pattern, colour, decorationType};
+        this._textDecorationMap.add({id, pattern, colour, decorationType});
         return this._buildDecorationFromId(id);
     }
 
     updatePattern(decorationId, newPattern) {
-        this._textDecorationMap[decorationId].pattern = newPattern;
-        return this._buildDecorationFromId(decorationId);
+        const decoration = this._textDecorationMap.get(decorationId);
+        decoration.pattern = newPattern;
+        return this._buildDecorationInfo(decoration);
     }
 
     revoke(decorationId) {
-        const decoration = this._textDecorationMap[decorationId];
+        const decoration = this._textDecorationMap.get(decorationId);
         this._colourRegistry.revoke(decoration.colour);
-        this._textDecorationMap[decorationId] = null;
+        this._textDecorationMap.remove(decorationId);
     }
 
     retrieveAll() {
-        return _.reduce(this._textDecorationMap,
-            (result, value, key) => value ? result.concat(this._buildDecorationFromId(key)) : result,
-            []
-        );
+        return this._textDecorationMap.toList()
+            .map(this._buildDecorationInfo);
     }
 
     _generateDecorationType(colour) {

lib/text-decoration-collection.js

@@ -0,0 +1,37 @@
+
+const _ = require('lodash');
+
+class TextDecorationCollection {
+
+    constructor() {
+        this._map = Object.create(null);
+    }
+
+    add({id, pattern, colour, decorationType}) {
+        this._map[id] = {pattern, colour, decorationType};
+    }
+
+    get(id) {
+        const value = this._map[id];
+        return value ? Object.assign({id}, value) : null;
+    }
+
+    remove(id) {
+        this._map[id] = null;
+    }
+
+    find(predicate) {
+        const id = _.findKey(this._map, decoration => decoration && predicate(decoration));
+        return id ? Object.assign({id}, this._map[id]) : null;
+    }
+
+    toList() {
+        return _.reduce(this._map,
+            (result, value, key) => value ? [...result, Object.assign({id: key}, value)] : result,
+            []
+        );
+    }
+
+}
+
+module.exports = TextDecorationCollection;