From 6c0cac107fdeecf658aaed2746b7cc44dac328cd Mon Sep 17 00:00:00 2001
From: Innei <tukon479@gmail.com>
Date: Mon, 31 Jan 2022 18:21:37 +0800
Subject: [PATCH] fix: limit apply link char length

---
 src/modules/comment/comment.model.ts |  8 +++++++-
 src/modules/link/link.model.ts       | 13 ++++++++++++-
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/modules/comment/comment.model.ts b/src/modules/comment/comment.model.ts
index ec0a77a1..6c89ea00 100644
--- a/src/modules/comment/comment.model.ts
+++ b/src/modules/comment/comment.model.ts
@@ -7,6 +7,7 @@ import {
 } from '@typegoose/typegoose'
 import { BeAnObject } from '@typegoose/typegoose/lib/types'
 import { Query, Types } from 'mongoose'
+import { URL } from 'url'
 import { BaseModel } from '~/shared/model/base.model'
 import { getAvatar } from '~/utils'
 import { NoteModel } from '../note/note.model'
@@ -58,7 +59,12 @@ export class CommentModel extends BaseModel {
   @prop({ trim: true })
   mail?: string
 
-  @prop({ trim: true })
+  @prop({
+    trim: true,
+    set(val) {
+      return new URL(val).origin
+    },
+  })
   url?: string
 
   @prop({ required: true })
diff --git a/src/modules/link/link.model.ts b/src/modules/link/link.model.ts
index a42fe5ca..f311554a 100644
--- a/src/modules/link/link.model.ts
+++ b/src/modules/link/link.model.ts
@@ -1,7 +1,14 @@
 import { ApiProperty } from '@nestjs/swagger'
 import { modelOptions, prop } from '@typegoose/typegoose'
 import { Transform } from 'class-transformer'
-import { IsEmail, IsEnum, IsOptional, IsString, IsUrl } from 'class-validator'
+import {
+  IsEmail,
+  IsEnum,
+  IsOptional,
+  IsString,
+  IsUrl,
+  MaxLength,
+} from 'class-validator'
 import { range } from 'lodash'
 import { URL } from 'url'
 import { BaseModel } from '~/shared/model/base.model'
@@ -24,6 +31,7 @@ export enum LinkState {
 export class LinkModel extends BaseModel {
   @prop({ required: true, trim: true, unique: true })
   @IsString()
+  @MaxLength(30)
   /**
    * name is site name
    */
@@ -48,11 +56,13 @@ export class LinkModel extends BaseModel {
   @prop({ trim: true })
   // 对空字符串处理
   @Transform(({ value }) => (value === '' ? null : value))
+  @MaxLength(200)
   avatar?: string
 
   @IsOptional()
   @IsString()
   @prop({ trim: true })
+  @MaxLength(150)
   description?: string
 
   @IsOptional()
@@ -71,6 +81,7 @@ export class LinkModel extends BaseModel {
   @IsOptional()
   // 对空字符串处理
   @Transform(({ value }) => (value === '' ? null : value))
+  @MaxLength(50)
   email?: string
   get hide() {
     return this.state === LinkState.Audit