@atproto/bsky 0.0.249 → 0.0.250

package/dist/data-plane/bsync/index.js

@@ -1,376 +0,0 @@
-import assert from 'node:assert';
-import events from 'node:events';
-import { expressConnectMiddleware } from '@connectrpc/connect-express';
-import express from 'express';
-// eslint-disable-next-line import/default
-import httpTerminator from 'http-terminator';
-import { TID } from '@atproto/common';
-import { lexParse } from '@atproto/lex';
-import { AtUri } from '@atproto/syntax';
-import { app } from '../../lexicons/index.js';
-import { httpLogger } from '../../logger.js';
-import { Service } from '../../proto/bsync_connect.js';
-import { Method, MuteOperation_Type, } from '../../proto/bsync_pb.js';
-import { Namespaces } from '../../stash.js';
-import { countAll, excluded } from '../server/db/util.js';
-export class MockBsync {
-    constructor(server) {
-        this.server = server;
-        this.terminator = httpTerminator.createHttpTerminator({ server });
-    }
-    static async create(db, port) {
-        const app = express();
-        const routes = createRoutes(db);
-        app.use(expressConnectMiddleware({ routes }));
-        const server = app.listen(port);
-        await events.once(server, 'listening');
-        return new MockBsync(server);
-    }
-    async destroy() {
-        await this.terminator.terminate();
-    }
-    async [Symbol.asyncDispose]() {
-        await this.destroy();
-    }
-}
-const createRoutes = (db) => (router) => router.service(Service, {
-    async addMuteOperation(req) {
-        const { type, actorDid, subject } = req;
-        if (type === MuteOperation_Type.ADD) {
-            if (subject.startsWith('did:')) {
-                assert(actorDid !== subject, 'cannot mute yourself'); // @TODO pass message through in http error
-                await db.db
-                    .insertInto('mute')
-                    .values({
-                    mutedByDid: actorDid,
-                    subjectDid: subject,
-                    createdAt: new Date().toISOString(),
-                })
-                    .onConflict((oc) => oc.doNothing())
-                    .execute();
-            }
-            else {
-                const uri = new AtUri(subject);
-                if (uri.collection === app.bsky.graph.list.$type) {
-                    await db.db
-                        .insertInto('list_mute')
-                        .values({
-                        mutedByDid: actorDid,
-                        listUri: subject,
-                        createdAt: new Date().toISOString(),
-                    })
-                        .onConflict((oc) => oc.doNothing())
-                        .execute();
-                }
-                else {
-                    await db.db
-                        .insertInto('thread_mute')
-                        .values({
-                        mutedByDid: actorDid,
-                        rootUri: subject,
-                        createdAt: new Date().toISOString(),
-                    })
-                        .onConflict((oc) => oc.doNothing())
-                        .execute();
-                }
-            }
-        }
-        else if (type === MuteOperation_Type.REMOVE) {
-            if (subject.startsWith('did:')) {
-                await db.db
-                    .deleteFrom('mute')
-                    .where('mutedByDid', '=', actorDid)
-                    .where('subjectDid', '=', subject)
-                    .execute();
-            }
-            else {
-                const uri = new AtUri(subject);
-                if (uri.collection === app.bsky.graph.list.$type) {
-                    await db.db
-                        .deleteFrom('list_mute')
-                        .where('mutedByDid', '=', actorDid)
-                        .where('listUri', '=', subject)
-                        .execute();
-                }
-                else {
-                    await db.db
-                        .deleteFrom('thread_mute')
-                        .where('mutedByDid', '=', actorDid)
-                        .where('rootUri', '=', subject)
-                        .execute();
-                }
-            }
-        }
-        else if (type === MuteOperation_Type.CLEAR) {
-            await db.db
-                .deleteFrom('mute')
-                .where('mutedByDid', '=', actorDid)
-                .execute();
-            await db.db
-                .deleteFrom('list_mute')
-                .where('mutedByDid', '=', actorDid)
-                .execute();
-        }
-        return {};
-    },
-    async scanMuteOperations() {
-        throw new Error('not implemented');
-    },
-    async addNotifOperation(req) {
-        const { actorDid, priority } = req;
-        if (priority !== undefined) {
-            await db.db
-                .insertInto('actor_state')
-                .values({
-                did: actorDid,
-                priorityNotifs: priority,
-                lastSeenNotifs: new Date().toISOString(),
-            })
-                .onConflict((oc) => oc.column('did').doUpdateSet({ priorityNotifs: priority }))
-                .execute();
-        }
-        return {};
-    },
-    async scanNotifOperations() {
-        throw new Error('not implemented');
-    },
-    async putOperation(req) {
-        const { actorDid, namespace, key, method, payload } = req;
-        assert(method === Method.CREATE ||
-            method === Method.UPDATE ||
-            method === Method.DELETE, `Unsupported method: ${method}`);
-        const now = new Date().toISOString();
-        // Index all items into private_data.
-        await handleGenericOperation(db, req, now);
-        // Maintain bespoke indexes for certain namespaces.
-        try {
-            if (namespace ===
-                Namespaces.AppBskyNotificationDefsSubjectActivitySubscription.$type) {
-                await handleSubjectActivitySubscriptionOperation(db, req, now);
-            }
-            else if (namespace === Namespaces.AppBskyUnspeccedDefsAgeAssuranceEvent.$type) {
-                await handleAgeAssuranceEventOperation(db, req, now);
-            }
-            else if (namespace === Namespaces.AppBskyAgeassuranceDefsEvent.$type) {
-                await handleAgeAssuranceV2EventOperation(db, req, now);
-            }
-            else if (namespace === Namespaces.AppBskyBookmarkDefsBookmark.$type) {
-                await handleBookmarkOperation(db, req, now);
-            }
-            else if (namespace === Namespaces.AppBskyDraftDefsDraftWithId.$type) {
-                await handleDraftOperation(db, req, now);
-            }
-        }
-        catch (err) {
-            httpLogger.warn({ err, namespace }, 'mock bsync put operation failed');
-        }
-        return {
-            operation: {
-                id: TID.nextStr(),
-                actorDid,
-                namespace,
-                key,
-                method,
-                payload,
-            },
-        };
-    },
-    async scanOperations() {
-        throw new Error('not implemented');
-    },
-    async ping() {
-        return {};
-    },
-});
-// upsert into or remove from private_data
-const handleGenericOperation = async (db, req, now) => {
-    const { actorDid, namespace, key, method, payload } = req;
-    if (method === Method.CREATE || method === Method.UPDATE) {
-        await db.db
-            .insertInto('private_data')
-            .values({
-            actorDid,
-            namespace,
-            key,
-            payload: Buffer.from(payload).toString('utf8'),
-            indexedAt: now,
-            updatedAt: now,
-        })
-            .onConflict((oc) => oc.columns(['actorDid', 'namespace', 'key']).doUpdateSet({
-            payload: excluded(db.db, 'payload'),
-            updatedAt: excluded(db.db, 'updatedAt'),
-        }))
-            .execute();
-    }
-    else if (method === Method.DELETE) {
-        await db.db
-            .deleteFrom('private_data')
-            .where('actorDid', '=', actorDid)
-            .where('namespace', '=', namespace)
-            .where('key', '=', key)
-            .execute();
-    }
-    else {
-        assert.fail(`unexpected method ${method}`);
-    }
-};
-const handleSubjectActivitySubscriptionOperation = async (db, req, now) => {
-    const { actorDid, key, method, payload } = req;
-    if (method === Method.DELETE) {
-        return db.db
-            .deleteFrom('activity_subscription')
-            .where('creator', '=', actorDid)
-            .where('key', '=', key)
-            .execute();
-    }
-    const json = Buffer.from(payload).toString('utf8');
-    const parsed = lexParse(json);
-    const { subject, activitySubscription: { post, reply }, } = parsed;
-    if (method === Method.CREATE) {
-        return db.db
-            .insertInto('activity_subscription')
-            .values({
-            creator: actorDid,
-            subjectDid: subject,
-            key,
-            indexedAt: now,
-            post,
-            reply,
-        })
-            .execute();
-    }
-    return db.db
-        .updateTable('activity_subscription')
-        .where('creator', '=', actorDid)
-        .where('key', '=', key)
-        .set({
-        indexedAt: now,
-        post,
-        reply,
-    })
-        .execute();
-};
-const handleAgeAssuranceEventOperation = async (db, req, _now) => {
-    const { actorDid, method, payload } = req;
-    if (method !== Method.CREATE)
-        return;
-    const parsed = lexParse(Buffer.from(payload).toString('utf8'));
-    const { status, createdAt } = parsed;
-    const update = {
-        ageAssuranceStatus: status,
-        ageAssuranceLastInitiatedAt: status === 'pending' ? createdAt : undefined,
-    };
-    return db.db
-        .updateTable('actor')
-        .set(update)
-        .where('did', '=', actorDid)
-        .execute();
-};
-const handleAgeAssuranceV2EventOperation = async (db, req, _now) => {
-    const { actorDid, method, payload } = req;
-    if (method !== Method.CREATE)
-        return;
-    const parsed = lexParse(Buffer.from(payload).toString('utf8'));
-    const { status, createdAt, access, countryCode, regionCode } = parsed;
-    const update = {
-        ageAssuranceStatus: status,
-        ageAssuranceLastInitiatedAt: status === 'pending' ? createdAt : undefined,
-        ageAssuranceAccess: access,
-        ageAssuranceCountryCode: countryCode,
-        ageAssuranceRegionCode: regionCode,
-    };
-    return db.db
-        .updateTable('actor')
-        .set(update)
-        .where('did', '=', actorDid)
-        .execute();
-};
-const handleBookmarkOperation = async (db, req, now) => {
-    const { actorDid, key, method, payload } = req;
-    const updateAgg = (uri, dbTxn) => {
-        return dbTxn.db
-            .insertInto('post_agg')
-            .values({
-            uri,
-            bookmarkCount: dbTxn.db
-                .selectFrom('bookmark')
-                .where('bookmark.subjectUri', '=', uri)
-                .select(countAll.as('count')),
-        })
-            .onConflict((oc) => oc
-            .column('uri')
-            .doUpdateSet({ bookmarkCount: excluded(dbTxn.db, 'bookmarkCount') }))
-            .execute();
-    };
-    if (method === Method.CREATE) {
-        const parsed = lexParse(Buffer.from(payload).toString('utf8'));
-        const { subject: { uri, cid }, } = parsed;
-        await db.transaction(async (dbTxn) => {
-            await dbTxn.db
-                .insertInto('bookmark')
-                .values({
-                creator: actorDid,
-                key,
-                indexedAt: now,
-                subjectUri: uri,
-                subjectCid: cid,
-            })
-                .execute();
-            await updateAgg(uri, dbTxn);
-        });
-    }
-    if (method === Method.DELETE) {
-        await db.transaction(async (dbTxn) => {
-            const bookmark = await dbTxn.db
-                .selectFrom('bookmark')
-                .selectAll()
-                .where('creator', '=', actorDid)
-                .where('key', '=', key)
-                .executeTakeFirst();
-            if (bookmark) {
-                await dbTxn.db
-                    .deleteFrom('bookmark')
-                    .where('creator', '=', actorDid)
-                    .where('key', '=', key)
-                    .execute();
-                await updateAgg(bookmark.subjectUri, dbTxn);
-            }
-        });
-    }
-};
-const handleDraftOperation = async (db, req, now) => {
-    const { actorDid, key, method, payload } = req;
-    if (method === Method.CREATE) {
-        const payloadString = Buffer.from(payload).toString('utf8');
-        await db.db
-            .insertInto('draft')
-            .values({
-            creator: actorDid,
-            key,
-            createdAt: now,
-            updatedAt: now,
-            payload: payloadString,
-        })
-            .execute();
-    }
-    if (method === Method.UPDATE) {
-        const payloadString = Buffer.from(payload).toString('utf8');
-        await db.db
-            .updateTable('draft')
-            .where('creator', '=', actorDid)
-            .where('key', '=', key)
-            .set({
-            updatedAt: now,
-            payload: payloadString,
-        })
-            .execute();
-    }
-    if (method === Method.DELETE) {
-        await db.db
-            .deleteFrom('draft')
-            .where('creator', '=', actorDid)
-            .where('key', '=', key)
-            .execute();
-    }
-};
-//# sourceMappingURL=index.js.map

