Add directories list

Readme.md

@@ -7,6 +7,7 @@ Including:
 - [x] `buffers` current buffer list.
 - [x] `cmdhistory` history of commands.
 - [x] `colors` colors schemes.
+- [x] `directories` search directories from current cwd.
 - [x] `files` search files from current cwd.
 - [x] `filetypes` file types.
 - [x] `grep` grep text from current cwd.

package.json

@@ -48,6 +48,27 @@
             "type": "string"
           }
         },
+        "list.source.directories.command": {
+          "type": "string",
+          "default": "",
+          "description": "Command used for search for directories"
+        },
+        "list.source.directories.args": {
+          "type": "array",
+          "default": [],
+          "description": "Arguments for search command",
+          "items": {
+            "type": "string"
+          }
+        },
+        "list.source.directories.excludePatterns": {
+          "type": "array",
+          "default": [],
+          "description": "Minimatch patterns that should be excluded.",
+          "items": {
+            "type": "string"
+          }
+        },
         "list.source.files.command": {
           "type": "string",
           "default": "",

src/directories.ts

@@ -0,0 +1,58 @@
+import { ChildProcess, spawn } from 'child_process'
+import { workspace } from 'coc.nvim'
+import FilesList, { Task as FilesTask } from './files'
+import { executable } from './util'
+
+class Task extends FilesTask {
+  protected post_process_hook(process: ChildProcess) {
+    let post_process = spawn('sh', ['-c', 'xargs -I {} gdirname {} | sort | uniq'])
+    this.processes.push(post_process)
+    post_process.on('error', e => {
+      this.emit('error', e.message)
+    })
+    process.stdout.pipe(post_process.stdin)
+
+    return post_process.stdout
+  }
+}
+
+export default class DirectoriesList extends FilesList {
+  public task = Task
+  public readonly name = 'directories'
+  public description = 'Search directories by rg or ag'
+  public readonly detail = `Install ripgrep in your $PATH to have best experience.
+Directories are searched from current cwd by default.
+Provide directory names as arguments to search other directories.
+Use 'list.source.directories.command' configuration for custom search command.
+Use 'list.source.directories.args' configuration for custom command arguments.
+Note that rg ignore hidden directories by default.`
+  public options = [{
+    name: '-F, -folder',
+    description: 'Search directories from current workspace folder instead of cwd.'
+  }, {
+    name: '-W, -workspace',
+    description: 'Search directories from all workspace folders instead of cwd.'
+  }]
+
+  public getCommand(): { cmd: string, args: string[] } {
+    let config = workspace.getConfiguration('list.source.directories')
+    let cmd = config.get<string>('command', '')
+    let args = config.get<string[]>('args', [])
+    if (!cmd) {
+      if (executable('rg')) {
+        return { cmd: 'rg', args: this.getArgs(args, ['--color', 'never', '--files']) }
+      } else if (executable('ag')) {
+        return { cmd: 'ag', args: this.getArgs(args, ['-f', '-g', '.', '--nocolor']) }
+      } else if (process.platform == 'win32') {
+        return { cmd: 'dir', args: this.getArgs(args, ['/a-D', '/S', '/B']) }
+      } else if (executable('find')) {
+        return { cmd: 'find', args: this.getArgs(args, ['.', '-type', 'f']) }
+      } else {
+        throw new Error('Unable to find command for directories list.')
+        return null
+      }
+    } else {
+      return { cmd, args }
+    }
+  }
+}

src/files.ts

@@ -7,8 +7,8 @@ import path from 'path'
 import readline from 'readline'
 import { executable } from './util'
 
-class Task extends EventEmitter implements ListTask {
-  private processes: ChildProcess[] = []
+export class Task extends EventEmitter implements ListTask {
+  protected processes: ChildProcess[] = []
 
   public start(cmd: string, args: string[], cwds: string[], patterns: string[]): void {
     let remain = cwds.length
@@ -18,7 +18,7 @@ class Task extends EventEmitter implements ListTask {
       process.on('error', e => {
         this.emit('error', e.message)
       })
-      const rl = readline.createInterface(process.stdout)
+      const rl = readline.createInterface(this.post_process_hook(process))
       const range = Range.create(0, 0, 0, 0)
       let hasPattern = patterns.length > 0
       process.stderr.on('data', chunk => {
@@ -54,13 +54,18 @@ class Task extends EventEmitter implements ListTask {
       }
     }
   }
+
+  protected post_process_hook(process: ChildProcess) {
+    return process.stdout
+  }
 }
 
 export default class FilesList extends BasicList {
-  public readonly name = 'files'
+  public task = Task
+  public name = 'files'
   public readonly defaultAction = 'open'
   public description = 'Search files by rg or ag'
-  public readonly detail = `Install ripgrep in your $PATH to have best experience.
+  public detail = `Install ripgrep in your $PATH to have best experience.
 Files is searched from current cwd by default.
 Provide directory names as arguments to search other directories.
 Use 'list.source.files.command' configuration for custom search command.
@@ -147,7 +152,7 @@ Note that rg ignore hidden files by default.`
         }
       }
     }
-    let task = new Task()
+    let task = new this.task()
     let excludePatterns = this.getConfig().get<string[]>('excludePatterns', [])
     task.start(res.cmd, res.args.concat(searchArgs), cwds, excludePatterns)
     return task

src/index.ts

@@ -3,6 +3,7 @@ import BufferList from './buffers'
 import Cmdhistory from './cmdhistory'
 import Colors from './colors'
 import Commands from './commands'
+import DirectoriesList from './directories'
 import FilesList from './files'
 import Filetypes from './filetypes'
 import GrepList from './grep'
@@ -58,6 +59,9 @@ export async function activate(context: ExtensionContext): Promise<void> {
   if (!isDisabled('filetypes')) {
     subscriptions.push(listManager.registerList(new Filetypes(nvim)))
   }
+  if (!isDisabled('directories')) {
+    subscriptions.push(listManager.registerList(new DirectoriesList(nvim)))
+  }
   if (!isDisabled('files')) {
     subscriptions.push(listManager.registerList(new FilesList(nvim)))
   }