import {
  BufferGeometry,
  ColorRepresentation,
  DirectionalLight,
  Float32BufferAttribute,
  LineBasicMaterial,
  LineSegments,
  Object3D
} from 'three'

import { ObjectLayers } from '../constants/ObjectLayers'
import { setObjectLayers } from '../functions/setObjectLayers'

export default class EditorDirectionalLightHelper extends Object3D {
  color: ColorRepresentation
  lightPlane: LineSegments<BufferGeometry, LineBasicMaterial>
  targetLine: LineSegments<BufferGeometry, LineBasicMaterial>
  name: string
  directionalLight: DirectionalLight

  constructor(directionalLight: DirectionalLight, size?: number, color?: ColorRepresentation) {
    super()
    this.directionalLight = directionalLight
    this.name = 'directional-light-helper'
    if (color) this.color = color

    if (size === undefined) size = 1

    const material = new LineBasicMaterial()

    let geometry = new BufferGeometry()
    geometry.setAttribute(
      'position',
      new Float32BufferAttribute(
        [
          -size,
          size,
          0,
          size,
          size,
          0,
          size,
          size,
          0,
          size,
          -size,
          0,
          size,
          -size,
          0,
          -size,
          -size,
          0,
          -size,
          -size,
          0,
          -size,
          size,
          0,
          -size,
          size,
          0,
          size,
          -size,
          0,
          size,
          size,
          0,
          -size,
          -size,
          0
        ],
        3
      )
    )

    this.lightPlane = new LineSegments(geometry, material)
    this.add(this.lightPlane)

    geometry = new BufferGeometry()
    const t = size * 0.1
    geometry.setAttribute(
      'position',
      new Float32BufferAttribute([-t, t, 0, 0, 0, 1, t, t, 0, 0, 0, 1, t, -t, 0, 0, 0, 1, -t, -t, 0, 0, 0, 1], 3)
    )

    this.targetLine = new LineSegments(geometry, material)
    this.add(this.targetLine)

    setObjectLayers(this, ObjectLayers.NodeHelper)
  }

  update() {
    if (!this.directionalLight) return
    if (this.color !== undefined) {
      this.lightPlane.material.color.set(this.color)
      this.targetLine.material.color.set(this.color)
    } else {
      this.lightPlane.material.color.copy(this.directionalLight!.color)
      this.targetLine.material.color.copy(this.directionalLight!.color)
    }
  }

  dispose() {
    this.lightPlane.geometry.dispose()
    this.lightPlane.material.dispose()
    this.targetLine.geometry.dispose()
    this.targetLine.material.dispose()
  }
}