Kekkai/chart/chart.js

const ChartJSImage = require('chart.js-image');
const pool = require('../shared/database/src/postgresql.js');
const fs = require('fs');
const axios = require('axios');
const logger = require('../shared/logger/src/main.js');

/**
 * Graph generation
 * @param {String} from_currency - The currency that is being converted
 * @param {String} conv_currency - The currency to be converted into
 * @param {String} start_date - Start date of the period
 * @param {String} end_date - End date of the period
 * @returns {Promise<Error|string>}
 */
async function gen_chart(from_currency, conv_currency, start_date, end_date) {
    const data = await pool.query('SELECT date, rate FROM currency WHERE ' +
        '(date BETWEEN $3 AND $4) AND from_currency = $1 AND conv_currency = $2 ORDER BY date ', [
        from_currency.toUpperCase(),
        conv_currency.toUpperCase(),
        start_date,
        end_date
    ]);

    if (!data['rows'][0]) return 'Missing data';

    const date = [];
    const rate = [];

    for (let i = 0; i < data.rows.length; i++) {
        date.push(data.rows[i].date.toLocaleDateString());
        rate.push(data.rows[i].rate);
    }

    const chart = ChartJSImage().chart({
        type: 'line',
        options: {
            title: {
                  display: true,
                  text: `${from_currency} / ${conv_currency}`,
              },
            scales: [
                {
                    yAxes: [
                        {
                            ticks: {
                                beginAtZero: false,
                            }
                        }
                    ]
                }
            ]
        },
        data: {
            labels: date,
            datasets: [
                {
                    borderColor: rate[0] < rate[rate.length-1] ? 'rgb(24, 218, 39)' : 'rgb(243, 85, 50)',
                    backgroundColor:  rate[0] < rate[rate.length-1] ? 'rgb(36, 175, 47)' : 'rgb(218, 56, 24)',
                    data: rate,
                    borderWidth: 2,
                },
            ],
        },
        scales: {
           xAxes: [
               {
                   scaleLabel: {
                       display: true,
                       labelString: 'Day'
                   },
               },
           ],
            yAxes: [
                {
                    stacked: false,
                    scaleLabel: {
                        display: true,
                        labelString: 'Rate'
                    },
                },
            ],
        },
    }).width(1000).height(1000);

    logger.debug(chart.toURL());

    return chart.toURL();
}


/**
 * Saving a graph to a file
 * @param {String} url  - URL (or Buffer) to the chart
 * @param {String} filename - filename
 */
function save_chart(url, filename) {
    if (!fs.existsSync('../charts')) fs.mkdirSync('../charts');
    if (!url.startsWith('https://')) throw new Error('The passed parameter is not a URL');

    logger.info(`The schedule has been saved. The path of the graph 'chart/${filename}'`);
    axios({url, responseType: 'stream',}).then(
        response =>
            new Promise((resolve, reject) => {
                response.data
                    .pipe(fs.createWriteStream(`../charts/${filename}`))
                    .on('finish', () => resolve())
                    .on('error', e => reject(e));
            })
    );
}

module.exports = { gen_chart, save_chart };