1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
import { Interface, ToInterface, Cast, FieldCtor, ToConstructor } from "./Types";
import { Doc } from "./Doc";
export type makeInterface<T extends Interface, U extends Doc = Doc> = Partial<ToInterface<T>> & U;
export function makeInterface<T extends Interface, U extends Doc>(schema: T): (doc: U) => makeInterface<T, U> {
return function (doc: any) {
return new Proxy(doc, {
get(target, prop) {
const field = target[prop];
if (prop in schema) {
return Cast(field, (schema as any)[prop]);
}
return field;
}
});
};
}
export type makeStrictInterface<T extends Interface> = Partial<ToInterface<T>>;
export function makeStrictInterface<T extends Interface>(schema: T): (doc: Doc) => makeStrictInterface<T> {
const proto = {};
for (const key in schema) {
const type = schema[key];
Object.defineProperty(proto, key, {
get() {
return Cast(this.__doc[key], type as any);
},
set(value) {
value = Cast(value, type as any);
if (value !== undefined) {
this.__doc[key] = value;
return;
}
throw new TypeError("Expected type " + type);
}
});
}
return function (doc: any) {
const obj = Object.create(proto);
obj.__doc = doc;
return obj;
};
}
export function createSchema<T extends Interface>(schema: T): T & { prototype: ToConstructor<Doc> } {
schema.prototype = Doc;
return schema as any;
}
|