BetterDiscordApp-v2/scripts/inject.js

61 lines
2.7 KiB
JavaScript

const args = process.argv;
const fs = require('fs');
const path = require('path');
const useBdRelease = args[2] && args[2].toLowerCase() === 'release';
const releaseInput = useBdRelease ? args[3] && args[3].toLowerCase() : args[2] && args[2].toLowerCase();
const release = releaseInput === 'canary' ? 'Discord Canary' : releaseInput === 'ptb' ? 'Discord PTB' : 'Discord';
console.log(`Injecting into version ${release}`);
const discordPath = (function() {
if (process.platform === 'win32') {
const basedir = path.join(process.env.LOCALAPPDATA, release.replace(/ /g, ''));
if (!fs.existsSync(basedir)) throw new Error(`Cannot find directory for ${release}`);
const version = fs.readdirSync(basedir).filter(f => fs.lstatSync(path.join(basedir, f)).isDirectory() && f.split('.').length > 1).sort().reverse()[0];
return path.join(basedir, version, 'resources');
} else if (process.platform === 'darwin') {
const appPath = releaseInput === 'canary' ? path.join('/Applications', 'Discord Canary.app')
: releaseInput === 'ptb' ? path.join('/Applications', 'Discord PTB.app')
: useBdRelease && args[3] ? args[3] ? args[2] : args[2]
: path.join('/Applications', 'Discord.app');
return path.join(appPath, 'Contents', 'Resources');
} else if (process.platform === 'linux') {
return path.join('/usr', 'share', release.toLowerCase().replace(/ /g, '-'), 'resources');
}
})();
if (!fs.existsSync(discordPath)) throw new Error(`Cannot find directory for ${release}`);
console.log(`Found ${release} in ${discordPath}`);
const appPath = path.join(discordPath, 'app');
const packageJson = path.join(appPath, 'package.json');
const indexJs = path.join(appPath, 'index.js');
if (!fs.existsSync(appPath)) fs.mkdirSync(appPath);
if (fs.existsSync(packageJson)) fs.unlinkSync(packageJson);
if (fs.existsSync(indexJs)) fs.unlinkSync(indexJs);
const bdPath = useBdRelease ? path.resolve(__dirname, '..', 'release') : path.resolve(__dirname, '..');
console.log(`Writing package.json`);
fs.writeFileSync(packageJson, JSON.stringify({
name: 'betterdiscord',
description: 'BetterDiscord',
main: 'index.js',
private: true
}, null, 4));
console.log(`Writing index.js`);
fs.writeFileSync(indexJs, `const path = require('path');
const fs = require('fs');
const Module = require('module');
const electron = require('electron');
const basePath = path.join(__dirname, '..', 'app.asar');
electron.app.getAppPath = () => basePath;
Module._load(basePath, null, true);
electron.app.on('ready', () => new (require('${bdPath.replace(/\\/g, '\\\\').replace(/'/g, '\\\'')}').BetterDiscord)());
`);
console.log(`Injection successful, please restart ${release}.`);