51 lines
1.5 KiB
TypeScript
51 lines
1.5 KiB
TypeScript
|
import {
|
||
|
|
OutputSchema as RepoEvent,
|
||
|
|
isCommit,
|
||
|
|
} from './lexicon/types/com/atproto/sync/subscribeRepos'
|
||
|
|
import { FirehoseSubscriptionBase, getOpsByType } from './util/subscription'
|
||
|
|
|
||
|
|
export class FirehoseSubscription extends FirehoseSubscriptionBase {
|
||
|
|
async handleEvent(evt: RepoEvent) {
|
||
|
|
if (!isCommit(evt)) return
|
||
|
|
const ops = await getOpsByType(evt)
|
||
|
|
|
||
|
|
// This logs the text of every post off the firehose.
|
||
|
|
// Just for fun :)
|
||
|
|
// Delete before actually using
|
||
|
|
for (const post of ops.posts.creates) {
|
||
|
|
console.log(post.record.text)
|
||
|
|
}
|
||
|
|
|
||
|
|
const postsToDelete = ops.posts.deletes.map((del) => del.uri)
|
||
|
|
const postsToCreate = ops.posts.creates
|
||
|
|
.filter((create) => {
|
||
|
|
return create.record.text.match('^/[a-z]') || create.record.text.match('^@ai ') || create.record.text.match('/ai ');
|
||
|
|
//return create.record.text.toLowerCase().includes('alf')
|
||
|
|
})
|
||
|
|
.map((create) => {
|
||
|
|
// map alf-related posts to a db row
|
||
|
|
return {
|
||
|
|
uri: create.uri,
|
||
|
|
cid: create.cid,
|
||
|
|
replyParent: create.record?.reply?.parent.uri ?? null,
|
||
|
|
replyRoot: create.record?.reply?.root.uri ?? null,
|
||
|
|
indexedAt: new Date().toISOString(),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
if (postsToDelete.length > 0) {
|
||
|
|
await this.db
|
||
|
|
.deleteFrom('post')
|
||
|
|
.where('uri', 'in', postsToDelete)
|
||
|
|
.execute()
|
||
|
|
}
|
||
|
|
if (postsToCreate.length > 0) {
|
||
|
|
await this.db
|
||
|
|
.insertInto('post')
|
||
|
|
.values(postsToCreate)
|
||
|
|
.onConflict((oc) => oc.doNothing())
|
||
|
|
.execute()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|