editor/packages/nodes/src/fence/geometry.ts

import { type GeometryContext, getMaterialPresetByRef } from '@pascal-app/core'
import {
  applyMaterialPresetToMaterials,
  type ColorPreset,
  createDefaultMaterial,
  createMaterial,
  createSurfaceRoleMaterial,
  generateFenceSlotGeometries,
  type RenderShading,
  resolveMaterialRef,
  resolveSlotDefaultMaterial,
} from '@pascal-app/viewer'
import { FrontSide, Group, type Material, Mesh, type Texture } from 'three'
import type { FenceNode } from './schema'
import { FENCE_PANEL_SLOT_DEFAULT, FENCE_RAIL_SLOT_DEFAULT, type FenceSlotId } from './slots'

/**
 * Stage B builder for fence. Splits the geometry into two paintable slots —
 * `panel` (posts / base / infill) and `rail` (the cap rail) — each its own Mesh
 * with a `userData.slotId` so the unified slot paint resolves and previews per
 * part.
 *
 * Per slot the material resolves: `node.slots[slotId]` (a shared scene material
 * or `library:` finish) → the legacy inline `node.material` / `materialPreset`
 * (pre-slot-model scenes, applied to both parts) → the declared slot default.
 * Textures-off collapses both parts to the themed joinery role.
 *
 * Phase 6 cleanup moves the geometry math out of the legacy
 * `viewer/src/systems/fence/fence-system.tsx` into this folder once the legacy
 * system file is deleted. Until then `generateFenceSlotGeometries` is publicly
 * re-exported from viewer.
 */
type FenceMaterial = Material & {
  alphaMap?: Texture | null
  depthWrite: boolean
  opacity: number
  transparent: boolean
}

const SLOT_DEFAULTS: Record<FenceSlotId, string> = {
  panel: FENCE_PANEL_SLOT_DEFAULT,
  rail: FENCE_RAIL_SLOT_DEFAULT,
}

const fenceMaterialCache = new Map<string, Material>()

function getFenceSlotMaterial(
  node: FenceNode,
  slotId: FenceSlotId,
  shading: RenderShading,
  textures: boolean,
  colorPreset: ColorPreset,
  sceneTheme: string | undefined,
  sceneMaterials: GeometryContext['materials'],
): Material {
  if (!textures) {
    return createSurfaceRoleMaterial('joinery', colorPreset, FrontSide, sceneTheme)
  }

  const slotRef = node.slots?.[slotId]
  if (slotRef) {
    const resolved = resolveMaterialRef(slotRef, sceneMaterials, shading)
    if (resolved) return resolved
  }

  if (node.materialPreset || node.material) {
    return getLegacyFenceMaterial(node, shading)
  }

  return resolveSlotDefaultMaterial(SLOT_DEFAULTS[slotId], shading, 0.8)
}

function getLegacyFenceMaterial(node: FenceNode, shading: RenderShading): Material {
  const cacheKey = JSON.stringify({
    shading,
    material: node.material ?? null,
    materialPreset: node.materialPreset ?? null,
  })
  const cached = fenceMaterialCache.get(cacheKey)
  if (cached) return cached

  const preset = getMaterialPresetByRef(node.materialPreset)
  const material = preset
    ? createDefaultMaterial('#ffffff', 0.5, shading)
    : node.material
      ? createMaterial(node.material, shading).clone()
      : createDefaultMaterial('#ffffff', 0.9, shading)

  if (preset) {
    applyMaterialPresetToMaterials(material, preset)
  }

  const fenceMaterial = material as FenceMaterial
  fenceMaterial.transparent = false
  fenceMaterial.opacity = 1
  fenceMaterial.alphaMap = null
  fenceMaterial.side = FrontSide
  fenceMaterial.depthWrite = true
  fenceMaterial.needsUpdate = true

  fenceMaterialCache.set(cacheKey, material)
  return material
}

export function buildFenceGeometry(
  node: FenceNode,
  ctx?: GeometryContext,
  shading: RenderShading = 'rendered',
  textures = true,
  colorPreset: ColorPreset = 'clay',
  sceneTheme?: string,
): Group {
  const group = new Group()
  const geometries = generateFenceSlotGeometries(node)

  for (const slotId of ['panel', 'rail'] as const) {
    const geometry = geometries[slotId]
    if (geometry.getAttribute('position') === undefined) continue
    const material = getFenceSlotMaterial(
      node,
      slotId,
      shading,
      textures,
      colorPreset,
      sceneTheme,
      ctx?.materials,
    )
    const mesh = new Mesh(geometry, material)
    mesh.castShadow = true
    mesh.receiveShadow = true
    mesh.userData.slotId = slotId
    group.add(mesh)
  }

  return group
}