Support schema associations in yaml files #204

- association done using modeline "# yaml-language-server: $schema=<mySchemaURL"> Signed-off-by: Aurélien Pupier <[email protected]>
redhat-developer · Jul 21, 2020 · 4427479 · 4427479
1 parent 7346049
commit 4427479
Show file tree

Hide file tree

Showing 4 changed files with 191 additions and 1 deletion.
diff --git a/src/languageservice/parser/yamlParser07.ts b/src/languageservice/parser/yamlParser07.ts
@@ -18,6 +18,10 @@ import { ASTNode } from '../jsonASTTypes';
 import { ErrorCode } from 'vscode-json-languageservice';
 import { emit } from 'process';
 
+const YAML_DIRECTIVE_PREFIX = '%';
+const YAML_COMMENT_PREFIX = '#';
+const YAML_DATA_INSTANCE_SEPARATOR = '---';
+
 /**
  * These documents are collected into a final YAMLDocument
  * and passed to the `parseYAML` caller.
@@ -29,13 +33,15 @@ export class SingleYAMLDocument extends JSONDocument {
     public warnings: YAMLDocDiagnostic[];
     public isKubernetes: boolean;
     public currentDocIndex: number;
+    public lineComments: string[];
 
     constructor(lines: number[]) {
         super(null, []);
         this.lines = lines;
         this.root = null;
         this.errors = [];
         this.warnings = [];
+        this.lineComments = [];
     }
 
     public getSchemas(schema, doc, node) {
@@ -101,7 +107,25 @@ export function parse(text: string, customTags = []): YAMLDocument {
     // Generate the SingleYAMLDocs from the AST nodes
     const startPositions = getLineStartPositions(text);
     const yamlDocs: SingleYAMLDocument[] = yamlNodes.map(node => nodeToSingleDoc(node, startPositions, text));
+
+    parseLineComments(text, yamlDocs);
 
     // Consolidate the SingleYAMLDocs
     return new YAMLDocument(yamlDocs);
 }
+
+function parseLineComments(text: string, yamlDocs: SingleYAMLDocument[]) {
+    const lines = text.split(/[\r\n]+/g);
+    let yamlDocCount = 0;
+    lines.forEach(line => {
+        if (line.startsWith(YAML_DIRECTIVE_PREFIX) && yamlDocCount === 0) {
+            yamlDocCount--;
+        }
+        if (line === YAML_DATA_INSTANCE_SEPARATOR) {
+            yamlDocCount++;
+        }
+        if (line.startsWith(YAML_COMMENT_PREFIX)) {
+            yamlDocs[yamlDocCount].lineComments.push(line);
+        }
+    });
+}
diff --git a/src/languageservice/services/yamlSchemaService.ts b/src/languageservice/services/yamlSchemaService.ts
@@ -14,6 +14,10 @@ import { URI } from 'vscode-uri';
 
 import * as nls from 'vscode-nls';
 import { convertSimple2RegExpPattern } from '../utils/strings';
+import { TextDocument } from 'vscode-languageserver';
+import { SingleYAMLDocument } from '../parser/yamlParser07';
+import { stringifyObject } from '../utils/json';
+import { getNodeValue } from '../parser/jsonParser07';
 const localize = nls.loadMessageBundle();
 
 export declare type CustomSchemaProvider = (uri: string) => Thenable<string | string[]>;
@@ -216,9 +220,15 @@ export class YAMLSchemaService extends JSONSchemaService {
 
     public getSchemaForResource (resource: string, doc = undefined): Thenable<ResolvedSchema> {
         const resolveSchema = () => {
-
             const seen: { [schemaId: string]: boolean } = Object.create(null);
             const schemas: string[] = [];
+
+            const schemaFromModeline = this.getSchemaFromModeline(doc);
+            if(schemaFromModeline !== undefined) {
+                schemas.push(schemaFromModeline);
+                seen[schemaFromModeline] = true;
+            }
+
             for (const entry of this.filePatternAssociations) {
                 if (entry.matchesPattern(resource)) {
                     for (const schemaId of entry.getURIs()) {
@@ -282,6 +292,28 @@ export class YAMLSchemaService extends JSONSchemaService {
             return resolveSchema();
         }
     }
+
+    /**
+     * Retrieve schema if declared as modeline
+     * @param doc 
+     */
+    private getSchemaFromModeline(doc: any) : string{
+        if (doc instanceof SingleYAMLDocument) {
+            const modelineDeclaration = '# yaml-language-server:';
+            const yamlLanguageServerModeline = doc.lineComments.find(lineComment => lineComment.startsWith(modelineDeclaration));
+            if (yamlLanguageServerModeline != undefined) {
+                const schemaKey = '$schema=';
+                const indexOfJsonSchemaParameter = yamlLanguageServerModeline.indexOf(schemaKey);
+                if (yamlLanguageServerModeline.indexOf(schemaKey) != -1) {
+                    const startIndex = indexOfJsonSchemaParameter + schemaKey.length;
+                    const indexOfNextSpace = yamlLanguageServerModeline.indexOf(' ', startIndex);
+                    const endIndex = indexOfNextSpace !== -1 ? indexOfNextSpace : yamlLanguageServerModeline.length;
+                    return yamlLanguageServerModeline.substring(startIndex, endIndex);
+                }
+            }
+        }
+        return undefined;
+    }
 
     private async resolveCustomSchema (schemaUri, doc) {
         const unresolvedSchema = await this.loadSchema(schemaUri);

diff --git a/test/autoCompletion.withschemainfile.test.ts b/test/autoCompletion.withschemainfile.test.ts
@@ -0,0 +1,50 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Red Hat. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+import { TextDocument } from 'vscode-languageserver';
+import { getLanguageService } from '../src/languageservice/yamlLanguageService';
+import { toFsPath, schemaRequestService, workspaceContext }  from './utils/testHelper';
+import assert = require('assert');
+import path = require('path');
+
+const languageService = getLanguageService(schemaRequestService, workspaceContext, [], null);
+
+const languageSettings = {
+    schemas: [],
+    completion: true
+};
+
+const uri = toFsPath(path.join(__dirname, './fixtures/testArrayMaxProperties.json'));
+languageService.configure(languageSettings);
+
+suite('Auto Completion Tests with schema defined in file', () => {
+
+    describe('yamlCompletion with schema defined in file', function () {
+
+            function setup(content: string) {
+                return TextDocument.create('file://~/Desktop/vscode-k8s/test-schemainfile.yaml', 'yaml', 0, content);
+            }
+
+            function parseSetup(content: string, position) {
+                const testTextDocument = setup(content);
+                return languageService.doComplete(testTextDocument, testTextDocument.positionAt(position), false);
+            }
+
+            it('Provide completion from schema declared in file', done => {
+                const content = `# yaml-language-server: $schema=${uri}\n- `;
+                const completion = parseSetup(content, content.length);
+                completion.then(function (result) {
+                    assert.equal(result.items.length, 3);
+                }).then(done, done);
+            });
+
+            it('Provide completion from schema declared in file with several attributes', done => {
+                const content = `# yaml-language-server: $schema=${uri} anothermodeline=value\n- `;
+                const completion = parseSetup(content, content.length);
+                completion.then(function (result) {
+                    assert.equal(result.items.length, 3);
+                }).then(done, done);
+            });
+    });
+});
diff --git a/test/parser.test.ts b/test/parser.test.ts
@@ -0,0 +1,84 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Red Hat. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+import assert = require('assert');
+import {parse} from './../src/languageservice/parser/yamlParser07';
+import { YAMLDocument } from '../src/languageservice/parser/yamlParser07';
+import { getNodeValue } from '../src/languageservice/parser/jsonParser07';
+
+
+suite('Test parser', () => {
+
+    describe('test parser', function () {
+
+            it('parse emtpy text', () => {
+                const parsedDocument = parse('');
+                assert(parsedDocument.documents.length === 0, 'A document has been created for an empty text');
+            });
+
+            it('parse single document', () => {
+                const parsedDocument = parse('test');
+                assert(parsedDocument.documents.length === 1, `A single document shoudl be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+            });
+
+            it('parse single document with directives', () => {
+                const parsedDocument = parse('%TAG demo\n---\ntest');
+                assert(parsedDocument.documents.length === 1, `A single document shoudl be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+            });
+
+            it('parse 2 documents', () => {
+                const parsedDocument = parse('test\n---\ntest2');
+                assert(parsedDocument.documents.length === 2, `2 documents should be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[0].root.value === 'test');
+                assert(parsedDocument.documents[1].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[1].root.children.length}`);
+                assert(parsedDocument.documents[1].root.value === 'test2');
+            });
+
+            it('parse 3 documents', () => {
+                const parsedDocument = parse('test\n---\ntest2\n---\ntest3');
+                assert(parsedDocument.documents.length === 3, `3 documents should be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[0].root.value === 'test');
+                assert(parsedDocument.documents[1].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[1].root.children.length}`);
+                assert(parsedDocument.documents[1].root.value === 'test2');
+                assert(parsedDocument.documents[2].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[2].root.children.length}`);
+                assert(parsedDocument.documents[2].root.value === 'test3');
+            });
+
+            it('parse single document with comment', () => {
+                const parsedDocument = parse('# a comment\ntest');
+                assert(parsedDocument.documents.length === 1, `A single document should be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[0].lineComments.length === 1);
+                assert(parsedDocument.documents[0].lineComments[0] === '# a comment');
+            });
+
+            it('parse 2 documents with comment', () => {
+                const parsedDocument = parse('# a comment\ntest\n---\n# a second comment\ntest2');
+                assert(parsedDocument.documents.length === 2, `2 documents should be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[0].lineComments.length === 1);
+                assert(parsedDocument.documents[0].lineComments[0] === '# a comment');
+
+                assert(parsedDocument.documents[1].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[1].lineComments.length === 1);
+                assert(parsedDocument.documents[1].lineComments[0] === '# a second comment');
+            });
+
+            it('parse 2 documents with comment and a directive', () => {
+                const parsedDocument = parse('%TAG demo\n---\n# a comment\ntest\n---\n# a second comment\ntest2');
+                assert(parsedDocument.documents.length === 2, `2 documents should be available but there are ${parsedDocument.documents.length}`);
+                assert(parsedDocument.documents[0].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[0].lineComments.length === 1);
+                assert(parsedDocument.documents[0].lineComments[0] === '# a comment');
+
+                assert(parsedDocument.documents[1].root.children.length === 0, `There should no children available but there are ${parsedDocument.documents[0].root.children.length}`);
+                assert(parsedDocument.documents[1].lineComments.length === 1);
+                assert(parsedDocument.documents[1].lineComments[0] === '# a second comment');
+            });
+    });
+});