Lightcord/modules/discord_cloudsync/index.js

139 lines
4.3 KiB
JavaScript
Raw Normal View History

2020-09-05 22:50:45 +02:00
const EventEmitter = require('events');
const {CloudSync: CloudSyncNative} = require('./discord_cloudsync_'+process.platform+'.node');
const BACKSLASH_REGEX = /\\/g;
const ID_REGEX = /^[0-9]+$/;
const APPLICATION_STORAGE_ENDPOINT = (applicationId, branchId) =>
`/applications/${applicationId}/branches/${branchId}/storage`;
const DefaultCloudSyncConfiguration = Object.freeze({
STORAGE_MANIFEST: (normalizedInstallPath, userId) => `${normalizedInstallPath}/storage/${userId}/MANIFEST`,
INSTALL_DIR: (normalizedInstallPath) => `${normalizedInstallPath}/content`,
ROOT_PATTERN: ['**/*'],
ROOT_STORAGE_PATH: (normalizedInstallPath, userId) => `${normalizedInstallPath}/storage/${userId}/content`,
ROOT_ID: 'sdk',
ROOT_PLATFORMS: {
macos: 'macos',
windows: 'windows',
linux: 'linux',
},
});
let dispatch;
2020-09-05 22:50:45 +02:00
function makeCallback(resolve, reject) {
return (err, result) => {
if (err != null && err !== '') {
reject(new Error(JSON.parse(err)));
} else {
resolve(result != null && result !== '' ? JSON.parse(result) : null);
}
};
}
function ensureDispatch() {
if (dispatch != null) {
return;
}
dispatch = DiscordNative.nativeModules.requireModule('discord_dispatch');
}
function normalizePathForCloudSync(installPath) {
return installPath.replace(BACKSLASH_REGEX, '/');
}
async function generateConfig({userId, userToken, applicationId, branchId, forceHash}) {
if (!ID_REGEX.test(userId)) {
throw new Error(`invalid user: ${userId}`);
}
const dispatchApplication = dispatch.getLastState().applications[applicationId];
if (!ID_REGEX.test(applicationId) || dispatchApplication == null) {
throw new Error(`invalid application: ${applicationId}`);
}
const dispatchBranch = dispatchApplication[branchId];
if (!ID_REGEX.test(branchId) || dispatchBranch == null) {
throw new Error(`invalid branch: ${branchId}`);
}
const normalizedInstallPath = normalizePathForCloudSync(dispatchBranch.install_path);
const dispatchStorage = dispatchBranch.storage;
const apiEndpoint = await DiscordNative.http.getAPIEndpoint();
return {
forceHash,
manifestPath: DefaultCloudSyncConfiguration.STORAGE_MANIFEST(normalizedInstallPath, userId),
roots:
dispatchStorage.roots != null
? dispatchStorage.roots
: [
{
id: DefaultCloudSyncConfiguration.ROOT_ID,
paths: Object.keys(DefaultCloudSyncConfiguration.ROOT_PLATFORMS).map((platform) => ({
platform,
path: DefaultCloudSyncConfiguration.ROOT_STORAGE_PATH(normalizedInstallPath, userId),
})),
patterns: DefaultCloudSyncConfiguration.ROOT_PATTERN,
},
],
storage: {
baseURL: `${apiEndpoint}${APPLICATION_STORAGE_ENDPOINT(applicationId, branchId)}`,
token: userToken,
},
replacements: {
INSTALLDIR: DefaultCloudSyncConfiguration.INSTALL_DIR(normalizedInstallPath),
USERID: userId,
BRANCHID: branchId,
},
};
}
// TODO: remove deprecated config support
function upgradeConfig(id, config) {
const [applicationId, branchId] = id.split(':');
const {
forceHash,
storage: {token: userToken},
replacements: {USERID: userId},
} = config;
return {applicationId, branchId, forceHash, userId, userToken};
}
2020-09-05 22:50:45 +02:00
class CloudSync extends EventEmitter {
constructor() {
super();
this._cloudSync = new CloudSyncNative((state) => this.emit('state', JSON.parse(state)));
2020-09-05 22:50:45 +02:00
}
async sync(optionsOrOldId, oldConfig) {
ensureDispatch();
let options;
if (typeof optionsOrOldId === 'string') {
options = upgradeConfig(optionsOrOldId, oldConfig);
} else {
options = optionsOrOldId;
}
// TODO: remove deprecated config support
const config = dispatch.getLastState != null ? await generateConfig(options) : oldConfig;
const id = dispatch.getLastState != null ? `${options.applicationId}:${options.branchId}` : optionsOrOldId;
2020-09-05 22:50:45 +02:00
return new Promise((resolve, reject) =>
this._cloudSync.command(JSON.stringify({type: 'SYNC', id, config}), makeCallback(resolve, reject))
);
}
}
function cloudSyncConstructor() {
const instance = new CloudSync();
return {
on: instance.on.bind(instance),
sync: instance.sync.bind(instance),
};
}
module.exports = cloudSyncConstructor;