Support allowed/suggested value providers

This commit is contained in:
Christopher Schleiden
2022-11-28 16:15:57 -08:00
parent 71619bee95
commit a9cdcdba80
8 changed files with 98 additions and 46 deletions
+6 -3
View File
@@ -1,7 +1,7 @@
import {complete} from "./complete";
import {WorkflowContext} from "./context/workflow-context";
import {getPositionFromCursor} from "./test-utils/cursor-position";
import {ValueProviderConfig} from "./value-providers/config";
import {ValueProviderConfig, ValueProviderKind} from "./value-providers/config";
describe("completion", () => {
it("runs-on", async () => {
@@ -180,8 +180,11 @@ jobs:
const input = "on: push\njobs:\n build:\n runs-on: |";
const config: ValueProviderConfig = {
"runs-on": async (_: WorkflowContext) => {
return [{label: "my-custom-label"}];
"runs-on": {
kind: ValueProviderKind.SuggestedValues,
get: async (_: WorkflowContext) => {
return [{label: "my-custom-label"}];
}
}
};
const result = await complete(...getPositionFromCursor(input), config);
+2 -2
View File
@@ -104,7 +104,7 @@ async function getValues(
const existingValues = getExistingValues(token, parent);
if (token?.definition?.key) {
const customValues = await valueProviderConfig?.[token.definition.key]?.(workflowContext);
const customValues = await valueProviderConfig?.[token.definition.key]?.get(workflowContext);
if (customValues) {
return filterAndSortCompletionOptions(customValues, existingValues);
@@ -117,7 +117,7 @@ async function getValues(
(parent.definition?.key && defaultValueProviders[parent.definition.key]);
if (valueProvider) {
const values = await valueProvider(workflowContext);
const values = await valueProvider.get(workflowContext);
return filterAndSortCompletionOptions(values, existingValues);
}
+39 -3
View File
@@ -59,7 +59,7 @@ jobs:
} as Diagnostic);
});
it("single value not returned by value provider", async () => {
it("single value not returned by suggested value provider", async () => {
const result = await validate(
createDocument(
"wf.yaml",
@@ -76,7 +76,7 @@ jobs:
expect(result.length).toBe(1);
expect(result[0]).toEqual({
message: "Value 'does-not-exist' is not valid",
severity: DiagnosticSeverity.Error,
severity: DiagnosticSeverity.Warning,
range: {
end: {
character: 27,
@@ -109,7 +109,7 @@ jobs:
expect(result.length).toBe(1);
expect(result[0]).toEqual({
message: "Value 'does-not-exist' is not valid",
severity: DiagnosticSeverity.Error,
severity: DiagnosticSeverity.Warning,
range: {
end: {
character: 20,
@@ -122,4 +122,40 @@ jobs:
}
} as Diagnostic);
});
it("single value not returned by allowed value provider", async () => {
const result = await validate(
createDocument(
"wf.yaml",
`on: push
jobs:
test:
runs-on: ubuntu-latest
steps:
- run: echo
build:
runs-on: ubuntu-latest
needs: test2
steps:
- run: echo`
),
defaultValueProviders
);
expect(result.length).toBe(1);
expect(result[0]).toEqual({
message: "Value 'test2' is not valid",
severity: DiagnosticSeverity.Error,
range: {
end: {
character: 16,
line: 8
},
start: {
character: 11,
line: 8
}
}
} as Diagnostic);
});
});
+13 -16
View File
@@ -24,7 +24,7 @@ import {getWorkflowContext, WorkflowContext} from "./context/workflow-context";
import {AccessError, wrapDictionary} from "./expression-validation/error-dictionary";
import {nullTrace} from "./nulltrace";
import {findToken} from "./utils/find-token";
import {Value, ValueProviderConfig} from "./value-providers/config";
import {ValueProviderConfig, ValueProviderKind} from "./value-providers/config";
import {defaultValueProviders} from "./value-providers/default";
/**
@@ -124,26 +124,23 @@ async function additionalValidations(
if (valueProviderConfig && token.range && token.definition?.key) {
const defKey = token.definition.key;
let customValues: Value[] | undefined;
const customValueProvider = valueProviderConfig[defKey];
if (customValueProvider) {
customValues = await customValueProvider(getProviderContext(documentUri, template, root, token));
} else {
const defaultValueProvider = defaultValueProviders[defKey];
if (defaultValueProvider) {
customValues = await defaultValueProvider(getProviderContext(documentUri, template, root, token));
}
// Try a custom value provider first
let valueProvider = valueProviderConfig[defKey];
if (!valueProvider) {
// fall back to default
valueProvider = defaultValueProviders[defKey];
}
if (customValues) {
if (valueProvider) {
const customValues = await valueProvider.get(getProviderContext(documentUri, template, root, token));
if (isSequence(token)) {
for (let i = 0; i < token.count; ++i) {
const entry = token.get(i);
if (isString(entry)) {
if (!customValues.map(x => x.label).includes(entry.value)) {
invalidValue(diagnostics, entry);
invalidValue(diagnostics, entry, valueProvider.kind);
}
}
}
@@ -151,7 +148,7 @@ async function additionalValidations(
if (isString(token)) {
if (!customValues.map(x => x.label).includes(token.value)) {
invalidValue(diagnostics, token);
invalidValue(diagnostics, token, valueProvider.kind);
}
}
}
@@ -159,10 +156,10 @@ async function additionalValidations(
}
}
function invalidValue(diagnostics: Diagnostic[], token: StringToken) {
function invalidValue(diagnostics: Diagnostic[], token: StringToken, kind: ValueProviderKind) {
diagnostics.push({
message: `Value '${token.value}' is not valid`,
severity: DiagnosticSeverity.Error,
severity: kind === ValueProviderKind.AllowedValues ? DiagnosticSeverity.Error : DiagnosticSeverity.Warning,
range: mapRange(token.range)
});
}
@@ -5,7 +5,15 @@ export interface Value {
description?: string;
}
export type ValueProvider = (context: WorkflowContext) => Promise<Value[]>;
export enum ValueProviderKind {
AllowedValues,
SuggestedValues
}
export type ValueProvider = {
kind: ValueProviderKind;
get: (context: WorkflowContext) => Promise<Value[]>;
};
export interface ValueProviderConfig {
[definitionKey: string]: ValueProvider;
@@ -1,25 +1,28 @@
import {WorkflowContext} from "../context/workflow-context";
import {Value, ValueProviderConfig} from "./config";
import {ValueProviderConfig, ValueProviderKind} from "./config";
import {needs} from "./needs";
import {stringsToValues} from "./strings-to-values";
export const defaultValueProviders: ValueProviderConfig = {
needs,
"runs-on": async (_: WorkflowContext) =>
stringsToValues([
"ubuntu-latest",
"ubuntu-18.04",
"ubuntu-16.04",
"windows-latest",
"windows-2019",
"windows-2016",
"macos-latest",
"macos-10.15",
"macos-10.14",
"macos-10.13",
"self-hosted"
])
needs: {
kind: ValueProviderKind.AllowedValues,
get: needs
},
"runs-on": {
kind: ValueProviderKind.SuggestedValues,
get: async (_: WorkflowContext) =>
stringsToValues([
"ubuntu-latest",
"ubuntu-18.04",
"ubuntu-16.04",
"windows-latest",
"windows-2019",
"windows-2016",
"macos-latest",
"macos-10.15",
"macos-10.14",
"macos-10.13",
"self-hosted"
])
}
};
export function stringsToValues(labels: string[]): Value[] {
return labels.map(x => ({label: x}));
}
@@ -5,7 +5,7 @@ import {OneOfDefinition} from "@github/actions-workflow-parser/templates/schema/
import {StringDefinition} from "@github/actions-workflow-parser/templates/schema/string-definition";
import {getWorkflowSchema} from "@github/actions-workflow-parser/workflows/workflow-schema";
import {Value} from "./config";
import {stringsToValues} from "./default";
import {stringsToValues} from "./strings-to-values";
export function definitionValues(def: Definition): Value[] {
const schema = getWorkflowSchema();
@@ -0,0 +1,5 @@
import {Value} from "./config";
export function stringsToValues(labels: string[]): Value[] {
return labels.map(x => ({label: x}));
}