/
SafeArrayKeyChecker.php
40 lines (35 loc) · 1.17 KB
/
SafeArrayKeyChecker.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
<?php
namespace Psalm\Example\Plugin;
use PhpParser\Node\Expr\ArrayItem;
use Psalm\Internal\Analyzer\StatementsAnalyzer;
use Psalm\Plugin\EventHandler\Event\AddRemoveTaintsEvent;
use Psalm\Plugin\EventHandler\RemoveTaintsInterface;
class SafeArrayKeyChecker implements RemoveTaintsInterface
{
/**
* Called to see what taints should be removed
*
* @return list<string>
*/
public static function removeTaints(AddRemoveTaintsEvent $event): array
{
$item = $event->getExpr();
$statements_analyzer = $event->getStatementsSource();
if (!($item instanceof ArrayItem) || !($statements_analyzer instanceof StatementsAnalyzer)) {
return [];
}
$item_key_value = '';
if ($item->key) {
if ($item_key_type = $statements_analyzer->node_data->getType($item->key)) {
$key_type = $item_key_type;
if ($key_type->isSingleStringLiteral()) {
$item_key_value = $key_type->getSingleStringLiteral()->value;
}
}
}
if ($item_key_value === 'safe_key') {
return ['html'];
}
return [];
}
}