2017-11-24 00:22:23 +00:00
|
|
|
// Import the required functions & object types from various packages.
|
2017-11-23 13:06:21 +00:00
|
|
|
import { Client } from 'discord-rpc';
|
|
|
|
import { basename, extname } from 'path';
|
2017-11-23 15:21:46 +00:00
|
|
|
import { ExtensionContext, commands, window, workspace, Uri, TextDocumentChangeEvent, TextDocument } from 'vscode';
|
2017-11-23 23:53:16 +00:00
|
|
|
import { setInterval, clearInterval } from 'timers';
|
2017-11-23 13:06:21 +00:00
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// Define the RPC variable and its type.
|
2017-11-23 21:14:16 +00:00
|
|
|
let rpc: Client;
|
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// `Activate` is fired when the extension is enabled. This SHOULD only fire once.
|
2017-11-23 13:06:21 +00:00
|
|
|
export function activate(context: ExtensionContext) {
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Get the workspace's configuration for "discord".
|
2017-11-23 15:21:46 +00:00
|
|
|
const config = workspace.getConfiguration('discord');
|
2017-11-23 13:06:21 +00:00
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// Obtain whether or not the extension is activated.
|
2017-11-23 23:53:16 +00:00
|
|
|
if (config.get('enabled')) {
|
|
|
|
initRPC(config.get('clientID'));
|
|
|
|
}
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Register the `discord.enable` command, and set the `enabled` config option to true.
|
2017-11-23 23:53:16 +00:00
|
|
|
const enabler = commands.registerCommand('discord.enable', () => {
|
|
|
|
config.update('enabled', true);
|
|
|
|
initRPC(config.get('clientID'));
|
2017-11-23 13:06:21 +00:00
|
|
|
});
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Register the `discord.disable` command, and set the `enabled` config option to false.
|
2017-11-23 23:53:16 +00:00
|
|
|
const disabler = commands.registerCommand('discord.disable', () => {
|
|
|
|
config.update('enabled', false);
|
|
|
|
rpc.setActivity({});
|
|
|
|
});
|
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// Push the new commands into the subscriptions.
|
2017-11-23 23:53:16 +00:00
|
|
|
context.subscriptions.push(enabler, disabler);
|
2017-11-23 13:06:21 +00:00
|
|
|
}
|
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// `Deactivate` is fired whenever the extension is deactivated.
|
2017-11-23 21:14:16 +00:00
|
|
|
export function deactivate(context: ExtensionContext) {
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// If there's an RPC Client initalized, destroy it.
|
2017-11-23 21:14:16 +00:00
|
|
|
if (rpc) rpc.destroy();
|
|
|
|
}
|
2017-11-23 13:06:21 +00:00
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
// This function updates the activity (The Client's Rich Presence status).
|
2017-11-23 21:14:16 +00:00
|
|
|
function setActivity(): void {
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Do not continue if RPC isn't initalized.
|
2017-11-23 13:06:21 +00:00
|
|
|
if (!rpc) return;
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Create a JSON Object with the user's activity information.
|
2017-11-23 13:06:21 +00:00
|
|
|
const activity = {
|
2017-11-23 21:17:11 +00:00
|
|
|
details: window.activeTextEditor
|
|
|
|
? `Editing ${basename(window.activeTextEditor.document.fileName)}`
|
|
|
|
: 'Idle.',
|
|
|
|
state: window.activeTextEditor
|
|
|
|
? `Workspace: ${workspace.getWorkspaceFolder(window.activeTextEditor.document.uri).name}`
|
|
|
|
: 'Idling.',
|
2017-11-23 13:47:42 +00:00
|
|
|
startTimestamp: new Date().getTime() / 1000,
|
2017-11-23 21:17:11 +00:00
|
|
|
largeImageKey: window.activeTextEditor
|
|
|
|
? extname(basename(window.activeTextEditor.document.fileName)).substring(1)
|
|
|
|
|| basename(window.activeTextEditor.document.fileName).substring(1)
|
|
|
|
|| 'file'
|
|
|
|
: 'vscode-big',
|
|
|
|
largeImageText: window.activeTextEditor
|
|
|
|
? window.activeTextEditor.document.languageId
|
|
|
|
: 'Idling',
|
2017-11-23 13:06:21 +00:00
|
|
|
smallImageKey: 'vscode',
|
2017-11-23 13:47:42 +00:00
|
|
|
smallImageText: 'Visual Studio Code',
|
2017-11-23 13:06:21 +00:00
|
|
|
instance: false
|
|
|
|
};
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Update the user's activity to the `activity` variable.
|
2017-11-23 19:43:10 +00:00
|
|
|
rpc.setActivity(activity);
|
2017-11-23 13:06:21 +00:00
|
|
|
}
|
2017-11-23 23:53:16 +00:00
|
|
|
|
2017-11-24 00:22:23 +00:00
|
|
|
/// Initalize the RPC systems.
|
2017-11-23 23:53:16 +00:00
|
|
|
function initRPC(clientID: string): void {
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Update the RPC variable with a new RPC Client.
|
2017-11-23 23:53:16 +00:00
|
|
|
rpc = new Client({ transport: 'ipc' });
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Once the RPC Client is ready, set the activity.
|
2017-11-23 23:53:16 +00:00
|
|
|
rpc.once('ready', () => {
|
|
|
|
setActivity();
|
|
|
|
workspace.onDidChangeTextDocument((e: TextDocumentChangeEvent) => setActivity());
|
|
|
|
});
|
2017-11-24 00:22:23 +00:00
|
|
|
|
|
|
|
// Log in to the RPC Client, and check whether or not it errors.
|
2017-11-23 23:53:16 +00:00
|
|
|
rpc.login(clientID).catch(error =>
|
|
|
|
error.message.includes('ENOENT')
|
|
|
|
? window.showErrorMessage('No Discord Client detected!')
|
|
|
|
: window.showErrorMessage(`Could not connect to discord via rpc: ${error.message}`)
|
|
|
|
);
|
|
|
|
}
|