extension.ts

import ms from "ms";
import { StatusBarAlignment, window } from "vscode";
import type { ExtensionContext } from "vscode";

import configureAws from "./plugins/configure-aws.ts";
import logs from "./plugins/logs.ts";
import manage from "./plugins/manage.ts";
import setup from "./plugins/setup.ts";
import statusBar from "./plugins/status-bar.ts";
import { PluginManager } from "./plugins.ts";
import { createCliStatusTracker } from "./utils/cli.ts";
import { createLocalStackContainerStatusTracker } from "./utils/localstack-container.ts";
import {
	createHealthStatusTracker,
	createLocalStackInstanceStatusTracker,
} from "./utils/localstack-instance.ts";
import { getOrCreateExtensionSessionId } from "./utils/manage.ts";
import { createSetupStatusTracker } from "./utils/setup.ts";
import { createTelemetry } from "./utils/telemetry.ts";
import { createTimeTracker } from "./utils/time-tracker.ts";

const plugins = new PluginManager([
	setup,
	configureAws,
	manage,
	statusBar,
	logs,
]);

export async function activate(context: ExtensionContext) {
	const outputChannel = window.createOutputChannel("LocalStack", {
		log: true,
	});
	context.subscriptions.push(outputChannel);

	const cliStatusTracker = createCliStatusTracker(outputChannel);
	context.subscriptions.push(cliStatusTracker);

	const timeTracker = createTimeTracker({ outputChannel });

	const {
		containerStatusTracker,
		localStackStatusTracker,
		setupStatusTracker,
		statusBarItem,
		telemetry,
	} = await timeTracker.run("extension.dependencies", async () => {
		const statusBarItem = window.createStatusBarItem(
			StatusBarAlignment.Left,
			-1,
		);
		context.subscriptions.push(statusBarItem);
		statusBarItem.text = "$(loading~spin) LocalStack";
		statusBarItem.show();

		const containerStatusTracker = createLocalStackContainerStatusTracker(
			"localstack-main",
			outputChannel,
			timeTracker,
		);
		context.subscriptions.push(containerStatusTracker);

		const healthCheckStatusTracker = createHealthStatusTracker(timeTracker);
		const localStackStatusTracker = createLocalStackInstanceStatusTracker(
			containerStatusTracker,
			healthCheckStatusTracker,
			outputChannel,
		);
		context.subscriptions.push(localStackStatusTracker);

		const setupStatusTracker = await timeTracker.run(
			"setup-status",
			async () => {
				return await createSetupStatusTracker(
					outputChannel,
					timeTracker,
					cliStatusTracker,
				);
			},
		);
		context.subscriptions.push(setupStatusTracker);

		const telemetry = await timeTracker.run("telemetry", async () => {
			const sessionId = await getOrCreateExtensionSessionId(context);
			return createTelemetry(outputChannel, sessionId);
		});

		return {
			statusBarItem,
			containerStatusTracker,
			localStackStatusTracker,
			setupStatusTracker,
			telemetry,
		};
	});

	await timeTracker.run("extension.activatePlugins", async () => {
		await plugins.activate({
			context,
			outputChannel,
			statusBarItem,
			cliStatusTracker,
			containerStatusTracker,
			localStackStatusTracker,
			setupStatusTracker,
			telemetry,
			timeTracker,
		});
	});
}

export async function deactivate() {
	await plugins.deactivate();
}