import-videos-folder.js

const config       = require('./src/app/chatbot.json');
const database     = require('./src/js/chatbot/database');
const locales      = require('./src/js/chatbot/locales');

const fs           = require('fs');
const glob         = require('glob');
const mediainfo    = require('mediainfo-wrapper');
const moment       = require('moment');
const {v4: uuidv4} = require('uuid');
const yargs        = require('yargs');

let addedVideosAmount = 0;
let localRegExp = /(.*)(\.mp4)$/i;
let options = {};
let playlists = [];
let slashesRegExp = /(\/|\\)+/g;
let time = moment().unix();
let videosFolder = config.videosFolder.replace(slashesRegExp, '/');

const argv = yargs
    .option('backup', {
        alias: 'b',
        default: false,
        description: 'Create an additional backup',
        type: 'boolean'
    })
    .option('channel', {
        alias: 'c',
        description: 'Channel name which owns the videos (required)',
        type: 'string'
    })
    .option('identity', {
        alias: 'i',
        description: 'Channel id / Room id (required if channel is not in databse)',
        type: 'number'
    })
    .option('locale', {
        alias: 'l',
        default: 'en',
        description: 'Locale to use in date generation and log messages',
        type: 'string'
    })
    .option('log', {
        default: true,
        description: 'Show logs in CLI',
        type: 'boolean'
    })
    .option('subname', {
        alias: 'sn',
        default: true,
        description: 'Add date as sub name',
        type: 'boolean'
    })
    .help()
    .alias('help', 'h')
    .argv;

if (argv.channel) {
    argv.channel = argv.channel.toLowerCase();
}

if (argv.locale) {
    locales.changeLanguage(argv.locale);
}

function log(message) {
    if (argv.log === true) {
        console.log(message);
    }
}

function importSummary(videos, index) {
    // if all files processed and no video was added
    if (index === videos.length && !addedVideosAmount) {
        log(locales.t('ivf-all-video-exists'));
    } else if (index === videos.length && addedVideosAmount) {
        const durationObject = moment.duration(moment().unix() - time, 's');
        const hours = ('0' + durationObject.hours()).substr(-2);
        const minutes = ('0' + durationObject.minutes()).substr(-2);
        const seconds = ('0' + durationObject.seconds()).substr(-2);
        const duration = hours + ':' + minutes + ':' + seconds + ' h';

        log('----- ' + locales.t('summary') + ' -----');
        log(locales.t('ivf-added-videos', [
            addedVideosAmount,
            locales.t('video', {count: addedVideosAmount}),
            playlists.length,
            locales.t('playlist', {count: playlists.length}),
            duration
        ]));
    }
}

