Add environment variable support (#14)

* Add environment variable support

* Use Vim environment instead

* Use vars instead

* Add/fix for windows sep and powershell/cmd env style

Co-authored-by: Luma <world@luma.email>

Author: Shougo

autoload/ddc_file/internal.vim

@@ -2,8 +2,8 @@ function! s:line_to_file_full(line, is_posix) abort
   return matchlist(
     \ a:line,
     \ a:is_posix
-       \ ? '[[:fname:]/]*$'
-       \ : '[[:fname:]\\]*$'
+       \ ? '\c\%(^\$\%(env:\)\?\)\?[[:fname:]/]*$'
+       \ : '\c\%(^\$\%(env:\)\?\)\?[[:fname:]\\]*$'
   \ )[0]
 endfunction
 
@@ -56,6 +56,9 @@ function! ddc_file#internal#_test() abort
     call assert_equal('../bb/aa', s:line_to_file_full('aa ff ../bb/aa', v:true))
     call assert_equal('a', s:line_to_file_full('\a', v:true))
     call assert_equal('./a', s:line_to_file_full('\./a', v:true))
+    call assert_equal('$Foo/bar', s:line_to_file_full('$Foo/bar', v:true))
+    call assert_equal('/bar', s:line_to_file_full('$Foo /bar', v:true))
+    call assert_equal('Foo/bar', s:line_to_file_full('$ Foo/bar', v:true))
     call assert_equal('', s:full_to_base_prefix('', v:true))
     call assert_equal('', s:full_to_base_prefix('a', v:true))
     call assert_equal('', s:full_to_base_prefix('abc', v:true))
@@ -103,6 +106,14 @@ function! ddc_file#internal#_test() abort
     call assert_equal('..\bb\aa', s:line_to_file_full('aa ff ..\bb\aa', v:false))
     call assert_equal('/a', s:line_to_file_full('/a', v:false))
     call assert_equal('/.\a', s:line_to_file_full('/.\a', v:false))
+    call assert_equal('$Foo/bar', s:line_to_file_full('$Foo/bar', v:false))
+    call assert_equal('/bar', s:line_to_file_full('$Foo /bar', v:false))
+    call assert_equal('Foo/bar', s:line_to_file_full('$ Foo/bar', v:false))
+    call assert_equal('$env:Foo/bar', s:line_to_file_full('$env:Foo/bar', v:false))
+    call assert_equal('/root/$env:Foo/bar', s:line_to_file_full('/root/$env:Foo/bar', v:false))
+    call assert_equal('$Env:Foo/bar', s:line_to_file_full('$Env:Foo/bar', v:false))
+    call assert_equal('/bar', s:line_to_file_full('$env:Foo /bar', v:false))
+    call assert_equal('Foo/bar', s:line_to_file_full('$env: Foo/bar', v:false))
     call assert_equal('', s:full_to_base_prefix('', v:false))
     call assert_equal('', s:full_to_base_prefix('a', v:false))
     call assert_equal('', s:full_to_base_prefix('abc', v:false))

denops/@ddc-sources/file.ts

@@ -5,6 +5,7 @@ import {
   GatherCandidatesArguments,
   homeDir,
   path as univPath,
+  vars,
   wrapA,
 } from "../@ddc-file/deps.ts";
 import * as util from "../@ddc-file/util.ts";
@@ -86,7 +87,7 @@ export class Source extends BaseSource<Params> {
     );
 
     // e.g. '/home/ubuntu/config' for inputFileFull = '~/config'
-    const inputFileFullExpanded = (() => {
+    const inputFileFullExpanded = await (async () => {
       const home = homeDir();
       const last = inputFileFull.endsWith(path.sep) ? path.sep : "";
       {
@@ -109,6 +110,22 @@ export class Source extends BaseSource<Params> {
           return path.join(home, inputFileFull.slice(pat.length)) + last;
         }
       }
+      {
+        const pat = new RegExp(
+          `^(?:\\$(?:env:)?(\\w+)|%(\\w+)%)${path.sep === "/" ? "/" : "\\\\"}`,
+          "i",
+        );
+        const m = inputFileFull.match(pat);
+        if (m) {
+          const env = await vars.environment.get(
+            args.denops,
+            m[1] || m[2],
+          ) as string;
+          if (env) {
+            return path.join(env, inputFileFull.slice(m[0].length)) + last;
+          }
+        }
+      }
       return inputFileFull;
     })();