siimee/backend/lib/services/user.js

'use strict'
require('dotenv').config()
const crypto = require('crypto')
const Util = require('util')
const Jwt = require('@hapi/jwt')
const JWT = require('jsonwebtoken')
const Schmervice = require('@hapipal/schmervice')
const SecurePassword = require('secure-password')

// Configuration for Brevo
const SibApiV3Sdk = require('sib-api-v3-sdk')
const defaultClient = SibApiV3Sdk.ApiClient.instance
const apiKey = defaultClient.authentications['api-key']
apiKey.apiKey = process.env.BREVO_KEY

const apiInstance = new SibApiV3Sdk.TransactionalEmailsApi()

const hashEmail = async email => {
    try {
        return crypto.createHmac('sha256', '').update(email).digest('hex')
    } catch (err) {
        console.error('ERROR :=>', err)
        return undefined
    }
}
// const emailsSent = {}
const hasher = async (pwd, steak) => {
    const hash = await pwd.hash(steak)
    const result = await pwd.verify(steak, hash)
    let squirtle = null

    switch (result) {
        case SecurePassword.INVALID_UNRECOGNIZED_HASH:
            return console.error(
                'This hash was not made with secure-password. Attempt legacy algorithm',
            )
        case SecurePassword.INVALID:
            return console.log('Invalid password')
        case SecurePassword.VALID:
            return result
        case SecurePassword.VALID_NEEDS_REHASH:
            console.log('Yay you made it, wait for us to improve your safety')
            try {
                squirtle = await pwd.hash(steak)
                // console.log('improvedHash', squirtle)
                // const saveHash = Auth.insert({user_email: matchingEmails}, ).into('token')
                return squirtle
            } catch (err) {
                console.error(
                    'You are authenticated, but we could not improve your safety this time around',
                )
            }
            break
    }
}

