import { Cube, FACES, COLORS } from '../src/utils/Cube.js';
import assert from 'assert';

console.log('Running Cube Integrity Tests...');

const cube = new Cube();

// Helper: Count colors on all faces
const countColors = () => {
  const counts = {
    [COLORS.WHITE]: 0,
    [COLORS.YELLOW]: 0,
    [COLORS.ORANGE]: 0,
    [COLORS.RED]: 0,
    [COLORS.GREEN]: 0,
    [COLORS.BLUE]: 0,
    [COLORS.BLACK]: 0 // Should be ignored or internal
  };

  cube.cubies.forEach(cubie => {
    Object.values(cubie.faces).forEach(color => {
      if (counts[color] !== undefined) {
        counts[color]++;
      }
    });
  });

  return counts;
};

// Helper: Verify solved state counts
const verifyCounts = (counts) => {
  // Each face has 9 stickers. 6 faces.
  // 9 * 6 = 54 colored stickers.
  // 27 cubies * 6 faces = 162 total faces.
  // 162 - 54 = 108 black faces (internal).
  
  assert.strictEqual(counts[COLORS.WHITE], 9, 'White count should be 9');
  assert.strictEqual(counts[COLORS.YELLOW], 9, 'Yellow count should be 9');
  assert.strictEqual(counts[COLORS.ORANGE], 9, 'Orange count should be 9');
  assert.strictEqual(counts[COLORS.RED], 9, 'Red count should be 9');
  assert.strictEqual(counts[COLORS.GREEN], 9, 'Green count should be 9');
  assert.strictEqual(counts[COLORS.BLUE], 9, 'Blue count should be 9');
};

// Helper: Verify piece integrity
// Corners: 8 corners, each has 3 colors.
// Edges: 12 edges, each has 2 colors.
// Centers: 6 centers, each has 1 color.
// Core: 1 core, 0 colors.
const verifyPieceTypes = () => {
  let corners = 0;
  let edges = 0;
  let centers = 0;
  let cores = 0;

  cube.cubies.forEach(cubie => {
    const coloredFaces = Object.values(cubie.faces).filter(c => c !== COLORS.BLACK).length;
    if (coloredFaces === 3) corners++;
    else if (coloredFaces === 2) edges++;
    else if (coloredFaces === 1) centers++;
    else if (coloredFaces === 0) cores++;
    else assert.fail(`Invalid cubie with ${coloredFaces} colors at (${cubie.x},${cubie.y},${cubie.z})`);
  });

  assert.strictEqual(corners, 8, 'Should have 8 corners');
  assert.strictEqual(edges, 12, 'Should have 12 edges');
  assert.strictEqual(centers, 6, 'Should have 6 centers');
  assert.strictEqual(cores, 1, 'Should have 1 core');
};

// Helper: Verify specific relative positions of centers (they never change relative to each other)
// Up (White) opposite Down (Yellow)
// Front (Green) opposite Back (Blue)
// Left (Orange) opposite Right (Red)
const verifyCenters = () => {
  const centers = cube.cubies.filter(c => 
    Object.values(c.faces).filter(f => f !== COLORS.BLACK).length === 1
  );
  
  // Find center by color
  const findCenter = (color) => centers.find(c => Object.values(c.faces).includes(color));
  
  const white = findCenter(COLORS.WHITE);
  const yellow = findCenter(COLORS.YELLOW);
  const green = findCenter(COLORS.GREEN);
  const blue = findCenter(COLORS.BLUE);
  const orange = findCenter(COLORS.ORANGE);
  const red = findCenter(COLORS.RED);

  // Check opposites
  // Distance between opposites should be 2 (e.g. y=1 and y=-1)
  // And they should be on same axis
  
  // Note: After rotations, x/y/z coordinates change.
  // But relative vectors should hold?
  // Actually, centers DO rotate around the core.
  // But White is always opposite Yellow.
  // So vector(White) + vector(Yellow) == (0,0,0).
  
  const checkOpposite = (c1, c2, name) => {
    assert.strictEqual(c1.x + c2.x, 0, `${name} X mismatch`);
    assert.strictEqual(c1.y + c2.y, 0, `${name} Y mismatch`);
    assert.strictEqual(c1.z + c2.z, 0, `${name} Z mismatch`);
  };

  checkOpposite(white, yellow, 'White-Yellow');
  checkOpposite(green, blue, 'Green-Blue');
  checkOpposite(orange, red, 'Orange-Red');
};


// --- Test Execution ---

// 1. Initial State
console.log('Test 1: Initial State Integrity');
verifyCounts(countColors());
verifyPieceTypes();
verifyCenters();
console.log('PASS Initial State');

// 2. Single Rotation (R)
console.log('Test 2: Single Rotation (R)');
cube.rotateLayer('x', 1, -1); // R
verifyCounts(countColors());
verifyPieceTypes();
verifyCenters();
console.log('PASS Single Rotation');

// 3. Multiple Rotations (R U R' U')
console.log('Test 3: Sexy Move (R U R\' U\')');
cube.reset();
cube.move("R");
cube.move("U");
cube.move("R'");
cube.move("U'");
verifyCounts(countColors());
verifyPieceTypes();
verifyCenters();
console.log('PASS Sexy Move');

// 4. Random Rotations (Fuzzing)
console.log('Test 4: 100 Random Moves');
cube.reset();
const axes = ['x', 'y', 'z'];
const indices = [-1, 0, 1];
const dirs = [1, -1];

for (let i = 0; i < 100; i++) {
  const axis = axes[Math.floor(Math.random() * axes.length)];
  const index = indices[Math.floor(Math.random() * indices.length)];
  const dir = dirs[Math.floor(Math.random() * dirs.length)];
  cube.rotateLayer(axis, index, dir);
}
verifyCounts(countColors());
verifyPieceTypes();
verifyCenters();
console.log('PASS 100 Random Moves');

console.log('ALL INTEGRITY TESTS PASSED');