From c42bf77644243192a83e1d4f0fee6cb9be2498b3 Mon Sep 17 00:00:00 2001
From: Shigma <1700011071@pku.edu.cn>
Date: Fri, 4 Sep 2020 21:55:54 +0800
Subject: [PATCH] feat(core): narrow parsedArgv.session typings

---
 packages/koishi-core/src/command.ts          | 8 ++++++--
 packages/koishi-core/src/plugins/validate.ts | 6 +++---
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/koishi-core/src/command.ts b/packages/koishi-core/src/command.ts
index e36f933683..5af3ee6ec2 100644
--- a/packages/koishi-core/src/command.ts
+++ b/packages/koishi-core/src/command.ts
@@ -84,7 +84,7 @@ export interface ParsedLine<O = {}> {
 
 export interface ParsedArgv<U extends User.Field = never, G extends Group.Field = never, O = {}> extends Partial<ParsedLine<O>> {
   command: Command<U, G, O>
-  session: Session<U, G>
+  session: ArgvSession<U, G, O>
   next?: NextFunction
 }
 
@@ -93,6 +93,10 @@ export interface ExecuteArgv extends Partial<ParsedLine> {
   next?: NextFunction
 }
 
+export interface ArgvSession<U extends User.Field, G extends Group.Field, O extends {}> extends Session<U, G> {
+  $argv?: ParsedArgv<U, G, O>
+}
+
 export interface CommandConfig<U extends User.Field = never, G extends Group.Field = never> {
   /** description */
   description?: string
@@ -107,7 +111,7 @@ export type FieldCollector<T extends TableType, K = keyof Tables[T], O extends {
   | ((argv: ParsedArgv<never, never, O>, fields: Set<keyof Tables[T]>) => void)
 
 export type CommandAction<U extends User.Field = never, G extends Group.Field = never, O = {}> =
-  (this: Command<U, G>, config: ParsedArgv<U, G, O>, ...args: string[]) => void | string | Promise<void | string>
+  (this: Command<U, G, O>, argv: ParsedArgv<U, G, O>, ...args: string[]) => void | string | Promise<void | string>
 
 export class Command<U extends User.Field = never, G extends Group.Field = never, O = {}> {
   config: CommandConfig<U, G>
diff --git a/packages/koishi-core/src/plugins/validate.ts b/packages/koishi-core/src/plugins/validate.ts
index 007ba3f524..6e320e7cb0 100644
--- a/packages/koishi-core/src/plugins/validate.ts
+++ b/packages/koishi-core/src/plugins/validate.ts
@@ -9,9 +9,9 @@ import { Message } from './message'
 export type UserType<T, U extends User.Field = User.Field> = T | ((user: Pick<User, U>) => T)
 
 declare module '../command' {
-  interface Command<U, G> {
-    _checkers: ((session: Session<U, G>) => string | boolean)[]
-    before(checker: (session: Session<U, G>) => string | boolean): this
+  interface Command<U, G, O> {
+    _checkers: ((session: ArgvSession<U, G, O>) => string | boolean)[]
+    before(checker: (session: ArgvSession<U, G, O>) => string | boolean): this
     getConfig<K extends keyof CommandConfig>(key: K, session: Session): Exclude<CommandConfig[K], (user: User) => any>
   }