use Koha::Auth::TwoFactorAuth;

    my $invalid_otp_token;

    my $require_2FA = ( C4::Context->preference('TwoFactorAuthentication') && $type ne "OPAC" ) ? 1 : 0;

    my $auth_challenge_complete;

        if ( $return eq 'ok' || $return eq 'additional-auth-needed' ) {

            $userid = $session->param('id');

        }

        $additional_auth_needed = ( $return eq 'additional-auth-needed' ) ? 1 : 0;

        # We are at the second screen if the waiting-for-2FA is set in session

        # and otp_token param has been passed

        if (   $require_2FA

            && $additional_auth_needed

            && ( my $otp_token = $query->param('otp_token') ) )

        {

            my $patron    = Koha::Patrons->find( { userid => $userid } );

            my $auth      = Koha::Auth::TwoFactorAuth::get_auth( { patron => $patron } );

            my $verified = $auth->verify($otp_token);

            $auth->clear;

            if ( $verified ) {

                # The token is correct, the user is fully logged in!

                $additional_auth_needed = 0;

                $session->param( 'waiting-for-2FA', 0 );

                $return = "ok";

                $auth_challenge_complete = 1;

               # This is an ugly trick to pass the test

               # $query->param('koha_login_context') && ( $q_userid ne $userid )

               # few lines later

                $q_userid = $userid;

            }

            else {

                $invalid_otp_token = 1;

            }

        }

        $session   = undef;

    if ($logout) {

        # voluntary logout the user

        # check wether the user was using their shibboleth session or a local one

        my $shibSuccess = C4::Context->userenv->{'shibboleth'};

        if ( $session ) {

            $session->delete();

            $session->flush;

        }

        C4::Context->_unset_userenv($sessionID);

        if ($cas and $caslogout) {

            logout_cas($query, $type);

        }

        # If we are in a shibboleth session (shibboleth is enabled, a shibboleth match attribute is set and matches koha matchpoint)

        if ( $shib and $shib_login and $shibSuccess) {

            logout_shib($query);

        }

        $sessionID = undef;

        $session   = undef;

        $userid    = undef;

        $additional_auth_needed = 0;

    }

    if ( $require_2FA && ( $loggedin && !$auth_challenge_complete)) {

        my $patron = Koha::Patrons->find({userid => $userid});

        if ( $patron->auth_method eq 'two-factor' ) {

            # Ask for the OTP token

            $additional_auth_needed = 1;

            $session->param('waiting-for-2FA', 1);

            %info = ();# We remove the warnings/errors we may have set incorrectly before

        }

    }

    my $patron = Koha::Patrons->find({ userid => $q_userid }); # Not necessary logged in!

    my @inputs_to_clean = qw( userid password ticket logout.x otp_token );

    if ( $additional_auth_needed ) {

        $template->param(

            TwoFA_prompt => 1,

            invalid_otp_token => $invalid_otp_token,

        );

    }

=item "additional-auth-needed -- User is in an authentication process that is not finished

                return ( "additional-auth-needed", $session )

                    if $session->param('waiting-for-2FA');

package Koha::Auth::TwoFactorAuth;

use Modern::Perl;

use Auth::GoogleAuth;

sub get_auth {

    my ($params) = @_;

    my $patron   = $params->{patron};

    my $secret   = $params->{secret};

    my $secret32 = $params->{secret32};

    if (!$secret && !$secret32){

        $secret32 = $patron->secret;

    }

    my $issuer = $patron->library->branchname;

    my $key_id = sprintf "%s_%s",

      $issuer, ( $patron->email || $patron->userid );

    return Auth::GoogleAuth->new(

        {

            ( $secret   ? ( secret   => $secret )   : () ),

            ( $secret32 ? ( secret32 => $secret32 ) : () ),

            issuer => $issuer,

            key_id => $key_id,

        }

    );

}

1;

