import { ItemView, WorkspaceLeaf } from 'obsidian';
import { mount, unmount } from 'svelte';
import ChatWindow from 'Components/ChatWindow.svelte';
import TopBar from 'Components/TopBar.svelte';
import type { StatusBarService } from 'Services/StatusBarService';
import { Resolve } from 'Services/DependencyService';
import { Services } from 'Services/Services';

export const VIEW_TYPE_MAIN = 'ai-agent-main-view';

export class MainView extends ItemView {

  private statusBarService: StatusBarService = Resolve<StatusBarService>(Services.StatusBarService);

  constructor(leaf: WorkspaceLeaf) {
    super(leaf);
  }

  topBar: ReturnType<typeof TopBar> | undefined;
  input: ReturnType<typeof ChatWindow> | undefined;

  getViewType() {
    return VIEW_TYPE_MAIN;
  }

  getDisplayText() {
    return "AI Agent";
  }

  getIcon(): string {
    return 'sparkles';
  }

  async onOpen() {
    const container = this.contentEl;
    container.empty();

    // Mount TopBar with reference to ChatWindow's focus function
    this.topBar = mount(TopBar, {
      target: container,
      props: {
        leaf: this.leaf,
        onNewConversation: () => {
          this.input?.resetChatArea();
          this.input?.focusInput();
        }
      }
    });

    // Mount ChatWindow first
    this.input = mount(ChatWindow, {
      target: container,
      props: {}
    });
  }

  async onClose() {
    if (this.topBar) {
      unmount(this.topBar);
    }
    if (this.input) {
      unmount(this.input);
    }
    this.statusBarService.removeStatusBarMessage();
  }
}