feat(shares): implement shares service

Author: ycdzj

src/shares/dto/share-info.dto.ts

@@ -0,0 +1,28 @@
+import { Share, ShareType } from '../entities/share.entity';
+
+export class ShareInfoDto {
+  id: string;
+  namespaceId: string;
+  resourceId: string;
+  enabled: boolean;
+  requireLogin: boolean;
+  passwordEnabled: boolean;
+  shareType: ShareType;
+  expiresAt: Date | null;
+
+  static fromEntity(share: Share | null): ShareInfoDto {
+    if (!share) {
+      return new ShareInfoDto();
+    }
+    const dto = new ShareInfoDto();
+    dto.id = share.id;
+    dto.namespaceId = share.namespaceId;
+    dto.resourceId = share.resourceId;
+    dto.enabled = share.enabled;
+    dto.requireLogin = share.requireLogin;
+    dto.passwordEnabled = !!share.password;
+    dto.shareType = share.shareType;
+    dto.expiresAt = share.expiresAt;
+    return dto;
+  }
+}

src/shares/dto/update-share-info-req.dto.ts

@@ -0,0 +1,9 @@
+import { ShareType } from '../entities/share.entity';
+
+export class UpdateShareInfoReqDto {
+  enabled?: boolean;
+  requireLogin?: boolean;
+  password?: string | null;
+  shareType?: ShareType;
+  expiresAt?: Date | null;
+}

src/shares/entities/share.entity.ts

@@ -15,7 +15,7 @@ export class Share extends Base {
 
   @BeforeInsert()
   generateId?() {
-    this.id = generateId(16);
+    this.id = generateId(6);
   }
 
   @Column()
@@ -24,6 +24,9 @@ export class Share extends Base {
   @Column()
   resourceId: string;
 
+  @Column()
+  enabled: boolean;
+
   @Column()
   requireLogin: boolean;
 

src/shares/shares.service.ts

@@ -1,7 +1,9 @@
 import { Injectable } from '@nestjs/common';
 import { InjectRepository } from '@nestjs/typeorm';
-import { Share } from './entities/share.entity';
+import { Share, ShareType } from './entities/share.entity';
 import { Repository } from 'typeorm';
+import { ShareInfoDto } from './dto/share-info.dto';
+import { UpdateShareInfoReqDto } from './dto/update-share-info-req.dto';
 
 @Injectable()
 export class SharesService {
@@ -10,4 +12,57 @@ export class SharesService {
     private readonly shareRepo: Repository<Share>,
   ) {}
 
+  async getShareInfo(
+    namespaceId: string,
+    resourceId: string,
+  ): Promise<ShareInfoDto> {
+    const share = await this.shareRepo.findOne({
+      where: {
+        namespaceId,
+        resourceId,
+      },
+    });
+    return ShareInfoDto.fromEntity(share);
+  }
+
+  async updateShareInfo(
+    namespaceId: string,
+    resourceId: string,
+    req: UpdateShareInfoReqDto,
+  ): Promise<ShareInfoDto> {
+    let share = await this.shareRepo.findOne({
+      where: {
+        namespaceId,
+        resourceId,
+      },
+    });
+    if (!share) {
+      share = this.shareRepo.create({
+        namespaceId,
+        resourceId,
+        enabled: false,
+        requireLogin: true,
+        shareType: ShareType.ALL,
+        password: null,
+        expiresAt: null,
+      });
+    }
+    if (req.enabled !== undefined) {
+      share.enabled = req.enabled;
+    }
+    if (req.requireLogin !== undefined) {
+      share.requireLogin = req.requireLogin;
+    }
+    if (req.password !== undefined) {
+      share.password = req.password;
+    }
+    if (req.shareType !== undefined) {
+      share.shareType = req.shareType;
+    }
+    if (req.expiresAt !== undefined) {
+      share.expiresAt = req.expiresAt;
+    }
+    const savedShare = await this.shareRepo.save(share);
+    return ShareInfoDto.fromEntity(savedShare);
+  }
 }