import { describe, it, expect } from 'vitest';
import {
  LocalCharacterPredictor,
  setFrameTimeMs,
} from '../frontend/src/scripts/helper/prediction/local-character-predictor';
import { ClientCharacterWorld } from '../frontend/src/scripts/helper/prediction/client-character-world';
import {
  stepCharacterMovement,
  tickPlanetDetachment,
  headRadius,
  feetRadius,
} from '../shared/src/physics/character-movement';
import { followVec2 } from '../shared/src/helper/ease';
import { settings } from '../shared/src/settings';

const stepSeconds = settings.targetPhysicsDeltaTimeInSeconds;
const snapshotMs = settings.updateMessageInterval * 1000;
const heartbeatMs = settings.clientSendInterval * 1000;
const planetRadius = 800;

// The client's frame clock is rebased to its first frame (sdf-2d passes currentTime - startTime),
// so it shares no origin with the server's clock. Anything that only works when the two happen to
// agree is not a fix.
const clientClockOffsetMs = 987_654.321;

const vertices = Array.from({ length: settings.planetEdgeCount }, (_, i) => {
  const angle = (i / settings.planetEdgeCount) * -Math.PI * 2;
  return [planetRadius * Math.cos(angle), planetRadius * Math.sin(angle)];
});

const planetAt = (rotation: number) =>
  ({ id: 1, vertices, snapshotRotation: rotation, snapshotRotationSpeed: 0 }) as never;

const body = (center: Array<number>, radius: number) => ({
  center: center.slice(),
  radius,
  velocity: [0, 0],
  lastNormal: [0, 1],
  restitution: 0,
});

interface Conditions {
  fps: number;
  /** One way. The round trip is twice this. */
  wireMs: number;
  /** Client CPU spent between the frame timestamp and the socket emit at the end of the loop. */
  emitLagMs: number;
  /** Peak random variation in frame duration, as a fraction of the nominal frame. */
  frameJitter: number;
  /** Probability that a frame is missed entirely and the next one is twice as long. */
  dropRate: number;
  /** Server work that can delay a socket callback past the packet's arrival. */
  serverBusyMs: number;
}

interface Result {
  /** Peak-to-peak variation in apparent speed, over its mean. Zero means perfectly even motion. */
  speedSpread: number;
  /** Peak-to-peak wobble of the character's screen position about its own mean, in world units. */
  screenWobble: number;
  /** How far the rendered character trails the true server pose, in world units. */
  lagUnits: number;
}

