danog/psalm
1
0
Fork 0
mirror of https://github.com/danog/psalm.git synced 2024-11-30 04:39:00 +01:00
Code Issues Projects Releases Wiki Activity

Fix auto comletion by partial property or method

Browse Source
This commit is contained in:
Ivan Sidorov 2023-10-26 21:19:39 +00:00
parent 2726ad4d6f
commit d6faff2844
3 changed files with 246 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/Psalm/Codebase.php
View File

@ -1742,6 +1742,9 @@ final class Codebase
$offset = $position->toOffset($file_contents);
$literal_part = $this->getBeginedLiteralPart($file_path, $position);
$begin_literal_offset = $offset - strlen($literal_part);
[$reference_map, $type_map] = $this->analyzer->getMapsForFile($file_path);
if (!$reference_map && !$type_map) {
@ -1776,7 +1779,7 @@ final class Codebase
}
}
if ($offset - $end_pos === 2 || $offset - $end_pos === 3) {
if ($begin_literal_offset - $end_pos === 2) {
$candidate_gap = substr($file_contents, $end_pos, 2);
if ($candidate_gap === '->' || $candidate_gap === '::') {
@ -1801,6 +1804,11 @@ final class Codebase
return [$possible_reference, '::', $offset];
}
if ($offset <= $end_pos && substr($file_contents, $begin_literal_offset - 2, 2) === '::') {
$class_name = explode('::', $possible_reference)[0];
return [$class_name, '::', $offset];
}
// Only continue for references that are partial / don't exist.
if ($possible_reference[0] !== '*') {
continue;
@ -1816,6 +1824,23 @@ final class Codebase
return null;
}
public function getBeginedLiteralPart(string $file_path, Position $position): string
{
$is_open = $this->file_provider->isOpen($file_path);
if (!$is_open) {
throw new UnanalyzedFileException($file_path . ' is not open');
}
$file_contents = $this->getFileContents($file_path);
$offset = $position->toOffset($file_contents);
preg_match('/\$?\w+$/', substr($file_contents, 0, $offset), $matches);
return $matches[0] ?? '';
}
public function getTypeContextAtPosition(string $file_path, Position $position): ?Union
{
$file_contents = $this->getFileContents($file_path);
@ -1957,6 +1982,26 @@ final class Codebase
return $completion_items;
}
/**
* @param list<CompletionItem> $items
* @return list<CompletionItem>
*/
public function filterCompletionItemsByBeginLiteralPart(array $items, string $literal_part): array
{
if (!$literal_part) {
return $items;
}
$res = [];
foreach ($items as $item) {
if ($item->insertText && strpos($item->insertText, $literal_part) === 0) {
$res[] = $item;
}
}
return $res;
}
/**
* @return list<CompletionItem>
*/

3
src/Psalm/Internal/LanguageServer/Server/TextDocument.php
View File

@ -297,6 +297,7 @@ class TextDocument
try {
$completion_data = $this->codebase->getCompletionDataAtPosition($file_path, $position);
$literal_part = $this->codebase->getBeginedLiteralPart($file_path, $position);
if ($completion_data) {
[$recent_type, $gap, $offset] = $completion_data;
@ -305,6 +306,8 @@ class TextDocument
->textDocument->completion->completionItem->snippetSupport ?? false;
$completion_items =
$this->codebase->getCompletionItemsForClassishThing($recent_type, $gap, $snippetSupport);
$completion_items =
$this->codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
} elseif ($gap === '[') {
$completion_items = $this->codebase->getCompletionItemsForArrayKeys($recent_type);
} else {

198
tests/LanguageServer/CompletionTest.php
View File

@ -15,6 +15,7 @@ use Psalm\Tests\TestCase;
use Psalm\Tests\TestConfig;
use Psalm\Type;
use function array_map;
use function count;
class CompletionTest extends TestCase
@ -370,7 +371,7 @@ class CompletionTest extends TestCase
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$this->assertNull($codebase->getCompletionDataAtPosition('somefile.php', new Position(16, 41)));
$this->assertSame(['B\C', '->', 456], $codebase->getCompletionDataAtPosition('somefile.php', new Position(16, 41)));
}
public function testCompletionOnTemplatedThisProperty(): void
@ -725,6 +726,201 @@ class CompletionTest extends TestCase
$this->assertSame('baz()', $completion_items[1]->insertText);
}
public function testObjectPropertyOnAppendToEnd(): void
{
$codebase = $this->codebase;
$config = $codebase->config;
$config->throw_exception = false;
$this->addFile(
'somefile.php',
'<?php
namespace B;
class A {
public $aProp = 123;
public $bProp = 234;
public function bar() {
$this->aPr
}
}',
);
$codebase->file_provider->openFile('somefile.php');
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$position = new Position(8, 34);
$completion_data = $codebase->getCompletionDataAtPosition('somefile.php', $position);
$literal_part = $codebase->getBeginedLiteralPart('somefile.php', $position);
$this->assertSame(['B\A&static', '->', 223], $completion_data);
$completion_items = $codebase->getCompletionItemsForClassishThing($completion_data[0], $completion_data[1], true);
$completion_items = $codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
$completion_item_texts = array_map(fn($item) => $item->insertText, $completion_items);
$this->assertSame(['aProp'], $completion_item_texts);
}
public function testObjectPropertyOnReplaceEndPart(): void
{
$codebase = $this->codebase;
$config = $codebase->config;
$config->throw_exception = false;
$this->addFile(
'somefile.php',
'<?php
namespace B;
class A {
public $aProp1 = 123;
public $aProp2 = 234;
public function bar() {
$this->aProp2;
}
}',
);
$codebase->file_provider->openFile('somefile.php');
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$position = new Position(8, 34);
$completion_data = $codebase->getCompletionDataAtPosition('somefile.php', $position);
$literal_part = $codebase->getBeginedLiteralPart('somefile.php', $position);
$this->assertSame(['B\A&static', '->', 225], $completion_data);
$completion_items = $codebase->getCompletionItemsForClassishThing($completion_data[0], $completion_data[1], true);
$completion_items = $codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
$completion_item_texts = array_map(fn($item) => $item->insertText, $completion_items);
$this->assertSame(['aProp1', 'aProp2'], $completion_item_texts);
}
public function testSelfPropertyOnAppendToEnd(): void
{
$codebase = $this->codebase;
$config = $codebase->config;
$config->throw_exception = false;
$this->addFile(
'somefile.php',
'<?php
namespace B;
class A {
public static $aProp = 123;
public static $bProp = 234;
public function bar() {
self::$aPr
}
}',
);
$codebase->file_provider->openFile('somefile.php');
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$position = new Position(8, 34);
$completion_data = $codebase->getCompletionDataAtPosition('somefile.php', $position);
$literal_part = $codebase->getBeginedLiteralPart('somefile.php', $position);
$this->assertSame(['B\A', '::', 237], $completion_data);
$completion_items = $codebase->getCompletionItemsForClassishThing($completion_data[0], $completion_data[1], true);
$completion_items = $codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
$completion_item_texts = array_map(fn($item) => $item->insertText, $completion_items);
$this->assertSame(['$aProp'], $completion_item_texts);
}
public function testStaticPropertyOnAppendToEnd(): void
{
$codebase = $this->codebase;
$config = $codebase->config;
$config->throw_exception = false;
$this->addFile(
'somefile.php',
'<?php
namespace B;
class A {
public static $aProp = 123;
public static $bProp = 234;
public function bar() {
static::$aPr
}
}',
);
$codebase->file_provider->openFile('somefile.php');
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$position = new Position(8, 36);
$completion_data = $codebase->getCompletionDataAtPosition('somefile.php', $position);
$literal_part = $codebase->getBeginedLiteralPart('somefile.php', $position);
$this->assertSame(['B\A', '::', 239], $completion_data);
$completion_items = $codebase->getCompletionItemsForClassishThing($completion_data[0], $completion_data[1], true);
$completion_items = $codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
$completion_item_texts = array_map(fn($item) => $item->insertText, $completion_items);
$this->assertSame(['$aProp'], $completion_item_texts);
}
public function testStaticPropertyOnReplaceEndPart(): void
{
$codebase = $this->codebase;
$config = $codebase->config;
$config->throw_exception = false;
$this->addFile(
'somefile.php',
'<?php
namespace B;
class A {
public static $aProp1 = 123;
public static $aProp2 = 234;
public function bar() {
self::$aProp2;
}
}',
);
$codebase->file_provider->openFile('somefile.php');
$codebase->scanFiles();
$this->analyzeFile('somefile.php', new Context());
$position = new Position(8, 34);
$completion_data = $codebase->getCompletionDataAtPosition('somefile.php', $position);
$literal_part = $codebase->getBeginedLiteralPart('somefile.php', $position);
$this->assertSame(['B\A', '::', 239], $completion_data);
$completion_items = $codebase->getCompletionItemsForClassishThing($completion_data[0], $completion_data[1], true);
$completion_items = $codebase->filterCompletionItemsByBeginLiteralPart($completion_items, $literal_part);
$completion_item_texts = array_map(fn($item) => $item->insertText, $completion_items);
$this->assertSame(['$aProp1', '$aProp2'], $completion_item_texts);
}
public function testCompletionOnNewExceptionWithoutNamespace(): void
{
$codebase = $this->codebase;