initial changes for value provider

This commit is contained in:
Laura Yu
2022-11-14 22:34:57 +00:00
parent cadaeee4a1
commit 3f7cfbfdb1
6 changed files with 79 additions and 13 deletions
+10 -3
View File
@@ -7,7 +7,10 @@ import { Octokit } from "@octokit/rest";
import { CompletionItem, DocumentUri, Position } from "vscode-languageserver";
import { TextDocument } from "vscode-languageserver-textdocument";
import { RepositoryContext } from "./initializationOptions";
import { getJobNames } from "./value-providers/needs";
import { getRunnerLabels } from "./value-providers/runs-on";
import { WorkflowTemplate } from "@github/actions-workflow-parser";
export async function onCompletion(
position: Position,
@@ -17,8 +20,8 @@ export async function onCompletion(
repoWorkflowMap: Map<string, RepositoryContext>
): Promise<CompletionItem[]> {
const config: ValueProviderConfig = {
getCustomValues: async (key: string, context: WorkflowContext) =>
getCustomValues(key, context, sessionToken, repoWorkflowMap),
getCustomValues: async (key: string, context: WorkflowContext, template: WorkflowTemplate | undefined) =>
getCustomValues(key, context, sessionToken, repoWorkflowMap, template),
};
return await complete(document, position, config);
}
@@ -27,7 +30,8 @@ async function getCustomValues(
key: string,
context: WorkflowContext,
sessionToken: string | undefined,
repoWorkspaceMap: Map<string, RepositoryContext>
repoWorkspaceMap: Map<string, RepositoryContext>,
template: WorkflowTemplate | undefined
) {
if (!sessionToken) {
return;
@@ -44,4 +48,7 @@ async function getCustomValues(
});
return await getRunnerLabels(octokit, repo.owner, repo.name);
}
if (key === "needs" && template) {
return await getJobNames(template);
}
}
@@ -0,0 +1,19 @@
import { Value } from "@github/actions-languageservice/value-providers/config";
import { WorkflowTemplate } from "@github/actions-workflow-parser/model/workflow-template";
export async function getJobNames(
template: WorkflowTemplate
): Promise<Value[]> {
const labels = new Set<string>([
// "dummy-job",
]);
const jobsValues = template.jobs.jobNames;
if (jobsValues) {
for (const job of jobsValues?.values() || []) {
labels.add(job);
}
}
return Array.from(labels).map((label) => ({ label }));
}