import type * as monaco from "monaco-editor/esm/vs/editor/editor.api";
import { Document as DocumentInfo } from "./api/proto/exa/language_server_pb/language_server_pb";
import { PromiseClient } from "@connectrpc/connect";
import { Status } from "./Status";
import { MonacoCompletionProvider } from "./CompletionProvider";
import { LanguageServerService } from "./api/proto/exa/language_server_pb/language_server_connect";

declare module "monaco-editor" {
  namespace editor {
    interface ICodeEditor {
      _commandService: { executeCommand(command: string): unknown };
    }
  }
}

export class InlineCompletionProvider
  implements monaco.languages.InlineCompletionsProvider
{
  private numCompletionsProvided: number;
  readonly completionProvider: MonacoCompletionProvider;

  constructor(
    grpcClient: PromiseClient<typeof LanguageServerService>,
    readonly setCompletionCount: (count: number) => void,
    setCodeiumStatus: (status: Status) => void,
    setCodeiumStatusMessage: (message: string) => void,
    readonly getEditorDocuments: (
      currentEditorModel: monaco.editor.ITextModel
    ) => DocumentInfo[],
    apiKey?: string | undefined
  ) {
    this.numCompletionsProvided = 0;
    this.completionProvider = new MonacoCompletionProvider(
      grpcClient,
      setCodeiumStatus,
      setCodeiumStatusMessage,
      apiKey
    );
  }

  freeInlineCompletions() {
    // nothing
  }

  async provideInlineCompletions(
    model: monaco.editor.ITextModel,
    position: monaco.Position,
    _context: monaco.languages.InlineCompletionContext,
    token: monaco.CancellationToken
  ) {
    this.updateOtherDocuments(this.getEditorDocuments(model));
    const completions = await this.completionProvider.provideInlineCompletions(
      model,
      position,
      token
    );
    // Only count completions provided if non-empty (i.e. exclude cancelled
    // requests).
    // TODO(nick): don't count cached results either.
    // TODO(nick): better distinguish warning and error states.
    if (completions) {
      this.numCompletionsProvided += 1;
      this.setCompletionCount(this.numCompletionsProvided);
    }
    return completions;
  }

  public acceptedLastCompletion(completionId: string) {
    this.completionProvider.acceptedLastCompletion(completionId);
  }

  public updateOtherDocuments(otherDocuments: DocumentInfo[]) {
    this.completionProvider.otherDocuments = otherDocuments;
  }
}