import { Doodler, IDoodlerOptions } from "./canvas.ts";
import { OriginVector, Point } from "./geometry/vector.ts";
import { postInit } from "./postInit.ts";
import { easeInOut } from "./timing/EaseInOut.ts";
import { map } from "./timing/Map.ts";

type TouchEventCallback = (e: TouchEvent) => void;

export class ZoomableDoodler extends Doodler {
  private scale = 1;
  dragging = false;
  private origin: Point = {
    x: 0,
    y: 0,
  };

  mouse = {
    x: 0,
    y: 0,
  };

  private previousTouchLength?: number;

  private touchTimer?: number;

  private hasDoubleTapped = false;
  private zooming = false;
  scaleAround: Point = { x: 0, y: 0 };

  maxScale = 4;

  constructor(options: IDoodlerOptions, postInit?: postInit) {
    super(options, postInit);

    this._canvas.addEventListener("wheel", (e) => {
      this.scaleAtMouse(e.deltaY < 0 ? 1.1 : .9);
      if (this.scale === 1) {
        this.origin.x = 0;
        this.origin.y = 0;
      }
    });
    this._canvas.addEventListener("dblclick", (e) => {
      e.preventDefault();
      this.scale = 1;
      this.origin.x = 0;
      this.origin.y = 0;
      this.ctx.setTransform(1, 0, 0, 1, 0, 0);
    });
    this._canvas.addEventListener("mousedown", (e) => {
      e.preventDefault();
      this.dragging = true;
    });
    this._canvas.addEventListener("mouseup", (e) => {
      e.preventDefault();
      this.dragging = false;
    });
    this._canvas.addEventListener("mouseleave", (_e) => {
      this.dragging = false;
    });
    this._canvas.addEventListener("mousemove", (e) => {
      const prev = this.mouse;
      this.mouse = {
        x: e.offsetX,
        y: e.offsetY,
      };
      if (this.dragging && !this.dragTarget) this.drag(prev);
    });

    this._canvas.addEventListener("touchstart", (e) => {
      e.preventDefault();
      if (e.touches.length === 1) {
        const t1 = e.touches.item(0);
        if (t1) {
          this.mouse = this.getTouchOffset({
            x: t1.clientX,
            y: t1.clientY,
          });
        }
        // this.touchTimer = setTimeout(() => {
        //   this.dragging = true;
        // }, 100)
      } else {
        clearTimeout(this.touchTimer);
      }
    });
    this._canvas.addEventListener("touchend", (e) => {
      if (e.touches.length !== 2) {
        this.previousTouchLength = undefined;
      }

      switch (e.touches.length) {
        case 1:
          break;
        case 0:
          if (!this.zooming) {
            this.events.get("touchend")?.map((cb) => cb(e));
          }
          break;
      }

      this.dragging = e.touches.length === 1;
      clearTimeout(this.touchTimer);
    });
    this._canvas.addEventListener("touchmove", (e) => {
      e.preventDefault();

      if (e.touches.length === 2) {
        const t1 = e.touches.item(0);
        const t2 = e.touches.item(1);

        if (t1 && t2) {
          const vect = OriginVector.from(
            this.getTouchOffset({
              x: t1.clientX,
              y: t1.clientY,
            }),
            {
              x: t2.clientX,
              y: t2.clientY,
            },
          );

          if (this.previousTouchLength) {
            const diff = this.previousTouchLength - vect.mag();
            this.scaleAt(vect.halfwayPoint, diff < 0 ? 1.01 : .99);
            this.scaleAround = { ...vect.halfwayPoint };
          }
          this.previousTouchLength = vect.mag();
        }
      }

      if (e.touches.length === 1) {
        this.dragging === true;
        const t1 = e.touches.item(0);
        if (t1) {
          const prev = this.mouse;
          this.mouse = this.getTouchOffset({
            x: t1.clientX,
            y: t1.clientY,
          });
          this.drag(prev);
        }
      }
    });

    this._canvas.addEventListener("touchstart", (e) => {
      if (e.touches.length !== 1) return false;

      if (!this.hasDoubleTapped) {
        this.hasDoubleTapped = true;
        setTimeout(() => this.hasDoubleTapped = false, 300);
        return false;
      }

      // this.ctx.setTransform(1, 0, 0, 1, 0, 0);
      // this.scale = 1;
      // this.origin.x = 0;
      // this.origin.y = 0;

      if (this.scale > 1) {
        this.frameCounter = map(this.scale, this.maxScale, 1, 0, 59);
        this.zoomDirection = -1;
      } else {
        this.frameCounter = 0;
        this.zoomDirection = 1;
      }
      if (this.zoomDirection > 0) {
        this.scaleAround = { ...this.mouse };
      }

      this.events.get("doubletap")?.map((cb) => cb(e));
    });
  }

