commit d5b45a1f725bb866fe31e4d1d39c910273360728 · cduck.me/statusphere-example-app-fork

+5 -14

src/context.ts

···

       3
       3
        
       import { pino } from 'pino'

     

       4
       4
        
       

     

       5
       5
        
       import { createOAuthClient } from '#/auth/client'

     

       6
       6
       -
       import { Database } from '#/db'

     

       7
       7
       -
       import { createDb } from '#/db'

     

       8
       8
       -
       import {

     

       9
       9
       -
         BidirectionalResolver,

     

       10
       10
       -
         createBidirectionalResolver,

     

       11
       11
       -
         createIdResolver,

     

       12
       12
       -
       } from '#/id-resolver'

     

       6
       6
       +
       import { createDb, Database } from '#/db'

     

       13
       7
        
       import { createIngester } from '#/ingester'

     

       14
       8
        
       

     

       15
       9
        
       /**

     
···

       20
       14
        
         ingester: Firehose

     

       21
       15
        
         logger: pino.Logger

     

       22
       16
        
         oauthClient: NodeOAuthClient

     

       23
       23
       -
         resolver: BidirectionalResolver

     

       17
       17
       +
         identityResolver: NodeOAuthClient['identityResolver']

     

       24
       18
        
       }

     

       25
       19
        
       

     

       26
       20
        
       export async function createAppContext(): Promise<AppContext> {

     

       27
       27
       -
         const logger = pino({ name: 'server start' })

     

       28
       28
       -
       

     

       29
       21
        
         const db = await createDb()

     

       30
       22
        
         const oauthClient = await createOAuthClient(db)

     

       31
       31
       -
         const baseIdResolver = createIdResolver()

     

       32
       32
       -
         const ingester = createIngester(db, baseIdResolver)

     

       33
       33
       -
         const resolver = createBidirectionalResolver(baseIdResolver)

     

       23
       23
       +
         const ingester = createIngester(db)

     

       24
       24
       +
         const logger = pino({ name: 'server' })

     

       34
       25
        
       

     

       35
       26
        
         return {

     

       36
       27
        
           db,

     

       37
       28
        
           ingester,

     

       38
       29
        
           logger,

     

       39
       30
        
           oauthClient,

     

       40
       40
       -
           resolver,

     

       31
       31
       +
           identityResolver: oauthClient.identityResolver,

     

       41
       32
        
         }

     

       42
       33
        
       }

-41

src/id-resolver.ts

···

       1
       1
       -
       import { IdResolver, MemoryCache } from '@atproto/identity'

     

       2
       2
       -
       

     

       3
       3
       -
       const HOUR = 60e3 * 60

     

       4
       4
       -
       const DAY = HOUR * 24

     

       5
       5
       -
       

     

       6
       6
       -
       export function createIdResolver() {

     

       7
       7
       -
         return new IdResolver({

     

       8
       8
       -
           didCache: new MemoryCache(HOUR, DAY),

     

       9
       9
       -
         })

     

       10
       10
       -
       }

     

       11
       11
       -
       

     

       12
       12
       -
       export interface BidirectionalResolver {

     

       13
       13
       -
         resolveDidToHandle(did: string): Promise<string>

     

       14
       14
       -
         resolveDidsToHandles(dids: string[]): Promise<Record<string, string>>

     

       15
       15
       -
       }

     

       16
       16
       -
       

     

       17
       17
       -
       export function createBidirectionalResolver(resolver: IdResolver) {

     

       18
       18
       -
         return {

     

       19
       19
       -
           async resolveDidToHandle(did: string): Promise<string> {

     

       20
       20
       -
             const didDoc = await resolver.did.resolveAtprotoData(did)

     

       21
       21
       -
             const resolvedHandle = await resolver.handle.resolve(didDoc.handle)

     

       22
       22
       -
             if (resolvedHandle === did) {

     

       23
       23
       -
               return didDoc.handle

     

       24
       24
       -
             }

     

       25
       25
       -
             return did

     

       26
       26
       -
           },

     

       27
       27
       -
       

     

       28
       28
       -
           async resolveDidsToHandles(

     

       29
       29
       -
             dids: string[]

     

       30
       30
       -
           ): Promise<Record<string, string>> {

     

       31
       31
       -
             const didHandleMap: Record<string, string> = {}

     

       32
       32
       -
             const resolves = await Promise.all(

     

       33
       33
       -
               dids.map((did) => this.resolveDidToHandle(did).catch((_) => did))

     

       34
       34
       -
             )

     

       35
       35
       -
             for (let i = 0; i < dids.length; i++) {

     

       36
       36
       -
               didHandleMap[dids[i]] = resolves[i]

     

       37
       37
       -
             }

     

       38
       38
       -
             return didHandleMap

     

       39
       39
       -
           },

     

       40
       40
       -
         }

     

       41
       41
       -
       }

+11 -6

src/ingester.ts

···

       1
       1
       -
       import pino from 'pino'

     

       2
       2
       -
       import { IdResolver } from '@atproto/identity'

     

       3
       3
       -
       import { Event, Firehose } from '@atproto/sync'

     

       4
       1
        
       import type { Database } from '#/db'

     

       5
       2
        
       import * as Status from '#/lexicon/types/xyz/statusphere/status'

     

       3
       3
       +
       import { IdResolver, MemoryCache } from '@atproto/identity'

     

       4
       4
       +
       import { Event, Firehose } from '@atproto/sync'

     

       5
       5
       +
       import pino from 'pino'

     

       6
       6
        
       

     

       7
       7
       -
       export function createIngester(db: Database, idResolver: IdResolver) {

     

       7
       7
       +
       const HOUR = 60e3 * 60

     

       8
       8
       +
       const DAY = HOUR * 24

     

       9
       9
       +
       

     

       10
       10
       +
       export function createIngester(db: Database) {

     

       8
       11
        
         const logger = pino({ name: 'firehose ingestion' })

     

       9
       12
        
         return new Firehose({

     

       10
       10
       -
           idResolver,

     

       13
       13
       +
           idResolver: new IdResolver({

     

       14
       14
       +
             didCache: new MemoryCache(HOUR, DAY),

     

       15
       15
       +
           }),

     

       11
       16
        
           handleEvent: async (evt: Event) => {

     

       12
       17
        
             // Watch for write events

     

       13
       18
        
             if (evt.event === 'create' || evt.event === 'update') {

     
···

       34
       39
        
                     oc.column('uri').doUpdateSet({

     

       35
       40
        
                       status: record.status,

     

       36
       41
        
                       indexedAt: now.toISOString(),

     

       37
       37
       -
                     })

     

       42
       42
       +
                     }),

     

       38
       43
        
                   )

     

       39
       44
        
                   .execute()

     

       40
       45
        
               }

+14 -4

src/routes.ts

···

       162
       162
        
                   .executeTakeFirst()

     

       163
       163
        
               : undefined

     

       164
       164
        
       

     

       165
       165
       -
             // Map user DIDs to their domain-name handles

     

       166
       166
       -
             const didHandleMap = await ctx.resolver.resolveDidsToHandles(

     

       167
       167
       -
               statuses.map((s) => s.authorDid),

     

       168
       168
       -
             )

     

       165
       165
       +
             // Map (unique) user DIDs to their domain-name handles

     

       166
       166
       +
             const uniqueDids = [...new Set(statuses.map((s) => s.authorDid))]

     

       167
       167
       +
       

     

       168
       168
       +
             const didHandleMap: Record<string, string | undefined> =

     

       169
       169
       +
               Object.fromEntries(

     

       170
       170
       +
                 await Promise.all(

     

       171
       171
       +
                   uniqueDids.map((did) =>

     

       172
       172
       +
                     ctx.identityResolver.resolve(did).then(

     

       173
       173
       +
                       (r) => [did, r.handle],

     

       174
       174
       +
                       () => [did, undefined],

     

       175
       175
       +
                     ),

     

       176
       176
       +
                   ),

     

       177
       177
       +
                 ),

     

       178
       178
       +
               )

     

       169
       179
        
       

     

       170
       180
        
             if (!agent) {

     

       171
       181
        
               // Serve the logged-out view