Class IndividualHatMap

src/core/IndividualHatMap.ts:11–82

}

export class IndividualHatMap implements ReadOnlyHatMap {
  private isExpired: boolean = false;
  private documentTokenLists: Map<string, Token[]> = new Map();
  private deregisterFunctions: (() => void)[] = [];

  private map: {
    [decoratedCharacter: string]: Token;
  } = {};

  constructor(private graph: Graph) {}

  private getDocumentTokenList(document: TextDocument) {
    const key = document.uri.toString();
    let currentValue = this.documentTokenLists.get(key);

    if (currentValue == null) {
      currentValue = [];
      this.documentTokenLists.set(key, currentValue);
      this.deregisterFunctions.push(
        this.graph.rangeUpdater.registerRangeInfoList(document, currentValue)
      );
    }

    return currentValue;
  }

  clone() {
    const ret = new IndividualHatMap(this.graph);

    this.getEntries().forEach(([key, token]) => {
      ret.addTokenByKey(key, { ...token });
    });

    return ret;
  }

  getEntries() {
    this.checkExpired();
    return Object.entries(this.map);
  }

  private addTokenByKey(key: string, token: Token) {
    this.map[key] = token;
    this.getDocumentTokenList(token.editor.document).push(token);
  }

  addToken(hatStyle: HatStyleName, character: string, token: Token) {
    this.addTokenByKey(HatTokenMap.getKey(hatStyle, character), token);
  }

  getToken(hatStyle: HatStyleName, character: string) {
    this.checkExpired();
    return this.map[HatTokenMap.getKey(hatStyle, character)];
  }

  clear() {
    this.map = {};
    this.documentTokenLists = new Map();