Merge pull request #15 from vim-denops/load-scriptfile

Add `load()` to source local/remote Vim script

Author: lambdalisue

deps.ts

@@ -1,3 +1,7 @@
+export * as path from "https://deno.land/std@0.95.0/path/mod.ts";
+export * as hash from "https://deno.land/std@0.95.0/hash/mod.ts";
+export * as fs from "https://deno.land/std@0.95.0/fs/mod.ts";
+
 export type {
   Context,
   Dispatcher,

vim/load.ts

@@ -0,0 +1,71 @@
+import { Denops, fs, hash, path } from "../deps.ts";
+import { execute } from "./execute.ts";
+
+/**
+ * Load Vim script in local/remote URL
+ */
+export async function load(
+  denops: Denops,
+  url: URL,
+): Promise<void> {
+  const scriptPath = await ensureLocalFile(url);
+  await execute(
+    denops,
+    "execute printf('source %s', fnameescape(scriptPath)) ",
+    { scriptPath },
+  );
+}
+
+async function ensureLocalFile(url: URL): Promise<string> {
+  if (url.protocol === "file:") {
+    return path.fromFileUrl(url);
+  }
+  const cacheDir = await getOrCreateCacheDir();
+  const filename = getLocalFilename(url);
+  const filepath = path.join(cacheDir, filename);
+  if (await fs.exists(filepath)) {
+    return filepath;
+  }
+  const response = await fetch(url);
+  if (response.status !== 200) {
+    throw new Error(`Failed to fetch '${url}'`);
+  }
+  const content = await response.arrayBuffer();
+  await Deno.writeFile(
+    filepath,
+    new Uint8Array(content),
+    { mode: 0o700 }, // Do NOT allow groups/others to read the file
+  );
+  return filepath;
+}
+
+function getLocalFilename(url: URL): string {
+  const h = hash.createHash("sha256");
+  h.update(url.href);
+  const basename = path.basename(url.pathname);
+  return `${h.digest()}-${basename}`;
+}
+
+async function getOrCreateCacheDir(): Promise<string> {
+  const cacheDir = Deno.build.os === "windows"
+    ? getCacheDirWindows()
+    : getCacheDirUnix();
+  await Deno.mkdir(cacheDir, { recursive: true });
+  return cacheDir;
+}
+
+function getCacheDirUnix(): string {
+  const root = Deno.env.get("HOME");
+  if (!root) {
+    throw new Error("`HOME` environment variable is not defined.");
+  }
+  return path.join(root, ".cache", "denops_std", "load");
+}
+
+function getCacheDirWindows(): string {
+  const root = Deno.env.get("LOCALAPPDATA");
+  if (!root) {
+    throw new Error("`LOCALAPPDATA` environment variable is not defined.");
+  }
+  return path.join(root, "denops_std", "load");
+}

vim/mod.ts

@@ -1,4 +1,5 @@
 export * from "./autocmd.ts";
 export * from "./execute.ts";
 export * from "./variable.ts";
+export * from "./load.ts";
 export * from "./vim.ts";

vim/vim.ts

@@ -2,6 +2,7 @@ import { Context, Denops, Dispatcher } from "../deps.ts";
 import { execute } from "./execute.ts";
 import { autocmd, AutocmdHelper } from "./autocmd.ts";
 import { VariableHelper } from "./variable.ts";
+import { load } from "./load.ts";
 
 export class Vim {
   private static instance?: Vim;
@@ -60,6 +61,10 @@ export class Vim {
     await autocmd(this.#denops, group, main);
   }
 
+  async load(url: URL): Promise<void> {
+    await load(this.#denops, url);
+  }
+
   register(dispatcher: Dispatcher): void {
     this.#denops.extendDispatcher(dispatcher);
   }