import { Decoration } from "@tiptap/pm/view";
import { Annotation } from "../../contracts";

export class AnnotationDecoration<K> implements Annotation<K> {
  private decoration!: any;

  constructor(decoration: Decoration) {
    this.decoration = decoration;
  }

  get displayName(): string {
    return this.decoration.type.spec.data.displayName;
  }
  get tag(): string {
    return this.decoration.type.spec.data.tag;
  }

  get id() {
    return this.decoration.type.spec.id;
  }

  get from() {
    return this.decoration.from;
  }

  get to() {
    return this.decoration.to;
  }

  get selectedText() {
    return this.decoration.type.spec.data.selectedText;
  }

  get data(): K {
    return this.decoration.type.spec.data;
  }

  get HTMLAttributes() {
    return this.decoration.type.attrs;
  }

  toString() {
    return JSON.stringify({
      id: this.id,
      data: this.data,
      from: this.from,
      to: this.to,
      HTMLAttributes: this.HTMLAttributes,
    });
  }
}