  worldToScreen(x: number, y: number) {
    x = x * this.scale + this.origin.x;
    y = y * this.scale + this.origin.y;
    return { x, y };
  }
  screenToWorld(x: number, y: number) {
    x = (x - this.origin.x) / this.scale;
    y = (y - this.origin.y) / this.scale;
    return { x, y };
  }
  scaleAtMouse(scaleBy: number) {
    if (this.scale === this.maxScale && scaleBy > 1) return;
    this.scaleAt({
      x: this.mouse.x,
      y: this.mouse.y,
    }, scaleBy);
  }
  scaleAt(p: Point, scaleBy: number) {
    this.scale = Math.min(Math.max(this.scale * scaleBy, 1), this.maxScale);
    this.origin.x = p.x - (p.x - this.origin.x) * scaleBy;
    this.origin.y = p.y - (p.y - this.origin.y) * scaleBy;
    this.constrainOrigin();
  }
  moveOrigin(motion: Point) {
    if (this.scale > 1) {
      this.origin.x += motion.x;
      this.origin.y += motion.y;
      this.constrainOrigin();
    }
  }
  drag(prev: Point) {
    if (this.scale > 1) {
      const xOffset = this.mouse.x - prev.x;
      const yOffset = this.mouse.y - prev.y;
      this.origin.x += xOffset;
      this.origin.y += yOffset;
      this.constrainOrigin();
    }
  }
  constrainOrigin() {
    this.origin.x = Math.min(
      Math.max(
        this.origin.x,
        (-this._canvas.width * this.scale) + this._canvas.width,
      ),
      0,
    );
    this.origin.y = Math.min(
      Math.max(
        this.origin.y,
        (-this._canvas.height * this.scale) + this._canvas.height,
      ),
      0,
    );
  }

  draw() {
    this.ctx.setTransform(
      this.scale,
      0,
      0,
      this.scale,
      this.origin.x,
      this.origin.y,
    );
    this.animateZoom();
    this.ctx.fillStyle = this.bg;
    this.ctx.fillRect(0, 0, this.width / this.scale, this.height / this.scale);
    super.draw();
  }

  getTouchOffset(p: Point) {
    const { x, y } = this._canvas.getBoundingClientRect();
    const offsetX = p.x - x;
    const offsetY = p.y - y;

    return {
      x: offsetX,
      y: offsetY,
    };
  }

  onDrag(e: MouseEvent): void {
    const d = {
      ...e,
      movementX: e.movementX / this.scale,
      movementY: e.movementY / this.scale,
    };
    super.onDrag(d);
    const { x, y } = this.screenToWorld(e.offsetX, e.offsetY);
    this.mouseX = x;
    this.mouseY = y;
  }

  zoomDirection = -1;
  frameCounter = 60;
  animateZoom() {
    if (this.frameCounter < 60) {
      const frame = easeInOut(map(this.frameCounter, 0, 59, 0, 1));
      switch (this.zoomDirection) {
        case 1:
          {
            this.scale = map(frame, 0, 1, 1, this.maxScale);
          }
          break;
        case -1:
          {
            this.scale = map(frame, 0, 1, this.maxScale, 1);
          }
          break;
      }
      this.origin.x = this.scaleAround.x - (this.scaleAround.x * this.scale);
      this.origin.y = this.scaleAround.y - (this.scaleAround.y * this.scale);
      this.constrainOrigin();

      this.frameCounter++;
    }
  }

  events: Map<string, TouchEventCallback[]> = new Map();
  registerEvent(
    eventName: "touchend" | "touchstart" | "touchmove" | "doubletap",
    cb: TouchEventCallback,
  ) {
    let events = this.events.get(eventName);
    if (!events) events = this.events.set(eventName, []).get(eventName)!;
    events.push(cb);
  }
}