import 'dart:math' as math;

import 'package:flutter/material.dart';
import 'package:flutter/scheduler.dart';
import 'package:wolf_dart/classes/linear_coordinates.dart';
import 'package:wolf_dart/classes/matrix.dart';
import 'package:wolf_dart/features/difficulty/difficulty.dart';
import 'package:wolf_dart/features/entities/collectible.dart';
import 'package:wolf_dart/features/entities/enemies/enemy.dart';
import 'package:wolf_dart/features/entities/entity.dart';
import 'package:wolf_dart/features/entities/entity_registry.dart';
import 'package:wolf_dart/features/input/input_manager.dart';
import 'package:wolf_dart/features/map/wolf_map.dart';
import 'package:wolf_dart/features/player/player.dart';
import 'package:wolf_dart/features/renderer/raycast_painter.dart';
import 'package:wolf_dart/features/renderer/weapon_painter.dart';
import 'package:wolf_dart/features/ui/hud.dart';
import 'package:wolf_dart/sprite_gallery.dart';

class WolfRenderer extends StatefulWidget {
  const WolfRenderer({
    super.key,
    required this.difficulty,
    this.showSpriteGallery = false,
    this.isDemo = true,
  });

  final Difficulty difficulty;
  final bool showSpriteGallery;
  final bool isDemo;

  @override
  State<WolfRenderer> createState() => _WolfRendererState();
}

