@evanp/activitypub-bot 0.22.0 → 0.24.0

package/README.md

@@ -97,6 +97,24 @@ The path to the [config file](#config-file) for this server, which defines the u
 
 Falls back to `BOTS_CONFIG_FILE` environment variable. The default is to use the shipped default bot config file, which defines an [OKBot](#okbot) named `ok` and a [DoNothingBot](#donothingbot) named `null`.
 
+#### --delivery
+
+The number of background delivery workers to run for this server. These workers accept and process remote activities that arrive at the server. If the server has a lot of traffic, and it's taking a while for bots to do their thing, increase this number.
+
+Falls back to the `DELIVERY` environment variable. Default is 2.
+
+#### --distribution
+
+The number of background distribution workers to run for this server.  These workers distribute bot activities to followers and other recipients on remote servers.  If it's taking a while for bot activities to get to their recipients, increase this number.
+
+Falls back to the `DISTRIBUTION` environment variable. Default is 8.
+
+#### --index-file
+
+Path to the HTML file to show for the home page of your server. The activitypub.bot server doesn't support any other files, so any images or CSS stylesheets or JavaScript in this page have to be hosted elsewhere. Or just skip them!
+
+Falls back to the 'INDEX_FILE' environment variable. The default index file is in `web/index.html` and just says that this is an activitypub.bot server with a link to the GitHub repo.
+
 ### Config file
 
 The config file defines the bots provided by this server.

package/bin/activitypub-bot.js

@@ -12,6 +12,9 @@ const { values } = parseArgs({
     port: { type: 'string' },
     'bots-config-file': { type: 'string' },
     'log-level': { type: 'string' },
+    delivery: { type: 'string' },
+    distribution: { type: 'string' },
+    'index-file': { type: 'string' },
     help: { type: 'boolean', short: 'h' }
   },
   allowPositionals: false
@@ -26,13 +29,16 @@ Options:
   --port <number>            Port to listen on
   --bots-config-file <path>  Path to bots config module
   --log-level <level>        Log level (e.g., info, debug)
+  --delivery <number>        Number of background delivery workers
+  --distribution <number>    Number of background distribution workers
+  --index-file <path>        HTML page to show at root path
   -h, --help                 Show this help
 `)
   process.exit(0)
 }
 
 const normalize = (value) => (value === '' ? undefined : value)
-const parsePort = (value) => {
+const parseNumber = (value) => {
   if (!value) return undefined
   const parsed = Number.parseInt(value, 10)
   return Number.isNaN(parsed) ? undefined : parsed
@@ -40,21 +46,31 @@ const parsePort = (value) => {
 
 const baseDir = dirname(fileURLToPath(import.meta.url))
 const DEFAULT_BOTS_CONFIG_FILE = resolve(baseDir, '..', 'bots', 'index.js')
+const DEFAULT_INDEX_FILE = resolve(baseDir, '..', 'web', 'index.html')
 
 const DATABASE_URL = normalize(values['database-url']) || process.env.DATABASE_URL || 'sqlite::memory:'
 const ORIGIN = normalize(values.origin) || process.env.ORIGIN || 'https://activitypubbot.test'
-const PORT = parsePort(normalize(values.port)) || parsePort(process.env.PORT) || 9000 // HAL
+const PORT = parseNumber(normalize(values.port)) || parseNumber(process.env.PORT) || 9000 // HAL
 const BOTS_CONFIG_FILE =
   normalize(values['bots-config-file']) || process.env.BOTS_CONFIG_FILE || DEFAULT_BOTS_CONFIG_FILE
 const LOG_LEVEL =
   normalize(values['log-level']) ||
   process.env.LOG_LEVEL ||
   (process.env.NODE_ENV === 'test' ? 'silent' : 'info')
+const DELIVERY = parseNumber(values.delivery) || parseNumber(process.env.DELIVERY) || 2
+const DISTRIBUTION = parseNumber(values.distribution) || parseNumber(process.env.DISTRIBUTION) || 8
+const INDEX_FILE = values['index-file'] || process.env.INDEX_FILE || DEFAULT_INDEX_FILE
 
 const bots = (await import(BOTS_CONFIG_FILE)).default
 
 const app = await makeApp({
-  databaseUrl: DATABASE_URL, origin: ORIGIN, bots, logLevel: LOG_LEVEL
+  databaseUrl: DATABASE_URL,
+  origin: ORIGIN,
+  bots,
+  logLevel: LOG_LEVEL,
+  deliveryWorkerCount: DELIVERY,
+  distributionWorkerCount: DISTRIBUTION,
+  indexFileName: INDEX_FILE
 })
 
 const server = app.listen(parseInt(PORT), () => {

package/lib/bots/relayclient.js

@@ -2,10 +2,12 @@ import Bot from '../bot.js'
 
 export default class RelayClientBot extends Bot {
   #relay
+  #unsubscribe
 
-  constructor (username, relay) {
+  constructor (username, relay, unsubscribe = null) {
     super(username)
     this.#relay = relay
+    this.#unsubscribe = !!unsubscribe
   }
 
   get fullname () {
@@ -18,21 +20,61 @@ export default class RelayClientBot extends Bot {
 
   async initialize (context) {
     super.initialize(context)
-    if (!(await this._context.hasData(`follow:${this.#relay}`))) {
-      await this.#followRelay()
+    this._context.logger.debug(
+      { relay: this.#relay, unsubscribe: this.#unsubscribe },
+      'Initialising relay client'
+    )
+    if (this.#unsubscribe) {
+      if (await this._context.hasData(`follow:${this.#relay}`)) {
+        await this.#unfollowRelay()
+      }
+    } else {
+      if (!(await this._context.hasData(`follow:${this.#relay}`))) {
+        await this.#followRelay()
+      }
     }
   }
 
   async #followRelay () {
+    this._context.logger.debug(
+      { relay: this.#relay },
+      'Following relay'
+    )
     const activity = await this._context.doActivity({
       to: this.#relay,
       type: 'Follow',
       object: 'https://www.w3.org/ns/activitystreams#Public'
     })
+    this._context.logger.debug(
+      { relay: this.#relay, activity: activity.id },
+      'Saving follow for later'
+    )
     this._context.setData(`follow:${this.#relay}`, activity.id)
   }
 
+  async #unfollowRelay () {
+    const activityId = await this._context.getData(`follow:${this.#relay}`)
+    this._context.logger.debug(
+      { relay: this.#relay, activity: activityId },
+      'Unfollowing relay'
+    )
+    await this._context.doActivity({
+      to: this.#relay,
+      type: 'Undo',
+      object: {
+        id: activityId,
+        type: 'Follow',
+        object: 'https://www.w3.org/ns/activitystreams#Public'
+      }
+    })
+    this._context.logger.debug(
+      { relay: this.#relay },
+      'Clearing follow data'
+    )
+    this._context.setData(`follow:${this.#relay}`, null)
+  }
+
   async actorOK (actorId, activity) {
-    return (actorId === this.#relay)
+    return (actorId === this.#relay && !this.#unsubscribe)
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@evanp/activitypub-bot",
-  "version": "0.22.0",
+  "version": "0.24.0",
   "description": "server-side ActivityPub bot framework",
   "type": "module",
   "main": "lib/index.js",