65 lines
2.3 KiB
TypeScript
65 lines
2.3 KiB
TypeScript
import { ExtensionContext, workspace, env, version } from "vscode";
|
|
import { TabbyAgent, PartialAgentConfig, DataStore } from "tabby-agent";
|
|
|
|
function getWorkspaceConfiguration(): PartialAgentConfig {
|
|
const configuration = workspace.getConfiguration("tabby");
|
|
const config: PartialAgentConfig = {};
|
|
const endpoint = configuration.get<string>("api.endpoint");
|
|
if (endpoint && endpoint.trim().length > 0) {
|
|
config.server = {
|
|
endpoint,
|
|
};
|
|
}
|
|
const anonymousUsageTrackingDisabled = configuration.get<boolean>("usage.anonymousUsageTracking", false);
|
|
config.anonymousUsageTracking = {
|
|
disable: anonymousUsageTrackingDisabled,
|
|
};
|
|
return config;
|
|
}
|
|
|
|
var instance: TabbyAgent;
|
|
|
|
export function agent(): TabbyAgent {
|
|
if (!instance) {
|
|
throw new Error("Tabby Agent not initialized");
|
|
}
|
|
return instance;
|
|
}
|
|
|
|
export async function createAgentInstance(context: ExtensionContext): Promise<TabbyAgent> {
|
|
if (!instance) {
|
|
const extensionDataStore: DataStore = {
|
|
data: {},
|
|
load: async function () {
|
|
this.data = context.globalState.get("data", {});
|
|
},
|
|
save: async function () {
|
|
context.globalState.update("data", this.data);
|
|
},
|
|
};
|
|
const agent = await TabbyAgent.create({ dataStore: env.appHost === "desktop" ? undefined : extensionDataStore });
|
|
const initPromise = agent.initialize({
|
|
config: getWorkspaceConfiguration(),
|
|
client: `${env.appName} ${env.appHost} ${version}, ${context.extension.id} ${context.extension.packageJSON.version}`,
|
|
});
|
|
workspace.onDidChangeConfiguration(async (event) => {
|
|
await initPromise;
|
|
const configuration = workspace.getConfiguration("tabby");
|
|
if (event.affectsConfiguration("tabby.api.endpoint")) {
|
|
const endpoint = configuration.get<string>("api.endpoint");
|
|
if (endpoint && endpoint.trim().length > 0) {
|
|
agent.updateConfig("server.endpoint", endpoint);
|
|
} else {
|
|
agent.clearConfig("server.endpoint");
|
|
}
|
|
}
|
|
if (event.affectsConfiguration("tabby.usage.anonymousUsageTracking")) {
|
|
const anonymousUsageTrackingDisabled = configuration.get<boolean>("usage.anonymousUsageTracking", false);
|
|
agent.updateConfig("anonymousUsageTracking.disable", anonymousUsageTrackingDisabled);
|
|
}
|
|
});
|
|
instance = agent;
|
|
}
|
|
return instance;
|
|
}
|