push activity when backup codes are generated
Signed-off-by: Christoph Wurst <christoph@winzerhof-wurst.at>
This commit is contained in:
parent
7ae9442f3d
commit
9300312046
5 changed files with 123 additions and 4 deletions
|
@ -26,6 +26,7 @@
|
|||
</settings>
|
||||
<providers>
|
||||
<provider>OCA\TwoFactorBackupCodes\Activity\GenericProvider</provider>
|
||||
<provider>OCA\TwoFactorBackupCodes\Activity\Provider</provider>
|
||||
</providers>
|
||||
</activity>
|
||||
</info>
|
||||
|
|
|
@ -56,7 +56,6 @@ class GenericProvider implements IProvider {
|
|||
switch ($event->getSubject()) {
|
||||
case 'twofactor_success':
|
||||
$params = $event->getSubjectParameters();
|
||||
error_log(json_encode($params['provider']));
|
||||
$event->setParsedSubject($l->t('You successfully logged in using two-factor authentication (%1$s)', [
|
||||
$params['provider'],
|
||||
]));
|
||||
|
|
67
apps/twofactor_backupcodes/lib/Activity/Provider.php
Normal file
67
apps/twofactor_backupcodes/lib/Activity/Provider.php
Normal file
|
@ -0,0 +1,67 @@
|
|||
<?php
|
||||
|
||||
/**
|
||||
* @author Christoph Wurst <christoph@winzerhof-wurst.at>
|
||||
* @copyright Copyright (c) 2016 Christoph Wurst <christoph@winzerhof-wurst.at>
|
||||
*
|
||||
* Two-factor backup codes
|
||||
*
|
||||
* This code is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License, version 3,
|
||||
* as published by the Free Software Foundation.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License, version 3,
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*
|
||||
*/
|
||||
|
||||
namespace OCA\TwoFactorBackupCodes\Activity;
|
||||
|
||||
use InvalidArgumentException;
|
||||
use OCP\Activity\IEvent;
|
||||
use OCP\Activity\IProvider;
|
||||
use OCP\ILogger;
|
||||
use OCP\IURLGenerator;
|
||||
use OCP\L10N\IFactory as L10nFactory;
|
||||
|
||||
class Provider implements IProvider {
|
||||
|
||||
/** @var L10nFactory */
|
||||
private $l10n;
|
||||
|
||||
/** @var IURLGenerator */
|
||||
private $urlGenerator;
|
||||
|
||||
/** @var ILogger */
|
||||
private $logger;
|
||||
|
||||
public function __construct(L10nFactory $l10n, IURLGenerator $urlGenerator, ILogger $logger) {
|
||||
$this->logger = $logger;
|
||||
$this->urlGenerator = $urlGenerator;
|
||||
$this->l10n = $l10n;
|
||||
}
|
||||
|
||||
public function parse($language, IEvent $event, IEvent $previousEvent = null) {
|
||||
if ($event->getApp() !== 'twofactor_backupcodes') {
|
||||
throw new InvalidArgumentException();
|
||||
}
|
||||
|
||||
$l = $this->l10n->get('twofactor_backupcodes', $language);
|
||||
|
||||
switch ($event->getSubject()) {
|
||||
case 'codes_generated':
|
||||
$event->setParsedSubject($l->t('You created backup codes for your account'));
|
||||
$event->setIcon($this->urlGenerator->getAbsoluteURL($this->urlGenerator->imagePath('core', 'actions/password.svg')));
|
||||
break;
|
||||
default:
|
||||
throw new InvalidArgumentException();
|
||||
}
|
||||
return $event;
|
||||
}
|
||||
|
||||
}
|
|
@ -24,6 +24,7 @@ namespace OCA\TwoFactorBackupCodes\Service;
|
|||
|
||||
use OCA\TwoFactorBackupCodes\Db\BackupCode;
|
||||
use OCA\TwoFactorBackupCodes\Db\BackupCodeMapper;
|
||||
use OCP\Activity\IManager;
|
||||
use OCP\IUser;
|
||||
use OCP\Security\IHasher;
|
||||
use OCP\Security\ISecureRandom;
|
||||
|
@ -39,10 +40,14 @@ class BackupCodeStorage {
|
|||
/** @var ISecureRandom */
|
||||
private $random;
|
||||
|
||||
public function __construct(BackupCodeMapper $mapper, ISecureRandom $random, IHasher $hasher) {
|
||||
/** @var IManager */
|
||||
private $activityManager;
|
||||
|
||||
public function __construct(BackupCodeMapper $mapper, ISecureRandom $random, IHasher $hasher, IManager $activityManager) {
|
||||
$this->mapper = $mapper;
|
||||
$this->hasher = $hasher;
|
||||
$this->random = $random;
|
||||
$this->activityManager = $activityManager;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -68,9 +73,27 @@ class BackupCodeStorage {
|
|||
array_push($result, $code);
|
||||
}
|
||||
|
||||
$this->publishEvent($user, 'codes_generated');
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
/**
|
||||
* Push an event the user's activity stream
|
||||
*
|
||||
* @param IUser $user
|
||||
* @param string $event
|
||||
*/
|
||||
private function publishEvent(IUser $user, $event) {
|
||||
$activity = $this->activityManager->generateEvent();
|
||||
$activity->setApp('twofactor_backupcodes')
|
||||
->setType('twofactor')
|
||||
->setAuthor($user->getUID())
|
||||
->setAffectedUser($user->getUID());
|
||||
$activity->setSubject($event);
|
||||
$this->activityManager->publish($activity);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param IUser $user
|
||||
* @return bool
|
||||
|
|
|
@ -25,6 +25,8 @@ namespace OCA\TwoFactorBackupCodes\Tests\Unit\Service;
|
|||
use OCA\TwoFactorBackupCodes\Db\BackupCode;
|
||||
use OCA\TwoFactorBackupCodes\Db\BackupCodeMapper;
|
||||
use OCA\TwoFactorBackupCodes\Service\BackupCodeStorage;
|
||||
use OCP\Activity\IEvent;
|
||||
use OCP\Activity\IManager;
|
||||
use OCP\IUser;
|
||||
use OCP\Security\IHasher;
|
||||
use OCP\Security\ISecureRandom;
|
||||
|
@ -41,6 +43,9 @@ class BackupCodeStorageTest extends TestCase {
|
|||
/** @var IHasher|\PHPUnit_Framework_MockObject_MockObject */
|
||||
private $hasher;
|
||||
|
||||
/** @var IManager|\PHPUnit_Framework_MockObject_MockObject */
|
||||
private $activityManager;
|
||||
|
||||
/** @var BackupCodeStorage */
|
||||
private $storage;
|
||||
|
||||
|
@ -52,14 +57,16 @@ class BackupCodeStorageTest extends TestCase {
|
|||
->getMock();
|
||||
$this->random = $this->getMockBuilder(ISecureRandom::class)->getMock();
|
||||
$this->hasher = $this->getMockBuilder(IHasher::class)->getMock();
|
||||
$this->storage = new BackupCodeStorage($this->mapper, $this->random, $this->hasher);
|
||||
$this->activityManager = $this->createMock(IManager::class);
|
||||
$this->storage = new BackupCodeStorage($this->mapper, $this->random, $this->hasher, $this->activityManager);
|
||||
}
|
||||
|
||||
public function testCreateCodes() {
|
||||
$user = $this->getMockBuilder(IUser::class)->getMock();
|
||||
$number = 5;
|
||||
$event = $this->createMock(IEvent::class);
|
||||
|
||||
$user->expects($this->once())
|
||||
$user->expects($this->any())
|
||||
->method('getUID')
|
||||
->will($this->returnValue('fritz'));
|
||||
$this->random->expects($this->exactly($number))
|
||||
|
@ -77,6 +84,28 @@ class BackupCodeStorageTest extends TestCase {
|
|||
$this->mapper->expects($this->exactly($number))
|
||||
->method('insert')
|
||||
->with($this->equalTo($row));
|
||||
$this->activityManager->expects($this->once())
|
||||
->method('generateEvent')
|
||||
->will($this->returnValue($event));
|
||||
$event->expects($this->once())
|
||||
->method('setApp')
|
||||
->with('twofactor_backupcodes')
|
||||
->will($this->returnSelf());
|
||||
$event->expects($this->once())
|
||||
->method('setType')
|
||||
->with('twofactor')
|
||||
->will($this->returnSelf());
|
||||
$event->expects($this->once())
|
||||
->method('setAuthor')
|
||||
->with('fritz')
|
||||
->will($this->returnSelf());
|
||||
$event->expects($this->once())
|
||||
->method('setAffectedUser')
|
||||
->with('fritz')
|
||||
->will($this->returnSelf());
|
||||
$this->activityManager->expects($this->once())
|
||||
->method('publish')
|
||||
->will($this->returnValue($event));
|
||||
|
||||
$codes = $this->storage->createCodes($user, $number);
|
||||
$this->assertCount($number, $codes);
|
||||
|
|
Loading…
Reference in a new issue