Improved censorship plugin

plugins/Censorship/index.js

@@ -34,6 +34,18 @@ module.exports = {
       description: 'Select custom replacement style to use this',
       default: 'uwu',
     },
+    keepFirstLetter: {
+      type: 'boolean',
+      label: 'Keep first letter',
+      description: 'Show the first letter of a censored word instead of masking it',
+      default: false,
+    },
+    keepLastLetter: {
+      type: 'boolean',
+      label: 'Keep last letter',
+      description: 'Show the last letter of a censored word instead of masking it',
+      default: false,
+    },
   },
 
   renderer: fs.readFileSync(path.join(__dirname, 'renderer.js'), 'utf8'),

plugins/Censorship/renderer.js

@@ -30,6 +30,8 @@
   var matcher = null;
   var style = 'stars';
   var replacement = 'uwu';
+  var keepFirst = false;
+  var keepLast = false;
   var applying = false;
 
   function settings() {
@@ -58,20 +60,32 @@
 
   function termPattern(term) {
     var pattern = term.split(/\s+/).map(esc).join('\\s+');
-    if (isWordChar(term[0])) pattern = '(?<![\\p{L}\\p{N}_])' + pattern;
-    if (isWordChar(term[term.length - 1])) pattern += '(?![\\p{L}\\p{N}_])';
+    var chars = Array.from(term);
+    if (isWordChar(chars[0])) pattern = '(?<![\\p{L}\\p{N}_])' + pattern;
+    if (isWordChar(chars[chars.length - 1])) pattern += '(?![\\p{L}\\p{N}_])';
     return pattern;
   }
 
   var compiledFor = null;
   function compile() {
     var cfg = settings();
-    var sig = String(cfg.terms) + '\u0000' + String(cfg.style) + '\u0000' + String(cfg.replacement);
+    var sig =
+      String(cfg.terms) +
+      '\u0000' +
+      String(cfg.style) +
+      '\u0000' +
+      String(cfg.replacement) +
+      '\u0000' +
+      String(cfg.keepFirstLetter) +
+      '\u0000' +
+      String(cfg.keepLastLetter);
     if (sig === compiledFor) return;
     compiledFor = sig;
     var terms = splitTerms(cfg.terms);
     style = String(cfg.style);
     replacement = String(cfg.replacement) || 'uwu';
+    keepFirst = !!cfg.keepFirstLetter;
+    keepLast = !!cfg.keepLastLetter;
 
     if (!terms.length) {
       matcher = null;
@@ -86,7 +100,23 @@
   }
 
   function repeated(match, char) {
-    return match.replace(/\S/g, char);
+    var chars = Array.from(match);
+    var first = -1;
+    var last = -1;
+    for (var i = 0; i < chars.length; i++) {
+      if (/\S/.test(chars[i])) {
+        if (first === -1) first = i;
+        last = i;
+      }
+    }
+    return chars
+      .map(function (c, index) {
+        if (!/\S/.test(c)) return c;
+        if (keepFirst && index === first) return c;
+        if (keepLast && index === last) return c;
+        return char;
+      })
+      .join('');
   }
 
   function mask(match) {