using NBShader;
using UnityEditor;
using UnityEngine;

namespace NBShaderEditor
{
    public enum FxLightMode
    {
        UnLit = 0,
        BlinnPhong = 1,
        HalfLambert = 2,
        PBR = 3,
        SixWay = 4,
        UnKnownOrMixedValue = -1
    }

    public class LightBigBlockItem : BigBlockItem
    {
        private readonly NBShaderRootItem _nbRootItem;
        private readonly FxLightModePopupItem _lightModeItem;
        private readonly ToggleItem _specularToggleItem;
        private readonly ColorItem _specularColorItem;
        private readonly VectorComponentItem _specularSmoothnessItem;
        private readonly VectorComponentItem _pbrMetallicItem;
        private readonly VectorComponentItem _pbrSmoothnessItem;
        private readonly PropertyToggleBlockItem _bumpBlock;
        private readonly PropertyToggleBlockItem _matCapBlock;
        private readonly TextureItem _sixWayPositiveItem;
        private readonly TextureItem _sixWayNegativeItem;
        private readonly ToggleItem _sixWayAbsorptionToggleItem;
        private readonly VectorComponentItem _sixWayAbsorptionStrengthItem;
        private readonly TextureItem _sixWayEmissionRampItem;
        private readonly VectorComponentItem _sixWayEmissionPowItem;
        private readonly ColorItem _sixWayEmissionColorItem;

