1 | <?php |
||
2 | |||
3 | namespace Firesphere\YubiAuth\Extensions; |
||
4 | |||
5 | use SilverStripe\Forms\FieldList; |
||
6 | use SilverStripe\Forms\NumericField; |
||
7 | use SilverStripe\ORM\DataExtension; |
||
8 | |||
9 | /** |
||
10 | * Class YubiAuthMemberExtension |
||
11 | * |
||
12 | * Enable yubikey authentication disabling temporarily |
||
13 | * |
||
14 | * @property MemberExtension $owner |
||
15 | * @property string $Yubikey |
||
16 | * @property int $NoYubikeyCount |
||
17 | */ |
||
18 | class MemberExtension extends DataExtension |
||
19 | { |
||
20 | private static $db = [ |
||
0 ignored issues
–
show
introduced
by
![]() |
|||
21 | 'Yubikey' => 'Varchar(255)', |
||
22 | 'NoYubikeyCount' => 'Int' |
||
23 | ]; |
||
24 | |||
25 | /** |
||
26 | * @inheritdoc |
||
27 | * @param array $labels |
||
28 | */ |
||
29 | public function updateFieldLabels(&$labels) |
||
30 | { |
||
31 | $labels['Yubikey'] = _t(self::class . '.YUBIKEY', 'Yubikey code'); |
||
32 | $labels['NoYubikeyCount'] = _t(self::class . '.NOYUBIKEYCOUNT', 'Login count without yubikey'); |
||
33 | } |
||
34 | |||
35 | /** |
||
36 | * @inheritdoc |
||
37 | * @param FieldList $fields |
||
38 | */ |
||
39 | public function updateCMSFields(FieldList $fields) |
||
40 | { |
||
41 | $fields->removeByName(['NoYubikeyCount', 'Yubikey']); |
||
42 | $yubiCount = NumericField::create('NoYubikeyCount'); |
||
43 | |||
44 | $fields->addFieldsToTab('Root.MFA', [$yubiCount]); |
||
45 | |||
46 | return $fields; |
||
47 | } |
||
48 | |||
49 | /** |
||
50 | * @inheritdoc |
||
51 | */ |
||
52 | public function onBeforeWrite() |
||
53 | { |
||
54 | // Empty the yubikey field on member write, if the yubiauth is not required |
||
55 | // Maybe the user lost the key? So a new one will be set next time it's logged in with key |
||
56 | if (!$this->owner->MFAEnabled) { |
||
0 ignored issues
–
show
|
|||
57 | $this->owner->Yubikey = ''; |
||
58 | } |
||
59 | } |
||
60 | } |
||
61 |