class _WolfRendererState extends State<WolfRenderer>
    with SingleTickerProviderStateMixin {
  final InputManager inputManager = InputManager();
  late Ticker _gameLoop;
  final FocusNode _focusNode = FocusNode();
  late WolfMap gameMap;
  late Matrix<int> currentLevel;

  final double fov = math.pi / 3;

  late Player player;

  bool _isLoading = true;

  double damageFlashOpacity = 0.0;

  List<Entity> entities = [];

  Map<String, double> doorOffsets = {};
  Map<String, int> doorStates = {};

  @override
  void initState() {
    super.initState();
    _initGame(demo: widget.isDemo);
  }

  Future<void> _initGame({bool demo = true}) async {
    gameMap = demo ? await WolfMap.loadDemo() : await WolfMap.load();
    currentLevel = gameMap.levels[0].wallGrid;

    final Matrix<int> objectLevel = gameMap.levels[0].objectGrid;

    for (int y = 0; y < 64; y++) {
      for (int x = 0; x < 64; x++) {
        int objId = objectLevel[y][x];

        if (objId >= 19 && objId <= 22) {
          double spawnAngle = 0.0;
          switch (objId) {
            case 19:
              spawnAngle = 3 * math.pi / 2;
              break;
            case 20:
              spawnAngle = 0.0;
              break;
            case 21:
              spawnAngle = math.pi / 2;
              break;
            case 22:
              spawnAngle = math.pi;
              break;
          }
          player = Player(
            x: x + 0.5,
            y: y + 0.5,
            angle: spawnAngle,
          );
        } else {
          Entity? newEntity = EntityRegistry.spawn(
            objId,
            x + 0.5,
            y + 0.5,
            widget.difficulty.level,
            gameMap.sprites.length,
          );

          if (newEntity != null) {
            entities.add(newEntity);
          }
        }
      }
    }

    for (int y = 0; y < 64; y++) {
      for (int x = 0; x < 64; x++) {
        int id = currentLevel[y][x];
        if ((id >= 1 && id <= 63) || (id >= 90 && id <= 101)) {
          // Leave walls and doors solid
        } else {
          currentLevel[y][x] = 0;
        }
      }
    }

    _bumpPlayerIfStuck();
    _gameLoop = createTicker(_tick)..start();
    _focusNode.requestFocus();

    setState(() {
      _isLoading = false;
    });
  }

  @override
  void dispose() {
    _gameLoop.dispose();
    _focusNode.dispose();
    super.dispose();
  }

  void _bumpPlayerIfStuck() {
    int pX = player.x.toInt();
    int pY = player.y.toInt();

    if (pY < 0 ||
        pY >= currentLevel.length ||
        pX < 0 ||
        pX >= currentLevel[0].length ||
        currentLevel[pY][pX] > 0) {
      double shortestDist = double.infinity;
      LinearCoordinates nearestSafeSpot = (x: 1.5, y: 1.5);

      for (int y = 0; y < currentLevel.length; y++) {
        for (int x = 0; x < currentLevel[y].length; x++) {
          if (currentLevel[y][x] == 0) {
            double safeX = x + 0.5;
            double safeY = y + 0.5;
            double dist = math.sqrt(
              math.pow(safeX - player.x, 2) + math.pow(safeY - player.y, 2),
            );

            if (dist < shortestDist) {
              shortestDist = dist;
              nearestSafeSpot = (x: safeX, y: safeY);
            }
          }
        }
      }
      player.x = nearestSafeSpot.x;
      player.y = nearestSafeSpot.y;
    }
  }

  bool _isWalkable(int x, int y) {
    if (currentLevel[y][x] == 0) return true;
    if (currentLevel[y][x] >= 90) {
      String key = '$x,$y';
      if (doorOffsets[key] != null && doorOffsets[key]! > 0.7) {
        return true;
      }
    }
    return false;
  }

  // --- ORCHESTRATOR ---
  void _tick(Duration elapsed) {
    // 1. Process intentions and receive movement vectors
    final movement = _processInputs(elapsed);

    // 2. Explicit State Updates
    player.updateWeaponSwitch();
    _updateDoors();
    _applyMovementAndCollision(movement.dx, movement.dy);
    _updateEntities(elapsed);

    // Explicit reassignment from a pure(r) function
    damageFlashOpacity = _calculateScreenEffects(damageFlashOpacity);

    // 3. Combat
    player.updateWeapon(
      currentTime: elapsed.inMilliseconds,
      entities: entities,
      isWalkable: _isWalkable,
    );

    // 4. Render
    setState(() {});
  }

  void _takeDamage(int damage) {
    player.takeDamage(damage);
    damageFlashOpacity = 0.5;
  }

  // Returns movement deltas instead of modifying class variables
  ({double dx, double dy}) _processInputs(Duration elapsed) {
    inputManager.update();

    const double moveSpeed = 0.16;
    const double turnSpeed = 0.12;
    double dx = 0;
    double dy = 0;

    if (inputManager.requestedWeapon != null) {
      player.requestWeaponSwitch(inputManager.requestedWeapon!);
    }

    if (inputManager.isFiring) {
      player.fire(elapsed.inMilliseconds);
    }

    if (inputManager.isMovingForward) {
      dx += math.cos(player.angle) * moveSpeed;
      dy += math.sin(player.angle) * moveSpeed;
    }
    if (inputManager.isMovingBackward) {
      dx -= math.cos(player.angle) * moveSpeed;
      dy -= math.sin(player.angle) * moveSpeed;
    }

    if (inputManager.isTurningLeft) player.angle -= turnSpeed;
    if (inputManager.isTurningRight) player.angle += turnSpeed;

    if (player.angle < 0) player.angle += 2 * math.pi;
    if (player.angle > 2 * math.pi) player.angle -= 2 * math.pi;

    if (inputManager.isInteracting) {
      int targetX = (player.x + math.cos(player.angle)).toInt();
      int targetY = (player.y + math.sin(player.angle)).toInt();

      if (targetY > 0 &&
          targetY < currentLevel.length &&
          targetX > 0 &&
          targetX < currentLevel[0].length) {
        if (currentLevel[targetY][targetX] >= 90) {
          String key = '$targetX,$targetY';
          if (!doorStates.containsKey(key) || doorStates[key] == 0) {
            doorStates[key] = 1;
          }
        }
      }
    }

    return (dx: dx, dy: dy);
  }

  void _updateDoors() {
    doorStates.forEach((key, state) {
      if (state == 1) {
        doorOffsets[key] = (doorOffsets[key] ?? 0.0) + 0.02;
        if (doorOffsets[key]! >= 1.0) {
          doorOffsets[key] = 1.0;
          doorStates[key] = 2;
        }
      }
    });
  }

  // Now receives dx and dy explicitly
  void _applyMovementAndCollision(double dx, double dy) {
    const double margin = 0.3;

    double newX = player.x + dx;
    int checkX = (dx > 0) ? (newX + margin).toInt() : (newX - margin).toInt();

    if (_isWalkable(checkX, player.y.toInt())) {
      player.x = newX;
    }

    double newY = player.y + dy;
    int checkY = (dy > 0) ? (newY + margin).toInt() : (newY - margin).toInt();

    if (_isWalkable(player.x.toInt(), checkY)) {
      player.y = newY;
    }
  }

  void _updateEntities(Duration elapsed) {
    List<Entity> itemsToRemove = [];

    for (Entity entity in entities) {
      if (entity is Enemy) {
        entity.update(
          elapsedMs: elapsed.inMilliseconds,
          player: player.position,
          isWalkable: _isWalkable,
          onDamagePlayer: _takeDamage,
        );
      } else if (entity is Collectible) {
        double dx = player.x - entity.x;
        double dy = player.y - entity.y;
        double dist = math.sqrt(dx * dx + dy * dy);

        if (dist < 0.5) {
          if (player.tryPickup(entity)) {
            itemsToRemove.add(entity);
          }
        }
      }
    }

    if (itemsToRemove.isNotEmpty) {
      entities.removeWhere((e) => itemsToRemove.contains(e));
    }
  }

  // Takes an input and returns a value instead of implicitly changing state
  double _calculateScreenEffects(double currentOpacity) {
    if (currentOpacity > 0) {
      return math.max(0.0, currentOpacity - 0.05);
    }
    return currentOpacity;
  }

  @override
  Widget build(BuildContext context) {
    if (_isLoading) {
      return const Center(child: CircularProgressIndicator(color: Colors.teal));
    }

    if (widget.showSpriteGallery) {
      return SpriteGallery(sprites: gameMap.sprites);
    }

    return Scaffold(
      backgroundColor: Colors.black,
      body: KeyboardListener(
        focusNode: _focusNode,
        autofocus: true,
        onKeyEvent: (_) {},
        child: Column(
          children: [
            Expanded(
              child: LayoutBuilder(
                builder: (context, constraints) {
                  return Center(
                    child: AspectRatio(
                      aspectRatio: 16 / 10,
                      child: Stack(
                        children: [
                          CustomPaint(
                            size: Size(
                              constraints.maxWidth,
                              constraints.maxHeight,
                            ),
                            painter: RaycasterPainter(
                              map: currentLevel,
                              textures: gameMap.textures,
                              player: player,
                              fov: fov,
                              doorOffsets: doorOffsets,
                              entities: entities,
                              sprites: gameMap.sprites,
                            ),
                          ),
                          Positioned(
                            bottom: -20,
                            left: 0,
                            right: 0,
                            child: Center(
                              child: Transform.translate(
                                offset: Offset(
                                  // Replaced hidden step variables with a direct intention check!
                                  (inputManager.isMovingForward ||
                                          inputManager.isMovingBackward)
                                      ? math.sin(
                                              DateTime.now()
                                                      .millisecondsSinceEpoch /
                                                  100,
                                            ) *
                                            12
                                      : 0,
                                  player.weaponAnimOffset,
                                ),
                                child: SizedBox(
                                  width: 500,
                                  height: 500,
                                  child: CustomPaint(
                                    painter: WeaponPainter(
                                      sprite:
                                          gameMap.sprites[player
                                              .currentWeapon
                                              .currentSprite],
                                    ),
                                  ),
                                ),
                              ),
                            ),
                          ),
                          if (damageFlashOpacity > 0)
                            Positioned.fill(
                              child: Container(
                                color: Colors.red.withValues(
                                  alpha: damageFlashOpacity,
                                ),
                              ),
                            ),
                        ],
                      ),
                    ),
                  );
                },
              ),
            ),
            Hud(player: player),
          ],
        ),
      ),
    );
  }
}