paxaus.ts

import { QueryParams } from '../lexicon/types/app/bsky/feed/getFeedSkeleton'
import { AppContext } from '../config'
import { AlgoManager } from '../addn/algoManager'
import dotenv from 'dotenv'
import { Post } from '../db/schema'
import dbClient from '../db/dbClient'
import { Database } from '../db'
import { BskyAgent } from '@atproto/api'

dotenv.config()

// max 15 chars
export const shortname = 'paxaus'

export const handler = async (ctx: AppContext, params: QueryParams) => {
  const builder = await dbClient.getLatestPostsForTag({
    tag: shortname,
    limit: params.limit,
    cursor: params.cursor,
  })

  let feed = builder.map((row) => ({
    post: row.uri,
  }))

  let cursor: string | undefined
  const last = builder.at(-1)
  if (last) {
    cursor = `${new Date(last.indexedAt).getTime()}::${last.cid}`
  }

  return {
    cursor,
    feed,
  }
}

export class manager extends AlgoManager {
  public name: string = shortname

  public matchTerms: string[] = ['pax aus', 'paxaus', 'pax australia']

  public re: RegExp

  constructor(db: Database, agent: BskyAgent) {
    super(db, agent)

    this.re = new RegExp(
      `^.*\\b(${this.matchTerms.join('|')})(es|s)?\\b.*$`,
      'ims',
    )
  }

  public async periodicTask() {
    await this.db.removeTagFromOldPosts(
      this.name,
      new Date().getTime() - 7 * 24 * 60 * 60 * 1000,
    )
  }

  public async filter_post(post: Post): Promise<Boolean> {
    if (post.replyRoot !== null) return false
    if (this.agent === null) {
      await this.start()
    }
    if (this.agent === null) return false

    let match = false
    let matchParts: string[] = []

    if (post.embed?.images) {
      post.embed.images.forEach((image) => {
        matchParts.push(image.alt.replace('\n', ' '))
      })
    }

    if (post.embed?.alt) {
      matchParts.push(post.embed.alt.replace('\n', ' '))
    }

    if (post.embed?.media?.alt) {
      matchParts.push(post.embed.media.alt.replace('\n', ' '))
    }

    if (post.tags) {
      matchParts.push(post.tags.join(' '))
    }

    matchParts.push(post.text.replace('\n', ' '))

    const matchString = matchParts.join(' ')

    if (matchString.match(this.re) !== null) {
      match = true
    }

    return match
  }
}