feat: import contacts file

Author: marian2js

apps/api/src/chat/resolvers/contact.resolver.ts

@@ -7,6 +7,7 @@ import { getXmtpContacts } from '@chainjet/tools/dist/messages'
 import { BadRequestException, Logger, UnauthorizedException, UseGuards, UseInterceptors } from '@nestjs/common'
 import { Args, Mutation, Resolver } from '@nestjs/graphql'
 import { Authorizer, AuthorizerInterceptor, InjectAuthorizer } from '@ptc-org/nestjs-query-graphql'
+import AWS from 'aws-sdk'
 import { getAddress, isAddress } from 'ethers/lib/utils'
 import { GraphQLBoolean, GraphQLString } from 'graphql'
 import { ObjectId } from 'mongoose'
@@ -19,6 +20,13 @@ import { UserService } from '../../users/services/user.service'
 import { Contact, CreateContactInput, UpdateContactInput } from '../entities/contact'
 import { ContactService } from '../services/contact.service'
 
+AWS.config.update({
+  accessKeyId: process.env.AWS_ACCESS_KEY_ID,
+  secretAccessKey: process.env.AWS_SECRET_ACCESS_KEY,
+  region: 'us-east-2',
+})
+const s3 = new AWS.S3()
+
 @Resolver(() => Contact)
 @UseGuards(GraphqlGuard)
 @UseInterceptors(AuthorizerInterceptor(Contact))
@@ -69,6 +77,85 @@ export class ContactResolver extends BaseResolver(Contact, {
     }
   }
 
+  @Mutation(() => GraphQLString)
+  async generateContactsPresignedUrl(
+    @UserId() userId: ObjectId,
+    @Args({ name: 'id', type: () => GraphQLString }) id: string,
+  ): Promise<string> {
+    if (!userId) {
+      throw new UnauthorizedException('Not logged in')
+    }
+    const user = await this.userService.findOne({ _id: userId })
+    if (!user) {
+      throw new BadRequestException('User not found')
+    }
+    const threeDays = new Date()
+    threeDays.setDate(threeDays.getDate() + 3)
+    const params = {
+      Bucket: 'chainjet-contacts',
+      Fields: {
+        key: `${userId}/${id}.txt`,
+      },
+      Expires: 60,
+      Metadata: {
+        Expires: threeDays.toISOString(),
+      },
+      Conditions: [['content-length-range', 0, 52428800]], // up to 50MB
+    }
+    return new Promise((resolve, reject) => {
+      s3.createPresignedPost(params, function (err, data) {
+        if (err) {
+          reject(err)
+        } else {
+          resolve(JSON.stringify(data))
+        }
+      })
+    })
+  }
+
+  @Mutation(() => ResultPayload)
+  async addContactsFile(
+    @UserId() userId: ObjectId,
+    @Args({ name: 'id', type: () => GraphQLString }) id: string,
+    @Args({ name: 'tags', type: () => [GraphQLString], nullable: true }) tags?: string[],
+    @Args({ name: 'limitToPlan', type: () => GraphQLBoolean, nullable: true }) limitToPlan?: boolean,
+  ): Promise<ResultPayload> {
+    if (!userId) {
+      throw new UnauthorizedException('Not logged in')
+    }
+    const user = await this.userService.findOne({ _id: userId })
+    if (!user) {
+      throw new BadRequestException('User not found')
+    }
+
+    const params = {
+      Bucket: 'chainjet-contacts',
+      Key: `${userId}/${id}.txt`,
+    }
+
+    let fileContent = ''
+    try {
+      const data = await s3.getObject(params).promise()
+      fileContent = data.Body?.toString('utf-8') || ''
+    } catch (error) {
+      throw new Error(`Failed to import the contacts: ${error.message}`)
+    }
+    const addresses = fileContent.split('\n')
+    if (!addresses || addresses.length === 0) {
+      throw new BadRequestException('Addresses cannot be empty')
+    }
+    if (addresses.some((address) => !address || !isAddress(address))) {
+      throw new BadRequestException(
+        `Address ${addresses.find((address) => !address || !isAddress(address))} is not valid`,
+      )
+    }
+    this.logger.log(`Importing ${addresses.length} addresses - user ${user.id}`)
+    await this.contactService.addContacts(addresses, user, tags, limitToPlan)
+    return {
+      success: true,
+    }
+  }
+
   @Mutation(() => ResultPayload)
   async importXmtpContacts(
     @UserId() userId: ObjectId,

generated/graphql.ts

@@ -1703,6 +1703,8 @@ export interface IMutation {
     updateOneContact(input: UpdateOneContactInput): Contact | Promise<Contact>;
     deleteOneContact(input: DeleteOneContactInput): ContactDeleteResponse | Promise<ContactDeleteResponse>;
     addContacts(addresses: string[], tags?: Nullable<string[]>, limitToPlan?: Nullable<boolean>): ResultPayload | Promise<ResultPayload>;
+    generateContactsPresignedUrl(id: string): string | Promise<string>;
+    addContactsFile(id: string, tags?: Nullable<string[]>, limitToPlan?: Nullable<boolean>): ResultPayload | Promise<ResultPayload>;
     importXmtpContacts(tags?: Nullable<string[]>, limitToPlan?: Nullable<boolean>): ResultPayload | Promise<ResultPayload>;
     importPoapContacts(eventId: string, tags?: Nullable<string[]>, limitToPlan?: Nullable<boolean>): ResultPayload | Promise<ResultPayload>;
     importLensFollowersContacts(handle: string, tags?: Nullable<string[]>, limitToPlan?: Nullable<boolean>): ResultPayload | Promise<ResultPayload>;

generated/schema.graphql

@@ -1823,6 +1823,8 @@ type Mutation {
   updateOneContact(input: UpdateOneContactInput!): Contact!
   deleteOneContact(input: DeleteOneContactInput!): ContactDeleteResponse!
   addContacts(addresses: [String!]!, tags: [String!], limitToPlan: Boolean): ResultPayload!
+  generateContactsPresignedUrl(id: String!): String!
+  addContactsFile(id: String!, tags: [String!], limitToPlan: Boolean): ResultPayload!
   importXmtpContacts(tags: [String!], limitToPlan: Boolean): ResultPayload!
   importPoapContacts(eventId: String!, tags: [String!], limitToPlan: Boolean): ResultPayload!
   importLensFollowersContacts(handle: String!, tags: [String!], limitToPlan: Boolean): ResultPayload!