127 lines
4.2 KiB
TypeScript
127 lines
4.2 KiB
TypeScript
import { commands, ExtensionContext, StatusBarAlignment, StatusBarItem, window, workspace, extensions } from 'vscode';
|
|
import RPCClient from './client/RPCClient';
|
|
import Logger from './structures/Logger';
|
|
import { GitExtension } from './git';
|
|
|
|
const sleep = (wait: number) => new Promise(resolve => setTimeout(resolve, wait));
|
|
let loginTimeout: NodeJS.Timer;
|
|
|
|
const statusBarIcon: StatusBarItem = window.createStatusBarItem(StatusBarAlignment.Left);
|
|
statusBarIcon.text = '$(pulse) Connecting to Discord...';
|
|
|
|
const config = workspace.getConfiguration('discord');
|
|
const rpc = new RPCClient(config.get<string>('clientID')!, statusBarIcon);
|
|
|
|
export async function activate(context: ExtensionContext) {
|
|
try {
|
|
const ext = extensions.getExtension<GitExtension>('vscode.git')!;
|
|
await ext.activate();
|
|
rpc.git = ext.exports.getAPI(1);
|
|
} catch {
|
|
// We loaded before the git extension, give it a bit to load
|
|
// In a perfect world this shouldn't happen
|
|
await sleep(2000);
|
|
const ext = extensions.getExtension<GitExtension>('vscode.git')!;
|
|
await ext.activate();
|
|
rpc.git = ext.exports.getAPI(1);
|
|
}
|
|
Logger.log('Discord Presence activated!');
|
|
|
|
let isWorkspaceExcluded = false;
|
|
const excludePatterns = config.get<string[]>('workspaceExcludePatterns');
|
|
if (excludePatterns?.length) {
|
|
for (const pattern of excludePatterns) {
|
|
const regex = new RegExp(pattern);
|
|
const folders = workspace.workspaceFolders;
|
|
if (!folders) break;
|
|
if (folders.some(folder => regex.test(folder.uri.fsPath))) {
|
|
isWorkspaceExcluded = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!isWorkspaceExcluded && config.get<boolean>('enabled')) {
|
|
statusBarIcon.show();
|
|
try {
|
|
await rpc.login();
|
|
} catch (error) {
|
|
Logger.log(`Encountered following error after trying to login:\n${error}`);
|
|
await rpc.dispose();
|
|
if (!config.get('silent')) {
|
|
if (error.message.includes('ENOENT')) window.showErrorMessage('No Discord Client detected!');
|
|
else window.showErrorMessage(`Couldn't connect to Discord via RPC: ${error.toString()}`);
|
|
}
|
|
rpc.statusBarIcon.text = '$(pulse) Reconnect to Discord';
|
|
rpc.statusBarIcon.command = 'discord.reconnect';
|
|
}
|
|
}
|
|
|
|
const enabler = commands.registerCommand('discord.enable', async () => {
|
|
await rpc.dispose();
|
|
config.update('enabled', true);
|
|
rpc.config = workspace.getConfiguration('discord');
|
|
rpc.statusBarIcon.text = '$(pulse) Connecting to Discord...';
|
|
rpc.statusBarIcon.show();
|
|
await rpc.login();
|
|
window.showInformationMessage('Enabled Discord Rich Presence for this workspace.');
|
|
});
|
|
|
|
const disabler = commands.registerCommand('discord.disable', async () => {
|
|
config.update('enabled', false);
|
|
rpc.config = workspace.getConfiguration('discord');
|
|
await rpc.dispose();
|
|
rpc.statusBarIcon.hide();
|
|
window.showInformationMessage('Disabled Discord Rich Presence for this workspace.');
|
|
});
|
|
|
|
const reconnecter = commands.registerCommand('discord.reconnect', async () => {
|
|
if (loginTimeout) clearTimeout(loginTimeout);
|
|
await rpc.dispose();
|
|
loginTimeout = setTimeout(async () => {
|
|
await rpc.login();
|
|
if (!config.get('silent')) window.showInformationMessage('Reconnecting to Discord RPC...');
|
|
rpc.statusBarIcon.text = '$(pulse) Reconnecting to Discord...';
|
|
rpc.statusBarIcon.command = 'discord.reconnect';
|
|
}, 1000);
|
|
});
|
|
|
|
const disconnect = commands.registerCommand('discord.disconnect', async () => {
|
|
await rpc.dispose();
|
|
rpc.statusBarIcon.text = '$(pulse) Reconnect to Discord';
|
|
rpc.statusBarIcon.command = 'discord.reconnect';
|
|
});
|
|
|
|
const allowSpectate = commands.registerCommand('discord.allowSpectate', async () => {
|
|
await rpc.allowSpectate();
|
|
});
|
|
|
|
const disableSpectate = commands.registerCommand('discord.disableSpectate', async () => {
|
|
await rpc.disableSpectate();
|
|
});
|
|
|
|
const allowJoinRequests = commands.registerCommand('discord.allowJoinRequests', async () => {
|
|
await rpc.allowJoinRequests();
|
|
});
|
|
|
|
const disableJoinRequests = commands.registerCommand('discord.disableJoinRequests', async () => {
|
|
await rpc.disableJoinRequests();
|
|
});
|
|
|
|
context.subscriptions.push(
|
|
enabler,
|
|
disabler,
|
|
reconnecter,
|
|
disconnect,
|
|
allowSpectate,
|
|
disableSpectate,
|
|
allowJoinRequests,
|
|
disableJoinRequests,
|
|
);
|
|
}
|
|
|
|
export async function deactivate() {
|
|
await rpc.dispose();
|
|
}
|
|
|
|
process.on('unhandledRejection', err => Logger.log(err as string));
|