BetterDiscordApp-v2/client/src/modules/packageinstaller.js

import EventListener from './eventlistener';
import asar from 'asar';
import fs from 'fs';
import path from 'path';

import { Modals } from 'ui';
import { Utils } from 'common';
import PluginManager from './pluginmanager';
import Globals from './globals';
import Security from './security';
import { ReactComponents } from './reactcomponents';
import Reflection from './reflection';
import DiscordApi from './discordapi';

export default class PackageInstaller {

    /**
     * Handler for drag and drop package install
     * @param {String} filePath Path to local file
     * @param {String} channelId Current channel id
     */
    static async dragAndDropHandler(filePath, channelId) {
        try {
            const config = JSON.parse(asar.extractFile(filePath, 'config.json').toString());
            const { info, main } = config;

            let icon = null;
            if (info.icon && info.icon_type) {
                const extractIcon = asar.extractFile(filePath, info.icon);
                icon = `data:${info.icon_type};base64,${Utils.arrayBufferToBase64(extractIcon)}`;
            }

            const isPlugin = info.type && info.type === 'plugin' || main.endsWith('.js');

            // Show install modal
            const modalResult = await Modals.installModal(isPlugin ? 'plugin' : 'theme', config, filePath, icon).promise;

            if (modalResult === 0) {
                // Upload it instead
            }

        } catch (err) {
            console.log(err);
        }
    }

    /**
     * Hash and verify a package
     * @param {Byte[]|String} bytesOrPath byte array of binary or path to local file
     * @param {String} id Package id
     */
    static async verifyPackage(bytesOrPath, id) {
        const bytes = typeof bytesOrPath === 'string' ? fs.readFileSync(bytesOrPath) : bytesOrPath;
        // Temporary hash to simulate response from server
        const tempVerified = ['2e3532ee366816adc37b0f478bfef35e03f96e7aeee9b115f5918ef6a4e94de8', '06a2eb4e37b926354ab80cd83207db67e544c932e9beddce545967a21f8db5aa'];
        const hashBytes = Security.hash('sha256', bytes, 'hex');

        return tempVerified.includes(hashBytes);
    }

    // TODO lots of stuff
    /**
     * Installs or updates defined package
     * @param {Byte[]|String} bytesOrPath byte array of binary or path to local file
     * @param {String} name Package name
     * @param {Boolean} update Does an older version already exist
     */
    static async installPackage(bytesOrPath, id, update = false) {
        try {
            const bytes = typeof bytesOrPath === 'string' ? fs.readFileSync(bytesOrPath) : bytesOrPath;

            const outputName = `${id}.bd`;
            const outputPath = path.join(Globals.getPath('plugins'), outputName);
            fs.writeFileSync(outputPath, bytes);

            let newContent = null;

            if (!update) {
                newContent = await PluginManager.preloadPackedContent(outputName);
            } else {
                newContent = await PluginManager.reloadContent(PluginManager.getPluginById(id));
            }

            return newContent;

        } catch (err) {
            throw err;
        }
    }

    /**
     * Patches Discord upload area for .bd files
     */
    static async uploadAreaPatch() {
        const { selector } = Reflection.resolve('uploadArea');
        this.UploadArea = await ReactComponents.getComponent('UploadArea', { selector });

        const reflect = Reflection.DOM(selector);
        const stateNode = reflect.getComponentStateNode(this.UploadArea);
        const callback = function (e) {
            if (!e.dataTransfer.files.length || !e.dataTransfer.files[0].name.endsWith('.bd')) return;
            e.preventDefault();
            e.stopPropagation();
            e.stopImmediatePropagation();
            stateNode.clearDragging();

            PackageInstaller.dragAndDropHandler(e.dataTransfer.files[0].path, DiscordApi.currentChannel.id);
        };

        // Remove their handler, add ours, then read theirs to give ours priority to stop theirs when we get a .bd file.
        reflect.element.removeEventListener('drop', stateNode.handleDrop);
        reflect.element.addEventListener('drop', callback);
        reflect.element.addEventListener('drop', stateNode.handleDrop);

        this.unpatchUploadArea = function () {
            reflect.element.removeEventListener('drop', callback);
        };
    }

}