requires 'Auth::GoogleAuth', '1.02';

    [% IF TwoFA_prompt %]Two-factor authentication[% END %]

[% ELSIF TwoFA_prompt %]

    <form action="[% script_name | html %]" method="post" name="loginform" id="loginform">

        <input type="hidden" name="koha_login_context" value="intranet" />

        [% FOREACH INPUT IN INPUTS %]

            <input type="hidden" name="[% INPUT.name | html %]" value="[% INPUT.value | html %]" />

        [% END %]

        [% IF invalid_otp_token %]

            <div class="dialog error">Invalid two-factor code</div>

        [% END %]

        <p><label for="otp_token">Two-factor authentication code:</label>

        <input type="text" name="otp_token" id="otp_token" class="input focus" value="" size="20" tabindex="1" />

        <p class="submit"><input id="submit-button" type="submit" value="Verify code" /></p>

        <a class="logout" id="logout" href="/cgi-bin/koha/mainpage.pl?logout.x=1">Log out</a>

    </form>

#!/usr/bin/perl

# This file is part of Koha.

#

# Koha is free software; you can redistribute it and/or modify it

# under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 3 of the License, or

# (at your option) any later version.

#

# Koha 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 General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with Koha; if not, see <http://www.gnu.org/licenses>.

use Modern::Perl;

use Test::More tests => 2;

use C4::Context;

use Koha::AuthUtils;

use Koha::Auth::TwoFactorAuth;

use t::lib::Mocks;

use t::lib::Selenium;

use t::lib::TestBuilder;

my @data_to_cleanup;

my $pref_value = C4::Context->preference('TwoFactorAuthentication');

