-
Notifications
You must be signed in to change notification settings - Fork 0
/
extension.js
119 lines (103 loc) · 3.11 KB
/
extension.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
const vscode = require('vscode');
const shell = require('shelljs');
/**
* @param {vscode.ExtensionContext} context
*/
function activate(context) {
console.log('Congratulations, your extension "s5" is now active!');
context.subscriptions.push(vscode.commands.registerCommand('s5.cipher', () => s5('cipher')));
context.subscriptions.push(vscode.commands.registerCommand('s5.decipher', () => s5('decipher')));
context.subscriptions.push(vscode.commands.registerCommand('s5.render', () => s5('render')));
}
exports.activate = activate;
function deactivate() {}
module.exports = {
activate,
deactivate
}
/**
* @param {string} command
*/
function s5(command) {
if (!shell.which('s5')) {
vscode.window.showErrorMessage('s5 command not found in $PATH');
return;
}
let config = vscode.workspace.getConfiguration('s5');
let provider = config.get('provider');
let cmd = 's5 ' + command + ' ' + provider;
switch(provider) {
case 'aes':
cmd += ' --key=' + config.get('aes.key');
break;
case 'aws':
if ( config.get('aws.accessKeyID').length > 0 ) {
shell.env["AWS_ACCESS_KEY_ID"] = config.get('aws.accessKeyID');
}
if ( config.get('aws.secretAccessKey').length > 0 ) {
shell.env["AWS_SECRET_ACCESS_KEY"] = config.get('aws.secretAccessKey');
}
if ( config.get('aws.sessionToken').length > 0 ) {
shell.env["AWS_SESSION_TOKEN"] = config.get('aws.sessionToken');
}
cmd += ' --kms-key-arn=' + config.get('aws.kmsKeyARN');
break;
case 'gcp':
cmd += ' --kms-key-name=' + config.get('gcp.kmsKeyName');
break;
case 'pgp':
shell.env["S5_PGP_PUBLIC_KEY_PATH"] = config.get('pgp.publicKeyPath');
shell.env["S5_PGP_PRIVATE_KEY_PATH"] = config.get('pgp.privateKeyPath');
break;
case 'vault':
if ( config.get('vault.address').length > 0 ) {
shell.env["VAULT_ADDRESS"] = config.get('vault.address');
}
if ( config.get('vault.token').length > 0 ) {
shell.env["VAULT_TOKEN"] = config.get('vault.token');
}
cmd += ' --transit-key=' + config.get('vault.transitKey');
break;
}
let editor = vscode.window.activeTextEditor;
if (!editor) {
vscode.window.showErrorMessage('You need to select a document/text to cipher/decipher');
return;
}
let document = editor.document;
switch(command) {
case 'cipher': case 'decipher':
let selection = editor.selection;
cmd += ' <<_EOL_\n' + document.getText(selection) + '\n_EOL_';
shell.exec(
cmd,
function(code, stdout, stderr) {
if (code !== 0) {
vscode.window.showErrorMessage('s5 error : ' + cmd + ' - ' + stdout + ' - ' + stderr);
return;
} else {
editor.edit(editBuilder => {
editBuilder.replace(selection, stdout);
});
}
}
);
break;
case 'render':
cmd += ' --in-place ' + document.uri.fsPath;
vscode.window.showInformationMessage(cmd);
shell.exec(
cmd,
function(code, stdout, stderr) {
if (code !== 0) {
vscode.window.showErrorMessage('s5 error : ' + cmd + ' - ' + stdout + ' - ' + stderr);
return;
}
}
);
break;
default:
vscode.window.showErrorMessage('undefined command : ' + command);
break;
}
}