// Runs a real server tick loop, a real wire, and a real render loop against the predictor, and
// reports how steady the character's *on-screen* travel is. On-screen, not in world space: the
// camera follows the character, so what the player sees is the residual between the two.
const measure = ({
  fps,
  wireMs,
  emitLagMs,
  frameJitter,
  dropRate,
  serverBusyMs,
}: Conditions): Result => {
  const world = new ClientCharacterWorld();
  world.sync([planetAt(0)]);

  const server: any = {
    head: body([0, planetRadius + 90], headRadius),
    leftFoot: body([-33, planetRadius + 35], feetRadius),
    rightFoot: body([33, planetRadius + 35], feetRadius),
    direction: 0,
    currentPlanet: world.surfaceById(1),
    secondsSinceOnSurface: 0,
    bodyVelocity: [0, 0],
  };

  const input = [1, 0];
  const predictor = new LocalCharacterPredictor();
  predictor.setStrength(settings.playerMaxStrength);

  // Deterministic frame-time noise: a real display never delivers a perfectly even cadence.
  let seed = 12345;
  const random = () => ((seed = (seed * 1103515245 + 12345) & 0x7fffffff) / 0x7fffffff);

  const nominalFrameMs = 1000 / fps;
  const toServer: Array<{ atMs: number; clientStampMs: number }> = [];
  const toClient: Array<{ atMs: number; ackAgeMs: number; clientStampMs: number; pose: any }> = [];
  const samples: Array<{
    screen: Array<number>;
    world: Array<number>;
    atMs: number;
    error: number;
  }> = [];

  // The client's own view of the character, and the camera that chases it.
  const camera = [0, planetRadius + 90];
  const cameraPreviousTarget = [0, planetRadius + 90];

  let simMs = 0;
  let nextSnapshotMs = 0;
  let nextFrameAtMs = 0;
  let lastHeartbeatClientMs = -Infinity;
  let lastInputClientTimeMs = 0;
  let lastInputReceiptMs = 0;
  let hasPressed = false;


  while (simMs < 8000) {
    tickPlanetDetachment(server, stepSeconds);
    stepCharacterMovement(server, world, input.slice() as never, stepSeconds);
    simMs += stepSeconds * 1000;

    while (toServer.length > 0 && toServer[0].atMs <= simMs) {
      const packet = toServer.shift()!;
      if (packet.clientStampMs > lastInputClientTimeMs) {
        lastInputClientTimeMs = packet.clientStampMs;
        // Socket callbacks run between physics frames, so this is the true arrival instant,
        // not the frame boundary — plus however long the frame in progress still had to run.
        lastInputReceiptMs = packet.atMs + serverBusyMs;
      }
    }

    if (simMs >= nextSnapshotMs) {
      nextSnapshotMs += snapshotMs;
      toClient.push({
        atMs: simMs + wireMs,
        clientStampMs: lastInputClientTimeMs,
        ackAgeMs: lastInputReceiptMs > 0 ? Math.max(0, simMs - lastInputReceiptMs) : 0,
        pose: {
          head: server.head.center.slice(),
          leftFoot: server.leftFoot.center.slice(),
          rightFoot: server.rightFoot.center.slice(),
          movement: {
            direction: server.direction,
            bodyVelocity: server.bodyVelocity.slice(),
            leftFootNormal: server.leftFoot.lastNormal.slice(),
            rightFootNormal: server.rightFoot.lastNormal.slice(),
            groundPlanetId: 1,
            secondsSinceOnSurface: server.secondsSinceOnSurface,
          },
        },
      });
    }

    while (toClient.length > 0 && toClient[0].atMs <= simMs) {
      const packet = toClient.shift()!;
      predictor.acknowledge(
        packet.clientStampMs,
        packet.pose.movement,
        -Infinity,
        packet.ackAgeMs,
      );
      predictor.setAuthoritative(
        { center: packet.pose.head, radius: headRadius } as never,
        { center: packet.pose.leftFoot, radius: feetRadius } as never,
        { center: packet.pose.rightFoot, radius: feetRadius } as never,
      );
    }

    while (nextFrameAtMs <= simMs) {
      const frameClientMs = nextFrameAtMs + clientClockOffsetMs;
      setFrameTimeMs(frameClientMs);

      // Movement input is edge-triggered, exactly as KeyboardListener sends it: recorded once
      // when the key goes down, then held by both sides.
      if (!hasPressed) {
        hasPressed = true;
        predictor.recordInput(input as never);
      }

      const dropped = random() < dropRate;
      const jitter = 1 + (random() * 2 - 1) * frameJitter;
      const frameMs = nominalFrameMs * jitter * (dropped ? 2 : 1);

      if (!dropped) {
        predictor.update([planetAt(0)], frameMs / 1000);
        const head = predictor.head.center;
        followVec2(
          camera as never,
          head,
          cameraPreviousTarget as never,
          frameMs / 1000,
          0.08,
          1500,
        );
        samples.push({
          screen: [head[0] - camera[0], head[1] - camera[1]],
          world: [head[0], head[1]],
          atMs: frameClientMs,
          error: Math.hypot(head[0] - server.head.center[0], head[1] - server.head.center[1]),
        });
      }

      // Stamped at the top of the frame, emitted at the bottom.
      if (frameClientMs - lastHeartbeatClientMs >= heartbeatMs - 1) {
        lastHeartbeatClientMs = frameClientMs;
        toServer.push({
          atMs: nextFrameAtMs + emitLagMs + wireMs,
          clientStampMs: frameClientMs,
        });
      }

      nextFrameAtMs += frameMs;
    }
  }

  const tail = samples.slice(Math.floor(samples.length / 2));

  // Apparent speed: world travel divided by how long the frame was on screen. A longer frame
  // should show proportionally more travel; anything else reads as a lurch.
  const speeds = tail
    .slice(1)
    .map(
      (s, i) =>
        Math.hypot(s.world[0] - tail[i].world[0], s.world[1] - tail[i].world[1]) /
        (s.atMs - tail[i].atMs),
    );
  const meanSpeed = speeds.reduce((a, b) => a + b, 0) / speeds.length;

  // Screen position: the camera settles to a constant lag, so anything left is visible wobble.
  const centre = [0, 1].map((axis) => {
    const values = tail.map((s) => s.screen[axis]);
    return values.reduce((a, b) => a + b, 0) / values.length;
  });
  const wobble = Math.max(
    ...tail.map((s) => Math.hypot(s.screen[0] - centre[0], s.screen[1] - centre[1])),
  );

  if ((globalThis as any).__dump) {
    console.log('--- apparent speed (u/ms) ---');
    speeds.slice(0, 40).forEach((v, i) => console.log(`${i} ${v.toFixed(4)}`));
  }
  return {
    speedSpread: (Math.max(...speeds) - Math.min(...speeds)) / meanSpeed,
    screenWobble: wobble,
    lagUnits: tail.reduce((sum, s) => sum + s.error, 0) / tail.length,
  };
};

