-
Notifications
You must be signed in to change notification settings - Fork 0
/
extension.js
98 lines (82 loc) · 2.54 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
// Copyright 2023 EPAM Systems
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
const vscode = require("vscode");
const { workspaceService, alitaService } = require("./services");
const { COMMAND, EXTERNAL_PROMPTS_PROVIDERS } = require("./constants/index");
const {
addContext,
addExample,
createPrompt,
predict,
addGoodPrediction,
initAlita,
syncPrompts,
onConfigChange
} = require("./commands");
async function activate(context) {
await vscode.commands.executeCommand("setContext", "alitacode.ExtentablePlatforms", EXTERNAL_PROMPTS_PROVIDERS);
try {
await onConfigChange();
} catch (error) {
console.error(error);
}
vscode.workspace.onDidChangeConfiguration(async (e) => {
await onConfigChange();
});
const predictSub = vscode.commands.registerCommand(
COMMAND.PREDICT,
predict.bind(null)
);
const initAlitaSub = vscode.commands.registerCommand(
COMMAND.INIT_ALITA,
initAlita
);
const syncPromptsSub = vscode.commands.registerCommand(
COMMAND.SYNC_PROMPTS,
syncPrompts
);
const createPromptSub = vscode.commands.registerCommand(
COMMAND.CREATE_PROMPT,
createPrompt.bind(null, workspaceService.promptsList)
);
const addContextSub = vscode.commands.registerCommand(
COMMAND.ADD_CONTEXT,
addContext.bind(null, workspaceService.promptsList)
);
const addExampleSub = vscode.commands.registerCommand(
COMMAND.ADD_EXAMPLE,
addExample.bind(null, workspaceService.promptsList)
);
const addGoodPredictionSub = vscode.commands.registerCommand(
COMMAND.ADD_GOOD_PREDICTION,
addGoodPrediction
);
context.subscriptions.push(predictSub);
context.subscriptions.push(createPromptSub);
context.subscriptions.push(addContextSub);
context.subscriptions.push(addExampleSub);
context.subscriptions.push(addGoodPredictionSub);
context.subscriptions.push(initAlitaSub);
context.subscriptions.push(syncPromptsSub);
const api = {
alitaService,
workspaceService
}
return api
}
function deactivate() {}
module.exports = {
activate,
deactivate,
};