Started reworking yamlCompletion and autoCompleter

apupier · Jun 12, 2017 · d0e10d2 · d0e10d2
1 parent 5707c3c
commit d0e10d2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 27 deletions.
diff --git a/server/src/languageService/services/autoCompleter.ts b/server/src/languageService/services/autoCompleter.ts
@@ -1,7 +1,8 @@
 import {SchemaToMappingTransformer} from "../schemaToMappingTransformer"
-import {TextDocument} from 'vscode-languageserver-types';
+import {TextDocument, CompletionList} from 'vscode-languageserver-types';
 import {JSONSchema} from "../jsonSchema";
 import {YAMLDocument, YAMLNode} from 'yaml-ast-parser';
+
 let AutoComplete = require('triesearch');
 
 export class AutoCompleter {
@@ -17,11 +18,15 @@ export class AutoCompleter {
     }
 
     public search(searchItem: String): Array<String>{
-        return this.autoCompleter.search(searchItem).map(x => x.value);
+        return this.autoCompleter.search(searchItem).map(x => ({
+            label: x.value.toString()
+        }));
     }
 
-    public searchAll(): Array<String>{
-        return Object.keys(this.kuberSchema);
+    public searchAll() {
+        return Object.keys(this.kuberSchema).map(x => ({
+            label: x.toString()
+        }));
     }
 
     public initData(data:Array<String>): void {
@@ -42,7 +47,7 @@ export class AutoCompleter {
             let results = this.kuberSchema[getParentNodeValue].map(x => x.children).reduce((a, b) => a.concat(b)).filter((value, index, self) => self.indexOf(value) === index);
             this.initData(results);
         }else{
-            this.initData(this.searchAll());
+            this.initData(Object.keys(this.kuberSchema));
         }
     }
 
@@ -65,8 +70,9 @@ export class AutoCompleter {
 
     public generateScalarAutocompletion(nodeValue: String){
         let results = this.kuberSchema[nodeValue.toString()].map(x => x.default).filter((value, index, self) => self.indexOf(value) === index && value !== undefined);
-        return results;
+        return results.map(x => ({
+            label: x.toString()
+        }));
     }
 
-
 }
diff --git a/server/src/languageService/services/yamlCompletion.ts b/server/src/languageService/services/yamlCompletion.ts
@@ -26,36 +26,23 @@ export class YamlCompletion {
       let node = findNode(<YAMLNode>doc, offset);
 
       if(node !== undefined && node.kind === Kind.SCALAR){
-        autoComplete.generateScalarAutocompletion(node.parent.key.value).map(x => result.items.push({
-            label: x.toString()
-        }));
-        return result;
+        return autoComplete.generateScalarAutocompletion(node.parent.key.value);
       }
 
-      if(node != undefined && node.value !== null && node.value.kind === Kind.SCALAR){
-        autoComplete.generateScalarAutocompletion(node.key.value).map(x => result.items.push({
-            label: x.toString()
-        }));
-        return result;
+      if(node != undefined && node.value !== null && node.value !== undefined && node.value.kind === Kind.SCALAR){
+        return autoComplete.generateScalarAutocompletion(node.key.value);
       }
 
       if(node === undefined || node.parent === null){
         //Its a root node
-        autoComplete.searchAll().map(x => result.items.push({
-            label: x.toString()
-        }));
+        return autoComplete.searchAll();
       }else{
         autoComplete.generateResults(node);
-        autoComplete.search(node.key.value).map(x => result.items.push({
-            label: x.toString()
-        }));
+        return autoComplete.search(node.key.value);
       }
-
-
-      return result;
+
     });
-  }
-
+  } 
 
 
 }