languageservices/languageservice/src/utils/workflow-cache.ts

import {convertWorkflowTemplate, parseWorkflow} from "@github/actions-workflow-parser";
import {WorkflowTemplateConverterOptions} from "@github/actions-workflow-parser/model/convert";
import {File} from "@github/actions-workflow-parser/workflows/file";
import {ParseWorkflowResult} from "@github/actions-workflow-parser";
import {WorkflowTemplate} from "@github/actions-workflow-parser";
import {nullTrace} from "../nulltrace";
import { CompletionConfig } from "../complete";

const parsedWorkflowCache = new Map<string, ParseWorkflowResult>();
const workflowTemplateCache = new Map<string, WorkflowTemplate>();

export function clearParsedCacheEntry(uri: string) {
  parsedWorkflowCache.delete(uri);
}

export function clearParsedCache() {
  parsedWorkflowCache.clear();
}

export function clearWorkflowTemplateCacheEntry(uri: string) {
  workflowTemplateCache.delete(uri);
}

export function clearWorkflowTemplateCache() {
  workflowTemplateCache.clear();
}

export function fetchOrParseWorkflow(file: File, uri: string): ParseWorkflowResult | undefined {
  let result = parsedWorkflowCache.get(uri);
  if (!result || !result.value) {
    result = parseWorkflow(file, nullTrace);
    if (!result.value) {
      return undefined;
    }
    parsedWorkflowCache.set(uri, result);
  }
  return result;
}

export async function fetchOrConvertWorkflowTemplate(file: File, parsedWorkflow: ParseWorkflowResult, uri: string, config?: CompletionConfig, options?: WorkflowTemplateConverterOptions): Promise<WorkflowTemplate> {
  let template = workflowTemplateCache.get(uri);
  if (!template) {
    template = await convertWorkflowTemplate(file.name, parsedWorkflow.context, parsedWorkflow.value!, config?.fileProvider, options);
    workflowTemplateCache.set(uri, template);
  }
  return template;
}