feat(core): support experimental Schema.dynamic()

koishijs · May 6, 2023 · 5d5ba30 · 5d5ba30
1 parent 767a1ce
commit 5d5ba30
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 41 deletions.
diff --git a/packages/core/src/filter.ts b/packages/core/src/filter.ts
@@ -1,51 +1,15 @@
 import { defineProperty } from 'cosmokit'
-import { Context, Schema, Session } from '@satorijs/core'
+import { Context, Session } from '@satorijs/core'
 import { Eval } from '@minatojs/core'
 import { Channel, User } from './database'
 
-declare global {
-  namespace Schemastery {
-    interface Static {
-      path(options?: Path.Options): Schema<string>
-      filter(): Schema<Computed<boolean>>
-      computed<X>(inner: X, options?: Computed.Options): Schema<Computed<TypeS<X>>, Computed<TypeT<X>>>
-    }
-
-    namespace Path {
-      interface Options {
-        filters?: Filter[]
-        allowCreate?: boolean
-      }
-
-      type Filter = FileFilter | 'file' | 'directory'
-
-      interface FileFilter {
-        name: string
-        extensions: string[]
-      }
-    }
-
-    export namespace Computed {
-      export interface Options {
-        userFields?: User.Field[]
-        channelFields?: Channel.Field[]
-      }
-    }
+export namespace Computed {
+  export interface Options {
+    userFields?: User.Field[]
+    channelFields?: Channel.Field[]
   }
 }
 
-Schema.filter = function filter() {
-  return Schema.any().role('filter')
-}
-
-Schema.computed = function computed(inner, options = {}) {
-  return Schema.union([inner, Schema.any().hidden()]).role('computed', options)
-}
-
-Schema.path = function path(options = {}) {
-  return Schema.string().role('path', options)
-}
-
 export type Computed<T> = T | Eval.Expr<T> | ((session: Session) => T)
 export type Filter = (session: Session) => boolean
 

diff --git a/packages/core/src/schema.ts b/packages/core/src/schema.ts
@@ -1,5 +1,47 @@
 import { Dict, remove } from 'cosmokit'
 import { Context, Schema } from '@satorijs/core'
+import { Computed } from './filter'
+
+declare global {
+  namespace Schemastery {
+    interface Static {
+      path(options?: Path.Options): Schema<string>
+      filter(): Schema<Computed<boolean>>
+      computed<X>(inner: X, options?: Computed.Options): Schema<Computed<TypeS<X>>, Computed<TypeT<X>>>
+      dynamic(name: string): Schema
+    }
+
+    namespace Path {
+      interface Options {
+        filters?: Filter[]
+        allowCreate?: boolean
+      }
+
+      type Filter = FileFilter | 'file' | 'directory'
+
+      interface FileFilter {
+        name: string
+        extensions: string[]
+      }
+    }
+  }
+}
+
+Schema.dynamic = function dynamic(name) {
+  return Schema.any().role('dynamic', { name }) as never
+}
+
+Schema.filter = function filter() {
+  return Schema.any().role('filter')
+}
+
+Schema.computed = function computed(inner, options = {}) {
+  return Schema.union([Schema.from(inner), Schema.any().hidden()]).role('computed', options)
+}
+
+Schema.path = function path(options = {}) {
+  return Schema.string().role('path', options)
+}
 
 const kSchemaOrder = Symbol('schema-order')
 
@@ -37,6 +79,15 @@ export class SchemaService {
   get(name: string) {
     return this._data[name] ||= Schema.intersect([])
   }
+
+  set(name: string, schema: Schema) {
+    this._data[name] = schema
+    this.ctx.emit('internal/schema', name)
+    this[Context.current]?.on('dispose', () => {
+      delete this._data[name]
+      this.ctx.emit('internal/schema', name)
+    })
+  }
 }
 
 Context.service('schema', SchemaService)