feat(command): pass command to completion handler

c4spar · Jun 14, 2020 · 1e8d51b · 1e8d51b
1 parent 8dfe004
commit 1e8d51b
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/packages/command/commands/completions/complete.ts b/packages/command/commands/completions/complete.ts
@@ -30,7 +30,7 @@ export class CompleteCommand extends DefaultCommand {
                 }
 
                 const completion: ICompleteSettings | undefined = cmd.getCompletion( action );
-                const result: string[] = await completion?.complete() ?? [];
+                const result: string[] = await completion?.complete( cmd ) ?? [];
 
                 if ( result?.length ) {
                     Deno.stdout.writeSync( encode( result.join( ' ' ) ) );

diff --git a/packages/command/lib/base-command.ts b/packages/command/lib/base-command.ts
@@ -294,7 +294,7 @@ export class BaseCommand<O = any, A extends Array<any> = any> {
         this.cmd.types.set( name, { ...options, name, handler } );
 
         if ( handler instanceof Type && typeof handler.complete !== 'undefined' ) {
-            this.complete( name, () => handler.complete?.() || [], options );
+            this.complete( name, ( cmd: BaseCommand ) => handler.complete?.( cmd ) || [], options );
         }
 
         return this;

diff --git a/packages/command/lib/types.ts b/packages/command/lib/types.ts
@@ -96,7 +96,7 @@ export interface ICompleteSettings extends ICompleteOptions {
 }
 
 /** Type parser method. */
-export type ICompleteHandler = () => string[] | Promise<string[]>;
+export type ICompleteHandler = ( cmd: BaseCommand ) => string[] | Promise<string[]>;
 
 export interface IHelpCommand<O = any> extends BaseCommand<O> {
     show( name?: string ): void;

diff --git a/packages/command/types/type.ts b/packages/command/types/type.ts
@@ -1,8 +1,9 @@
 import { IFlagArgument, IFlagOptions } from '../../flags/lib/types.ts';
+import { BaseCommand } from '../lib/base-command.ts';
 
 export abstract class Type<T> {
 
     public abstract parse( option: IFlagOptions, arg: IFlagArgument, value: string ): T
 
-    public complete?(): string[];
+    public complete?( cmd: BaseCommand ): string[];
 }