From 728b4d8a32ce11df0655bb552baaa8f02c698c90 Mon Sep 17 00:00:00 2001
From: Shigma <shigma10826@gmail.com>
Date: Mon, 6 May 2024 20:51:44 +0800
Subject: [PATCH] feat(minato): enhance typings, add overloads

---
 packages/core/src/database.ts  |  8 ++++++--
 packages/core/src/index.ts     |  4 ++++
 packages/core/src/query.ts     | 14 +++++++-------
 packages/core/src/selection.ts |  3 ++-
 4 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/packages/core/src/database.ts b/packages/core/src/database.ts
index 0d0d367c..7882bc5d 100644
--- a/packages/core/src/database.ts
+++ b/packages/core/src/database.ts
@@ -265,12 +265,16 @@ export class Database<S = {}, N = {}, C extends Context = Context> extends Servi
     return this.select(sel)
   }
 
+  async get<K extends Keys<S>>(table: K, query: Query<S[K]>): Promise<S[K][]>
+
   async get<K extends Keys<S>, P extends FlatKeys<S[K]> = any>(
     table: K,
     query: Query<S[K]>,
     cursor?: Driver.Cursor<P>,
-  ): Promise<FlatPick<S[K], P>[]> {
-    return this.select(table, query).execute(cursor)
+  ): Promise<FlatPick<S[K], P>[]>
+
+  async get<K extends Keys<S>>(table: K, query: Query<S[K]>, cursor?: any) {
+    return this.select(table, query).execute(cursor) as any
   }
 
   async eval<K extends Keys<S>, T>(table: K, expr: Selection.Callback<S[K], T, true>, query?: Query<S[K]>): Promise<T> {
diff --git a/packages/core/src/index.ts b/packages/core/src/index.ts
index 630cf5bc..54ace897 100644
--- a/packages/core/src/index.ts
+++ b/packages/core/src/index.ts
@@ -27,7 +27,11 @@ declare module 'cordis' {
   namespace Context {
     const Minato: unique symbol
     const Database: unique symbol
+    // https://github.com/typescript-eslint/typescript-eslint/issues/6720
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
     interface Minato<C extends Context = Context> {}
+    // https://github.com/typescript-eslint/typescript-eslint/issues/6720
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
     interface Database<C extends Context = Context> {}
   }
 }
diff --git a/packages/core/src/query.ts b/packages/core/src/query.ts
index fc45e361..0c5ecdb0 100644
--- a/packages/core/src/query.ts
+++ b/packages/core/src/query.ts
@@ -8,9 +8,9 @@ export type Query<T = any> = Query.Expr<Flatten<T>> | Query.Shorthand<Indexable>
 export namespace Query {
   export interface FieldExpr<T = any> {
     // logical
-    $or?: FieldQuery<T>[]
-    $and?: FieldQuery<T>[]
-    $not?: FieldQuery<T>
+    $or?: Field<T>[]
+    $and?: Field<T>[]
+    $not?: Field<T>
 
     // existence
     $exists?: boolean
@@ -28,7 +28,7 @@ export namespace Query {
     $lte?: Extract<T, Comparable>
 
     // list
-    $el?: T extends (infer U)[] ? FieldQuery<U> : never
+    $el?: T extends (infer U)[] ? Field<U> : never
     $size?: Extract<T, any[], number>
 
     // regexp
@@ -55,10 +55,10 @@ export namespace Query {
     | Extract<T, Indexable, T[]>
     | Extract<T, string, RegExp>
 
-  export type FieldQuery<T = any> = FieldExpr<T> | Shorthand<T>
+  export type Field<T = any> = FieldExpr<T> | Shorthand<T>
 
   export type Expr<T = any> = LogicalExpr<T> & {
-    [K in keyof T]?: null | FieldQuery<T[K]>
+    [K in keyof T]?: null | Field<T[K]>
   }
 }
 
@@ -102,7 +102,7 @@ const queryOperators: QueryOperators = {
   $size: (query, data) => data.length === query,
 }
 
-function executeFieldQuery(query: Query.FieldQuery, data: any) {
+function executeFieldQuery(query: Query.Field, data: any) {
   // shorthand syntax
   if (Array.isArray(query)) {
     return query.includes(data)
diff --git a/packages/core/src/selection.ts b/packages/core/src/selection.ts
index 4e5b78db..3b89b759 100644
--- a/packages/core/src/selection.ts
+++ b/packages/core/src/selection.ts
@@ -250,7 +250,8 @@ export class Selection<S = any> extends Executable<S, S[]> {
     return Eval.exec(selection._action('eval', expr))
   }
 
-  execute<K extends FlatKeys<S> = any>(cursor?: Driver.Cursor<K>): Promise<Extract<FlatPick<S, K>, S>[]>
+  execute(): Promise<S[]>
+  execute<K extends FlatKeys<S> = any>(cursor?: Driver.Cursor<K>): Promise<FlatPick<S, K>[]>
   execute<T>(callback: Selection.Callback<S, T, true>): Promise<T>
   async execute(cursor?: any) {
     if (typeof cursor === 'function') {