1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
namespace BringYourOwnIdeas\SecurityChecker\Extensions; |
4
|
|
|
|
5
|
|
|
use BringYourOwnIdeas\SecurityChecker\Models\SecurityAlert; |
6
|
|
|
use SilverStripe\ORM\ArrayList; |
7
|
|
|
use SilverStripe\ORM\DataExtension; |
8
|
|
|
use SilverStripe\View\ArrayData; |
9
|
|
|
|
10
|
|
|
class PackageSecurityExtension extends DataExtension |
11
|
|
|
{ |
12
|
|
|
private static $has_many = [ |
|
|
|
|
13
|
|
|
'SecurityAlerts' => SecurityAlert::class |
14
|
|
|
]; |
15
|
|
|
|
16
|
|
|
private static $summary_fields = [ |
|
|
|
|
17
|
|
|
'listSecurityAlertIdentifiers' => 'Security alerts', |
18
|
|
|
]; |
19
|
|
|
|
20
|
|
|
/** |
21
|
|
|
* Simply returns a comma separated list of active SecurityAlert Identifiers for this record. |
22
|
|
|
* Used in CSV exports as a type of brief indication (as opposed to full info) |
23
|
|
|
*/ |
24
|
|
|
public function listSecurityAlertIdentifiers() |
25
|
|
|
{ |
26
|
|
|
$alerts = $this->owner->SecurityAlerts()->Column('Identifier'); |
27
|
|
|
return $alerts ? implode(', ', $alerts) : null; |
28
|
|
|
} |
29
|
|
|
|
30
|
|
|
/** |
31
|
|
|
* updates the badges that render as part of the screen targeted |
32
|
|
|
* summary for this Package |
33
|
|
|
* |
34
|
|
|
* @param ArrayList $badges |
35
|
|
|
*/ |
36
|
|
|
public function updateBadges($badges) |
37
|
|
|
{ |
38
|
|
|
if ($this->owner->SecurityAlerts()->exists()) { |
39
|
|
|
$badges->push(ArrayData::create([ |
40
|
|
|
'Title' => _t(__CLASS__ . '.BADGE_SECURITY', 'RISK: Security'), |
41
|
|
|
'Type' => 'warning security-alerts__toggler', |
42
|
|
|
])); |
43
|
|
|
} |
44
|
|
|
} |
45
|
|
|
|
46
|
|
|
/** |
47
|
|
|
* Adds security alert notifications into the schema |
48
|
|
|
* |
49
|
|
|
* @param array &$schema |
50
|
|
|
* @return string |
51
|
|
|
*/ |
52
|
|
|
public function updateDataSchema(&$schema) |
53
|
|
|
{ |
54
|
|
|
// The keys from the SecurityAlert model that we need in the React component |
55
|
|
|
$keysToPass = ['Identifier', 'ExternalLink']; |
56
|
|
|
|
57
|
|
|
$alerts = []; |
58
|
|
|
foreach ($this->owner->SecurityAlerts()->toNestedArray() as $alert) { |
59
|
|
|
$alerts[] = array_intersect_key($alert, array_flip($keysToPass)); |
60
|
|
|
} |
61
|
|
|
|
62
|
|
|
$schema['securityAlerts'] = $alerts; |
63
|
|
|
} |
64
|
|
|
} |
65
|
|
|
|