        public LightBigBlockItem(NBShaderRootItem rootItem, ShaderGUIItem parentItem)
            : base(
                rootItem,
                parentItem,
                "_LightBigBlockItemFoldOut",
                () => Content("block.light", "Light", "Normal, MatCap and light mode controls"))
        {
            _nbRootItem = rootItem;
            _lightModeItem = new FxLightModePopupItem(rootItem, this);

            _specularToggleItem = new NBShaderKeywordToggleItem(
                rootItem,
                this,
                "_BlinnPhongSpecularToggle",
                "_SPECULAR_COLOR",
                () => Content("light.specular.toggle", "Specular"),
                isVisible: () => IsBlinnOrHalf(rootItem));

            _specularColorItem = new ColorItem(
                rootItem,
                this,
                "_SpecularColor",
                () => Content("light.specular.color", "Specular Color"),
                () => IsToggleOn(rootItem, "_BlinnPhongSpecularToggle") && IsBlinnOrHalf(rootItem));

            _specularSmoothnessItem = new VectorComponentItem(
                rootItem,
                this,
                "_MaterialInfo",
                1,
                () => Content("light.specular.smoothness", "Smoothness"),
                true,
                0f,
                1f,
                () => IsToggleOn(rootItem, "_BlinnPhongSpecularToggle") && IsBlinnOrHalf(rootItem));

            _pbrMetallicItem = new VectorComponentItem(
                rootItem,
                this,
                "_MaterialInfo",
                0,
                () => Content("light.pbr.metallic", "Metallic"),
                true,
                0f,
                1f,
                () => rootItem.Context.FxLightMode == FxLightMode.PBR);

            _pbrSmoothnessItem = new VectorComponentItem(
                rootItem,
                this,
                "_MaterialInfo",
                1,
                () => Content("light.pbr.smoothness", "Smoothness"),
                true,
                0f,
                1f,
                () => rootItem.Context.FxLightMode == FxLightMode.PBR);

            _bumpBlock = new PropertyToggleBlockItem(
                rootItem,
                this,
                "_BumpToggleFoldOut",
                "_BumpMapToggle",
                () => Content("light.bump.toggle", "Normal Map"),
                keyword: "_NORMALMAP",
                isVisible: () => rootItem.Context.FxLightMode != FxLightMode.SixWay);

            new TextureItem(
                rootItem,
                _bumpBlock,
                "_BumpTex",
                () => Content("light.bump.texture", "Normal Map"),
                drawScaleOffset: true);
            TextureRelatedFoldOutItem bumpTexRelatedFoldOut = new TextureRelatedFoldOutItem(
                rootItem,
                _bumpBlock,
                "_BumpTexFoldOut",
                "_BumpTex",
                () => Content("light.bump.related", "Normal Map Related"));
            new WrapModeItem(rootItem, bumpTexRelatedFoldOut, NBShaderFlags.FLAG_BIT_WRAPMODE_BUMPTEX, () => Content("light.bump.wrap", "Normal Map Wrap"));
            new UVModeSelectItem(
                rootItem,
                bumpTexRelatedFoldOut,
                "_BumpUVModeFoldOut",
                NBShaderFlags.FLAG_BIT_UVMODE_POS_0_BUMPMAP,
                0,
                () => Content("light.bump.uvmode", "Normal Map UV Source"),
                "_BumpTex");
            new ToggleItem(
                rootItem,
                bumpTexRelatedFoldOut,
                "_BumpMapMaskMode",
                () => Content("light.bump.maskMode", "Normal Map Multi Channel"),
                enabled => rootItem.SyncService.ApplyToggleFlag(NBShaderFlags.FLAG_BIT_PARTICLE_NORMALMAP_MASK_MODE, enabled));
            ShaderGUISliderItem bumpScaleItem = new ShaderGUISliderItem(rootItem, bumpTexRelatedFoldOut)
            {
                PropertyName = "_BumpScale",
                GuiContent = Content("light.bump.scale", "Normal Strength"),
                RangePropertyName = "BumpScaleRangeVec"
            };
            bumpScaleItem.InitTriggerByChild();

            _matCapBlock = new PropertyToggleBlockItem(
                rootItem,
                this,
                "_MatCapFoldOut",
                "_MatCapToggle",
                () => Content("light.matcap.toggle", "MatCap"),
                keyword: "_MATCAP",
                isVisible: () => rootItem.Context.FxLightMode != FxLightMode.SixWay);
            new TextureItem(rootItem, _matCapBlock, "_MatCapTex", () => Content("light.matcap.texture", "MatCap Texture"), "_MatCapColor", false);
            new VectorComponentItem(rootItem, _matCapBlock, "_MatCapInfo", 0, () => Content("light.matcap.blend", "Add/Multiply Blend"), true, 0f, 1f);

            _sixWayPositiveItem = new TextureItem(
                rootItem,
                this,
                "_RigRTBk",
                () => Content("light.sixway.positive", "SixWay Positive"),
                drawScaleOffset: false,
                isVisible: IsSixWay);

            _sixWayNegativeItem = new TextureItem(
                rootItem,
                this,
                "_RigLBtF",
                () => Content("light.sixway.negative", "SixWay Negative"),
                drawScaleOffset: false,
                isVisible: IsSixWay);

            _sixWayAbsorptionToggleItem = new NBShaderKeywordToggleItem(
                rootItem,
                this,
                "_SixWayColorAbsorptionToggle",
                "VFX_SIX_WAY_ABSORPTION",
                () => Content("light.sixway.absorption.toggle", "Light Color Absorption"),
                isVisible: IsSixWay);

            _sixWayAbsorptionStrengthItem = new VectorComponentItem(
                rootItem,
                this,
                "_SixWayInfo",
                0,
                () => Content("light.sixway.absorption.strength", "Absorption Strength"),
                true,
                0f,
                1f,
                () => IsSixWay() && IsToggleOn(rootItem, "_SixWayColorAbsorptionToggle"));

            _sixWayEmissionRampItem = new TextureItem(
                rootItem,
                this,
                "_SixWayEmissionRamp",
                () => Content("light.sixway.ramp", "SixWay Emission Ramp"),
                drawScaleOffset: false,
                afterDraw: SyncSixWayRampFlag,
                isVisible: IsSixWay);

            _sixWayEmissionPowItem = new VectorComponentItem(
                rootItem,
                this,
                "_SixWayInfo",
                1,
                () => Content("light.sixway.emissionPow", "SixWay Emission Pow"),
                false,
                isVisible: IsSixWay);

            _sixWayEmissionColorItem = new ColorItem(
                rootItem,
                this,
                "_SixWayEmissionColor",
                () => Content("light.sixway.color", "SixWay Emission Color"),
                IsSixWay);

            InitTriggerByChild();
        }

