//@ts-nocheck
import * as THREE from 'three'
import { blockEntityBrightness, DEFAULT_LIGHTMAP_PARAMS, type BlockLightmapParams } from './blockEntityLighting'

export type BlockEntityOverlayLight = {
  material: THREE.MeshBasicMaterial
  blockLightNorm: number
  skyLightNorm: number
}

export class BlockEntityLightRegistry {
  private readonly entries = new Set<BlockEntityOverlayLight>()
  private skyLevel = 1
  private lightmapParams: BlockLightmapParams = { ...DEFAULT_LIGHTMAP_PARAMS }

  register(entry: BlockEntityOverlayLight): void {
    this.entries.add(entry)
    this.applyBrightness(entry)
  }

  unregister(material: THREE.Material): void {
    for (const entry of this.entries) {
      if (entry.material === material) {
        this.entries.delete(entry)
        break
      }
    }
  }

  setSkyLevel(value: number): void {
    this.skyLevel = value
    this.refreshAll()
  }

  setLightmapParams(params: BlockLightmapParams): void {
    this.lightmapParams = { ...this.lightmapParams, ...params }
    this.refreshAll()
  }

  getSkyLevel(): number {
    return this.skyLevel
  }

  private refreshAll(): void {
    for (const entry of this.entries) {
      this.applyBrightness(entry)
    }
  }

  private applyBrightness(entry: BlockEntityOverlayLight): void {
    const brightness = blockEntityBrightness(entry.blockLightNorm, entry.skyLightNorm, this.skyLevel, this.lightmapParams)
    entry.material.color.setScalar(brightness)
  }
}

export function tintBannerMaterial(
  material: THREE.MeshBasicMaterial,
  blockLightNorm: number,
  skyLightNorm: number,
  skyLevel: number,
  lightmapParams: BlockLightmapParams = DEFAULT_LIGHTMAP_PARAMS
): number {
  const brightness = blockEntityBrightness(blockLightNorm, skyLightNorm, skyLevel, lightmapParams)
  material.color.setScalar(brightness)
  return brightness
}