package/dist/data-plane/bsync/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/data-plane/bsync/index.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,aAAa,CAAA;AAChC,OAAO,MAAM,MAAM,aAAa,CAAA;AAGhC,OAAO,EAAE,wBAAwB,EAAE,MAAM,6BAA6B,CAAA;AACtE,OAAO,OAAO,MAAM,SAAS,CAAA;AAC7B,0CAA0C;AAC1C,OAAO,cAAc,MAAM,iBAAiB,CAAA;AAC5C,OAAO,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAA;AACrC,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAA;AACvC,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAA;AACvC,OAAO,EAAE,GAAG,EAAE,MAAM,yBAAyB,CAAA;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAA;AAC5C,OAAO,EAAE,OAAO,EAAE,MAAM,8BAA8B,CAAA;AACtD,OAAO,EACL,MAAM,EACN,kBAAkB,GAEnB,MAAM,yBAAyB,CAAA;AAChC,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAA;AAE3C,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAA;AAEzD,MAAM,OAAO,SAAS;IAGpB,YAAmB,MAAmB;sBAAnB,MAAM;QACvB,IAAI,CAAC,UAAU,GAAG,cAAc,CAAC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAA;IACnE,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAY,EAAE,IAAY;QAC5C,MAAM,GAAG,GAAG,OAAO,EAAE,CAAA;QACrB,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,CAAC,CAAA;QAC/B,GAAG,CAAC,GAAG,CAAC,wBAAwB,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;QAC7C,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QAC/B,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,CAAA;QACtC,OAAO,IAAI,SAAS,CAAC,MAAM,CAAC,CAAA;IAC9B,CAAC;IAED,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAA;IACnC,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC;QACzB,MAAM,IAAI,CAAC,OAAO,EAAE,CAAA;IACtB,CAAC;CACF;AAED,MAAM,YAAY,GAAG,CAAC,EAAY,EAAE,EAAE,CAAC,CAAC,MAAqB,EAAE,EAAE,CAC/D,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE;IACtB,KAAK,CAAC,gBAAgB,CAAC,GAAG;QACxB,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;QACvC,IAAI,IAAI,KAAK,kBAAkB,CAAC,GAAG,EAAE,CAAC;YACpC,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/B,MAAM,CAAC,QAAQ,KAAK,OAAO,EAAE,sBAAsB,CAAC,CAAA,CAAC,2CAA2C;gBAChG,MAAM,EAAE,CAAC,EAAE;qBACR,UAAU,CAAC,MAAM,CAAC;qBAClB,MAAM,CAAC;oBACN,UAAU,EAAE,QAAQ;oBACpB,UAAU,EAAE,OAAO;oBACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC;qBACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;qBAClC,OAAO,EAAE,CAAA;YACd,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,CAAA;gBAC9B,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;oBACjD,MAAM,EAAE,CAAC,EAAE;yBACR,UAAU,CAAC,WAAW,CAAC;yBACvB,MAAM,CAAC;wBACN,UAAU,EAAE,QAAQ;wBACpB,OAAO,EAAE,OAAO;wBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC,CAAC;yBACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;yBAClC,OAAO,EAAE,CAAA;gBACd,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC,EAAE;yBACR,UAAU,CAAC,aAAa,CAAC;yBACzB,MAAM,CAAC;wBACN,UAAU,EAAE,QAAQ;wBACpB,OAAO,EAAE,OAAO;wBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;qBACpC,CAAC;yBACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;yBAClC,OAAO,EAAE,CAAA;gBACd,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,kBAAkB,CAAC,MAAM,EAAE,CAAC;YAC9C,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC/B,MAAM,EAAE,CAAC,EAAE;qBACR,UAAU,CAAC,MAAM,CAAC;qBAClB,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,CAAC;qBAClC,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,OAAO,CAAC;qBACjC,OAAO,EAAE,CAAA;YACd,CAAC;iBAAM,CAAC;gBACN,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,CAAA;gBAC9B,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;oBACjD,MAAM,EAAE,CAAC,EAAE;yBACR,UAAU,CAAC,WAAW,CAAC;yBACvB,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,CAAC;yBAClC,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC;yBAC9B,OAAO,EAAE,CAAA;gBACd,CAAC;qBAAM,CAAC;oBACN,MAAM,EAAE,CAAC,EAAE;yBACR,UAAU,CAAC,aAAa,CAAC;yBACzB,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,CAAC;yBAClC,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,OAAO,CAAC;yBAC9B,OAAO,EAAE,CAAA;gBACd,CAAC;YACH,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,kBAAkB,CAAC,KAAK,EAAE,CAAC;YAC7C,MAAM,EAAE,CAAC,EAAE;iBACR,UAAU,CAAC,MAAM,CAAC;iBAClB,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,CAAC;iBAClC,OAAO,EAAE,CAAA;YACZ,MAAM,EAAE,CAAC,EAAE;iBACR,UAAU,CAAC,WAAW,CAAC;iBACvB,KAAK,CAAC,YAAY,EAAE,GAAG,EAAE,QAAQ,CAAC;iBAClC,OAAO,EAAE,CAAA;QACd,CAAC;QAED,OAAO,EAAE,CAAA;IACX,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,GAAG;QACzB,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAA;QAClC,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,MAAM,EAAE,CAAC,EAAE;iBACR,UAAU,CAAC,aAAa,CAAC;iBACzB,MAAM,CAAC;gBACN,GAAG,EAAE,QAAQ;gBACb,cAAc,EAAE,QAAQ;gBACxB,cAAc,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACzC,CAAC;iBACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CACjB,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,QAAQ,EAAE,CAAC,CAC3D;iBACA,OAAO,EAAE,CAAA;QACd,CAAC;QACD,OAAO,EAAE,CAAA;IACX,CAAC;IAED,KAAK,CAAC,mBAAmB;QACvB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAG;QACpB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;QACzD,MAAM,CACJ,MAAM,KAAK,MAAM,CAAC,MAAM;YACtB,MAAM,KAAK,MAAM,CAAC,MAAM;YACxB,MAAM,KAAK,MAAM,CAAC,MAAM,EAC1B,uBAAuB,MAAM,EAAE,CAChC,CAAA;QAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAA;QAEpC,qCAAqC;QACrC,MAAM,sBAAsB,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;QAE1C,mDAAmD;QACnD,IAAI,CAAC;YACH,IACE,SAAS;gBACT,UAAU,CAAC,kDAAkD,CAAC,KAAK,EACnE,CAAC;gBACD,MAAM,0CAA0C,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YAChE,CAAC;iBAAM,IACL,SAAS,KAAK,UAAU,CAAC,qCAAqC,CAAC,KAAK,EACpE,CAAC;gBACD,MAAM,gCAAgC,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YACtD,CAAC;iBAAM,IACL,SAAS,KAAK,UAAU,CAAC,4BAA4B,CAAC,KAAK,EAC3D,CAAC;gBACD,MAAM,kCAAkC,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YACxD,CAAC;iBAAM,IAAI,SAAS,KAAK,UAAU,CAAC,2BAA2B,CAAC,KAAK,EAAE,CAAC;gBACtE,MAAM,uBAAuB,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YAC7C,CAAC;iBAAM,IAAI,SAAS,KAAK,UAAU,CAAC,2BAA2B,CAAC,KAAK,EAAE,CAAC;gBACtE,MAAM,oBAAoB,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YAC1C,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,EAAE,iCAAiC,CAAC,CAAA;QACxE,CAAC;QAED,OAAO;YACL,SAAS,EAAE;gBACT,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE;gBACjB,QAAQ;gBACR,SAAS;gBACT,GAAG;gBACH,MAAM;gBACN,OAAO;aACR;SACF,CAAA;IACH,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAA;IACpC,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,EAAE,CAAA;IACX,CAAC;CACF,CAAC,CAAA;AAEJ,0CAA0C;AAC1C,MAAM,sBAAsB,GAAG,KAAK,EAClC,EAAY,EACZ,GAAwB,EACxB,GAAW,EACX,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IACzD,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QACzD,MAAM,EAAE,CAAC,EAAE;aACR,UAAU,CAAC,cAAc,CAAC;aAC1B,MAAM,CAAC;YACN,QAAQ;YACR,SAAS;YACT,GAAG;YACH,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC9C,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;aACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CACjB,EAAE,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC;YACvD,OAAO,EAAE,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,SAAS,CAAC;YACnC,SAAS,EAAE,QAAQ,CAAC,EAAE,CAAC,EAAE,EAAE,WAAW,CAAC;SACxC,CAAC,CACH;aACA,OAAO,EAAE,CAAA;IACd,CAAC;SAAM,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QACpC,MAAM,EAAE,CAAC,EAAE;aACR,UAAU,CAAC,cAAc,CAAC;aAC1B,KAAK,CAAC,UAAU,EAAE,GAAG,EAAE,QAAQ,CAAC;aAChC,KAAK,CAAC,WAAW,EAAE,GAAG,EAAE,SAAS,CAAC;aAClC,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;aACtB,OAAO,EAAE,CAAA;IACd,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,IAAI,CAAC,qBAAqB,MAAM,EAAE,CAAC,CAAA;IAC5C,CAAC;AACH,CAAC,CAAA;AAED,MAAM,0CAA0C,GAAG,KAAK,EACtD,EAAY,EACZ,GAAwB,EACxB,GAAW,EACX,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IAE9C,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,OAAO,EAAE,CAAC,EAAE;aACT,UAAU,CAAC,uBAAuB,CAAC;aACnC,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;aACtB,OAAO,EAAE,CAAA;IACd,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IAClD,MAAM,MAAM,GACV,QAAQ,CAAyD,IAAI,CAAC,CAAA;IACxE,MAAM,EACJ,OAAO,EACP,oBAAoB,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,GACtC,GAAG,MAAM,CAAA;IAEV,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,OAAO,EAAE,CAAC,EAAE;aACT,UAAU,CAAC,uBAAuB,CAAC;aACnC,MAAM,CAAC;YACN,OAAO,EAAE,QAAQ;YACjB,UAAU,EAAE,OAAO;YACnB,GAAG;YACH,SAAS,EAAE,GAAG;YACd,IAAI;YACJ,KAAK;SACN,CAAC;aACD,OAAO,EAAE,CAAA;IACd,CAAC;IAED,OAAO,EAAE,CAAC,EAAE;SACT,WAAW,CAAC,uBAAuB,CAAC;SACpC,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;SAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;SACtB,GAAG,CAAC;QACH,SAAS,EAAE,GAAG;QACd,IAAI;QACJ,KAAK;KACN,CAAC;SACD,OAAO,EAAE,CAAA;AACd,CAAC,CAAA;AAED,MAAM,gCAAgC,GAAG,KAAK,EAC5C,EAAY,EACZ,GAAwB,EACxB,IAAY,EACZ,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IACzC,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM;QAAE,OAAM;IAEpC,MAAM,MAAM,GAAG,QAAQ,CACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CACtC,CAAA;IAED,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAAA;IAEpC,MAAM,MAAM,GAAG;QACb,kBAAkB,EAAE,MAAM;QAC1B,2BAA2B,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;KAC1E,CAAA;IAED,OAAO,EAAE,CAAC,EAAE;SACT,WAAW,CAAC,OAAO,CAAC;SACpB,GAAG,CAAC,MAAM,CAAC;SACX,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,QAAQ,CAAC;SAC3B,OAAO,EAAE,CAAA;AACd,CAAC,CAAA;AAED,MAAM,kCAAkC,GAAG,KAAK,EAC9C,EAAY,EACZ,GAAwB,EACxB,IAAY,EACZ,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IACzC,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM;QAAE,OAAM;IAEpC,MAAM,MAAM,GAAG,QAAQ,CACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CACtC,CAAA;IAED,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,MAAM,CAAA;IAErE,MAAM,MAAM,GAAG;QACb,kBAAkB,EAAE,MAAM;QAC1B,2BAA2B,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;QACzE,kBAAkB,EAAE,MAAM;QAC1B,uBAAuB,EAAE,WAAW;QACpC,sBAAsB,EAAE,UAAU;KACnC,CAAA;IAED,OAAO,EAAE,CAAC,EAAE;SACT,WAAW,CAAC,OAAO,CAAC;SACpB,GAAG,CAAC,MAAM,CAAC;SACX,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,QAAQ,CAAC;SAC3B,OAAO,EAAE,CAAA;AACd,CAAC,CAAA;AAED,MAAM,uBAAuB,GAAG,KAAK,EACnC,EAAY,EACZ,GAAwB,EACxB,GAAW,EACX,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IAE9C,MAAM,SAAS,GAAG,CAAC,GAAW,EAAE,KAAe,EAAE,EAAE;QACjD,OAAO,KAAK,CAAC,EAAE;aACZ,UAAU,CAAC,UAAU,CAAC;aACtB,MAAM,CAAC;YACN,GAAG;YACH,aAAa,EAAE,KAAK,CAAC,EAAE;iBACpB,UAAU,CAAC,UAAU,CAAC;iBACtB,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,GAAG,CAAC;iBACtC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;SAChC,CAAC;aACD,UAAU,CAAC,CAAC,EAAE,EAAE,EAAE,CACjB,EAAE;aACC,MAAM,CAAC,KAAK,CAAC;aACb,WAAW,CAAC,EAAE,aAAa,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,EAAE,eAAe,CAAC,EAAE,CAAC,CACvE;aACA,OAAO,EAAE,CAAA;IACd,CAAC,CAAA;IAED,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,QAAQ,CACrB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CACtC,CAAA;QACD,MAAM,EACJ,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GACtB,GAAG,MAAM,CAAA;QAEV,MAAM,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,KAAK,CAAC,EAAE;iBACX,UAAU,CAAC,UAAU,CAAC;iBACtB,MAAM,CAAC;gBACN,OAAO,EAAE,QAAQ;gBACjB,GAAG;gBACH,SAAS,EAAE,GAAG;gBACd,UAAU,EAAE,GAAG;gBACf,UAAU,EAAE,GAAG;aAChB,CAAC;iBACD,OAAO,EAAE,CAAA;YAEZ,MAAM,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,CAAA;QAC7B,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACnC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,EAAE;iBAC5B,UAAU,CAAC,UAAU,CAAC;iBACtB,SAAS,EAAE;iBACX,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;iBAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;iBACtB,gBAAgB,EAAE,CAAA;YAErB,IAAI,QAAQ,EAAE,CAAC;gBACb,MAAM,KAAK,CAAC,EAAE;qBACX,UAAU,CAAC,UAAU,CAAC;qBACtB,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;qBAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;qBACtB,OAAO,EAAE,CAAA;gBAEZ,MAAM,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAA;YAC7C,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAA;AAED,MAAM,oBAAoB,GAAG,KAAK,EAChC,EAAY,EACZ,GAAwB,EACxB,GAAW,EACX,EAAE;IACF,MAAM,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,CAAA;IAE9C,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;QAE3D,MAAM,EAAE,CAAC,EAAE;aACR,UAAU,CAAC,OAAO,CAAC;aACnB,MAAM,CAAC;YACN,OAAO,EAAE,QAAQ;YACjB,GAAG;YACH,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,aAAa;SACvB,CAAC;aACD,OAAO,EAAE,CAAA;IACd,CAAC;IAED,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;QAE3D,MAAM,EAAE,CAAC,EAAE;aACR,WAAW,CAAC,OAAO,CAAC;aACpB,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;aACtB,GAAG,CAAC;YACH,SAAS,EAAE,GAAG;YACd,OAAO,EAAE,aAAa;SACvB,CAAC;aACD,OAAO,EAAE,CAAA;IACd,CAAC;IAED,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;QAC7B,MAAM,EAAE,CAAC,EAAE;aACR,UAAU,CAAC,OAAO,CAAC;aACnB,KAAK,CAAC,SAAS,EAAE,GAAG,EAAE,QAAQ,CAAC;aAC/B,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,CAAC;aACtB,OAAO,EAAE,CAAA;IACd,CAAC;AACH,CAAC,CAAA","sourcesContent":["import assert from 'node:assert'\nimport events from 'node:events'\nimport http from 'node:http'\nimport { ConnectRouter } from '@connectrpc/connect'\nimport { expressConnectMiddleware } from '@connectrpc/connect-express'\nimport express from 'express'\n// eslint-disable-next-line import/default\nimport httpTerminator from 'http-terminator'\nimport { TID } from '@atproto/common'\nimport { lexParse } from '@atproto/lex'\nimport { AtUri } from '@atproto/syntax'\nimport { app } from '../../lexicons/index.js'\nimport { httpLogger } from '../../logger.js'\nimport { Service } from '../../proto/bsync_connect.js'\nimport {\n  Method,\n  MuteOperation_Type,\n  PutOperationRequest,\n} from '../../proto/bsync_pb.js'\nimport { Namespaces } from '../../stash.js'\nimport { Database } from '../server/db/index.js'\nimport { countAll, excluded } from '../server/db/util.js'\n\nexport class MockBsync {\n  private terminator: httpTerminator.HttpTerminator\n\n  constructor(public server: http.Server) {\n    this.terminator = httpTerminator.createHttpTerminator({ server })\n  }\n\n  static async create(db: Database, port: number) {\n    const app = express()\n    const routes = createRoutes(db)\n    app.use(expressConnectMiddleware({ routes }))\n    const server = app.listen(port)\n    await events.once(server, 'listening')\n    return new MockBsync(server)\n  }\n\n  async destroy() {\n    await this.terminator.terminate()\n  }\n\n  async [Symbol.asyncDispose]() {\n    await this.destroy()\n  }\n}\n\nconst createRoutes = (db: Database) => (router: ConnectRouter) =>\n  router.service(Service, {\n    async addMuteOperation(req) {\n      const { type, actorDid, subject } = req\n      if (type === MuteOperation_Type.ADD) {\n        if (subject.startsWith('did:')) {\n          assert(actorDid !== subject, 'cannot mute yourself') // @TODO pass message through in http error\n          await db.db\n            .insertInto('mute')\n            .values({\n              mutedByDid: actorDid,\n              subjectDid: subject,\n              createdAt: new Date().toISOString(),\n            })\n            .onConflict((oc) => oc.doNothing())\n            .execute()\n        } else {\n          const uri = new AtUri(subject)\n          if (uri.collection === app.bsky.graph.list.$type) {\n            await db.db\n              .insertInto('list_mute')\n              .values({\n                mutedByDid: actorDid,\n                listUri: subject,\n                createdAt: new Date().toISOString(),\n              })\n              .onConflict((oc) => oc.doNothing())\n              .execute()\n          } else {\n            await db.db\n              .insertInto('thread_mute')\n              .values({\n                mutedByDid: actorDid,\n                rootUri: subject,\n                createdAt: new Date().toISOString(),\n              })\n              .onConflict((oc) => oc.doNothing())\n              .execute()\n          }\n        }\n      } else if (type === MuteOperation_Type.REMOVE) {\n        if (subject.startsWith('did:')) {\n          await db.db\n            .deleteFrom('mute')\n            .where('mutedByDid', '=', actorDid)\n            .where('subjectDid', '=', subject)\n            .execute()\n        } else {\n          const uri = new AtUri(subject)\n          if (uri.collection === app.bsky.graph.list.$type) {\n            await db.db\n              .deleteFrom('list_mute')\n              .where('mutedByDid', '=', actorDid)\n              .where('listUri', '=', subject)\n              .execute()\n          } else {\n            await db.db\n              .deleteFrom('thread_mute')\n              .where('mutedByDid', '=', actorDid)\n              .where('rootUri', '=', subject)\n              .execute()\n          }\n        }\n      } else if (type === MuteOperation_Type.CLEAR) {\n        await db.db\n          .deleteFrom('mute')\n          .where('mutedByDid', '=', actorDid)\n          .execute()\n        await db.db\n          .deleteFrom('list_mute')\n          .where('mutedByDid', '=', actorDid)\n          .execute()\n      }\n\n      return {}\n    },\n\n    async scanMuteOperations() {\n      throw new Error('not implemented')\n    },\n\n    async addNotifOperation(req) {\n      const { actorDid, priority } = req\n      if (priority !== undefined) {\n        await db.db\n          .insertInto('actor_state')\n          .values({\n            did: actorDid,\n            priorityNotifs: priority,\n            lastSeenNotifs: new Date().toISOString(),\n          })\n          .onConflict((oc) =>\n            oc.column('did').doUpdateSet({ priorityNotifs: priority }),\n          )\n          .execute()\n      }\n      return {}\n    },\n\n    async scanNotifOperations() {\n      throw new Error('not implemented')\n    },\n\n    async putOperation(req) {\n      const { actorDid, namespace, key, method, payload } = req\n      assert(\n        method === Method.CREATE ||\n          method === Method.UPDATE ||\n          method === Method.DELETE,\n        `Unsupported method: ${method}`,\n      )\n\n      const now = new Date().toISOString()\n\n      // Index all items into private_data.\n      await handleGenericOperation(db, req, now)\n\n      // Maintain bespoke indexes for certain namespaces.\n      try {\n        if (\n          namespace ===\n          Namespaces.AppBskyNotificationDefsSubjectActivitySubscription.$type\n        ) {\n          await handleSubjectActivitySubscriptionOperation(db, req, now)\n        } else if (\n          namespace === Namespaces.AppBskyUnspeccedDefsAgeAssuranceEvent.$type\n        ) {\n          await handleAgeAssuranceEventOperation(db, req, now)\n        } else if (\n          namespace === Namespaces.AppBskyAgeassuranceDefsEvent.$type\n        ) {\n          await handleAgeAssuranceV2EventOperation(db, req, now)\n        } else if (namespace === Namespaces.AppBskyBookmarkDefsBookmark.$type) {\n          await handleBookmarkOperation(db, req, now)\n        } else if (namespace === Namespaces.AppBskyDraftDefsDraftWithId.$type) {\n          await handleDraftOperation(db, req, now)\n        }\n      } catch (err) {\n        httpLogger.warn({ err, namespace }, 'mock bsync put operation failed')\n      }\n\n      return {\n        operation: {\n          id: TID.nextStr(),\n          actorDid,\n          namespace,\n          key,\n          method,\n          payload,\n        },\n      }\n    },\n\n    async scanOperations() {\n      throw new Error('not implemented')\n    },\n\n    async ping() {\n      return {}\n    },\n  })\n\n// upsert into or remove from private_data\nconst handleGenericOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  now: string,\n) => {\n  const { actorDid, namespace, key, method, payload } = req\n  if (method === Method.CREATE || method === Method.UPDATE) {\n    await db.db\n      .insertInto('private_data')\n      .values({\n        actorDid,\n        namespace,\n        key,\n        payload: Buffer.from(payload).toString('utf8'),\n        indexedAt: now,\n        updatedAt: now,\n      })\n      .onConflict((oc) =>\n        oc.columns(['actorDid', 'namespace', 'key']).doUpdateSet({\n          payload: excluded(db.db, 'payload'),\n          updatedAt: excluded(db.db, 'updatedAt'),\n        }),\n      )\n      .execute()\n  } else if (method === Method.DELETE) {\n    await db.db\n      .deleteFrom('private_data')\n      .where('actorDid', '=', actorDid)\n      .where('namespace', '=', namespace)\n      .where('key', '=', key)\n      .execute()\n  } else {\n    assert.fail(`unexpected method ${method}`)\n  }\n}\n\nconst handleSubjectActivitySubscriptionOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  now: string,\n) => {\n  const { actorDid, key, method, payload } = req\n\n  if (method === Method.DELETE) {\n    return db.db\n      .deleteFrom('activity_subscription')\n      .where('creator', '=', actorDid)\n      .where('key', '=', key)\n      .execute()\n  }\n\n  const json = Buffer.from(payload).toString('utf8')\n  const parsed =\n    lexParse<app.bsky.notification.defs.SubjectActivitySubscription>(json)\n  const {\n    subject,\n    activitySubscription: { post, reply },\n  } = parsed\n\n  if (method === Method.CREATE) {\n    return db.db\n      .insertInto('activity_subscription')\n      .values({\n        creator: actorDid,\n        subjectDid: subject,\n        key,\n        indexedAt: now,\n        post,\n        reply,\n      })\n      .execute()\n  }\n\n  return db.db\n    .updateTable('activity_subscription')\n    .where('creator', '=', actorDid)\n    .where('key', '=', key)\n    .set({\n      indexedAt: now,\n      post,\n      reply,\n    })\n    .execute()\n}\n\nconst handleAgeAssuranceEventOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  _now: string,\n) => {\n  const { actorDid, method, payload } = req\n  if (method !== Method.CREATE) return\n\n  const parsed = lexParse<app.bsky.unspecced.defs.AgeAssuranceEvent>(\n    Buffer.from(payload).toString('utf8'),\n  )\n\n  const { status, createdAt } = parsed\n\n  const update = {\n    ageAssuranceStatus: status,\n    ageAssuranceLastInitiatedAt: status === 'pending' ? createdAt : undefined,\n  }\n\n  return db.db\n    .updateTable('actor')\n    .set(update)\n    .where('did', '=', actorDid)\n    .execute()\n}\n\nconst handleAgeAssuranceV2EventOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  _now: string,\n) => {\n  const { actorDid, method, payload } = req\n  if (method !== Method.CREATE) return\n\n  const parsed = lexParse<app.bsky.ageassurance.defs.Event>(\n    Buffer.from(payload).toString('utf8'),\n  )\n\n  const { status, createdAt, access, countryCode, regionCode } = parsed\n\n  const update = {\n    ageAssuranceStatus: status,\n    ageAssuranceLastInitiatedAt: status === 'pending' ? createdAt : undefined,\n    ageAssuranceAccess: access,\n    ageAssuranceCountryCode: countryCode,\n    ageAssuranceRegionCode: regionCode,\n  }\n\n  return db.db\n    .updateTable('actor')\n    .set(update)\n    .where('did', '=', actorDid)\n    .execute()\n}\n\nconst handleBookmarkOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  now: string,\n) => {\n  const { actorDid, key, method, payload } = req\n\n  const updateAgg = (uri: string, dbTxn: Database) => {\n    return dbTxn.db\n      .insertInto('post_agg')\n      .values({\n        uri,\n        bookmarkCount: dbTxn.db\n          .selectFrom('bookmark')\n          .where('bookmark.subjectUri', '=', uri)\n          .select(countAll.as('count')),\n      })\n      .onConflict((oc) =>\n        oc\n          .column('uri')\n          .doUpdateSet({ bookmarkCount: excluded(dbTxn.db, 'bookmarkCount') }),\n      )\n      .execute()\n  }\n\n  if (method === Method.CREATE) {\n    const parsed = lexParse<app.bsky.bookmark.defs.Bookmark>(\n      Buffer.from(payload).toString('utf8'),\n    )\n    const {\n      subject: { uri, cid },\n    } = parsed\n\n    await db.transaction(async (dbTxn) => {\n      await dbTxn.db\n        .insertInto('bookmark')\n        .values({\n          creator: actorDid,\n          key,\n          indexedAt: now,\n          subjectUri: uri,\n          subjectCid: cid,\n        })\n        .execute()\n\n      await updateAgg(uri, dbTxn)\n    })\n  }\n\n  if (method === Method.DELETE) {\n    await db.transaction(async (dbTxn) => {\n      const bookmark = await dbTxn.db\n        .selectFrom('bookmark')\n        .selectAll()\n        .where('creator', '=', actorDid)\n        .where('key', '=', key)\n        .executeTakeFirst()\n\n      if (bookmark) {\n        await dbTxn.db\n          .deleteFrom('bookmark')\n          .where('creator', '=', actorDid)\n          .where('key', '=', key)\n          .execute()\n\n        await updateAgg(bookmark.subjectUri, dbTxn)\n      }\n    })\n  }\n}\n\nconst handleDraftOperation = async (\n  db: Database,\n  req: PutOperationRequest,\n  now: string,\n) => {\n  const { actorDid, key, method, payload } = req\n\n  if (method === Method.CREATE) {\n    const payloadString = Buffer.from(payload).toString('utf8')\n\n    await db.db\n      .insertInto('draft')\n      .values({\n        creator: actorDid,\n        key,\n        createdAt: now,\n        updatedAt: now,\n        payload: payloadString,\n      })\n      .execute()\n  }\n\n  if (method === Method.UPDATE) {\n    const payloadString = Buffer.from(payload).toString('utf8')\n\n    await db.db\n      .updateTable('draft')\n      .where('creator', '=', actorDid)\n      .where('key', '=', key)\n      .set({\n        updatedAt: now,\n        payload: payloadString,\n      })\n      .execute()\n  }\n\n  if (method === Method.DELETE) {\n    await db.db\n      .deleteFrom('draft')\n      .where('creator', '=', actorDid)\n      .where('key', '=', key)\n      .execute()\n  }\n}\n"]}