Merge validation code into a single traversal

This commit is contained in:
Christopher Schleiden
2022-11-28 16:04:53 -08:00
parent a32295a09a
commit a495432d40
5 changed files with 151 additions and 89 deletions
+1 -1
View File
@@ -117,7 +117,7 @@ async function getValues(
(parent.definition?.key && defaultValueProviders[parent.definition.key]);
if (valueProvider) {
const values = valueProvider(workflowContext);
const values = await valueProvider(workflowContext);
return filterAndSortCompletionOptions(values, existingValues);
}
+66 -1
View File
@@ -1,6 +1,7 @@
import {Diagnostic} from "vscode-languageserver-types";
import {Diagnostic, DiagnosticSeverity} from "vscode-languageserver-types";
import {createDocument} from "./test-utils/document";
import {validate} from "./validate";
import {defaultValueProviders} from "./value-providers/default";
describe("validation", () => {
it("valid workflow", async () => {
@@ -57,4 +58,68 @@ jobs:
}
} as Diagnostic);
});
it("single value not returned by value provider", async () => {
const result = await validate(
createDocument(
"wf.yaml",
`on: push
jobs:
build:
runs-on: does-not-exist
steps:
- run: echo`
),
defaultValueProviders
);
expect(result.length).toBe(1);
expect(result[0]).toEqual({
message: "Value 'does-not-exist' is not valid",
severity: DiagnosticSeverity.Error,
range: {
end: {
character: 27,
line: 3
},
start: {
character: 13,
line: 3
}
}
} as Diagnostic);
});
it("value in sequence not returned by value provider", async () => {
const result = await validate(
createDocument(
"wf.yaml",
`on: push
jobs:
build:
runs-on:
- ubuntu-latest
- does-not-exist
steps:
- run: echo`
),
defaultValueProviders
);
expect(result.length).toBe(1);
expect(result[0]).toEqual({
message: "Value 'does-not-exist' is not valid",
severity: DiagnosticSeverity.Error,
range: {
end: {
character: 20,
line: 5
},
start: {
character: 6,
line: 5
}
}
} as Diagnostic);
});
});
+80 -83
View File
@@ -10,6 +10,8 @@ import {
WorkflowTemplate
} from "@github/actions-workflow-parser";
import {splitAllowedContext} from "@github/actions-workflow-parser/templates/allowed-context";
import {BasicExpressionToken} from "@github/actions-workflow-parser/templates/tokens/basic-expression-token";
import {StringToken} from "@github/actions-workflow-parser/templates/tokens/string-token";
import {TemplateToken} from "@github/actions-workflow-parser/templates/tokens/template-token";
import {TokenRange} from "@github/actions-workflow-parser/templates/tokens/token-range";
import {File} from "@github/actions-workflow-parser/workflows/file";
@@ -50,13 +52,15 @@ export async function validate(
// Errors will be updated in the context
const template = convertWorkflowTemplate(result.context, result.value);
// Validate with value providers
if (valueProviderConfig) {
await validateValueProviders(diagnostics, textDocument.uri, template, result, valueProviderConfig);
}
// Validate expressions
validateExpressions(diagnostics, result, contextProviderConfig);
// Validate expressions and value providers
await additionalValidations(
diagnostics,
textDocument.uri,
template,
result.value,
valueProviderConfig,
contextProviderConfig
);
}
// For now map parser errors directly to diagnostics
@@ -101,86 +105,34 @@ function mapRange(range: TokenRange | undefined): Range {
};
}
function validateExpressions(
diagnostics: Diagnostic[],
result: ParseWorkflowResult,
contextProviderConfig: ContextProviderConfig | undefined
) {
if (!result.value) {
return;
}
// Iterate over the parsed workflow
for (const token of TemplateToken.traverse(result.value)) {
if (isBasicExpression(token)) {
// Validate the expression
for (const expression of token.originalExpressions || [token]) {
const allowedContexts = token.definition?.readerContext || [];
const {namedContexts, functions} = splitAllowedContext(allowedContexts);
let expr: Expr | undefined;
try {
const l = new Lexer(expression.expression);
const lr = l.lex();
const p = new Parser(lr.tokens, namedContexts, functions);
expr = p.parse();
} catch {
// Ignore any error here, we should've caught this earlier in the parsing process
continue;
}
try {
const context = getContext(namedContexts, contextProviderConfig);
const e = new Evaluator(expr, wrapDictionary(context));
e.evaluate();
// Any invalid context access would've thrown an error via the `ErrorDictionary`, for now we don't have to check the actual
// result of the evaluation.
} catch (e) {
if (e instanceof AccessError) {
diagnostics.push({
message: `Context access might be invalid: ${e.keyName}`,
severity: DiagnosticSeverity.Warning,
range: mapRange(expression.range)
});
} else {
// Ignore error
}
}
}
}
}
}
async function validateValueProviders(
async function additionalValidations(
diagnostics: Diagnostic[],
documentUri: URI,
template: WorkflowTemplate,
result: ParseWorkflowResult,
valueProviderConfig: ValueProviderConfig
root: TemplateToken,
valueProviderConfig: ValueProviderConfig | undefined,
contextProviderConfig: ContextProviderConfig | undefined
) {
if (!result.value) {
return;
}
for (const token of TemplateToken.traverse(root)) {
// If this is an expression, validate it
if (isBasicExpression(token)) {
validateExpression(diagnostics, token, contextProviderConfig);
}
// Allowed values coming from the schema have already been validated. Only check if
// a value provider is defined for a token and if it is, validate the values match.
for (const token of TemplateToken.traverse(result.value)) {
if (token.range && token.definition?.key) {
// Allowed values coming from the schema have already been validated. Only check if
// a value provider is defined for a token and if it is, validate the values match.
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, result.value, token));
customValues = await customValueProvider(getProviderContext(documentUri, template, root, token));
} else {
const defaultValueProvider = defaultValueProviders[defKey];
if (defaultValueProvider) {
customValues = defaultValueProvider(getProviderContext(documentUri, template, result.value, token));
customValues = await defaultValueProvider(getProviderContext(documentUri, template, root, token));
}
}
@@ -191,11 +143,7 @@ async function validateValueProviders(
if (isString(entry)) {
if (!customValues.map(x => x.label).includes(entry.value)) {
diagnostics.push({
message: `Value '${entry.value}' is not allowed`,
severity: DiagnosticSeverity.Error,
range: mapRange(entry.range)
});
invalidValue(diagnostics, entry);
}
}
}
@@ -203,11 +151,7 @@ async function validateValueProviders(
if (isString(token)) {
if (!customValues.map(x => x.label).includes(token.value)) {
diagnostics.push({
message: `Value '${token.value}' is not allowed`,
severity: DiagnosticSeverity.Error,
range: mapRange(token.range)
});
invalidValue(diagnostics, token);
}
}
}
@@ -215,6 +159,14 @@ async function validateValueProviders(
}
}
function invalidValue(diagnostics: Diagnostic[], token: StringToken) {
diagnostics.push({
message: `Value '${token.value}' is not valid`,
severity: DiagnosticSeverity.Error,
range: mapRange(token.range)
});
}
function getProviderContext(
documentUri: URI,
template: WorkflowTemplate,
@@ -230,3 +182,48 @@ function getProviderContext(
);
return getWorkflowContext(documentUri, template, path);
}
function validateExpression(
diagnostics: Diagnostic[],
token: BasicExpressionToken,
contextProviderConfig: ContextProviderConfig | undefined
) {
// Validate the expression
for (const expression of token.originalExpressions || [token]) {
const allowedContexts = token.definition?.readerContext || [];
const {namedContexts, functions} = splitAllowedContext(allowedContexts);
let expr: Expr | undefined;
try {
const l = new Lexer(expression.expression);
const lr = l.lex();
const p = new Parser(lr.tokens, namedContexts, functions);
expr = p.parse();
} catch {
// Ignore any error here, we should've caught this earlier in the parsing process
continue;
}
try {
const context = getContext(namedContexts, contextProviderConfig);
const e = new Evaluator(expr, wrapDictionary(context));
e.evaluate();
// Any invalid context access would've thrown an error via the `ErrorDictionary`, for now we don't have to check the actual
// result of the evaluation.
} catch (e) {
if (e instanceof AccessError) {
diagnostics.push({
message: `Context access might be invalid: ${e.keyName}`,
severity: DiagnosticSeverity.Warning,
range: mapRange(expression.range)
});
} else {
// Ignore error
}
}
}
}
@@ -1,10 +1,10 @@
import {WorkflowContext} from "../context/workflow-context";
import {Value} from "./config";
import {Value, ValueProviderConfig} from "./config";
import {needs} from "./needs";
export const defaultValueProviders: {[key: string]: (workflowContext: WorkflowContext) => Value[]} = {
export const defaultValueProviders: ValueProviderConfig = {
needs,
"runs-on": () =>
"runs-on": async (_: WorkflowContext) =>
stringsToValues([
"ubuntu-latest",
"ubuntu-18.04",
@@ -1,7 +1,7 @@
import {WorkflowContext} from "../context/workflow-context";
import {Value} from "./config";
export function needs(context: WorkflowContext): Value[] {
export async function needs(context: WorkflowContext): Promise<Value[]> {
if (!context.template) {
return [];
}