Merge pull request #126 from Watts-Lab/expandtemplates

Expand templates

Author: darrenmo12309

deliberation-empirica

@@ -65,6 +65,10 @@
       {
         "command": "deliberation-lab-tools.openStagePreview",
         "title": "Deliberation Lab Tools: Open Stage Preview"
+      },
+      {
+        "command": "deliberation-lab-tools.previewExpandedTemplates",
+        "title": "Preview: Expanded Treatment File"
       }
     ],
     "menus": {
@@ -78,6 +82,11 @@
           "command": "deliberation-lab-tools.openStagePreview",
           "when": "editorLangId == treatmentsYaml",
           "group": "deliberation-lab-tools"
+        },
+        {
+          "command": "deliberation-lab-tools.previewExpandedTemplates",
+          "when": "editorLangId == treatmentsYaml",
+          "group": "deliberation-lab-tools"
         }
       ]
     }

package.json

@@ -1,7 +1,9 @@
 import { get } from "http";
 import * as vscode from "vscode";
 import { getExtensionUri } from "./contextStore";
-import { load as loadYaml } from "js-yaml";
+import { dump as dumpYaml, load as loadYaml } from "js-yaml";
+import { EXP_SCHEME } from "./fillTemplates";
+import { fillTemplates } from "./fillTemplates";
 
 // Command to create default treatments YAML file
 export const defaultYaml = vscode.commands.registerCommand("deliberation-lab-tools.defaultTreatmentsYaml", async () => {
@@ -312,4 +314,31 @@ function getNonce(): string {
         text += possible.charAt(Math.floor(Math.random() * possible.length));
     }
     return text;
-}
+}
+
+export const expandedTemplatesPreview = vscode.commands.registerCommand("deliberation-lab-tools.previewExpandedTemplates", async (resource?: vscode.Uri) => {
+      const src = resource ?? vscode.window.activeTextEditor?.document.uri;
+      if (!src) return;
+
+      const doc = await vscode.workspace.openTextDocument(src);
+      if (doc.languageId !== "treatmentsYaml") {
+        vscode.window.showWarningMessage("This preview is only for treatments YAML files.");
+        return;
+      }
+
+      const previewUri = vscode.Uri.parse(
+        `${EXP_SCHEME}:${src.path}.expanded.treatments.yaml?src=${encodeURIComponent(src.toString())}`
+      );
+
+      const vdoc = await vscode.workspace.openTextDocument(previewUri);
+
+      if (vdoc.languageId !== "treatmentsYaml") {
+        await vscode.languages.setTextDocumentLanguage(vdoc, "treatmentsYaml");
+      }
+
+      await vscode.window.showTextDocument(vdoc, {
+        preview: true,
+        preserveFocus: false,
+        viewColumn: vscode.ViewColumn.Beside,
+      });
+    })

src/commands.ts

@@ -4,10 +4,11 @@ import { parseYaml } from "./parsers/parseYaml";
 import { parseMarkdown } from "./parsers/parseMarkdown";
 import { parseDlConfig } from "./parsers/parseDlConfig";
 import { parseBatchConfig } from "./parsers/parseBatchConfig";
-import { defaultMarkdown, inlineSuggestion, defaultYaml, markdownPreview } from "./commands";
+import { defaultMarkdown, inlineSuggestion, defaultYaml, markdownPreview, expandedTemplatesPreview } from "./commands";
+import { ExpandedTemplatesProvider, EXP_SCHEME } from "./fillTemplates";
 import { setExtensionContext } from "./contextStore";
 import { FileFixCodeActionProvider } from "./codeActionProvider";
-
+import * as yaml from "js-yaml";
 // should this be named yamlDiagnostics if also using markdown?
 export const diagnosticCollection = vscode.languages.createDiagnosticCollection("yamlDiagnostics");
 
@@ -51,12 +52,12 @@ export async function activate(context: vscode.ExtensionContext) {
         await parseDocument(event);
       }
     }),
-  // for changing document
-  vscode.workspace.onDidChangeTextDocument(async (event) => {
-    if (event?.document !== undefined) {
-      parseDocument(event?.document);
-    };
-  }),
+    // for changing document
+    vscode.workspace.onDidChangeTextDocument(async (event) => {
+      if (event?.document !== undefined) {
+        parseDocument(event?.document);
+      };
+    }),
 
     // When we switch to a document open in another tab
     vscode.window.onDidChangeActiveTextEditor(async (event) => {
@@ -90,4 +91,22 @@ export async function activate(context: vscode.ExtensionContext) {
 
   // Open Markdown preview
   context.subscriptions.push(markdownPreview);
+
+  // Open expanded templates preview
+  const expandedProvider = new ExpandedTemplatesProvider();
+  context.subscriptions.push(
+    vscode.workspace.registerTextDocumentContentProvider(EXP_SCHEME, expandedProvider)
+  );
+
+  // Open preview command
+  context.subscriptions.push(
+    expandedTemplatesPreview
+  );
+
+  // Auto-refresh the preview when the source changes
+  context.subscriptions.push(
+    vscode.workspace.onDidChangeTextDocument((e) => {
+      expandedProvider.refreshForSource(e.document.uri);
+    })
+  );
 }