tethys.backend/app/Controllers/Http/Auth/AuthController.ts

import type { HttpContextContract } from '@ioc:Adonis/Core/HttpContext';
import User from 'App/Models/User';
// import Hash from '@ioc:Adonis/Core/Hash';
// import InvalidCredentialException from 'App/Exceptions/InvalidCredentialException';
import AuthValidator from 'App/Validators/AuthValidator';

import TwoFactorAuthProvider from 'App/Services/TwoFactorAuthProvider';
// import { LoginState } from 'Contracts/enums';
// import { StatusCodes } from 'http-status-codes';

export default class AuthController {
    //   login function
    public async login({ request, response, auth, session }: HttpContextContract) {
        // console.log({
        // 	registerBody: request.body(),
        // });
        await request.validate(AuthValidator);

        // const plainPassword = await request.input('password');
        // const email = await request.input('email');
        // grab uid and password values off request body
        const { email, password } = request.only(['email', 'password']);

        try {
            // // attempt to verify credential and login user
            // await auth.use('web').attempt(email, plainPassword);

            const user = await auth.use('web').verifyCredentials(email, password);
            if (user.isTwoFactorEnabled) {
                // session.put("login.id", user.id);
                // return view.render("pages/two-factor-challenge");

                session.flash('user_id', user.id);
                return response.redirect().back();
            
                // let state = LoginState.STATE_VALIDATED;
                // return response.status(StatusCodes.OK).json({
                //     state: state,
                //     new_user_id: user.id,
                // });
            }          
            await auth.login(user);
        } catch (error) {
            // if login fails, return vague form message and redirect back
            session.flash('message', 'Your username, email, or password is incorrect');
            return response.redirect().back();
        }

        // otherwise, redirect todashboard
        response.redirect('/apps/dashboard');
    }

    public async twoFactorChallenge({ request, session, auth, response }) {
        const { code, recoveryCode, login_id } = request.only(['code', 'recoveryCode', 'login_id']);
        // const user = await User.query().where('id', session.get('login.id')).firstOrFail();
        const user = await User.query().where('id', login_id).firstOrFail();
      
        if (code) {
            const isValid = await TwoFactorAuthProvider.validate(user, code);
            if (isValid) {
                // login user and redirect to dashboard
                await auth.login(user);
                response.redirect('/apps/dashboard');
            } else {
                session.flash('message', 'Your tow factor code is incorrect');
                return response.redirect().back();
            }
        } else if (recoveryCode) {
            const codes = user?.twoFactorRecoveryCodes ?? [];
            if (codes.includes(recoveryCode)) {
                user.twoFactorRecoveryCodes = codes.filter((c) => c !== recoveryCode);
                await user.save();
                await auth.login(user);
                response.redirect('/apps/dashboard');
            }
        }
    }

    //   logout function
    public async logout({ auth, response }: HttpContextContract) {
        // await auth.logout();
        await auth.use('web').logout();
        response.redirect('/app/login');
        // return response.status(200);
    }
}