friendica/src/Module/TwoFactor/Recovery.php

<?php

namespace Friendica\Module\TwoFactor;

use Friendica\BaseModule;
use Friendica\Core\L10n;
use Friendica\Core\Renderer;
use Friendica\Core\Session;
use Friendica\Model\TwoFactorRecoveryCode;

/**
 * // Page 1a: Recovery code verification
 *
 * @package Friendica\Module\TwoFactor
 */
class Recovery extends BaseModule
{
	public static function init()
	{
		if (!local_user()) {
			return;
		}
	}

	public static function post()
	{
		if (!local_user()) {
			return;
		}

		if (defaults($_POST, 'action', null) == 'recover') {
			self::checkFormSecurityTokenRedirectOnError('2fa', 'twofactor_recovery');

			$a = self::getApp();

			$recovery_code = defaults($_POST, 'recovery_code', '');

			if (TwoFactorRecoveryCode::existsForUser(local_user(), $recovery_code)) {
				TwoFactorRecoveryCode::markUsedForUser(local_user(), $recovery_code);
				Session::set('2fa', true);
				notice(L10n::t('Remaining recovery codes: %d', TwoFactorRecoveryCode::countValidForUser(local_user())));

				// Resume normal login workflow
				Session::setAuthenticatedForUser($a, $a->user, true, true);
			} else {
				notice(L10n::t('Invalid code, please retry.'));
			}
		}
	}

	public static function content()
	{
		if (!local_user()) {
			self::getApp()->internalRedirect();
		}

		// Already authenticated with 2FA token
		if (Session::get('2fa')) {
			self::getApp()->internalRedirect();
		}

		return Renderer::replaceMacros(Renderer::getMarkupTemplate('twofactor/recovery.tpl'), [
			'$form_security_token' => self::getFormSecurityToken('twofactor_recovery'),

			'$title'            => L10n::t('Two-factor recovery'),
			'$message'          => L10n::t('<p>You can enter one of your one-time recovery codes in case you lost access to your mobile device.</p>'),
			'$recovery_message' => L10n::t('Don’t have your phone? <a href="%s">Enter a two-factor recovery code</a>', '2fa/recovery'),
			'$recovery_code'    => ['recovery_code', L10n::t('Please enter a recovery code'), '', '', '', 'placeholder="000000-000000"'],
			'$recovery_label'   => L10n::t('Submit recovery code and complete login'),
		]);
	}
}