iptvnator/api.ts

import { app, ipcMain } from 'electron';
import { parse } from 'iptv-playlist-parser';
import axios from 'axios';
import { guid } from '@datorama/akita';
import { Playlist } from './src/app/home/playlist.interface';
import Nedb from 'nedb-promises-ts';

const fs = require('fs');
const join = require('path').join;
const openAboutWindow = require('about-window').default;
const userData = app.getPath('userData');
const db = new Nedb<Playlist>({
    filename: `${userData}/db/data.db`,
    autoload: true,
});

export class Api {
    constructor() {
        ipcMain.on('parse-playlist-by-url', (event, args) => {
            axios.get(args.url).then((result) => {
                const array = result.data.split('\n');
                const parsedPlaylist = this.parsePlaylist(array);
                const playlistObject = this.createPlaylistObject(
                    args.title,
                    parsedPlaylist,
                    args.url
                );
                this.insertToDb(playlistObject);
                event.sender.send('parse-response', {
                    payload: playlistObject,
                });
            });
        });

        ipcMain.on('parse-playlist', (event, args) => {
            const parsedPlaylist = this.parsePlaylist(args.playlist);
            const playlistObject = this.createPlaylistObject(
                args.title,
                parsedPlaylist
            );
            this.insertToDb(playlistObject);
            event.sender.send('parse-response', { payload: playlistObject });
        });

        ipcMain.on('playlists-all', async (event, args) => {
            const playlists = await db.find(
                {},
                { count: 1, title: 1, _id: 1, url: 1, importDate: 1 }
            );
            event.sender.send('playlist-all-result', {
                payload: playlists,
            });
        });

        ipcMain.on('playlist-by-id', async (event, args) => {
            const playlist = await db.findOne({ _id: args.id });
            event.sender.send('parse-response', {
                payload: playlist,
            });
        });

        ipcMain.on('playlist-remove-by-id', async (event, args) => {
            const removed = await db.remove({ _id: args.id });
            if (removed) {
                event.sender.send('playlist-remove-by-id-result', {
                    message: 'playlist was removed',
                });
            }
        });

        ipcMain.on('show-about', () => {
            openAboutWindow({
                icon_path: join(__dirname, 'dist/assets/icons/icon.png'),
                copyright: 'Copyright (c) 2020 4gray',
                package_json_dir: __dirname,
            });
        });

        // open playlist from file system
        ipcMain.on('open-file', (event, args) => {
            fs.readFile(args.filePath, 'utf-8', (err, data) => {
                if (err) {
                    console.log(
                        'An error ocurred reading the file :' + err.message
                    );
                    return;
                }
                const array = (data as string).split('\n');
                const parsedPlaylist = this.parsePlaylist(array);
                const playlistObject = this.createPlaylistObject(
                    args.fileName,
                    parsedPlaylist
                );
                this.insertToDb(playlistObject);
                event.sender.send('parse-response', {
                    payload: playlistObject,
                });
            });
        });

        ipcMain.on('update-favorites', async (event, args) => {
            const updated = await db.update(
                { id: args.id },
                { $set: { favorites: args.favorites } }
            );
            if (!updated.numAffected || updated.numAffected === 0) {
                console.error('Error: Favorites were not updated');
            }
        });
    }

    /**
     * Saves playlist to the localStorage
     * @param name name of the playlist
     * @param playlist playlist to save
     * @param url url of the playlist
     */
    createPlaylistObject(name: string, playlist: any, url?: string): Playlist {
        return {
            id: guid(),
            _id: guid(),
            filename: name,
            title: name,
            count: playlist.items.length,
            playlist,
            importDate: new Date().toISOString(),
            lastUsage: new Date().toISOString(),
            favorites: [],
            ...(url ? { url } : {}),
        };
    }

    /**
     * Parses string based array to playlist object
     * @param m3uArray m3u playlist as array with strings
     */
    parsePlaylist(m3uArray: any[]): any {
        const playlistAsString = m3uArray.join('\n');
        return parse(playlistAsString);
    }

    /**
     * Inserts new playlist to the database
     * @param playlist playlist to add
     */
    insertToDb(playlist) {
        db.insert(playlist).then((response) => {
            console.log('playlist was saved...', response._id);
        });
    }
}