andy-stack_vaultkeeper-ai/main.ts

import { WorkspaceLeaf, Plugin } from 'obsidian';
import { AIProviderModel } from './Enums/ApiProvider';
import { MainView, VIEW_TYPE_MAIN } from 'Views/MainView';
import { RegisterAiProvider, RegisterDependencies } from 'Services/ServiceRegistration';
import { AIAgentSettingTab } from 'AIAgentSettingTab';
import { Services } from 'Services/Services';
import type { StatusBarService } from 'Services/StatusBarService';
import { DeregisterAllServices, Resolve } from 'Services/DependencyService';
import type { VaultService } from 'Services/VaultService';
import { Path } from 'Enums/Path';
import { Copy } from 'Enums/Copy';

interface IAIAgentSettings {
	firstTimeStart: boolean;

	model: string;
	apiKey: string;
	exclusions: string[];

	userInstruction: string;
}

const DEFAULT_SETTINGS: IAIAgentSettings = {
	firstTimeStart: true,

	model: AIProviderModel.ClaudeSonnet_4_5,
	apiKey: "",
	exclusions: [],

	userInstruction: ""
}

export default class AIAgentPlugin extends Plugin {
	public settings: IAIAgentSettings;
	
	public async onload() {
		// KaTeX CSS is bundled with the plugin to comply with CSP
		require('katex/dist/katex.min.css');
		// Plugin styles
		require('./styles.css');

		await this.loadSettings();

		RegisterDependencies(this);

		this.registerView(
			VIEW_TYPE_MAIN,
			(leaf) => new MainView(leaf)
		);

		this.addCommand({
			id: 'ai-agent',
			name: 'AI Agent',
			callback: () => {
				this.activateView();
			}
		});

		this.addRibbonIcon('sparkles', 'AI Agent', (_: MouseEvent) => {
			this.activateView();
		});

		this.addSettingTab(new AIAgentSettingTab(this.app, this));

		this.app.workspace.onLayoutReady(async () => {
			await this.setup(this);
		});
	}

	public async onunload() {
		Resolve<StatusBarService>(Services.StatusBarService).removeStatusBarMessage();
		DeregisterAllServices();
	}

	public async activateView() {
		const { workspace } = this.app;

		let leaf: WorkspaceLeaf | null = null;
		const leaves = workspace.getLeavesOfType(VIEW_TYPE_MAIN);

		if (leaves.length > 0) {
			leaf = leaves[0];
		} else {
			leaf = workspace.getRightLeaf(false);
			await leaf?.setViewState({ type: VIEW_TYPE_MAIN, active: true });
		}

		if (leaf != null) {
			workspace.revealLeaf(leaf);
		}
	}

	public async loadSettings() {
		this.settings = Object.assign({}, DEFAULT_SETTINGS, await this.loadData());
	}

	async saveSettings() {
		await this.saveData(this.settings);
		RegisterAiProvider(this);
	}

	// create example user instruction (on first launch only)
	private async setup(plugin: AIAgentPlugin) {
		if (!plugin.settings.firstTimeStart) {
			return;
		}
		plugin.settings.firstTimeStart = false;
		await plugin.saveSettings();

		const vaultService: VaultService = Resolve<VaultService>(Services.VaultService);

		await vaultService.create(Path.ExampleUserInstructions, Copy.EXAMPLE_USER_INSTRUCTION, true);
	}
}