using UnityEngine;

[ExecuteAlways]
public class MarsAtmosphereSystem : MonoBehaviour
{
    public float marsRadiusUnits = 3.3895f;
    public float heightMultiplier = 8f;

    void OnEnable()
    {
        if (!Application.isPlaying && transform.childCount == 0)
            BuildLayers();
    }

    [ContextMenu("Перебудувати шари")]
    public void BuildLayers()
    {
        for (int i = transform.childCount - 1; i >= 0; i--)
        {
            Transform child = transform.GetChild(i);
            if (child.name.StartsWith("Layer_"))
            {
                if (Application.isPlaying) Destroy(child.gameObject);
                else DestroyImmediate(child.gameObject);
            }
        }

        CreateLayer("Layer_Troposphere", 200f, new Color(0.9f, 0.5f, 0.2f), new Color(0.7f, 0.3f, 0.1f), 1.5f, 0.4f);
        CreateLayer("Layer_Mesosphere", 400f, new Color(0.85f, 0.45f, 0.15f), new Color(0.6f, 0.25f, 0.08f), 1.2f, 0.3f);
        CreateLayer("Layer_Thermosphere", 600f, new Color(0.8f, 0.4f, 0.12f), new Color(0.5f, 0.2f, 0.06f), 0.9f, 0.2f);
        CreateLayer("Layer_Exosphere", 800f, new Color(0.7f, 0.35f, 0.1f), new Color(0.4f, 0.15f, 0.04f), 0.6f, 0.1f);
    }

    void CreateLayer(string layerName, float heightKm, Color rimColor, Color coreColor, float rimIntensity, float coreIntensity)
    {
        float radiusUnits = marsRadiusUnits + (heightKm / 1000f) * heightMultiplier;

        GameObject obj = GameObject.CreatePrimitive(PrimitiveType.Sphere);
        obj.name = layerName;
        obj.transform.parent = this.transform;
        obj.transform.localPosition = Vector3.zero;
        obj.transform.localScale = Vector3.one * radiusUnits * 2f;

        if (Application.isPlaying) Destroy(obj.GetComponent<Collider>());
        else DestroyImmediate(obj.GetComponent<Collider>());

        Material mat = new Material(Shader.Find("Custom/AtmosphereLayer"));
        mat.SetColor("_RimColor", rimColor);
        mat.SetColor("_CoreColor", coreColor);
        mat.SetFloat("_RimPower", 2.5f);
        mat.SetFloat("_RimIntensity", rimIntensity);
        mat.SetFloat("_CoreIntensity", coreIntensity);
        mat.SetFloat("_PulseSpeed", 0.3f);
        mat.SetFloat("_PulseStrength", 0.1f);

        obj.GetComponent<Renderer>().material = mat;
    }

    public void SetLayerIntensity(string layerName, float level)
    {
        Transform layer = transform.Find(layerName);
        if (layer == null) return;
        Renderer r = layer.GetComponent<Renderer>();
        if (r == null) return;

        float rimBase = 0f;
        float coreBase = 0f;

        switch (layerName)
        {
            case "Layer_Troposphere": rimBase = 1.5f; coreBase = 0.4f; break;
            case "Layer_Mesosphere": rimBase = 1.2f; coreBase = 0.3f; break;
            case "Layer_Thermosphere": rimBase = 0.9f; coreBase = 0.2f; break;
            case "Layer_Exosphere": rimBase = 0.6f; coreBase = 0.1f; break;
        }

        r.material.SetFloat("_RimIntensity", Mathf.Lerp(0f, rimBase, level));
        r.material.SetFloat("_CoreIntensity", Mathf.Lerp(0f, coreBase, level));
    }

    public void SetLayerActive(string layerName, bool active)
    {
        Transform layer = transform.Find(layerName);
        if (layer != null) layer.gameObject.SetActive(active);
    }
    public void SetIntensity(float level)
    {
        SetLayerIntensity("Layer_Troposphere", level);
        SetLayerIntensity("Layer_Mesosphere", level);
        SetLayerIntensity("Layer_Thermosphere", level);
        SetLayerIntensity("Layer_Exosphere", level);
    }
}