feat: implement Discord profile data sync

Author: beeman

libs/api/bot/data-access/src/lib/api-bot-instances.service.ts

@@ -73,6 +73,22 @@ export class ApiBotInstancesService {
     return this.core.data.bot.findMany({ where: { status: BotStatus.Active } })
   }
 
+  async getDefaultBot(): Promise<Bot | null> {
+    return this.core.data.bot.findFirst({ where: { id: this.core.config.authDiscordClientId } })
+  }
+
+  async getDefaultBotInstance(): Promise<DiscordBot> {
+    const bot = await this.getDefaultBot()
+    if (!bot) {
+      throw new Error(`Can't get default bot`)
+    }
+    const instance = this.getBotInstance(bot.id)
+    if (!instance) {
+      throw new Error(`Can't get default bot instance`)
+    }
+    return instance
+  }
+
   async getBotRole({
     botId,
     roleId,

libs/api/bot/data-access/src/lib/api-bot.service.ts

@@ -1,13 +1,45 @@
 import { Injectable } from '@nestjs/common'
+import { IdentityProvider, Prisma } from '@prisma/client'
+import { ApiCoreService } from '@pubkey-link/api-core-data-access'
 import { ApiBotDataAdminService } from './api-bot-data-admin.service'
 import { ApiBotDataUserService } from './api-bot-data-user.service'
 import { ApiBotInstancesService } from './api-bot-instances.service'
 
 @Injectable()
 export class ApiBotService {
   constructor(
+    readonly core: ApiCoreService,
     readonly admin: ApiBotDataAdminService,
     readonly instances: ApiBotInstancesService,
     readonly user: ApiBotDataUserService,
   ) {}
+
+  async syncDiscordIdentity({ providerId }: { providerId: string }) {
+    const bot = await this.instances.getDefaultBotInstance()
+    const user = await bot.getUser(providerId)
+
+    if (!user) {
+      throw new Error(`Can't find discord user with id ${providerId}`)
+    }
+
+    const identity = await this.core.data.identity.findFirst({
+      where: {
+        provider: IdentityProvider.Discord,
+        providerId,
+      },
+    })
+
+    if (!identity) {
+      throw new Error(`Can't find discord identity for id ${providerId}`)
+    }
+    await this.core.data.identity.update({
+      where: { id: identity.id },
+      data: {
+        name: user.username,
+        profile: user.toJSON() as Prisma.InputJsonValue,
+        owner: { update: { avatarUrl: user?.avatarURL()?.toString() } },
+      },
+    })
+    return true
+  }
 }

libs/api/bot/util/src/lib/discord-bot.ts

@@ -1,6 +1,6 @@
 import { REST } from '@discordjs/rest'
 import { Logger } from '@nestjs/common'
-import { Client, Guild, GuildMember, MessageCreateOptions, NonThreadGuildBasedChannel } from 'discord.js'
+import { Client, Guild, GuildMember, MessageCreateOptions, NonThreadGuildBasedChannel, User } from 'discord.js'
 import { createDiscordClient, createDiscordRestClient } from './discord/client'
 
 export interface RESTDiscordRoleConnection {
@@ -210,6 +210,10 @@ export class DiscordBot {
 
     return (roles ?? []) as RESTDiscordRoleConnection[]
   }
+
+  async getUser(providerId: string): Promise<User | null> {
+    return this.client?.users?.fetch(providerId, { force: true }) ?? null
+  }
 }
 
 export interface BotApplication {

libs/api/identity/data-access/src/lib/api-identity-data-admin.service.ts

@@ -1,13 +1,18 @@
 import { Injectable } from '@nestjs/common'
 import { Identity as PrismaIdentity, IdentityProvider, UserStatus } from '@prisma/client'
+import { ApiBotService } from '@pubkey-link/api-bot-data-access'
 import { ApiCoreService } from '@pubkey-link/api-core-data-access'
 import { ApiNetworkAssetService } from '@pubkey-link/api-network-asset-data-access'
 import { AdminCreateIdentityInput } from './dto/admin-create-identity.input'
 import { AdminFindManyIdentityInput } from './dto/admin-find-many-identity.input'
 
 @Injectable()
 export class ApiIdentityDataAdminService {
-  constructor(private readonly core: ApiCoreService, private readonly networkAsset: ApiNetworkAssetService) {}
+  constructor(
+    private readonly bot: ApiBotService,
+    private readonly core: ApiCoreService,
+    private readonly networkAsset: ApiNetworkAssetService,
+  ) {}
 
   async createIdentity(input: AdminCreateIdentityInput): Promise<PrismaIdentity> {
     const found = await this.core.data.identity.findUnique({
@@ -107,15 +112,26 @@ export class ApiIdentityDataAdminService {
     if (!identity) {
       throw new Error(`Identity ${identityId} not found`)
     }
-    if (identity.provider !== IdentityProvider.Solana) {
-      throw new Error(`Identity ${identityId} not supported`)
+
+    switch (identity.provider) {
+      case IdentityProvider.Discord:
+        return this.bot
+          .syncDiscordIdentity({ providerId: identity.providerId })
+          .then((res) => !!res)
+          .catch((err) => {
+            console.log('Error syncing identity', err)
+            return false
+          })
+      case IdentityProvider.Solana:
+        return this.networkAsset.sync
+          .syncIdentity({ owner: identity.providerId })
+          .then((res) => !!res)
+          .catch((err) => {
+            console.log('Error syncing identity', err)
+            return false
+          })
+      default:
+        throw new Error(`Identity provider ${identity.provider} not supported`)
     }
-    return this.networkAsset.sync
-      .syncIdentity({ owner: identity.providerId })
-      .then((res) => !!res)
-      .catch((err) => {
-        console.log('Error syncing identity', err)
-        return false
-      })
   }
 }

libs/api/identity/data-access/src/lib/api-identity.data-access.module.ts

@@ -1,5 +1,6 @@
 import { Module } from '@nestjs/common'
 import { ApiAuthDataAccessModule } from '@pubkey-link/api-auth-data-access'
+import { ApiBotDataAccessModule } from '@pubkey-link/api-bot-data-access'
 import { ApiCoreDataAccessModule } from '@pubkey-link/api-core-data-access'
 import { ApiNetworkAssetDataAccessModule } from '@pubkey-link/api-network-asset-data-access'
 import { ApiNetworkDataAccessModule } from '@pubkey-link/api-network-data-access'
@@ -12,6 +13,7 @@ import { ApiIdentityService } from './api-identity.service'
 @Module({
   imports: [
     ApiAuthDataAccessModule,
+    ApiBotDataAccessModule,
     ApiNetworkDataAccessModule,
     ApiCoreDataAccessModule,
     ApiNetworkAssetDataAccessModule,

libs/api/identity/feature/src/lib/api-identity.resolver.ts

@@ -6,7 +6,11 @@ import { getIdentityUrl, Identity } from '@pubkey-link/api-identity-data-access'
 export class ApiIdentityResolver {
   @ResolveField(() => String, { nullable: true })
   avatarUrl(@Parent() identity: Identity) {
-    return (identity.profile as { avatarUrl?: string })?.avatarUrl ?? null
+    return (
+      (identity.profile as { avatarUrl?: string })?.avatarUrl ??
+      (identity.profile as { avatarURL?: string })?.avatarURL ??
+      null
+    )
   }
 
   @ResolveField(() => Boolean, { nullable: true })

libs/web/identity/data-access/src/lib/use-admin-find-many-identity.ts

@@ -65,15 +65,20 @@ export function useAdminFindManyIdentity({ ownerId, provider }: { ownerId?: stri
       })
     },
     syncIdentity: async (identity: Identity) => {
-      return sdk.adminSyncIdentity({ identityId: identity.id }).then(async (res) => {
-        if (res) {
-          toastSuccess('Identity synced')
-          await query.refetch()
-          return true
-        }
-        toastError('Error syncing identity')
-        return false
-      })
+      return sdk
+        .adminSyncIdentity({ identityId: identity.id })
+        .then(async (res) => {
+          if (res) {
+            toastSuccess('Identity synced')
+            await query.refetch()
+            return true
+          }
+          toastError('Error syncing identity')
+          return false
+        })
+        .catch((err) => {
+          toastError(err)
+        })
     },
   }
 }