import { Renderer } from "../../core/rendering/Renderer";
import { Group, Mesh, Object3D, PerspectiveCamera } from "three";
import { QuestEntityModel } from "../model/QuestEntityModel";
import { quest_editor_store } from "../stores/QuestEditorStore";
import { QuestModelManager } from "./QuestModelManager";
import { Disposer } from "../../core/observable/Disposer";
import { QuestEntityControls } from "./QuestEntityControls";
import { EntityUserData } from "./conversion/entities";

export class QuestRenderer extends Renderer {
    get debug(): boolean {
        return super.debug;
    }

    set debug(debug: boolean) {
        if (this.debug !== debug) {
            super.debug = debug;
            this._render_geometry.visible = debug;
            this.schedule_render();
        }
    }

    private _collision_geometry = new Object3D();

    get collision_geometry(): Object3D {
        return this._collision_geometry;
    }

    set collision_geometry(collision_geometry: Object3D) {
        this.scene.remove(this.collision_geometry);
        this._collision_geometry = collision_geometry;
        this.scene.add(collision_geometry);
    }

    private _render_geometry = new Object3D();

    get render_geometry(): Object3D {
        return this._render_geometry;
    }

    set render_geometry(render_geometry: Object3D) {
        this.scene.remove(this._render_geometry);
        this._render_geometry = render_geometry;
        render_geometry.visible = this.debug;
        this.scene.add(render_geometry);
    }

    private _entity_models = new Object3D();

    get entity_models(): Object3D {
        return this._entity_models;
    }

    private readonly disposer = new Disposer();
    private readonly perspective_camera: PerspectiveCamera;
    private readonly entity_to_mesh = new Map<QuestEntityModel, Mesh>();
    private readonly model_manager = this.disposer.add(new QuestModelManager(this));
    private readonly entity_controls = this.disposer.add(new QuestEntityControls(this));

    constructor() {
        super(new PerspectiveCamera(60, 1, 10, 10000));

        this.perspective_camera = this.camera as PerspectiveCamera;

        this.disposer.add_all(
            quest_editor_store.current_quest.observe(this.load_models),
            quest_editor_store.current_area.observe(this.load_models),
            quest_editor_store.debug.observe(({ value }) => (this.debug = value)),
        );

        this.dom_element.addEventListener("mousedown", this.entity_controls.on_mouse_down);
        this.dom_element.addEventListener("mouseup", this.entity_controls.on_mouse_up);
        this.dom_element.addEventListener("mousemove", this.entity_controls.on_mouse_move);
    }

    dispose(): void {
        super.dispose();
        this.disposer.dispose();
    }

    set_size(width: number, height: number): void {
        this.perspective_camera.aspect = width / height;
        this.perspective_camera.updateProjectionMatrix();
        super.set_size(width, height);
    }

    reset_entity_models(): void {
        this.scene.remove(this._entity_models);
        this._entity_models = new Group();
        this.scene.add(this._entity_models);
        this.entity_to_mesh.clear();
    }

    add_entity_model(model: Mesh): void {
        const entity = (model.userData as EntityUserData).entity;
        this._entity_models.add(model);
        this.entity_to_mesh.set(entity, model);

        if (entity === quest_editor_store.selected_entity.val) {
            this.entity_controls.try_highlight(entity);
        }

        this.schedule_render();
    }

    get_entity_mesh(entity: QuestEntityModel): Mesh | undefined {
        return this.entity_to_mesh.get(entity);
    }

    private load_models = () => {
        this.model_manager.load_models(
            quest_editor_store.current_quest.val,
            quest_editor_store.current_area.val,
        );
    };
}