        public override void DrawBlock()
        {
            if (IsAnyLightModeAllowed())
            {
                _lightModeItem.OnGUI();
            }

            if (IsTierAllowed("_SPECULAR_COLOR"))
            {
                _specularToggleItem.OnGUI();
                _specularColorItem.OnGUI();
                _specularSmoothnessItem.OnGUI();
            }

            if (IsTierAllowed("_FX_LIGHT_MODE_PBR"))
            {
                _pbrMetallicItem.OnGUI();
                _pbrSmoothnessItem.OnGUI();
            }

            if (IsTierAllowed("_NORMALMAP"))
            {
                _bumpBlock.OnGUI();
            }

            if (IsTierAllowed("_MATCAP"))
            {
                _matCapBlock.OnGUI();
            }

            if (IsTierAllowed("_FX_LIGHT_MODE_SIX_WAY"))
            {
                _sixWayPositiveItem.OnGUI();
                _sixWayNegativeItem.OnGUI();
                DrawSixWayWarning();
                _sixWayEmissionRampItem.OnGUI();
                _sixWayEmissionPowItem.OnGUI();
                _sixWayEmissionColorItem.OnGUI();
            }

            if (IsTierAllowed("VFX_SIX_WAY_ABSORPTION"))
            {
                _sixWayAbsorptionToggleItem.OnGUI();
                _sixWayAbsorptionStrengthItem.OnGUI();
            }
        }

        private bool IsTierAllowed(string keyword)
        {
            return _nbRootItem.Context == null || _nbRootItem.Context.IsKeywordAllowed(keyword);
        }

        private bool IsAnyLightModeAllowed()
        {
            return _nbRootItem.Context == null ||
                   _nbRootItem.Context.IsAnyKeywordAllowed(
                       "_FX_LIGHT_MODE_UNLIT",
                       "_FX_LIGHT_MODE_BLINN_PHONG",
                       "_FX_LIGHT_MODE_HALF_LAMBERT",
                       "_FX_LIGHT_MODE_PBR",
                       "_FX_LIGHT_MODE_SIX_WAY");
        }

        private bool IsSixWay()
        {
            return _nbRootItem.Context.FxLightMode == FxLightMode.SixWay;
        }

        private void DrawSixWayWarning()
        {
            if (!IsSixWay())
            {
                return;
            }

            DrawLayoutHelpBox(
                NBShaderInspectorLocalization.GetInspectorText(
                    "light.sixway.uvWarning.message",
                    "六路UV跟随主贴图UV及颜色"),
                MessageType.Warning);
        }

        private void SyncSixWayRampFlag(MaterialProperty rampProperty)
        {
            if (rampProperty.hasMixedValue)
            {
                return;
            }

            _nbRootItem.SyncService.ApplyToggleFlag(
                NBShaderFlags.FLAG_BIT_PARTICLE_1_SIXWAY_RAMPMAP,
                rampProperty.textureValue != null,
                1);
        }

        private static bool IsToggleOn(NBShaderRootItem rootItem, string propertyName)
        {
            return rootItem.Context.IsToggleOn(propertyName);
        }

        private static bool IsBlinnOrHalf(NBShaderRootItem rootItem)
        {
            return rootItem.Context.FxLightMode == FxLightMode.BlinnPhong ||
                   rootItem.Context.FxLightMode == FxLightMode.HalfLambert;
        }

        private static GUIContent Content(string key, string fallback, string tip = "")
        {
            return NBShaderInspectorLocalization.MakeInspectorContent(key, fallback, tip);
        }
    }

    public class FxLightModePopupItem : ShaderGUIPopUpItem
    {
        private static readonly string[] LightModeOptions =
        {
            "Unlit",
            "BlinnPhong",
            "HalfLambert",
            "PBR",
            "SixWay"
        };

        private readonly NBShaderRootItem _nbRootItem;

        public FxLightModePopupItem(NBShaderRootItem rootItem, ShaderGUIItem parentItem) : base(rootItem, parentItem)
        {
            _nbRootItem = rootItem;
            PropertyName = "_FxLightMode";
            GuiContent = NBShaderInspectorLocalization.MakeContent("inspector.light.mode.label", "Light Mode");
            PopUpNames = NBShaderInspectorLocalization.GetInspectorOptions("light.mode", LightModeOptions);
            InitTriggerByChild();
        }

        public override void OnGUI()
        {
            base.OnGUI();
        }

        public override void OnEndChange()
        {
            base.OnEndChange();
            _nbRootItem.SyncService.ApplyLightMode((FxLightMode)PropertyInfo.Property.floatValue);
            _nbRootItem.Context.Refresh();
        }
    }
}