Shutdown when the socket is closed (felixfbecker#191)

Author: felixfbecker

bin/php-language-server.php

@@ -68,10 +68,12 @@
                 exit(1);
             } else if ($pid === 0) {
                 // Child process
-                $ls = new LanguageServer(
-                    new ProtocolStreamReader($socket),
-                    new ProtocolStreamWriter($socket)
-                );
+                $reader = new ProtocolStreamReader($socket);
+                $writer = new ProtocolStreamWriter($socket);
+                $reader->on('close', function () {
+                    fwrite(STDOUT, "Connection closed\n");
+                });
+                $ls = new LanguageServer($reader, $writer);
                 Loop\run();
                 // Just for safety
                 exit(0);

src/LanguageServer.php

@@ -65,6 +65,10 @@ public function __construct(ProtocolReader $reader, ProtocolWriter $writer)
     {
         parent::__construct($this, '/');
         $this->protocolReader = $reader;
+        $this->protocolReader->on('close', function () {
+            $this->shutdown();
+            $this->exit();
+        });
         $this->protocolReader->on('message', function (Message $msg) {
             coroutine(function () use ($msg) {
                 // Ignore responses, this is the handler for requests and notifications

src/ProtocolReader.php

@@ -8,6 +8,8 @@
 /**
  * Must emit a "message" event with a Protocol\Message object as parameter
  * when a message comes in
+ *
+ * Must emit a "close" event when the stream closes
  */
 interface ProtocolReader extends EmitterInterface
 {

src/ProtocolStreamReader.php

@@ -25,7 +25,17 @@ public function __construct($input)
     {
         $this->input = $input;
 
+        $this->on('close', function () {
+            Loop\removeReadStream($this->input);
+        });
+
         Loop\addReadStream($this->input, function () {
+            if (feof($this->input)) {
+                // If stream_select reported a status change for this stream,
+                // but the stream is EOF, it means it was closed.
+                $this->emit('close');
+                return;
+            }
             while (($c = fgetc($this->input)) !== false && $c !== '') {
                 $this->buffer .= $c;
                 switch ($this->parsingMode) {