123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378
import { Ddc } from "./ddc.ts";import { ContextBuilder } from "./context.ts";import {  Context,  DdcEvent,  DdcExtType,  DdcOptions,  DdcUserData,} from "./types.ts";import {  batch,  Denops,  ensureNumber,  ensureObject,  ensureString,  fn,  op,  vars,} from "./deps.ts";import { createCallbackContext } from "./callback.ts";
type RegisterArg = {  path: string;  name: string;  type: DdcExtType;};
export async function main(denops: Denops) {  const ddc: Ddc = new Ddc();  const contextBuilder = new ContextBuilder();  const cbContext = createCallbackContext();
  denops.dispatcher = {    async register(arg1: unknown): Promise<void> {      const arg = ensureObject(arg1) as RegisterArg;      if (arg.type == "source") {        await ddc.registerSource(denops, arg.path, arg.name);      } else if (arg.type == "filter") {        await ddc.registerFilter(denops, arg.path, arg.name);      }    },    alias(arg1: unknown, arg2: unknown, arg3: unknown): Promise<void> {      ddc.registerAlias(        ensureString(arg1) as DdcExtType,        ensureString(arg2),        ensureString(arg3),      );      return Promise.resolve();    },    setGlobal(arg1: unknown): Promise<void> {      const options = ensureObject(arg1);      contextBuilder.setGlobal(options);      return Promise.resolve();    },    setFiletype(arg1: unknown, arg2: unknown): Promise<void> {      const options = ensureObject(arg1);      const filetype = ensureString(arg2);      contextBuilder.setFiletype(filetype, options);      return Promise.resolve();    },    setContext(arg1: unknown, arg2: unknown): Promise<void> {      const filetype = ensureString(arg1);      const id = ensureString(arg2);      contextBuilder.setContext(filetype, id);      return Promise.resolve();    },    setBuffer(arg1: unknown, arg2: unknown): Promise<void> {      const options = ensureObject(arg1);      const bufnr = ensureNumber(arg2);      contextBuilder.setBuffer(bufnr, options);      return Promise.resolve();    },    patchGlobal(arg1: unknown): Promise<void> {      const options = ensureObject(arg1);      contextBuilder.patchGlobal(options);      return Promise.resolve();    },    patchFiletype(arg1: unknown, arg2: unknown): Promise<void> {      const options = ensureObject(arg1);      const filetype = ensureString(arg2);      contextBuilder.patchFiletype(filetype, options);      return Promise.resolve();    },    patchBuffer(arg1: unknown, arg2: unknown): Promise<void> {      const options = ensureObject(arg1);      const bufnr = ensureNumber(arg2);      contextBuilder.patchBuffer(bufnr, options);      return Promise.resolve();    },    getGlobal(): Promise<Partial<DdcOptions>> {      return Promise.resolve(contextBuilder.getGlobal());    },    getFiletype(): Promise<Record<string, Partial<DdcOptions>>> {      return Promise.resolve(contextBuilder.getFiletype());    },    getContext(): Promise<Record<string, string>> {      return Promise.resolve(contextBuilder.getContext());    },    getBuffer(): Promise<Record<number, Partial<DdcOptions>>> {      return Promise.resolve(contextBuilder.getBuffer());    },    getCurrent(): Promise<DdcOptions> {      return Promise.resolve(contextBuilder.getCurrent(denops));    },    async manualComplete(arg1: unknown): Promise<void> {      const sources = arg1 as string[];
      const [skip, context, options] = await contextBuilder        .createContext(denops, "Manual");      if (skip) return;
      const mode = await fn.mode(denops);      if (mode == "c") {        // Use cmdlineSources instead        options.sources = options.cmdlineSources;      }      if (sources.length != 0) {        options.sources = sources;      }
      // Call onEvent() is needed to load sources      await ddc.onEvent(        denops,        context,        cbContext.createOnCallback(),        options,      );
      cbContext.revoke();      await doCompletion(denops, context, options);    },    async onEvent(arg1: unknown): Promise<void> {      const event = ensureString(arg1) as DdcEvent;      const [skip, context, options] = await contextBuilder        .createContext(denops, event);
      const mode = await fn.mode(denops);      if (mode == "c") {        // Use cmdlineSources instead        options.sources = options.cmdlineSources;      }
      await ddc.onEvent(        denops,        context,        cbContext.createOnCallback(),        options,      );      if (skip) return;
      cbContext.revoke();
      if (event != "InsertEnter" && await fn.mode(denops) == "n") {        return;      }
      if (options.completionMenu == "native") {        // Check for CompleteDone        const skipComplete = await vars.g.get(          denops,          "ddc#_skip_complete",        ) as boolean;        if (skipComplete) {          return;        }      } else if (options.completionMenu == "pum.vim") {        // Check for pum.vim        const skipComplete = await denops.call("pum#skip_complete") as boolean;        if (skipComplete) {          // Note: pum#skip_complete() does not close the popupmenu.          return;        }      }
      if (await checkSkipCompletion(event, context, options)) {        await cancelCompletion(denops, context, options);
        return;      }
      // Check auto complete delay.      if (options.autoCompleteDelay > 0) {        // Cancel previous completion        await cancelCompletion(denops, context, options);
        await new Promise((resolve) =>          setTimeout(            resolve,            options.autoCompleteDelay,          )        );
        const changedTick = vars.b.get(denops, "changedtick") as Promise<          number        >;        if (context.changedTick != await changedTick) {          // Input is changed.  Skip invalid completion.          return;        }      }
      await doCompletion(denops, context, options);    },    // deno-lint-ignore require-await    async onCallback(id: unknown, payload: unknown): Promise<void> {      cbContext.emit(ensureString(id), payload);    },    async onCompleteDone(arg1: unknown, arg2: unknown): Promise<void> {      const sourceName = ensureString(arg1);      const userData = ensureObject(arg2) as DdcUserData;      const [skip, context, options] = await contextBuilder        .createContext(denops, "CompleteDone");      if (skip) return;
      cbContext.revoke();      await ddc.onCompleteDone(        denops,        context,        cbContext.createOnCallback(),        options,        sourceName,        userData,      );    },  };
  async function checkSkipCompletion(    event: DdcEvent,    context: Context,    options: DdcOptions,  ): Promise<boolean> {    // Note: Don't complete when backspace by default, because of completion flicker.    const prevInput = await vars.g.get(denops, "ddc#_prev_input") as string;    const checkBackSpace = (!options.backspaceCompletion &&      context.input != prevInput &&      context.input.length + 1 == prevInput.length &&      prevInput.startsWith(context.input));    if (checkBackSpace && options.completionMode == "popupmenu") {      await vars.g.set(denops, "ddc#_prev_input", context.input);      return true;    }
    // Skip special buffers.    const buftype = await op.buftype.getLocal(denops);    const mode = await fn.mode(denops);    if (      buftype != "" && !options.specialBufferCompletion && mode != "c"    ) {      return true;    }
    // Check indentkeys.    // Note: re-indentation does not work for native popupmenu    const indentkeys = (await op.indentkeys.getLocal(denops)).split(",");    if (      options.completionMenu == "native" &&      mode == "i" &&      indentkeys.filter((pattern) => pattern == "!^F").length > 0    ) {      for (        const found of indentkeys.map((p) => p.match(/^0?=~?(.+)$/))      ) {        if (!found) {          continue;        }
        if (context.input.endsWith(found[1])) {          // Skip completion and reindent if matched.          await denops.call("ddc#util#indent_current_line");          return true;        }      }    }
    if (options.autoCompleteEvents.indexOf(event) < 0) {      return true;    }
    return false;  }
  async function cancelCompletion(    denops: Denops,    context: Context,    options: DdcOptions,  ): Promise<void> {    const visible = await denops.call("ddc#map#pum_visible");
    await batch(denops, async (denops: Denops) => {      await vars.g.set(denops, "ddc#_event", context.event);      await vars.g.set(denops, "ddc#_complete_pos", -1);      await vars.g.set(denops, "ddc#_items", []);      if (visible && options.completionMode != "manual") {        // Close current popupmenu.        await denops.call("ddc#complete#_clear");      }    });  }
  async function doCompletion(    denops: Denops,    context: Context,    options: DdcOptions,  ): Promise<void> {    const [completePos, items] = await ddc.gatherResults(      denops,      context,      cbContext.createOnCallback(),      options,    );
    await (async function write() {      const visible = await denops.call("ddc#map#pum_visible");
      await batch(denops, async (denops: Denops) => {        await vars.g.set(denops, "ddc#_event", context.event);        await vars.g.set(denops, "ddc#_prev_input", context.input);        await vars.g.set(denops, "ddc#_complete_pos", completePos);        await vars.g.set(denops, "ddc#_items", items);        await vars.g.set(denops, "ddc#_sources", options.sources);        await vars.g.set(          denops,          "ddc#_overwrite_completeopt",          options.overwriteCompleteopt,        );        await vars.g.set(          denops,          "ddc#_completion_menu",          options.completionMenu,        );        await vars.g.set(denops, "ddc#_changedtick", context.changedTick);
        if (          options.completionMode == "popupmenu" ||          context.event == "Manual" ||          visible        ) {          await denops.call("ddc#complete");        } else if (options.completionMode == "inline") {          await denops.call("ddc#complete#_inline", options.inlineHighlight);        } else if (options.completionMode == "manual") {          // through        }      });    })();  }
  await batch(denops, async (denops: Denops) => {    await vars.g.set(denops, "ddc#_items", []);    await vars.g.set(denops, "ddc#_changedtick", 0);    await vars.g.set(denops, "ddc#_complete_pos", -1);    await vars.g.set(denops, "ddc#_completion_menu", "native");    await vars.g.set(denops, "ddc#_event", "Manual");    await vars.g.set(denops, "ddc#_inline_popup_id", -1);    await vars.g.set(denops, "ddc#_now", 0);    await vars.g.set(denops, "ddc#_overwrite_completeopt", false);    await vars.g.set(denops, "ddc#_popup_id", -1);    await vars.g.set(denops, "ddc#_prev_input", "");    await vars.g.set(denops, "ddc#_skip_complete", false);    await vars.g.set(denops, "ddc#_sources", []);
    await denops.cmd("doautocmd <nomodeline> User DDCReady");    await denops.cmd("autocmd! User DDCReady");    await denops.call("ddc#_on_event", "Initialize");
    ddc.registerAutocmd(denops, [      "InsertEnter",      "InsertLeave",      "TextChangedI",      "TextChangedP",    ]);    await denops.cmd(      "autocmd ddc CmdlineChanged * " +        "if getcmdtype() !=# ':' | call ddc#_on_event('CmdlineChanged') | endif",    );  });}