<script setup lang="ts">
import { useGLTF } from '@tresjs/cientos'
import { Box3, Color, MeshStandardMaterial, Vector3 } from 'three';
const props = defineProps(['modelUrl', 'position', 'removePos', 'target', 'color'])

const { scene } = await useGLTF(props.modelUrl)
scene.receiveShadow = true
scene.castShadow = true

const target = props.target ? new Vector3(...props.target) : new Vector3(0, 0, 0)
const box = new Box3();
box.expandByObject(scene.children[0]);
let size = new Vector3();
box.getSize(size)

const getMaterial = () => {
    return new MeshStandardMaterial({
        color: props.color ? new Color(props.color) : new Color('#9c9c9c'),
        roughness: 0.5,
        metalness: 0.5,
    })
}
const material = getMaterial()
function shadows_and_pos(scene: any) {
    scene.children.forEach((el: any) => {
        el.castShadow = true
        el.receiveShadow = true
        if (props.removePos) {
            el.translateX(-el.position.x)
            el.translateY(-el.position.y)
            el.translateZ(-el.position.z)
        }
        if (el.material && props.color) el.material = material
        shadows_and_pos(el)
    })
}

shadows_and_pos(scene)

const model = ref()
const { onLoop } = useRenderLoop()
let stepbase = 0.005
const axis = [
    { axis: 'x', func: 'translateX', },
    { axis: 'y', func: 'translateY', },
    { axis: 'z', func: 'translateZ', },
]
type vectorType = 'x' | 'y' | 'z';

onLoop(() => {
    if (model.value && props.target) {
        axis.forEach(element => {
            const point = model.value.position[element.axis]
            let step = stepbase * target[element.axis as vectorType]
            if (step !== 0) {
                if (Math.abs(point) >= Math.abs(target[element.axis as vectorType])) {
                    step = 0
                }
                model.value[element.func](step)
            }
        });
    }
})

watch(props, () => {
    if (props.color && `#${material.color.getHexString()}` !== props.color) {
        function set_material(scene: any) {
            scene.children.forEach((el: any) => {
                if (el.material && props.color) el.material.color = new Color(props.color)
                set_material(el)
            })
        }
        set_material(model.value)
    }
}, { deep: true })
</script>
<template>
    <TresGroup :position="props.position || [0, 0, 0]" ref="model">
        <primitive :object="scene.children[0]" cast-shadow receive-shadow />
    </TresGroup>
</template>