using Content.Server.Construction.Components;
using Content.Server.Power.Components;
using Content.Shared.Computer;
using Robust.Shared.Containers;

namespace Content.Server.Construction;

public sealed partial class ConstructionSystem
{
    private void InitializeComputer()
    {
        SubscribeLocalEvent<ComputerComponent, ComponentInit>(OnCompInit);
        SubscribeLocalEvent<ComputerComponent, MapInitEvent>(OnCompMapInit);
        SubscribeLocalEvent<ComputerComponent, PowerChangedEvent>(OnCompPowerChange);
    }

    private void OnCompInit(EntityUid uid, ComputerComponent component, ComponentInit args)
    {
        // Let's ensure the container manager and container are here.
        _container.EnsureContainer<Container>(uid, "board");

        if (TryComp<ApcPowerReceiverComponent>(uid, out var powerReceiver) &&
            TryComp<AppearanceComponent>(uid, out var appearance))
        {
            appearance.SetData(ComputerVisuals.Powered, powerReceiver.Powered);
        }
    }

    private void OnCompMapInit(EntityUid uid, ComputerComponent component, MapInitEvent args)
    {
        CreateComputerBoard(component);
    }

    private void OnCompPowerChange(EntityUid uid, ComputerComponent component, PowerChangedEvent args)
    {
        if (TryComp<AppearanceComponent>(uid, out var appearance))
        {
            appearance.SetData(ComputerVisuals.Powered, args.Powered);
        }
    }

    /// <summary>
    ///     Creates the corresponding computer board on the computer.
    ///     This exists so when you deconstruct computers that were serialized with the map,
    ///     you can retrieve the computer board.
    /// </summary>
    private void CreateComputerBoard(ComputerComponent component)
    {
        // Ensure that the construction component is aware of the board container.
        if (TryComp<ConstructionComponent>(component.Owner, out var construction))
            AddContainer(component.Owner, "board", construction);

        // We don't do anything if this is null or empty.
        if (string.IsNullOrEmpty(component.BoardPrototype))
            return;

        var container = _container.EnsureContainer<Container>(component.Owner, "board");

        // We already contain a board. Note: We don't check if it's the right one!
        if (container.ContainedEntities.Count != 0)
            return;

        var board = EntityManager.SpawnEntity(component.BoardPrototype, Transform(component.Owner).Coordinates);

        if(!container.Insert(board))
            Logger.Warning($"Couldn't insert board {board} to computer {component.Owner}!");
    }
}