fix formatting for files where types are inserted

README.md

@@ -134,6 +134,12 @@ configure({
 
 ## Release Notes
 
+## [1.6.1] - 2021-05-22
+
+### Fixed
+
+- Fix the automatic formatting in files where types are inserted
+
 ## [1.6.0] - 2021-05-20
 
 ### Added

packages/extension/package.json

@@ -4,7 +4,7 @@
   "publisher": "rikurouvila",
   "description": "🧪  Take samples of runtime values and turn them into type definitions automatically",
   "repository": "https://github.com/rikukissa/typehole",
-  "version": "1.6.0",
+  "version": "1.6.1",
   "private": true,
   "icon": "images/logo.png",
   "galleryBanner": {

packages/extension/src/listener.ts

@@ -35,13 +35,7 @@ function createServer() {
 
   fastify.post("/samples", async (request, reply) => {
     const body = request.body as any;
-    log(
-      body.id,
-      "-",
-      "New sample",
-      JSON.stringify(request.body).substr(0, 30),
-      "received"
-    );
+    log(body.id, "-", "New sample", JSON.stringify(request.body), "received");
 
     const samples = addSample(body.id, body.sample);
     const typeString = samplesToType(samples);
@@ -200,7 +194,16 @@ async function updateTypes(hole: Typehole, types: string, fileName: string) {
   vscode.workspace.applyEdit(workEdits);
 
   try {
-    await vscode.commands.executeCommand("editor.action.formatDocument");
+    const workEdits = new vscode.WorkspaceEdit();
+    const edits = await vscode.commands.executeCommand<vscode.TextEdit[]>(
+      "vscode.executeFormatDocumentProvider",
+      document.uri
+    );
+
+    if (edits) {
+      workEdits.set(document.uri, edits);
+      vscode.workspace.applyEdit(workEdits);
+    }
   } catch (err) {
     error("Formatting the document failed", err.message);
   }

packages/runtime/package.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.6.0",
+  "version": "1.6.1",
   "name": "typehole",
   "repository": "rikukissa/typehole",
   "description": "Turn runtime types into static typescript types automatically",