borgwarehouse/pages/api/cronjob/checkStatus.js

// This API is design to be used by a cron (of your choice). Call it with curl for example
//(e.g : curl --request POST --url 'http://localhost:3000/api/cronjob/checkStatus' --header 'Authorization: Bearer 5173f388c0f4a0df92d1412c3036ddc897c22e4448')

//Lib
import { promises as fs } from 'fs';
import path from 'path';
const util = require('node:util');
const exec = util.promisify(require('node:child_process').exec);
import nodemailerSMTP from '../../../helpers/functions/nodemailerSMTP';
import emailAlertStatus from '../../../helpers/templates/emailAlertStatus';

export default async function handler(req, res) {
    if (req.headers.authorization == null) {
        res.status(401).json({
            status: 401,
            message: 'Unauthorized',
        });
        return;
    }

    const CRONJOB_KEY = process.env.CRONJOB_KEY;
    const ACTION_KEY = req.headers.authorization.split(' ')[1];

    if (req.method == 'POST' && ACTION_KEY === CRONJOB_KEY) {
        //Var
        let newRepoList;
        let repoListToSendAlert = [];
        let usersList;
        const date = Math.round(Date.now() / 1000);
        const jsonDirectory = path.join(process.cwd(), '/config');

        ////PART 1 : Status
        try {
            //Call the shell : getLastSave.sh
            //Find the absolute path of the shells directory
            const shellsDirectory = path.join(process.cwd(), '/helpers');
            //Exec the shell
            const { stdout, stderr } = await exec(
                `${shellsDirectory}/shells/getLastSave.sh`
            );
            if (stderr) {
                console.log('stderr:', stderr);
                res.status(500).json({
                    status: 500,
                    message:
                        'Error on getting the date for last save, contact the administrator.',
                });
                return;
            }
            //Parse the JSON output of getLastSave.sh to use it
            const lastSave = JSON.parse(stdout);

            //Find the absolute path of the json directory
            let repoList = await fs.readFile(
                jsonDirectory + '/repo.json',
                'utf8'
            );
            //Parse the repoList
            repoList = JSON.parse(repoList);

            //Rebuild a newRepoList with the lastSave timestamp updated and the status updated.
            newRepoList = repoList;
            for (let index in newRepoList) {
                const repoFiltered = lastSave.filter(
                    (x) => x.user === newRepoList[index].unixUser
                );
                if (repoFiltered.length === 1) {
                    //Write the timestamp of the last save
                    newRepoList[index].lastSave = repoFiltered[0].lastSave;
                    //Trigger the status if the last save is older than alert setting.
                    if (
                        date - newRepoList[index].lastSave >
                        newRepoList[index].alert
                    ) {
                        newRepoList[index].status = false;
                    } else if (
                        date - newRepoList[index].lastSave <
                        newRepoList[index].alert
                    ) {
                        newRepoList[index].status = true;
                    }
                }
            }
        } catch (err) {
            res.status(500).json({
                status: 500,
                message: "API error : can't update the status.",
            });
            return;
        }

        //// PART 2 : check if there is a repo that need an email alert
        try {
            //Here, a mail is sent every 24H (90000) if a repo has down status
            for (let index in newRepoList) {
                if (
                    !newRepoList[index].status &&
                    (!newRepoList[index].lastStatusAlertSend ||
                        date - newRepoList[index].lastStatusAlertSend > 90000)
                ) {
                    repoListToSendAlert.push(newRepoList[index].alias);
                    newRepoList[index].lastStatusAlertSend = date;
                }
            }
        } catch (err) {
            res.status(500).json({
                status: 500,
                message:
                    "API error : can't check if a repo needs an email alert.",
            });
            return;
        }

        //PART 3 : Save the new repoList
        try {
            //Stringify the repoList to write it into the json file.
            newRepoList = JSON.stringify(newRepoList);
            //Write the new json
            fs.writeFile(jsonDirectory + '/repo.json', newRepoList, (err) => {
                if (err) console.log(err);
            });
        } catch (err) {
            res.status(500).json({
                status: 500,
                message: "API error : can't write the new repoList.",
            });
            return;
        }
        //PART 4 : Send the mail alert
        if (repoListToSendAlert.length > 0) {
            // Read user informations
            try {
                //Read the email of the user
                usersList = await fs.readFile(
                    jsonDirectory + '/users.json',
                    'utf8'
                );
                //Parse the usersList
                usersList = JSON.parse(usersList);
            } catch (err) {
                res.status(500).json({
                    status: 500,
                    message: "API error : can't read user information.",
                });
                return;
            }
            // If the user has enabled email alerts
            if (usersList[0].emailAlert) {
                //Send mail
                //Create the SMTP Transporter
                const transporter = nodemailerSMTP();
                //Mail options
                const mailData = emailAlertStatus(
                    usersList[0].email,
                    usersList[0].username,
                    repoListToSendAlert
                );
                transporter.sendMail(mailData, function (err, info) {
                    if (err) {
                        console.log(err);
                        res.status(400).json({
                            message:
                                'An error occured while sending the email : ' +
                                err,
                        });
                        return;
                    } else {
                        console.log(info);
                    }
                });
            }
        }

        //PART 5 : Sucess
        res.status(200).json({
            success: 'Status cron has been executed.',
        });
        return;
    } else {
        res.status(401).json({
            status: 401,
            message: 'Unauthorized',
        });
        return;
    }
}