forked from fertigt/pycolonize
-
Notifications
You must be signed in to change notification settings - Fork 0
/
extension.js
80 lines (64 loc) · 2.46 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
const vscode = require('vscode')
function mustBeColonized(lineText) {
const functionKeywords = ['async def', 'def']
const pythonKeywords = ['class', 'elif', 'else', 'except', 'finally', 'for', 'if', 'try', 'while', 'with']
if (pythonKeywords.some(pythonKeyword => lineText.startsWith(pythonKeyword))) {
return 1
}
for (let functionKeyword of functionKeywords) {
if (lineText.startsWith(functionKeyword)) {
if (lineText.includes("(") && !lineText.includes(")"))
return 2
else if (!lineText.includes("(") && !lineText.includes(")"))
return 3
else {
return 1
}
}
}
return false
}
function colonize(option) {
var editor = vscode.window.activeTextEditor
if (!editor) return
vscode.commands.executeCommand('acceptSelectedSuggestion').then(() => {
var lineIndex = editor.selection.active.line
var lineObject = editor.document.lineAt(lineIndex)
var lineLength = lineObject.text.length
var lineWithoutWhitespaces = lineObject.text.substring(lineObject.firstNonWhitespaceCharacterIndex)
var colonize_option = mustBeColonized(lineWithoutWhitespaces)
if (colonize_option) {
if (lineObject.text.charAt(lineLength - 1) !== ':' && !lineObject.isEmptyOrWhitespace) {
var insertionSuccess = editor.edit((editBuilder) => {
if (colonize_option == 1) {
editBuilder.insert(new vscode.Position(lineIndex, lineLength), ':')
} else if (colonize_option == 2) {
editBuilder.insert(new vscode.Position(lineIndex, lineLength), '):')
} else if (colonize_option == 3) {
editBuilder.insert(new vscode.Position(lineIndex, lineLength), '():')
}
})
if (!insertionSuccess) return
}
}
if (option === 'hold') return
option === 'endline'
? vscode.commands.executeCommand('cursorEnd')
: vscode.commands.executeCommand('editor.action.insertLineAfter')
})
}
function activate(context) {
var endLineDisposable = vscode.commands.registerCommand('pycolonize.endline', () => {
colonize('endline')
})
var holdDisposable = vscode.commands.registerCommand('pycolonize.hold', () => {
colonize('hold')
})
var newLineDisposable = vscode.commands.registerCommand('pycolonize.newline', () => {
colonize('newline')
})
context.subscriptions.push(endLineDisposable)
context.subscriptions.push(newLineDisposable)
context.subscriptions.push(holdDisposable)
}
exports.activate = activate