Prettier category names

capsule-prototype/js/online-script.js

@@ -855,7 +855,7 @@ function openMosaic() {
     container.append(filterdiv);
     // TODO is there a better way to iterate over tags or documents ?
     for ( let [k, v] of  Object.entries(rekall.sortings.colors.categories)) {
-        let categoryName = v.categoryRaw;
+        let categoryName = rekall.sortings.colors.getCategoryName(k)
         let category = $('<div/>').addClass('mosaic_category').css('background', getTagGradientColor(v));
 
         let grid = $('<div/>').addClass('mosaic_category_grid');

capsule-prototype/js/rekall/Sorting.js

@@ -336,3 +336,13 @@ Sorting.prototype.getCategory = function(tag) {
 	var metadata = this.parseMeta(tag.getMetadata(this.metadataKey));
 	return this.categories[metadata];
 }
+
+Sorting.prototype.getCategoryName = function(metadata) {
+    if ('_rekall/link' === metadata) {
+        return 'Link';
+    }
+    if ('_rekall/marker' === metadata) {
+        return 'Note';
+    }
+    return metadata.substr(7).toUpperCase();
+}