perf(post): define more options in model

mogland · Jul 17, 2022 · 1f90938 · 1f90938
1 parent d53def3
commit 1f90938
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 5 deletions.
diff --git a/src/modules/post/post.controller.ts b/src/modules/post/post.controller.ts
@@ -22,6 +22,7 @@ import { Auth } from "~/common/decorator/auth.decorator";
 import { PostModel } from "./post.model";
 import { Types, PipelineStage } from "mongoose";
 import { MongoIdDto } from "~/shared/dto/id.dto";
+import { IsMaster } from "~/common/decorator/role.decorator";
 @Controller("posts")
 @ApiName
 export class PostController {
@@ -33,9 +34,18 @@ export class PostController {
   @Get('/')
   @Paginator
   @ApiOperation({ summary: "获取文章列表(附带分页器)" })
-  async getPaginate(@Query() query: PagerDto) {
+  async getPaginate(@Query() query: PagerDto, @IsMaster() isMaster: boolean) {
     const { size, select, page, year, sortBy, sortOrder } = query
-
+    let hideProperty
+    if (!isMaster) {
+      hideProperty = {
+        $project: {
+          hide: 0,
+          password: 0,
+          rss: 0,
+        },
+      }
+    }
     return this.postService.model.aggregatePaginate(
       this.postService.model.aggregate(
         [
@@ -46,7 +56,7 @@ export class PostController {
           },
           // @see https://stackoverflow.com/questions/54810712/mongodb-sort-by-field-a-if-field-b-null-otherwise-sort-by-field-c
           {
-            $addFields: {
+            $addFields: { 
               sortField: {
                 // create a new field called "sortField"
                 $cond: {
@@ -70,7 +80,7 @@ export class PostController {
                 },
           },
           {
-            $project: {
+            $project: { // project the fields we want to keep
               sortField: 0, // remove "sort" field if needed
             },
           },
@@ -100,6 +110,8 @@ export class PostController {
               preserveNullAndEmptyArrays: true, // if set to true, MongoDB will still create a document if the array is empty
             },
           },
+          // 移除 hide 字段
+          ...hideProperty,
         ].filter(Boolean) as PipelineStage[],
       ),
       {

diff --git a/src/modules/post/post.model.ts b/src/modules/post/post.model.ts
@@ -90,7 +90,7 @@ export class PostModel extends WriteBaseModel {
       return null;
     }
   })
-  @ApiProperty({ description: "文章置顶" })
+  @ApiProperty({ description: "文章pin日期" })
   pin?: Date | null;
 
   @prop()
@@ -130,6 +130,24 @@ export class PostModel extends WriteBaseModel {
   @IsOptional()
   @ApiProperty({ description: '文章修改时间' })
   modified?: Date
+
+  @prop({ type: Boolean, default: false })
+  @IsOptional()
+  @IsBoolean()
+  @ApiProperty({ description: "文章是否隐藏" })
+  hide?: boolean;
+
+  @prop({ type: String, default: null })
+  @IsOptional()
+  @IsString()
+  @ApiProperty({ description: "文章加密密码（若填写则启动加密）" })
+  password?: string;
+
+  @prop({ type: Boolean, default: true })
+  @IsOptional()
+  @IsBoolean()
+  @ApiProperty({ description: "文章是否公开在RSS输出" })
+  rss?: boolean;
 }