Add size limit back (#161)

src/ContentTooLargeException.php

@@ -0,0 +1,45 @@
+<?php
+declare(strict_types = 1);
+
+namespace LanguageServer;
+
+/**
+ * Thrown when the document content is not parsed because it exceeds the size limit
+ */
+class ContentTooLargeException extends \Exception
+{
+    /**
+     * The URI of the file that exceeded the limit
+     *
+     * @var string
+     */
+    public $uri;
+
+    /**
+     * The size of the file in bytes
+     *
+     * @var int
+     */
+    public $size;
+
+    /**
+     * The limit that was exceeded in bytes
+     *
+     * @var int
+     */
+    public $limit;
+
+    /**
+     * @param string     $uri      The URI of the file that exceeded the limit
+     * @param int        $size     The size of the file in bytes
+     * @param int        $limit    The limit that was exceeded in bytes
+     * @param \Throwable $previous The previous exception used for the exception chaining.
+     */
+    public function __construct(string $uri, int $size, int $limit, \Throwable $previous = null)
+    {
+        $this->uri = $uri;
+        $this->size = $size;
+        $this->limit = $limit;
+        parent::__construct("$uri exceeds size limit of $limit bytes ($size)", 0, $previous);
+    }
+}

src/LanguageServer.php

@@ -183,11 +183,16 @@ class LanguageServer extends AdvancedJsonRpc\Dispatcher
                 // Give LS to the chance to handle requests while indexing
                 yield timeout();
                 $this->client->window->logMessage(
-                    MessageType::INFO,
+                    MessageType::LOG,
                     "Parsing file $i/$count: {$textDocument->uri}"
                 );
                 try {
                     yield $this->project->loadDocument($textDocument->uri);
+                } catch (ContentTooLargeException $e) {
+                    $this->client->window->logMessage(
+                        MessageType::INFO,
+                        "Ignoring file {$textDocument->uri} because it exceeds size limit of {$e->limit} bytes ({$e->size})"
+                    );
                 } catch (Exception $e) {
                     $this->client->window->logMessage(
                         MessageType::ERROR,

src/Project.php

@@ -103,10 +103,20 @@ class Project
     public function loadDocument(string $uri): Promise
     {
         return coroutine(function () use ($uri) {
+            $limit = 150000;
             if ($this->clientCapabilities->xcontentProvider) {
                 $content = (yield $this->client->textDocument->xcontent(new TextDocumentIdentifier($uri)))->text;
+                $size = strlen($content);
+                if ($size > $limit) {
+                    throw new ContentTooLargeException($uri, $size, $limit);
+                }
             } else {
-                $content = file_get_contents(uriToPath($uri));
+                $path = uriToPath($uri);
+                $size = filesize($path);
+                if ($size > $limit) {
+                    throw new ContentTooLargeException($uri, $size, $limit);
+                }
+                $content = file_get_contents($path);
             }
             if (isset($this->documents[$uri])) {
                 $document = $this->documents[$uri];