feat: implement collection asset search

Author: beeman

api-schema.graphql

@@ -711,6 +711,7 @@ type Query {
   appConfig: AppConfig!
   me: User
   uptime: Float!
+  userCollectionAssetFindMany(input: UserCollectionAssetFindManyInput!): [CollectionAsset!]
   userCollectionFindMany(input: UserCollectionFindManyInput!): [Collection!]
   userCollectionFindOne(collectionId: String!): Collection
   userFindManyBotRoles(botId: String!, serverId: String!): [BotRole!]
@@ -879,6 +880,11 @@ input UserAddIdentityGrantInput {
   providerId: String!
 }
 
+input UserCollectionAssetFindManyInput {
+  collectionId: String!
+  search: String
+}
+
 input UserCollectionCreateInput {
   communityId: String!
   tokenId: String!

libs/api/collection/data-access/src/index.ts

@@ -1,5 +1,6 @@
 export * from './lib/api-collection.data-access.module'
 export * from './lib/api-collection.service'
+export * from './lib/api-collection-assets.service'
 export * from './lib/dto/user-collection-create-input'
 export * from './lib/dto/user-collection-find-many.input'
 export * from './lib/entity/collection-asset'

libs/api/collection/data-access/src/lib/api-collection-assets.service.ts

@@ -0,0 +1,72 @@
+import { BadRequestException, Injectable, Logger } from '@nestjs/common'
+import { ApiCacheService } from '@pubkey-link/api-cache-data-access'
+import { ApiCoreService } from '@pubkey-link/api-core-data-access'
+import { CollectionAsset } from '@pubkey-link/sdk'
+import { DAS } from 'helius-sdk'
+import slugify from 'slugify'
+import { UserCollectionCreateInput } from './dto/user-collection-create-input'
+import { UserCollectionAssetFindManyInput, UserCollectionFindManyInput } from './dto/user-collection-find-many.input'
+import { CollectionAssetAttribute } from './entity/collection-asset-attribute'
+import { Collection } from './entity/collection.entity'
+
+@Injectable()
+export class ApiCollectionAssetService {
+  constructor(private readonly core: ApiCoreService, private readonly cache: ApiCacheService) {}
+
+  async findMany({ collectionId, search }: UserCollectionAssetFindManyInput): Promise<CollectionAsset[]> {
+    const collection = await this.ensureCollection(collectionId)
+    if (!collection.token) {
+      throw new Error(`Token for collection ${collection.slug} not found`)
+    }
+    const resolver = this.cache.createTokenResolver(collection.token)
+    if (!resolver) {
+      throw new Error(`Resolver ${collection.token.account} not found`)
+    }
+
+    try {
+      const snapshot = await this.cache.assetsSnapshot({ cluster: collection.token.cluster, id: resolver.id })
+      const items: DAS.GetAssetResponse[] = (snapshot.items ?? []) as DAS.GetAssetResponse[]
+
+      const assets = items.map((asset) => ({
+        id: asset.id,
+        name: asset.content?.metadata?.name ?? '',
+        description: asset.content?.metadata?.description ?? '',
+        imageUrl: asset.content?.files?.[0]?.uri ?? '',
+        owner: asset.ownership.owner,
+        attributes: renameAttributes(asset.content?.metadata?.attributes ?? []),
+      }))
+
+      return assets.filter((assets) => {
+        if (!search?.length) {
+          return true
+        }
+
+        return (
+          assets.name.toLowerCase().includes(search.toLowerCase()) ||
+          assets.description.toLowerCase().includes(search.toLowerCase())
+        )
+      })
+    } catch (e) {
+      console.log('error', e)
+      throw e
+    }
+  }
+
+  private async ensureCollection(collectionId: string) {
+    const found = await this.core.data.collection.findUnique({
+      where: { id: collectionId },
+      include: { token: true },
+    })
+    if (!found) {
+      throw new Error(`Collection ${collectionId} not found`)
+    }
+    return found
+  }
+}
+
+function renameAttributes(items: { trait_type: string; value: string }[] = []): CollectionAssetAttribute[] {
+  return items.map((item) => ({
+    key: item.trait_type,
+    value: item.value,
+  }))
+}

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

@@ -2,10 +2,11 @@ import { Module } from '@nestjs/common'
 import { ApiCacheDataAccessModule } from '@pubkey-link/api-cache-data-access'
 import { ApiCoreDataAccessModule } from '@pubkey-link/api-core-data-access'
 import { ApiCollectionService } from './api-collection.service'
+import { ApiCollectionAssetService } from './api-collection-assets.service'
 
 @Module({
   imports: [ApiCoreDataAccessModule, ApiCacheDataAccessModule],
-  providers: [ApiCollectionService],
-  exports: [ApiCollectionService],
+  providers: [ApiCollectionService, ApiCollectionAssetService],
+  exports: [ApiCollectionService, ApiCollectionAssetService],
 })
 export class ApiCollectionDataAccessModule {}

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

@@ -53,7 +53,6 @@ export class ApiCollectionService {
       return {
         ...collection,
         attributes: accumulateAttributes(assets),
-        assets,
       }
     } catch (e) {
       console.log('error', e)

libs/api/collection/data-access/src/lib/dto/user-collection-find-many.input.ts

@@ -7,3 +7,11 @@ export class UserCollectionFindManyInput {
   @Field({ nullable: true })
   search?: string
 }
+
+@InputType()
+export class UserCollectionAssetFindManyInput {
+  @Field()
+  collectionId!: string
+  @Field({ nullable: true })
+  search?: string
+}

libs/api/collection/feature/src/lib/api-collection-asset-user.resolver.ts

@@ -0,0 +1,23 @@
+import { UseGuards } from '@nestjs/common'
+import { Args, Mutation, Query, Resolver } from '@nestjs/graphql'
+import { ApiAuthGraphQLUserGuard, CtxUserId } from '@pubkey-link/api-auth-data-access'
+import {
+  ApiCollectionAssetService,
+  ApiCollectionService,
+  Collection,
+  CollectionAsset,
+  UserCollectionAssetFindManyInput,
+  UserCollectionCreateInput,
+  UserCollectionFindManyInput,
+} from '@pubkey-link/api-collection-data-access'
+
+@Resolver()
+@UseGuards(ApiAuthGraphQLUserGuard)
+export class ApiCollectionAssetUserResolver {
+  constructor(private readonly service: ApiCollectionAssetService) {}
+
+  @Query(() => [CollectionAsset], { nullable: true })
+  userCollectionAssetFindMany(@Args('input') input: UserCollectionAssetFindManyInput) {
+    return this.service.findMany(input)
+  }
+}

libs/api/collection/feature/src/lib/api-collection.feature.module.ts

@@ -2,9 +2,10 @@ import { Module } from '@nestjs/common'
 import { ApiCollectionDataAccessModule } from '@pubkey-link/api-collection-data-access'
 import { ApiCollectionUserResolver } from './api-collection-user.resolver'
 import { ApiCollectionResolver } from './api-collection.resolver'
+import { ApiCollectionAssetUserResolver } from './api-collection-asset-user.resolver'
 
 @Module({
   imports: [ApiCollectionDataAccessModule],
-  providers: [ApiCollectionResolver, ApiCollectionUserResolver],
+  providers: [ApiCollectionResolver, ApiCollectionAssetUserResolver, ApiCollectionUserResolver],
 })
 export class ApiCollectionFeatureModule {}

libs/sdk/src/generated/graphql-sdk.ts

@@ -1080,6 +1080,7 @@ export type Query = {
   appConfig: AppConfig
   me?: Maybe<User>
   uptime: Scalars['Float']['output']
+  userCollectionAssetFindMany?: Maybe<Array<CollectionAsset>>
   userCollectionFindMany?: Maybe<Array<Collection>>
   userCollectionFindOne?: Maybe<Collection>
   userFindManyBotRoles?: Maybe<Array<BotRole>>
@@ -1222,6 +1223,10 @@ export type QueryAnonRequestIdentityChallengeArgs = {
   input: RequestIdentityChallengeInput
 }
 
+export type QueryUserCollectionAssetFindManyArgs = {
+  input: UserCollectionAssetFindManyInput
+}
+
 export type QueryUserCollectionFindManyArgs = {
   input: UserCollectionFindManyInput
 }
@@ -1509,6 +1514,11 @@ export type UserAddIdentityGrantInput = {
   providerId: Scalars['String']['input']
 }
 
+export type UserCollectionAssetFindManyInput = {
+  collectionId: Scalars['String']['input']
+  search?: InputMaybe<Scalars['String']['input']>
+}
+
 export type UserCollectionCreateInput = {
   communityId: Scalars['String']['input']
   tokenId: Scalars['String']['input']
@@ -2503,6 +2513,28 @@ export type UserCollectionFindManyQuery = {
   }> | null
 }
 
+export type UserCollectionAssetFindManyQueryVariables = Exact<{
+  input: UserCollectionAssetFindManyInput
+}>
+
+export type UserCollectionAssetFindManyQuery = {
+  __typename?: 'Query'
+  items?: Array<{
+    __typename?: 'CollectionAsset'
+    id: string
+    name: string
+    description: string
+    imageUrl: string
+    owner: string
+    attributes?: Array<{
+      __typename?: 'CollectionAssetAttribute'
+      key: string
+      value?: string | null
+      count?: number | null
+    }> | null
+  }> | null
+}
+
 export type UserCollectionFindOneQueryVariables = Exact<{
   collectionId: Scalars['String']['input']
 }>
@@ -2522,20 +2554,6 @@ export type UserCollectionFindOneQuery = {
       value?: string | null
       count?: number | null
     }> | null
-    assets?: Array<{
-      __typename?: 'CollectionAsset'
-      id: string
-      name: string
-      description: string
-      imageUrl: string
-      owner: string
-      attributes?: Array<{
-        __typename?: 'CollectionAssetAttribute'
-        key: string
-        value?: string | null
-        count?: number | null
-      }> | null
-    }> | null
     token?: {
       __typename?: 'NetworkToken'
       id: string
@@ -9774,21 +9792,25 @@ export const UserCollectionFindManyDocument = gql`
   }
   ${CollectionDetailsFragmentDoc}
 `
+export const UserCollectionAssetFindManyDocument = gql`
+  query userCollectionAssetFindMany($input: UserCollectionAssetFindManyInput!) {
+    items: userCollectionAssetFindMany(input: $input) {
+      ...CollectionAssetDetails
+    }
+  }
+  ${CollectionAssetDetailsFragmentDoc}
+`
 export const UserCollectionFindOneDocument = gql`
   query userCollectionFindOne($collectionId: String!) {
     item: userCollectionFindOne(collectionId: $collectionId) {
       ...CollectionDetails
       attributes {
         ...CollectionAssetAttributeDetails
       }
-      assets {
-        ...CollectionAssetDetails
-      }
     }
   }
   ${CollectionDetailsFragmentDoc}
   ${CollectionAssetAttributeDetailsFragmentDoc}
-  ${CollectionAssetDetailsFragmentDoc}
 `
 export const UserCollectionCreateDocument = gql`
   mutation userCollectionCreate($input: UserCollectionCreateInput!) {
@@ -10844,6 +10866,7 @@ const AdminCacheDetailDocumentString = print(AdminCacheDetailDocument)
 const AdminCacheResolveDocumentString = print(AdminCacheResolveDocument)
 const AdminCacheSyncDocumentString = print(AdminCacheSyncDocument)
 const UserCollectionFindManyDocumentString = print(UserCollectionFindManyDocument)
+const UserCollectionAssetFindManyDocumentString = print(UserCollectionAssetFindManyDocument)
 const UserCollectionFindOneDocumentString = print(UserCollectionFindOneDocument)
 const UserCollectionCreateDocumentString = print(UserCollectionCreateDocument)
 const UserCollectionDeleteDocumentString = print(UserCollectionDeleteDocument)
@@ -11700,6 +11723,27 @@ export function getSdk(client: GraphQLClient, withWrapper: SdkFunctionWrapper =
         variables,
       )
     },
+    userCollectionAssetFindMany(
+      variables: UserCollectionAssetFindManyQueryVariables,
+      requestHeaders?: GraphQLClientRequestHeaders,
+    ): Promise<{
+      data: UserCollectionAssetFindManyQuery
+      errors?: GraphQLError[]
+      extensions?: any
+      headers: Headers
+      status: number
+    }> {
+      return withWrapper(
+        (wrappedRequestHeaders) =>
+          client.rawRequest<UserCollectionAssetFindManyQuery>(UserCollectionAssetFindManyDocumentString, variables, {
+            ...requestHeaders,
+            ...wrappedRequestHeaders,
+          }),
+        'userCollectionAssetFindMany',
+        'query',
+        variables,
+      )
+    },
     userCollectionFindOne(
       variables: UserCollectionFindOneQueryVariables,
       requestHeaders?: GraphQLClientRequestHeaders,
@@ -14366,6 +14410,13 @@ export function UserAddIdentityGrantInputSchema(): z.ZodObject<Properties<UserAd
   })
 }
 
+export function UserCollectionAssetFindManyInputSchema(): z.ZodObject<Properties<UserCollectionAssetFindManyInput>> {
+  return z.object({
+    collectionId: z.string(),
+    search: z.string().nullish(),
+  })
+}
+
 export function UserCollectionCreateInputSchema(): z.ZodObject<Properties<UserCollectionCreateInput>> {
   return z.object({
     communityId: z.string(),

libs/sdk/src/graphql/feature-collection.graphql

@@ -32,15 +32,18 @@ query userCollectionFindMany($input: UserCollectionFindManyInput!) {
   }
 }
 
+query userCollectionAssetFindMany($input: UserCollectionAssetFindManyInput!) {
+  items: userCollectionAssetFindMany(input: $input) {
+    ...CollectionAssetDetails
+  }
+}
+
 query userCollectionFindOne($collectionId: String!) {
   item: userCollectionFindOne(collectionId: $collectionId) {
     ...CollectionDetails
     attributes {
       ...CollectionAssetAttributeDetails
     }
-    assets {
-      ...CollectionAssetDetails
-    }
   }
 }