+10

app/api/inngest/client.ts

reviewed

··· 51 51 documentUris?: string[]; 52 52 }; 53 53 }; 54 54 + "user/write-records-to-pds": { 55 55 + data: { 56 56 + did: string; 57 57 + records: Array<{ 58 58 + collection: string; 59 59 + rkey: string; 60 60 + record: unknown; 61 61 + }>; 62 62 + }; 63 63 + }; 54 64 }; 55 65 56 66 // Create a client to send and receive events

+70

app/api/inngest/functions/write_records_to_pds.ts

reviewed

··· 1 1 + import { inngest } from "../client"; 2 2 + import { restoreOAuthSession } from "src/atproto-oauth"; 3 3 + import { AtpBaseClient } from "lexicons/api"; 4 4 + 5 5 + // Batch size to avoid Inngest payload limits and PDS rate limits 6 6 + const BATCH_SIZE = 50; 7 7 + 8 8 + // Helper to create authenticated agent - must be called fresh in each step 9 9 + // (OAuth sessions cannot be serialized across Inngest steps) 10 10 + async function createAuthenticatedAgent(did: string): Promise<AtpBaseClient> { 11 11 + const result = await restoreOAuthSession(did); 12 12 + if (!result.ok) { 13 13 + throw new Error(`Failed to restore OAuth session: ${result.error.message}`); 14 14 + } 15 15 + return new AtpBaseClient(result.value.fetchHandler.bind(result.value)); 16 16 + } 17 17 + 18 18 + export const write_records_to_pds = inngest.createFunction( 19 19 + { id: "write-records-to-pds" }, 20 20 + { event: "user/write-records-to-pds" }, 21 21 + async ({ event, step }) => { 22 22 + const { did, records } = event.data; 23 23 + 24 24 + // Step 1: Verify OAuth session is valid before proceeding 25 25 + await step.run("verify-oauth-session", async () => { 26 26 + const result = await restoreOAuthSession(did); 27 27 + if (!result.ok) { 28 28 + throw new Error(`OAuth restore failed: ${result.error.message}`); 29 29 + } 30 30 + return { success: true }; 31 31 + }); 32 32 + 33 33 + // Step 2: Write records to PDS in batches 34 34 + // Split records into batches to avoid payload limits and rate limiting 35 35 + const batches: typeof records[] = []; 36 36 + for (let i = 0; i < records.length; i += BATCH_SIZE) { 37 37 + batches.push(records.slice(i, i + BATCH_SIZE)); 38 38 + } 39 39 + 40 40 + let totalWritten = 0; 41 41 + for (let batchIndex = 0; batchIndex < batches.length; batchIndex++) { 42 42 + const batch = batches[batchIndex]; 43 43 + const batchWritten = await step.run( 44 44 + `write-batch-${batchIndex}`, 45 45 + async () => { 46 46 + const agent = await createAuthenticatedAgent(did); 47 47 + let written = 0; 48 48 + for (const rec of batch) { 49 49 + await agent.com.atproto.repo.putRecord({ 50 50 + repo: did, 51 51 + collection: rec.collection, 52 52 + rkey: rec.rkey, 53 53 + record: rec.record as Record<string, unknown>, 54 54 + validate: false, 55 55 + }); 56 56 + written++; 57 57 + } 58 58 + return written; 59 59 + }, 60 60 + ); 61 61 + totalWritten += batchWritten; 62 62 + } 63 63 + 64 64 + return { 65 65 + success: true, 66 66 + recordsWritten: totalWritten, 67 67 + batchCount: batches.length, 68 68 + }; 69 69 + }, 70 70 + );

+2

app/api/inngest/route.tsx

reviewed

··· 12 12 cleanup_expired_oauth_sessions, 13 13 check_oauth_session, 14 14 } from "./functions/cleanup_expired_oauth_sessions"; 15 15 + import { write_records_to_pds } from "./functions/write_records_to_pds"; 15 16 16 17 export const { GET, POST, PUT } = serve({ 17 18 client: inngest, ··· 26 27 fix_standard_document_postref, 27 28 cleanup_expired_oauth_sessions, 28 29 check_oauth_session, 30 30 + write_records_to_pds, 29 31 ], 30 32 });