Proper collect comments from document

Signed-off-by: Yevhen Vydolob <[email protected]>

.mocharc.json

@@ -0,0 +1,9 @@
+{
+  "timeout": "5000",
+  "ui":  "bdd",
+  "extension": ["ts"],
+  "package": "./package.json",
+  "reporter": "spec",
+  "watch-files": ["./test/*.test.ts"],
+  "spec":  ["./test/*.test.ts"]
+}

.npmignore

@@ -10,3 +10,4 @@ coverage/
 .eslintrc.js
 .prettierrc.json
 .travis.yml
+.mocharc.json

src/languageservice/parser/yaml-documents.ts

@@ -44,7 +44,15 @@ export class SingleYAMLDocument extends JSONDocument {
       if (node?.commentBefore) {
         this._lineComments.push(`#${node.commentBefore}`);
       }
+
+      if (node?.comment) {
+        this._lineComments.push(`#${node.comment}`);
+      }
     });
+
+    if (this._internalDocument.comment) {
+      this._lineComments.push(`#${this._internalDocument.comment}`);
+    }
   }
 
   set internalDocument(document: Document) {

test/yamlSchemaService.test.ts

@@ -91,5 +91,16 @@ describe('YAML Schema Service', () => {
       expect(schema.schema.required).is.undefined;
       expect(schema.schema.definitions.bar.type).eqls('string');
     });
+
+    it('should handle modeline schema comment in the middle of file', () => {
+      const documentContent = `foo:\n  bar\n# yaml-language-server: $schema=https://json-schema.org/draft-07/schema#\naa:bbb\n`;
+      const content = `${documentContent}`;
+      const yamlDock = parse(content);
+
+      const service = new SchemaService.YAMLSchemaService(requestServiceMock);
+      service.getSchemaForResource('', yamlDock.documents[0]);
+
+      expect(requestServiceMock).calledOnceWith('https://json-schema.org/draft-07/schema#');
+    });
   });
 });