/** Class for methods used in the User plugin */
module.exports = class UserService extends Schmervice.Service {
    /**
     * Unsure of what our constructor does
     * @param  {...any} args
     */
    constructor(...args) {
        super(...args)
        const pwd = new SecurePassword()
        // TODO: Invalidate this cache somehow after a certain time period has
        // passed
        // TODO: Remove hashedEmails in preference of activeSessions
        this.hashedEmails = {
            // NOTE: key is email hash and value is timestamp in ms
            // abc123456: '123456689',
        }

        // this.activeSessions = [
        // {
        // user: {
        // useremail: email,
        // hashedEmail: hashedEmail,
        // username: name,
        // },
        // expiration: 1203984710234
        // },
        // token: 'tokenString + expirationDate + salt'
        // ]

        this.pwd = {
            hash: Util.promisify(pwd.hash.bind(pwd)),
            verify: Util.promisify(pwd.verify.bind(pwd)),
        }
    }

    /**
     * Use knex to find users with id column
     * @param {number} id
     * @param {*} txn
     * @returns
     */
    async findById(id, txn) {
        const { User } = this.server.models()

        return await User.query(txn)
            .throwIfNotFound()
            .first()
            .where({ user_id: id })
    }

    /**
     * Use knew to find first user with username
     * @param {*} username
     * @param {*} txn
     * @returns
     */
    async findByUsername(username, txn) {
        const { User } = this.server.models()

        return await User.query(txn)
            .throwIfNotFound()
            .first()
            .where({ user_name: username })
    }

    /**
     * Use knew to find first user with useremail
     * @param {*} username
     * @param {*} txn
     * @returns
     */
    async findByUserEmail(userEmail, txn) {
        const { User } = this.server.models()
        const user = await User.query(txn)
            .throwIfNotFound()
            .first()
            .where({ user_email: userEmail })
        return user
    }

    /**
     * Signup function
     * @param {*} param0
     * @param {*} txn
     * @returns
     */
    async signup({ password, userInfo, created_at }, txn) {
        const { User, Auth } = this.server.models()
        const matchingEmails = await User.query().where(
            'user_email',
            userInfo.user_email,
        )
        if (matchingEmails.length > 0) {
            throw `User ${userInfo.user_email} already exists: Cannot create a user without a unique email`
        }
        // Insert User Info to User table
        const insertUser = await User.query().insert(userInfo)
        // insert a row with blank password to be updated by changePassword()
        await Auth.query().insert({
            user_email: insertUser.user_email,
            created_at: created_at,
            token: null,
        })
        // update null token with hashed password
        await this.changePassword(insertUser.user_email, password, txn)
        return {
            user_id: insertUser.id,
            user_name: insertUser.user_name,
            user_email: insertUser.user_email,
            is_poster: insertUser.is_poster,
            is_admin: insertUser.is_admin,
            is_verified: insertUser.is_verified,
        }
    }

    /**
     * Updates user's info
     * @param {number} id
     * @param {*} param1
     * @param {*} txn
     * @returns
     */
    async update(id, { password, ...userInfo }, txn) {
        const { User } = this.server.models()

        if (Object.keys(userInfo).length > 0) {
            await User.query(txn)
                .throwIfNotFound()
                .where({ id })
                .patch(userInfo)
        }

        if (password) {
            await this.changePassword(id, password, txn)
        }

        return id
    }

    /**
     * Self explanatory
     * @param {*} param0
     * @param {*} txn
     * @returns
     */
    async login({ email, password }, txn) {
        const { User, Auth } = this.server.models()

        const user = await Auth.query(txn)
            .throwIfNotFound()
            .first()
            .where({ user_email: email })

        const bufferPepper = Buffer.from(process.env.PEPPER + password)

        /** Uncomment to run password check using SecurePassword */
        const passwordCheck = await this.pwd.verify(bufferPepper, user.token)
        if (passwordCheck === SecurePassword.VALID_NEEDS_REHASH) {
            await this.changePassword(user.user_email, password, txn)
        } else if (passwordCheck !== SecurePassword.VALID) {
            throw User.createNotFoundError()
        }

        return user
    }

    /**
     * Create a token to be sent in request headers
     * @param {User} user
     * @returns {Token}
     */
    // TODO: Put this logic in the routes, NOT here
    // createSessionToken(user, payload)
    // createAccessToken()
    //
    createToken(user) {
        const key = this.server.registrations['main-app-plugin'].options.jwtKey

        let token = Jwt.token.generate(
            {
                aud: 'urn:audience:test',
                iss: 'urn:issuer:test',
                // ...payload,
                email: user.email,
                name: user.name,
                seeking: user.seeking,
                salt: 'a;ldfkjas;l/dfkafnml;/cjkf',
                // profile_id: user.profile_id,
            },
            {
                key: key,
                algorithm: 'HS256',
            },
            {
                ttlSec: 4 * 60 * 60, // 7 days
            },
        )
        console.log('token :=>', token)
        token = Jwt.token.generate(
            {
                aud: 'urn:audience:test',
                iss: 'urn:issuer:test',
                // ...payload,
                email: user.email,
                name: user.name,
                seeking: user.seeking,
                salt: 'qpowieurpqowytqpoieryu',
                // profile_id: user.profile_id,
            },
            {
                key: key,
                algorithm: 'HS256',
            },
            {
                ttlSec: 4 * 60 * 60, // 7 days
            },
        )
        console.log('\n')
        console.log('token :=>', token)
        token = Jwt.token.generate(
            {
                aud: 'urn:audience:test',
                iss: 'urn:issuer:test',
                // ...payload,
                email: user.email,
                name: user.name,
                seeking: user.seeking,
                salt: 'a;ldfkjas;l/dfkafnml;/cjkf',
                // profile_id: user.profile_id,
            },
            {
                key: key,
                algorithm: 'HS256',
            },
            {
                ttlSec: 6 * 60 * 60, // 7 days
            },
        )
        console.log('token :=>', token)
        token = Jwt.token.generate(
            {
                aud: 'urn:audience:test',
                iss: 'urn:issuer:test',
                // ...payload,
                email: user.email,
                name: user.name,
                seeking: user.seeking,
                salt: 'a;ldfkjas;l/dfkafnml;/cjkf',
                // profile_id: user.profile_id,
            },
            {
                key: key,
                algorithm: 'HS256',
            },
            {
                ttlSec: 7 * 60 * 60, // 7 days
            },
        )
        console.log('token :=>', token)

        // TODO: keep userinfo and it's association with the sessionToken in state/memory
        // registerSession(user, sessionToken) // useremail, token
        // this.registerSession(user, token)
        return token
    }

    async registerSession(user, hashedEmail, token) {
        const sessionRequester = {
            user: user,
            hashedEmail: hashedEmail,
            token: token,
        }
        const alreadyExists = this.activeSessions.find(
            sessionRequester => sessionRequester.hashedEmail === hashedEmail,
        )
        if (!alreadyExists) {
            this.activeSessions.push(sessionRequester)
        }
    }
    /**
     * Validates whether a token has expired or not
     * @param {User} user
     * @returns {Token}
     */
    validateToken(token) {
        const key = this.server.registrations['main-app-plugin'].options.jwtKey
        // NOTE: reveals email...perhaps unhashed email belongs here instead...
        try {
            const decodedToken = Jwt.token.decode(token)
            Jwt.token.verify(decodedToken, key)
            return { isValid: true, payload: decodedToken.decoded.payload }
        } catch (err) {
            return { isValid: false, error: err.message }
        }
    }

    /**
     * Use knex to try to change password entry
     * @param {number} id
     * @param {string} password
     * @param {*} txn
     * @returns {number}
     */
    async changePassword(email, password, txn) {
        const { Auth } = this.server.models()

        const hashed = await this.pwd.hash(
            Buffer.from(process.env.PEPPER + password),
        )

        await Auth.query(txn)
            .throwIfNotFound()
            .where({ user_email: email })
            .patch({
                // user_email: email,
                token: hashed,
            })
        return email
    }

    async getPassword(email, txn) {
        const { Auth } = this.server.models()

        const passwordRow = await Auth.query(txn)
            .where('user_email', email)
            .first()

        return passwordRow ? passwordRow.token : null
    }

    async checkEmailRegistry(userEmail) {
        const hashedEmail = await hashEmail(userEmail)
        const now = Date.now()
        // hashedEmail needs to be derived by email, salt
        const expiration = this.hashedEmails[hashedEmail]
        console.log('this.hashedEmails :=>', this.hashedEmails)
        const emailIsRegistered = Object.keys(this.hashedEmails).includes(
            hashedEmail,
        )
        const emailIsExpired = now > expiration ? true : false
        console.log('emailIsRegistered :=>', emailIsRegistered)
        console.log('emailIsExpired :=>', emailIsExpired)
        if (emailIsRegistered && !emailIsExpired) {
            return true
        } else {
            // try {
            // delete this.hashedEmails[hashedEmail]
            // } catch (err) {
            // console.error('ERROR :=>', err)
            // }
            return false
        }
    }

    /**
     * Sends a Transactional Email via Brevo
     * @ returns {Object}
     */
    async emailSent(userEmail) {
        const hashedEmail = await hashEmail(userEmail)
        if (Object.keys(this.hashedEmails).includes(hashedEmail)) {
            return new Error('email address already in cache!!')
        }
        // Set expiration time for five minutes from now
        const duration = 1000 * 60 * 5
        this.hashedEmails[hashedEmail] = Date.now() + duration
        const sendSmtpEmail = {
            to: [
                {
                    email: userEmail,
                },
            ],
            templateId: 1,
            params: {
                // TODO: Change this in production...
                link: `localhost:3000/verify/${hashedEmail}`,
            },
        }

        await apiInstance.sendTransacEmail(sendSmtpEmail).then(
            data => {
                return {
                    wasSuccessfull: true,
                    data: data,
                }
            },
            error => {
                return {
                    wasSuccessfull: false,
                    error: error,
                }
            },
        )
    }
}