import { EnrichedToken } from "../../../formulas/composer_tokenizer";
import { rangeTokenize } from "../../../formulas/range_tokenizer";
import { Token } from "../../../formulas/tokenizer";
import { canonicalizeContent, localizeContent } from "../../../helpers/locale";
import { setXcToFixedReferenceType } from "../../../helpers/references";
import { AutoCompleteProviderDefinition } from "../../../registries/auto_completes/auto_complete_registry";
import { Color, UID, UnboundedZone, Zone } from "../../../types/misc";
import { Get } from "../../../types/store_engine";
import { AbstractComposerStore } from "../composer/abstract_composer_store";

export interface StandaloneComposerArgs {
  onConfirm: (content: string) => void;
  content: string;
  /**
   * the sheet id to which unqualified references (A1 vs Sheet1!A1)
   * will be resolved.
   */
  defaultStatic?: boolean;
  defaultRangeSheetId: UID;
  contextualAutocomplete?: AutoCompleteProviderDefinition;
  getContextualColoredSymbolToken?: (token: Token) => Color;
}

export class StandaloneComposerStore extends AbstractComposerStore {
  constructor(get: Get, private args: () => StandaloneComposerArgs) {
    super(get);
    this._currentContent = this.getComposerContent().text;
  }

  protected getAutoCompleteProviders(): AutoCompleteProviderDefinition[] {
    const providersDefinitions = super.getAutoCompleteProviders();
    const contextualAutocomplete = this.args().contextualAutocomplete;
    if (contextualAutocomplete) {
      providersDefinitions.push(contextualAutocomplete);
    }
    return providersDefinitions;
  }

  /**
   * Replace the current reference selected by the new one.
   * */
  protected getZoneReference(zone: Zone | UnboundedZone): string {
    const res = super.getZoneReference(zone);
    if (this.args().defaultStatic) {
      return setXcToFixedReferenceType(res, "colrow");
    }
    return res;
  }

  protected getComposerContent() {
    // References in the content might not be linked to the current active sheet
    // We here force the sheet name prefix for all references that are not in
    // the current active sheet
    const defaultRangeSheetId = this.args().defaultRangeSheetId;
    const content = rangeTokenize(this.args().content)
      .map((token) => {
        if (token.type === "REFERENCE") {
          const range = this.getters.getRangeFromSheetXC(defaultRangeSheetId, token.value);
          return this.getters.getRangeString(range, this.getters.getActiveSheetId());
        }
        return token.value;
      })
      .join("");
    return { text: localizeContent(content, this.getters.getLocale()) };
  }

  stopEdition() {
    this._stopEdition();
  }

  protected confirmEdition(content: string) {
    this.args().onConfirm(content);
  }

  protected getTokenColor(token: EnrichedToken): string {
    if (token.type === "SYMBOL") {
      const matchedColor = this.args().getContextualColoredSymbolToken?.(token);
      if (matchedColor) {
        return matchedColor;
      }
    }
    return super.getTokenColor(token);
  }

  hoverToken() {
    /**
     * Some functions can only be evaluated in the context of the grid, which is not feasible
     * in the standalone composer.
     */
    return;
  }

  protected getCurrentCanonicalContent(): string {
    return canonicalizeContent(this._currentContent, this.getters.getLocale());
  }
}