Clean up

Author: pokey

README.md

@@ -1 +1,22 @@
 # command-server README
+
+Creates an http server listening for commands. 
+
+## Features
+
+On startup, spawns an http server listening for commands. The port of the server for the active editor is written to a file named `vscode-port` in the operating system's default directory for temporary files.
+
+Accepts requests of the form:
+
+```http
+POST /execute-command HTTP/1.1
+
+{
+  "commandId": "some-command-id",
+  "args": [
+    "default"
+  ]
+}
+```
+
+## Release Notes

src/extension.ts

@@ -11,63 +11,66 @@ interface Command {
   commandId: string;
   args: any[];
   expectResponse: boolean;
-  timestamp: Date;
 }
 
-// this method is called when your extension is activated
-// your extension is activated the very first time the command is executed
-export function activate(context: vscode.ExtensionContext) {
-  // Use the console to output diagnostic information (console.log) and errors (console.error)
-  // This line of code will only be executed once when your extension is activated
-  console.log(
-    'Congratulations, your extension "command-server" is now active!'
-  );
-
-  var port: number | null = null;
-  //create a server object:
-  const server = http.createServer(function (req, res) {
-    console.log("Got request");
+function getBody(req: http.IncomingMessage) {
+  return new Promise<any>((resolve, reject) => {
     var body = "";
     req.on("data", function (chunk) {
       body += chunk;
     });
-    req.on("end", function () {
-      console.log("POSTed: " + body);
-      const { timestamp: rawTimestamp, ...rest } = JSON.parse(body);
-      const commandInfo = {
-        ...rest,
-        // timestamp: new Date(rawTimestamp),
-      };
-      console.dir(commandInfo);
+    req.on("end", () => resolve(JSON.parse(body)));
+  });
+}
 
-      vscode.commands.executeCommand(
-        commandInfo.commandId,
-        ...commandInfo.args
-      );
-      res.writeHead(200);
-      res.end("Hello World!");
-    });
+export function activate(context: vscode.ExtensionContext) {
+  var port: number | null = null;
+
+  const server = http.createServer(async function (req, res) {
+    if (!vscode.window.state.focused) {
+      res.writeHead(401);
+      res.end("This editor is not active");
+      return;
+    }
+
+    const commandInfo: Command = await getBody(req);
+
+    vscode.commands.executeCommand(commandInfo.commandId, ...commandInfo.args);
+
+    res.writeHead(200);
+    res.end();
   });
 
   server.listen(0, "localhost", function () {
     const address: AddressInfo = (server.address() as unknown) as AddressInfo;
     port = address.port;
+
     console.log("Listening on port " + address.port);
+
     if (vscode.window.state.focused) {
       writePort();
     }
   });
 
-  vscode.window.onDidChangeWindowState((event) => {
-    if (event.focused && port !== null) {
-      writePort();
+  const windowStateDisposable = vscode.window.onDidChangeWindowState(
+    (event) => {
+      if (event.focused && port !== null) {
+        writePort();
+      }
     }
-  });
+  );
 
   function writePort() {
     const path = join(tmpdir(), "vscode-port");
+    console.log(`Saved port ${port} to path ${path}`);
     writeFileSync(path, `${port}`);
   }
+
+  context.subscriptions.push(windowStateDisposable, {
+    dispose() {
+      server.close();
+    },
+  });
 }
 
 // this method is called when your extension is deactivated