SKIP: {

    eval { require Selenium::Remote::Driver; };

    skip "Selenium::Remote::Driver is needed for selenium tests.", 2 if $@;

    my $builder  = t::lib::TestBuilder->new;

    my $patron = $builder->build_object({ class => 'Koha::Patrons', value => { flags => 1 }});

    $patron->flags(1)->store; # superlibrarian permission

    my $password = Koha::AuthUtils::generate_password($patron->category);

    t::lib::Mocks::mock_preference( 'RequireStrongPassword', 0 );

    $patron->set_password({ password => $password });

    push @data_to_cleanup, $patron, $patron->category, $patron->library;

    my $s        = t::lib::Selenium->new({ login => $patron->userid, password => $password });

    my $driver   = $s->driver;

    subtest 'Setup' => sub {

        plan tests => 10;

        my $mainpage = $s->base_url . q|mainpage.pl|;

        $driver->get($mainpage);

        like( $driver->get_title, qr(Log in to Koha), 'Hitting the main page should redirect to the login form');

        fill_login_form($s);

        like( $driver->get_title, qr(Koha staff interface), 'Patron with flags superlibrarian should be able to login' );

        C4::Context->set_preference('TwoFactorAuthentication', 0);

        $driver->get($s->base_url . q|members/two_factor_auth.pl|);

        like( $driver->get_title, qr(Error 404), 'Must be redirected to 404 is the pref is off' );

        C4::Context->set_preference('TwoFactorAuthentication', 1);

        $driver->get($s->base_url . q|members/two_factor_auth.pl|);

        like( $driver->get_title, qr(Two-Factor Authentication), 'Must be on the page with the pref on' );

        is( $driver->find_element('//div[@class="two-factor-status"]')->get_text(), 'Status: Disabled', '2FA is disabled' );

        $driver->find_element('//form[@id="two-factor-auth"]//input[@type="submit"]')->click;

        ok($driver->find_element('//img[@id="qr_code"]'), 'There is a QR code');

        $s->fill_form({pin_code => 'wrong_code'});

        $s->submit_form;

        ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid pin code")]'));

        is( $patron->get_from_storage->secret, undef, 'secret is not set in DB yet' );

        my $secret32 = $driver->find_element('//form[@id="two-factor-auth"]//input[@name="secret32"]')->get_value();

        my $auth = Koha::Auth::TwoFactorAuth::get_auth({patron => $patron, secret32 => $secret32});

        my $code = $auth->code();

        $s->fill_form({pin_code => $code});

        $s->submit_form;

        is( $driver->find_element('//div[@class="two-factor-status"]')->get_text(), 'Status: Enabled', '2FA is enabled' );

        $patron = $patron->get_from_storage;

        is( $patron->secret, $secret32, 'secret is set in DB' );

    };

    subtest 'Login' => sub {

        plan tests => 19;

        my $mainpage = $s->base_url . q|mainpage.pl|;

        { # ok first try

            $driver->get($mainpage . q|?logout.x=1|);

            $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);

            like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );

            $driver->capture_screenshot('selenium_failure_2.png');

            fill_login_form($s);

            like( $driver->get_title, qr(Two-Factor Authentication), 'Must be on the second auth screen' );

            is( login_error($s), undef );

            my $auth = Koha::Auth::TwoFactorAuth::get_auth({patron => $patron});

            my $code = $auth->code();

            $auth->clear;

            $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys($code);

            $driver->find_element('//input[@type="submit"]')->click;

            like( $driver->get_title, qr(Checking out to ), 'Must be redirected to the original page' );

        }

        { # second try and logout

            $driver->get($mainpage . q|?logout.x=1|);

            $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);

            like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );

            fill_login_form($s);

            like( $driver->get_title, qr(Two-Factor Authentication), 'Must be on the second auth screen' );

            is( login_error($s), undef );

            $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys('wrong_code');

            $driver->find_element('//input[@type="submit"]')->click;

            ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid two-factor code")]'));

            is( login_error($s), undef );

            $driver->get($mainpage);

            like( $driver->get_title, qr(Two-Factor Authentication), 'Must still be on the second auth screen' );

            is( login_error($s), undef );

            $driver->find_element('//a[@id="logout"]')->click();

            like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );

            is( login_error($s), undef );

        }

        { # second try and success

            $driver->get($mainpage . q|?logout.x=1|);

            $driver->get($s->base_url . q|circ/circulation.pl?borrowernumber=|.$patron->borrowernumber);

            like( $driver->get_title, qr(Log in to Koha), 'Must be on the first auth screen' );

            like( login_error($s), qr(Session timed out) );

            fill_login_form($s);

            like( $driver->get_title, qr(Two-Factor Authentication), 'Must be on the second auth screen' );

            is( login_error($s), undef );

            $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys('wrong_code');

            $driver->find_element('//input[@type="submit"]')->click;

            ok($driver->find_element('//div[@class="dialog error"][contains(text(), "Invalid two-factor code")]'));

            my $auth = Koha::Auth::TwoFactorAuth::get_auth({patron => $patron});

            my $code = $auth->code();

            $auth->clear;

            $driver->find_element('//form[@id="loginform"]//input[@id="otp_token"]')->send_keys($code);

            $driver->find_element('//input[@type="submit"]')->click;

            like( $driver->get_title, qr(Checking out to ), 'Must be redirected to the original page' );

        }

    };

    $driver->quit();

};

END {

    $_->delete for @data_to_cleanup;

    C4::Context->set_preference('TwoFactorAuthentication', $pref_value);

};

sub login_error {

    my ( $s ) = @_;

    my $driver   = $s->driver;

    $s->remove_error_handler;

    my $login_error = eval {

        my $elt = $driver->find_element('//div[@id="login_error"]');

        return $elt->get_text if $elt && $elt->id;

    };

    $s->add_error_handler;

    return $login_error;

}

# Don't use the usual t::lib::Selenium->auth as we don't want the ->get($mainpage) to test the redirect

sub fill_login_form {

    my ( $s ) = @_;

    $s->fill_form({ userid => $s->login, password => $s->password });

    $s->driver->find_element('//input[@id="submit-button"]')->click;

}

            secret         => undef,