Upload Workspace logo during onboarding (#542)

* Upload image

* Upload image

* Fix tests

* Remove pictures from seeds

* Fix storybook

* Fix storybook

* Fix storybook

server/src/core/workspace/resolvers/workspace.resolver.spec.ts

@@ -1,8 +1,10 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { WorkspaceResolver } from './workspace.resolver';
 import { WorkspaceService } from '../services/workspace.service';
+import { FileUploadService } from 'src/core/file/services/file-upload.service';
+import { AbilityFactory } from 'src/ability/ability.factory';
 
-describe('WorkspaceMemberResolver', () => {
+describe('WorkspaceResolver', () => {
   let resolver: WorkspaceResolver;
 
   beforeEach(async () => {
@@ -10,6 +12,8 @@ describe('WorkspaceMemberResolver', () => {
       providers: [
         WorkspaceResolver,
         { provide: WorkspaceService, useValue: {} },
+        { provide: AbilityFactory, useValue: {} },
+        { provide: FileUploadService, useValue: {} },
       ],
     }).compile();
 

server/src/core/workspace/resolvers/workspace.resolver.ts

@@ -11,13 +11,25 @@ import { WorkspaceUpdateInput } from 'src/core/@generated/workspace/workspace-up
 import { JwtAuthGuard } from 'src/guards/jwt.auth.guard';
 import { Prisma } from '@prisma/client';
 import { assert } from 'src/utils/assert';
+import { FileUploadService } from 'src/core/file/services/file-upload.service';
+import { FileUpload, GraphQLUpload } from 'graphql-upload';
+import { streamToBuffer } from 'src/utils/stream-to-buffer';
+import { FileFolder } from 'src/core/file/interfaces/file-folder.interface';
+import { AbilityGuard } from 'src/guards/ability.guard';
+import { CheckAbilities } from 'src/decorators/check-abilities.decorator';
+import { UpdateWorkspaceAbilityHandler } from 'src/ability/handlers/workspace.ability-handler';
 
 @UseGuards(JwtAuthGuard)
 @Resolver(() => Workspace)
 export class WorkspaceResolver {
-  constructor(private readonly workspaceService: WorkspaceService) {}
+  constructor(
+    private readonly workspaceService: WorkspaceService,
+    private readonly fileUploadService: FileUploadService,
+  ) {}
 
   @Mutation(() => Workspace)
+  @UseGuards(AbilityGuard)
+  @CheckAbilities(UpdateWorkspaceAbilityHandler)
   async updateWorkspace(
     @Args('data') data: WorkspaceUpdateInput,
     @AuthWorkspace() workspace: Workspace,
@@ -51,4 +63,33 @@ export class WorkspaceResolver {
 
     return selectedWorkspace;
   }
+
+  @UseGuards(AbilityGuard)
+  @CheckAbilities(UpdateWorkspaceAbilityHandler)
+  @Mutation(() => String)
+  async uploadWorkspaceLogo(
+    @AuthWorkspace() workspace: Workspace,
+    @Args({ name: 'file', type: () => GraphQLUpload })
+    { createReadStream, filename, mimetype }: FileUpload,
+  ): Promise<string> {
+    const stream = createReadStream();
+    const buffer = await streamToBuffer(stream);
+    const fileFolder = FileFolder.WorkspaceLogo;
+
+    const { paths } = await this.fileUploadService.uploadImage({
+      file: buffer,
+      filename,
+      mimeType: mimetype,
+      fileFolder,
+    });
+
+    await this.workspaceService.update({
+      where: { id: workspace.id },
+      data: {
+        logo: paths[0],
+      },
+    });
+
+    return paths[0];
+  }
 }

server/src/core/workspace/workspace.module.ts

@@ -3,10 +3,12 @@ import { WorkspaceService } from './services/workspace.service';
 import { WorkspaceMemberService } from './services/workspace-member.service';
 import { WorkspaceMemberResolver } from './resolvers/workspace-member.resolver';
 import { WorkspaceResolver } from './resolvers/workspace.resolver';
+import { FileUploadService } from '../file/services/file-upload.service';
 
 @Module({
   providers: [
     WorkspaceService,
+    FileUploadService,
     WorkspaceMemberService,
     WorkspaceMemberResolver,
     WorkspaceResolver,