feat: model $setAttr

KHC-ZhiHao · Dec 3, 2022 · 57e77a6 · 57e77a6
1 parent b184264
commit 57e77a6
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 1 deletion.
diff --git a/core/model.ts b/core/model.ts
@@ -3,6 +3,12 @@ import Container from './container'
 import ModelUnit from './model-unit'
 import { EventCallback, MakeModelOptions } from './types'
 
+type KeysOfTypeStrict<T, U> = {
+    [P in keyof T]: T[P] extends U ? (U extends T[P] ? P : never) : never
+}[keyof T]
+
+type PickByTypeStrict<U, T> = Pick<T, KeysOfTypeStrict<T, U>>
+
 class Model {
     _model: ModelUnit
     _container!: Container
@@ -85,6 +91,17 @@ class Model {
         return this._model.getBody()
     }
 
+    $setAttr(data: Partial<Omit<PickByTypeStrict<number | boolean | string, this>, keyof Model>>) {
+        for (let key in data) {
+            if (key in this._model.body) {
+                let type = Utils.getType(this._model.body[key])
+                if (['string', 'number', 'boolean', 'empty'].includes(type)) {
+                    this._model.body[key] = (data as any)[key]
+                }
+            }
+        }
+    }
+
     $keys() {
         return this._model.getKeys()
     }

diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/model.d.ts b/dist/model.d.ts
@@ -2,6 +2,10 @@ import Utils from './utils';
 import Container from './container';
 import ModelUnit from './model-unit';
 import { EventCallback, MakeModelOptions } from './types';
+type KeysOfTypeStrict<T, U> = {
+    [P in keyof T]: T[P] extends U ? (U extends T[P] ? P : never) : never;
+}[keyof T];
+type PickByTypeStrict<U, T> = Pick<T, KeysOfTypeStrict<T, U>>;
 declare class Model {
     _model: ModelUnit;
     _container: Container;
@@ -29,6 +33,7 @@ declare class Model {
     $init(data?: any): this;
     $copy(options?: MakeModelOptions): this;
     $body(): any;
+    $setAttr(data: Partial<Omit<PickByTypeStrict<number | boolean | string, this>, keyof Model>>): void;
     $keys(): string[];
     $reset(key?: keyof Omit<this, keyof Model>): void;
     $rules(name: keyof Omit<this, keyof Model>): import("./types").RuleCallback[];

diff --git a/docs-code/structure/model.md b/docs-code/structure/model.md
@@ -489,6 +489,14 @@ model.$copy = function(options?: { save?: boolean }) => Model
 model.$body = function() => { [key: string]: any }
 ```
 
+### $setAttr
+
+批次更新常數屬性，意指更新屬性為 `number` | `string` | `boolean` | `null` | `undefined`  的值。
+
+```ts
+model.$setAttr = function(data: { [key: string]: any }) => void
+```
+
 ### $keys
 
 回傳 [body](#body) 與 [refs](#refs) 的 key 值。