const CONDITIONS: Array<[string, Conditions]> = [
  ['localhost, clean frames', { fps: 60, wireMs: 0.2, emitLagMs: 2, frameJitter: 0, dropRate: 0, serverBusyMs: 0 }],
  ['localhost, 144 Hz', { fps: 144, wireMs: 0.2, emitLagMs: 2, frameJitter: 0, dropRate: 0, serverBusyMs: 0 }],
  ['localhost, frame jitter', { fps: 60, wireMs: 0.2, emitLagMs: 2, frameJitter: 0.15, dropRate: 0, serverBusyMs: 0 }],
  ['localhost, 2% dropped', { fps: 60, wireMs: 0.2, emitLagMs: 2, frameJitter: 0.05, dropRate: 0.02, serverBusyMs: 0 }],
  ['localhost, variable CPU', { fps: 60, wireMs: 0.2, emitLagMs: 8, frameJitter: 0.15, dropRate: 0.02, serverBusyMs: 0 }],
  ['30 ms link', { fps: 60, wireMs: 15, emitLagMs: 2, frameJitter: 0.05, dropRate: 0, serverBusyMs: 0 }],
  ['120 ms link', { fps: 60, wireMs: 60, emitLagMs: 2, frameJitter: 0.05, dropRate: 0, serverBusyMs: 0 }],
];

describe('local character render smoothness', () => {
  it('keeps on-screen travel even under realistic clocks and frame pacing', () => {
    ;(globalThis as any).__dump = true
    measure(CONDITIONS[0][1])
    ;(globalThis as any).__dump = false
    const results: Array<[string, Result]> = CONDITIONS.map(([label, c]) => [label, measure(c)]);

    for (const [label, r] of results) {
      console.log(
        `${label.padEnd(26)} speed=${(r.speedSpread * 100).toFixed(1).padStart(6)}% ` +
          `wobble=${r.screenWobble.toFixed(2).padStart(6)}u ` +
          `lag=${r.lagUnits.toFixed(1).padStart(5)}u `,
      );
    }

    for (const [label, r] of results) {
      // Even travel: what the eye reads as smooth motion.
      expect(r.speedSpread, label).toBeLessThan(0.05);
      // No sub-pixel swimming against the camera, which follows the same character.
      expect(r.screenWobble, label).toBeLessThan(1);
    }

    // Prediction exists to cancel latency; a follower that lags undoes that on every link.
    const localhost = results.find(([label]) => label === 'localhost, clean frames')![1];
    expect(localhost.lagUnits).toBeLessThan(5);
  });
});