function addVideoRecursive(videos, channel, index) {
    let videosAmount = videos.length;

    // if is first function call
    if (!index) {
        log(locales.t('ivf-found-videos', [videosAmount, locales.t('video', {count: videosAmount})]));
    }

    let relFilePath = videos[index].replace(videosFolder, '');
    let join = 'JOIN playlist_video_join AS pvj ON pvj.video_id = v.id';
    let where = ['file = ?', 'channel_id = ?'];
    let prepare = [relFilePath, channel.id];

    database.find('name', 'video AS v', join, where, '', '', 1, prepare, function(videoRows) {
        // if video not exits
        if (!videoRows.length) {
            mediainfo(videos[index]).then(function(data) {
                // if data genaral is defined
                if (data.length && typeof data[0].general !== 'undefined') {
                    // milliseconds to seconds
                    let duration = ((data[0].general.duration[0] / 1000) -.5).toFixed(0);
                    let format = locales.t('date');
                    let playlist = relFilePath.indexOf('/') ? relFilePath.split('/')[0] : 'General';
                    let name = relFilePath
                        .split('/').pop()
                        .replace(localRegExp, '$1')
                        .replace(/(-|_| )([a-z])/gi, function($0, $1, $2) {
                            let prefix = ' ';

                            if ($1 === '-') {
                                prefix = ' - ';
                            }

                            return prefix + $2.toUpperCase();
                        });
                    name = name[0].toUpperCase() + name.slice(1);
                    let subName = '';

                    // if sub name flag is true
                    if (argv.subname) {
                        let isoDate = data[0].general.file_last_modification_date[0].replace(/^(UTC )([0-9-]+) ([0-9:.]+)$/, '$2T$3Z');
                        subName = moment(isoDate).format(format);
                    }

                    // if duration lower equal than 23:59:59 hours
                    if (duration <= 86399) {
                        let values = {
                            channelId: channel.id,
                            name: name,
                            subName: subName,
                            file: relFilePath,
                            duration: duration, // unix timestamp (seconds)
                            player: 'local',
                            updatedAt: time, // unix timestamp (seconds)
                            createdAt: time // unix timestamp (seconds)
                        };

                        database.insert('video', [values], function(insert) {
                            // if video was added
                            if (insert.lastID) {
                                addedVideosAmount++;
                                log(locales.t('ivf-added-video', [name, playlist, addedVideosAmount]));
                            }

                            let select = 'p.id, COUNT(pvj.playlist_id) AS videoQuantity';
                            let from = 'playlist AS p';
                            join = 'LEFT JOIN playlist_video_join AS pvj ON p.id = pvj.playlist_id ';
                            where = ['name = ?'];
                            prepare = [playlist];

                            database.find(select, from, join, where, '', '', 1, prepare, function(playlistRows) {
                                values = {
                                    uuid: uuidv4(),
                                    playlistId: null,
                                    videoId: insert.lastID,
                                    sort: 0,
                                    updatedAt: time, // unix timestamp (seconds)
                                    createdAt: time // unix timestamp (seconds)
                                };

                                // if playlist NOT exists
                                if (playlistRows[0].id === null) {
                                    let playlistValues = {
                                        channelId: channel.id,
                                        name: playlist,
                                        active: 0,
                                        updatedAt: time, // unix timestamp (seconds)
                                        createdAt: time // unix timestamp (seconds)
                                    };
                                    database.insert('playlist', [playlistValues], function(playlistInsert) {
                                        if (playlists.indexOf(playlist) === -1) {
                                            playlists.push(playlist);
                                            log(locales.t('ivf-added-playlist', [playlist]));
                                        }

                                        values.playlistId = playlistInsert.lastID;
                                        database.insert('playlist_video_join', [values], function(pvjInsert) {
                                            index++;
                                            if (typeof videos[index] !== 'undefined') {
                                                addVideoRecursive(videos, channel, index);
                                            }
                                            importSummary(videos, index);
                                        });
                                    });
                                } else {
                                    if (playlists.indexOf(playlist) === -1) {
                                        playlists.push(playlist);
                                        log(locales.t('ivf-found-playlist', [playlist]));
                                    }

                                    values.playlistId = playlistRows[0].id;
                                    values.sort = 1 * playlistRows[0].videoQuantity;
                                    database.insert('playlist_video_join', [values], function(pvjInsert) {
                                        index++;
                                        if (typeof videos[index] !== 'undefined') {
                                            addVideoRecursive(videos, channel, index);
                                        }
                                        importSummary(videos, index);
                                    });
                                }
                            });
                        });
                    }
                }
            }).catch(function(e) {
                console.error(e);
            });
        } else {
            index++;
            if (typeof videos[index] !== 'undefined') {
                addVideoRecursive(videos, channel, index);
            }
        }
        importSummary(videos, index);
    });
}

if (fs.existsSync(videosFolder) && typeof argv.channel === 'string' && argv.channel.length) {
    if (argv.backup === true) {
        database.backup('YYYY-MM-DD_HH-mm-ss');
    }

    glob(videosFolder + '*/*.mp4', options, function(error, videos) {
        if (error) {
            console.log(error);
        } else {
            let select = 'id, name, updated_at AS updatedAt, created_at AS createdAt';
            database.find(select, 'channel', '', ['name = ?'], '', '', 1, [argv.channel], function(rows) {
                if (rows.length) {
                    log(locales.t('ivf-found-channel', [argv.channel]));
                    addVideoRecursive(videos, rows[0], 0);
                } else if (typeof argv.identity === 'number' && argv.identity > 0) {
                    let values = {
                        id: argv.identity,
                        name: argv.channel,
                        updatedAt: time,
                        createdAt: time
                    };

                    database.insert('channel', [values], function(insert) {
                        values.id = insert.lastID;
                        addVideoRecursive(videos, values, 0);
                    });
                } else {
                    log(locales.t('ivf-identity-is-missing'));
                }
            });
        }
    });
} else if (!fs.existsSync(videosFolder)) {
    log(locales.t('ivf-folder-not-found', [config.videosFolder]));
} else if  (typeof argv.channel === 'undefined' || !argv.channel.length) {
    log(locales.t('ivf-channel-is-missing'));
}