tons of new avatars incl. random mode! big refactor & cleanup, part 1

Author: edfletcher

Dockerfile

@@ -1,7 +1,7 @@
 FROM node:18 as base
 WORKDIR /app/drc
 RUN apt update
-RUN apt -y install nmap
+RUN apt -y install nmap figlet
 RUN useradd -u 1001 -U -p discordrc drc
 COPY package*.json ./
 RUN npm install

config/default.js

@@ -69,6 +69,7 @@ const _config = {
       // options are: html, json, jsonl
       localQueryOutputFormat: 'html'
     },
+    avatarGenerator: 'robohash',
     allowedSpeakers: [],
     allowedSpeakersRoleId: null,
     // options are
@@ -150,8 +151,8 @@ const _config = {
   },
 
   figletBanners: {
-    enabled: false,
-    cacheDir: '.banners.cache',
+    enabled: true,
+    cacheDir: '/tmp/.discordrc.banners.cache',
     font: 'small'
   },
 
@@ -174,13 +175,6 @@ const _config = {
     rootRedirectUrl: 'https://discordrc.com'
   },
 
-  capture: {
-    enabled: true,
-    autoCaptureWindowMins: 5,
-    defaultCaptureWindowMins: 15,
-    cleanupLoopFreqSeconds: 17
-  },
-
   ipinfo: {
     token: null
   },

discord.js

@@ -224,24 +224,6 @@ setTimeout(alivenessCheck, 15 * 1000);
 
 const ignoreSquelched = [];
 
-const captureSpecs = {};
-
-if (config.capture.enabled) {
-  console.log(`Capture enabled: running cleanup loop on ${config.capture.cleanupLoopFreqSeconds} second frequency`);
-  setInterval(() => {
-    const nowNum = Number(new Date());
-    Object.entries(captureSpecs).forEach(([network, netSpec]) => {
-      Object.entries(netSpec).forEach(([channelId, chanSpec]) => {
-        if (chanSpec && chanSpec.exp > (nowNum / 100) && nowNum > chanSpec.exp) {
-          console.log(`Capture spec cleanup loop expiring ${network}:${channelId}`, chanSpec);
-          sendToBotChan(`\`SYSTEM\` Expiring channel capture for <#${channelId}> on \`${network}\` having captured ${chanSpec.captured} messages.`);
-          delete captureSpecs[network][channelId];
-        }
-      });
-    });
-  }, config.capture.cleanupLoopFreqSeconds * 1000);
-}
-
 setDefaultFont(config.figletBanners.font);
 
 console.log(`${PREFIX} Discord controller starting...`);
@@ -259,7 +241,11 @@ client.once('ready', async () => {
   client.user.setActivity('the IRC daemon...', { type: 'LISTENING' });
 
   const onExit = async (s) => {
-    sendToBotChan('\n```\n' + (await banner(s)) + '\n```\n');
+    sendToBotChan({
+      embeds: [
+        new MessageEmbed().setDescription('\n```\n' + (await banner(s)) + '\n```\n')
+      ]
+    }, true);
     client.user.setStatus('invisible');
     client.user.setActivity('nothing! (I\'m offline!)', { type: 'LISTENING' });
   };
@@ -376,7 +362,11 @@ client.once('ready', async () => {
     stbcServicer();
   }
 
-  sendToBotChan('\n```\n' + (await banner('Hello!')) + '\n```\n\n');
+  sendToBotChan({
+    embeds: [
+      new MessageEmbed().setDescription('\n```\n' + (await banner('Hello!')) + '\n```\n\n')
+    ]
+  }, true);
 
   siteCheck();
 
@@ -477,7 +467,6 @@ client.once('ready', async () => {
           registerButtonHandler,
           discordAuthor: data.author,
           ignoreSquelched,
-          captureSpecs,
           channelsById,
           categoriesByName,
           toChanId,
@@ -970,7 +959,6 @@ client.once('ready', async () => {
     ircReadyHandler,
     client,
     channelsById,
-    captureSpecs,
     allowedSpeakersMentionString,
     subscribedChans,
     listenedToMutate,

discord/avatarGenerators.js

@@ -0,0 +1,65 @@
+'use strict';
+
+const {
+  scopedRedisClient
+} = require('../util');
+
+async function DiceBearGenerator (style, fName) {
+  return `https://api.dicebear.com/5.x/${style}/png?seed=${fName}`;
+}
+
+const AvatarGenerators = {
+  robohash: async (fName) => `https://robohash.org/${fName}.png`,
+  dicebear_shapes: DiceBearGenerator.bind(null, 'shapes'),
+  dicebear_pixelart: DiceBearGenerator.bind(null, 'pixel-art'),
+  dicebear_identicon: DiceBearGenerator.bind(null, 'identicon'),
+  dicebear_personas: DiceBearGenerator.bind(null, 'personas'),
+  dicebear_micah: DiceBearGenerator.bind(null, 'micah'),
+  dicebear_miniavs: DiceBearGenerator.bind(null, 'miniavs'),
+  dicebear_bottts: DiceBearGenerator.bind(null, 'bottts'),
+  dicebear_botttsneutral: DiceBearGenerator.bind(null, 'bottts-neutral'),
+  dicebear_bigsmile: DiceBearGenerator.bind(null, 'big-smile'),
+  multiavatar: async (fName) => `https://api.multiavatar.com/${fName}.png`,
+  uiavatars: async (fName) => `https://ui-avatars.com/api/${fName}.png?name=${fName}&background=random&format=png`,
+  uiavatars_red: async (fName) => `https://ui-avatars.com/api/${fName}.png?name=${fName}&background=ff0000&format=png`,
+  random_style: RandomGenerator
+};
+
+const rKey = (p) => [p, 'randomAvatarStyles'].join(':');
+
+async function randomStyle () {
+  const excluded = JSON.parse(await scopedRedisClient((c, p) => c.get(rKey(p) + ':excludeRandomStyles'))) ?? [];
+  const styles = Object.keys(AvatarGenerators).filter((style) => !excluded.includes(style));
+  let setStyle;
+  // don't ever chose "random_style"!
+  while (!setStyle || setStyle === 'random_style') {
+    setStyle = styles[Math.floor(Math.random() * styles.length)];
+  }
+  return setStyle;
+}
+
+async function RandomGenerator (fName) {
+  let setStyle = await scopedRedisClient((c, p) => c.hget(rKey(p), fName));
+
+  if (!setStyle) {
+    setStyle = await randomStyle();
+    console.info(`User ${fName} did not have an avatar style: chose ${setStyle}`);
+    await scopedRedisClient((c, p) => c.hset(rKey(p), fName, setStyle));
+  }
+
+  return AvatarGenerators[setStyle](fName);
+}
+
+module.exports = {
+  AvatarGenerators,
+
+  createAvatarName (nick, network) {
+    return [nick, network].map((s) => s.replaceAll(/[^\d\w._-]+/g, '')).join('_');
+  },
+
+  randomStyle,
+
+  async excludeRandomStyles (...styles) {
+    return scopedRedisClient((c, p) => c.set(rKey(p) + ':excludeRandomStyles', JSON.stringify(styles)));
+  }
+};