///////////////////////////////////////////////////////////////////////////////
// Copyright (C) 2002-2026, Open Design Alliance (the "Alliance").
// All rights reserved.
//
// This software and its documentation and related materials are owned by
// the Alliance. The software may only be incorporated into application
// programs owned by members of the Alliance, subject to a signed
// Membership Agreement and Supplemental Software License Agreement with the
// Alliance. The structure and organization of this software are the valuable
// trade secrets of the Alliance and its suppliers. The software is also
// protected by copyright law and international treaty provisions. Application
// programs incorporating this software must include the following statement
// with their copyright notices:
//
//   This application incorporates Open Design Alliance software pursuant to a
//   license agreement with Open Design Alliance.
//   Open Design Alliance Copyright (C) 2002-2026 by Open Design Alliance.
//   All rights reserved.
//
// By use of this software, its documentation or related materials, you
// acknowledge and accept the above terms.
///////////////////////////////////////////////////////////////////////////////

import { AxesHelper, Box3, Vector3 } from "three";
import { IComponent, components, Viewer } from "@inweb/viewer-three";

class AxesHelperComponent implements IComponent {
  private viewer: Viewer;
  private axesHelper1: AxesHelper;
  private axesHelper2: AxesHelper;
  private modelHelpers: AxesHelper[];

  constructor(viewer: Viewer) {
    this.axesHelper1 = new AxesHelper(1);
    this.axesHelper2 = new AxesHelper(1);
    this.modelHelpers = [];

    // this.axesHelper1.setColors("#ccc", "#ccc", "#ccc");

    this.viewer = viewer;
    this.viewer.addEventListener("initialize", this.syncHelper);
    this.viewer.addEventListener("geometryend", this.syncHelper);
    this.viewer.addEventListener("clear", this.syncHelper);
  }

  dispose() {
    this.modelHelpers.forEach((helper) => {
      helper.removeFromParent();
      helper.dispose();
    });

    this.axesHelper1.removeFromParent();
    this.axesHelper1.dispose();

    this.axesHelper2.removeFromParent();
    this.axesHelper2.dispose();

    this.viewer.removeEventListener("initialize", this.syncHelper);
    this.viewer.removeEventListener("geometryend", this.syncHelper);
    this.viewer.removeEventListener("clear", this.syncHelper);
  }

  syncHelper = () => {
    this.modelHelpers.forEach((helper) => {
      helper.removeFromParent();
      helper.dispose();
    });
    this.modelHelpers.length = 0;

    this.axesHelper1.removeFromParent();
    this.axesHelper2.removeFromParent();

    const extentsSize = this.viewer.extents.getSize(new Vector3()).length() || 1;
    const extentsCenter = this.viewer.extents.getCenter(new Vector3());

    this.axesHelper1.position.set(0, 0, 0);
    this.axesHelper1.scale.setScalar(extentsSize * 1.5);
    this.viewer.helpers.add(this.axesHelper1);

    if (this.viewer.models.length < 1) return;

    this.axesHelper2.position.copy(extentsCenter);
    this.axesHelper2.scale.setScalar(extentsSize);
    this.viewer.helpers.add(this.axesHelper2);

    if (this.viewer.models.length < 2) return;

    this.viewer.models.forEach((model) => {
      const modelExtents = model.getExtents(new Box3());
      const modelSize = modelExtents.getSize(new Vector3()).length() || 1;
      const modelCenter = modelExtents.getCenter(new Vector3());

      const helper = new AxesHelper(modelSize);
      helper.position.copy(modelCenter);

      this.modelHelpers.push(helper);
      this.viewer.helpers.add(helper);
    });
  };
}

components.registerComponent("AxesHelperComponent", (viewer) => new AxesHelperComponent(viewer));
