1
0
Fork 0
php-language-server/src/ProtocolStreamWriter.php

95 lines
2.3 KiB
PHP
Raw Normal View History

2016-08-25 13:27:14 +00:00
<?php
declare(strict_types = 1);
namespace LanguageServer;
2016-08-25 15:03:29 +00:00
use LanguageServer\Protocol\Message;
2016-10-24 23:37:01 +00:00
use Sabre\Event\{
Loop,
Promise
};
use RuntimeException;
2016-08-25 15:03:29 +00:00
2016-08-25 13:27:14 +00:00
class ProtocolStreamWriter implements ProtocolWriter
{
2016-10-24 23:37:01 +00:00
/**
* @var resource $output
*/
2016-08-25 13:27:14 +00:00
private $output;
2016-10-24 21:08:59 +00:00
/**
2016-10-24 23:37:01 +00:00
* @var array $messages
2016-10-24 21:08:59 +00:00
*/
2016-10-24 23:37:01 +00:00
private $messages = [];
2016-10-24 21:08:59 +00:00
2016-08-25 13:27:14 +00:00
/**
* @param resource $output
*/
public function __construct($output)
{
$this->output = $output;
}
/**
* Sends a Message to the client
*
* @param Message $msg
* @return void
*/
public function write(Message $msg)
2016-08-25 13:27:14 +00:00
{
2016-10-24 23:37:01 +00:00
// if the message queue is currently empty, register a write handler.
if (empty($this->messages)) {
Loop\addWriteStream($this->output, [$this, 'writeData']);
}
$promise = new Promise();
$this->messages[] = [
'message' => (string)$msg,
'promise' => $promise
];
return $promise;
}
/**
* Writes pending messages to the output stream.
* Must be public to be able to be used as a callback.
*
* @return void
*/
2016-10-24 23:41:36 +00:00
public function writeData()
{
2016-10-24 23:37:01 +00:00
$message = $this->messages[0]['message'];
$promise = $this->messages[0]['promise'];
error_clear_last();
$bytesWritten = @fwrite($this->output, $message);
if ($bytesWritten === false) {
$error = error_get_last();
$promise->reject($error);
if ($error !== null) {
throw new RuntimeException('Could not write message: ' . error_get_last()['message']);
} else {
throw new RuntimeException('Could not write message');
}
2016-10-24 23:41:36 +00:00
} else if ($bytesWritten > 0) {
2016-10-24 23:37:01 +00:00
$message = substr($message, $bytesWritten);
}
// Determine if this message was completely sent
if (strlen($message) === 0) {
array_shift($this->messages);
// This was the last message in the queue, remove the write handler.
if (count($this->messages) === 0) {
Loop\removeWriteStream($this->output);
}
$promise->fulfill();
2016-10-24 23:41:36 +00:00
} else {
2016-10-24 23:37:01 +00:00
$this->messages[0]['message'] = $message;
}
2016-08-25 13:27:14 +00:00
}
}