export class AnimatedSprite {
  private frameHeight: number;
  private frameWidth: number;
  private currentFrame: number = 0;
  private elapsedTime: number = 0;
  private active: boolean = true;

  constructor(
    private image: CanvasImageSource,
    private frameCount: number,
    private frameDuration: number, // in milliseconds
    private looping: boolean = false,
    private originX: number,
    private originY: number,
  ) {
    if (frameCount <= 0) {
      throw new Error("Animated sprite should at least have one frame");
    }
    if ("height" in image && "width" in image) {
      this.frameHeight = (image as HTMLImageElement | HTMLCanvasElement).height;
      this.frameWidth = Math.floor(
        (image as HTMLImageElement | HTMLCanvasElement).width / frameCount,
      );
    } else {
      throw new Error(
        "Image source must have 'width' and 'height' properties.",
      );
    }
  }

  update(deltaTime: number) {
    if (!this.active) return;
    this.elapsedTime += deltaTime;
    if (this.elapsedTime >= this.frameDuration) {
      this.elapsedTime -= this.frameDuration;
      this.currentFrame++;

      if (this.currentFrame >= this.frameCount) {
        if (this.looping) {
          this.currentFrame = 0;
        } else {
          this.currentFrame = this.frameCount - 1;
          this.active = false;
        }
      }
    }
  }

  isActive(): boolean {
    return this.active;
  }

  lifeTime(): number | undefined {
    if (this.looping) {
      return undefined;
    }
    return this.frameDuration * this.frameCount;
  }

  draw(ctx: CanvasRenderingContext2D, x: number, y: number) {
    const drawX = x - this.originX;
    const drawY = y - this.originY;

    ctx.drawImage(
      this.image,
      this.currentFrame * this.frameWidth,
      0,
      this.frameWidth,
      this.frameHeight,
      drawX,
      drawY,
      this.frameWidth,
      this.frameHeight,
    );
  }

  reset() {
    this.currentFrame = 0;
    this.elapsedTime = 0;
  }

  setOrigin(xRatio: number, yRatio: number) {
    this.originX = xRatio;
    this.originY = yRatio;
  }
}