Fix: Update method, fileContent is optinal

unlight · Jun 30, 2018 · 0c7e6aa · 0c7e6aa
1 parent 84857f7
commit 0c7e6aa
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -92,12 +92,15 @@ export function createService({ compilerOptions, configFile }: createServiceOpti
     const service = ts.createLanguageService(servicesHost, ts.createDocumentRegistry());
 
     return {
-        update({ fileName, fileContent }: { fileName: string, fileContent: string }) {
+        update({ fileName, fileContent }: { fileName: string, fileContent?: string }) {
             fileName = fileName.replace(/\\/g, '/');
             let fileRef = files[fileName];
             if (!fileRef) {
                 files[fileName] = fileRef = { version: 0, snapshot: undefined };
             }
+            if (fileContent === undefined) {
+                fileContent = readFileSync(fileName, 'utf8');
+            }
             fileRef.snapshot = ts.ScriptSnapshot.fromString(fileContent);
             fileRef.version++;