Merge pull request #201 from github/joshmgross/cache-completion-separately

Cache transformed workflows separately

languageservice/src/complete.ts

@@ -65,8 +65,8 @@ export async function complete(
     content: newDoc.getText()
   };
 
-  const parsedWorkflow = fetchOrParseWorkflow(file, textDocument.uri);
-  if (!parsedWorkflow?.value) {
+  const parsedWorkflow = fetchOrParseWorkflow(file, textDocument.uri, true);
+  if (!parsedWorkflow.value) {
     return [];
   }
 

languageservice/src/e2e.test.ts

@@ -0,0 +1,28 @@
+import {complete} from "./complete";
+import {hover} from "./hover";
+import {registerLogger} from "./log";
+import {getPositionFromCursor} from "./test-utils/cursor-position";
+import {TestLogger} from "./test-utils/logger";
+import {clearCache} from "./utils/workflow-cache";
+
+registerLogger(new TestLogger());
+
+beforeEach(() => {
+  clearCache();
+});
+
+describe("end-to-end", () => {
+  it("empty workflow completion after hover", async () => {
+    const input = "|";
+
+    // Issue hover first to fill the cache
+    await hover(...getPositionFromCursor(input));
+
+    const result = await complete(...getPositionFromCursor(input));
+
+    expect(result).not.toBeUndefined();
+    expect(result.length).toEqual(8);
+    const labels = result.map(x => x.label);
+    expect(labels).toEqual(["concurrency", "defaults", "env", "jobs", "name", "on", "permissions", "run-name"]);
+  });
+});

languageservice/src/utils/workflow-cache.ts

@@ -17,7 +17,9 @@ const workflowTemplateCache = new Map<string, WorkflowTemplate>();
 
 export function clearCacheEntry(uri: string) {
   parsedWorkflowCache.delete(uri);
+  parsedWorkflowCache.delete(workflowKey(uri, true));
   workflowTemplateCache.delete(uri);
+  workflowTemplateCache.delete(workflowKey(uri, true));
 }
 
 export function clearCache() {
@@ -25,28 +27,49 @@ export function clearCache() {
   workflowTemplateCache.clear();
 }
 
-export function fetchOrParseWorkflow(file: File, uri: string): ParseWorkflowResult | undefined {
-  const cachedResult = parsedWorkflowCache.get(uri);
+/**
+ * Parses a workflow file and caches the result
+ * @param transformed Indicates whether the workflow has been transformed before parsing
+ * @returns the {@link ParseWorkflowResult}
+ */
+export function fetchOrParseWorkflow(file: File, uri: string, transformed = false): ParseWorkflowResult {
+  const key = workflowKey(uri, transformed);
+  const cachedResult = parsedWorkflowCache.get(key);
   if (cachedResult) {
     return cachedResult;
   }
   const result = parseWorkflow(file, nullTrace);
-  parsedWorkflowCache.set(uri, result);
+  parsedWorkflowCache.set(key, result);
   return result;
 }
 
+/**
+ * Converts a workflow template and caches the result
+ * @param transformed Indicates whether the workflow has been transformed before parsing
+ * @returns the converted {@link WorkflowTemplate}
+ */
 export async function fetchOrConvertWorkflowTemplate(
   context: TemplateContext,
   template: TemplateToken,
   uri: string,
   config?: CompletionConfig,
-  options?: WorkflowTemplateConverterOptions
+  options?: WorkflowTemplateConverterOptions,
+  transformed = false
 ): Promise<WorkflowTemplate> {
-  const cachedTemplate = workflowTemplateCache.get(uri);
+  const key = workflowKey(uri, transformed);
+  const cachedTemplate = workflowTemplateCache.get(key);
   if (cachedTemplate) {
     return cachedTemplate;
   }
   const workflowTemplate = await convertWorkflowTemplate(context, template, config?.fileProvider, options);
-  workflowTemplateCache.set(uri, workflowTemplate);
+  workflowTemplateCache.set(key, workflowTemplate);
   return workflowTemplate;
 }
+
+// Use a separate cache key for transformed workflows
+function workflowKey(uri: string, transformed: boolean): string {
+  if (transformed) {
+    return `transformed-${uri}`;
+  }
+  return uri;
+}