using Sirenix.OdinInspector;
using Sirenix.OdinInspector.Editor;
using Sirenix.Utilities;
using Sirenix.Utilities.Editor;
using System;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;

public class SillyComponent : MonoBehaviour
{
    [GridView(5)]
    public List<SomeClass> Test;
}

[Serializable, HideLabel]
public class SomeClass
{
    [HideInInspector]
    public string Name;

    [BoxGroup("$Name"), PreviewField(Sirenix.OdinInspector.ObjectFieldAlignment.Center), HideLabel]
    public Texture tex;
}


[DontApplyToListElements]
public class GridViewAttribute : Attribute
{
    public int ColumnCount;

    public GridViewAttribute(int columnCount)
    {
        this.ColumnCount = columnCount;
    }
}

public class SomeClassListDrawer : OdinAttributeDrawer<GridViewAttribute>
{
    private int colWidth;
    private GUITableRowLayoutGroup table; // Making lots of GUILayout groups in order to make a grid view is not very performant. The utility helps os minimize the number of layout groups needed.
    private LocalPersistentContext<bool> isVisible;
    private bool drawAsList;

    protected override bool CanDrawAttributeProperty(InspectorProperty property)
    {
        return property.ChildResolver is IOrderedCollectionResolver;
    }

    protected override void Initialize()
    {
        this.table = new GUITableRowLayoutGroup();
        this.table.MinScrollViewHeight = 600;
        this.table.MaxScrollViewHeight = 600;

        this.isVisible = this.GetPersistentValue("toggle", GeneralDrawerConfig.Instance.ExpandFoldoutByDefault);
        this.colWidth = 40;
    }

    protected override void DrawPropertyLayout(GUIContent label)
    {

        if (this.drawAsList)
        {
            if (GUILayout.Button("Draw Grid View"))
            {
                this.drawAsList = false;
            }

            this.CallNextDrawer(label);
            return;
        }

        var count = this.Property.Children.Count;
        var colCount = 5; // Make this dynamic based on width?
        var rowCount = Mathf.CeilToInt(count / (float)colCount);
        var colOffset = 0;
        var colWidth = (int)(this.Property.LastDrawnValueRect.width / colCount);

        var rect = SirenixEditorGUI.BeginIndentedVertical();
        EditorGUI.DrawRect(rect, SirenixGUIStyles.DarkEditorBackground);
        this.DrawToolbar(label);
        GUILayout.BeginVertical(SirenixGUIStyles.ContentPadding);
        this.table.BeginTable(rowCount);
        {
            for (int colIndex = 0; colIndex < colCount; colIndex++)
            {
                this.table.BeginColumn(colOffset, colWidth);
                for (int rowIndex = 0; rowIndex < rowCount; rowIndex++)
                {
                    var i = rowIndex * colCount + colIndex;

                    if (i < count)
                    {
                        this.table.BeginCell(rowIndex);
                        this.Property.Children[i].Draw();
                        this.table.EndCell(rowIndex);
                    }
                }
                this.table.EndColumn();
                colOffset += colWidth;
            }
        }
        this.table.EndTable();
        GUILayout.EndVertical();
        rect.width += 1;
        SirenixEditorGUI.DrawBorders(rect, 1);
        SirenixEditorGUI.EndIndentedVertical();
    }

    private void DrawToolbar(GUIContent label)
    {
        const int iconBtnSize = 23;
        var rect = GUILayoutUtility.GetRect(0, 22);

        // Draw background
        if (Event.current.type == EventType.Repaint)
        {
            SirenixGUIStyles.ToolbarBackground.Draw(rect, GUIContent.none, 0);
        }

        // Draw as list toggle
        var btnRect = rect.AlignRight(iconBtnSize);
        rect.xMax = btnRect.xMin;
        if (GUI.Button(btnRect, GUIContent.none, SirenixGUIStyles.ToolbarButton))
        {
            this.drawAsList = !this.drawAsList;
        }

        EditorIcons.HamburgerMenu.Draw(btnRect, 13);

        // Label
        if (label != null)
        {
            var labelRect = rect;
            labelRect.x += 5;
            labelRect.y += 3;
            labelRect.height = 16;
            if (this.Property.Children.Count > 0)
            {
                GUIHelper.PushHierarchyMode(false);
                this.isVisible.Value = SirenixEditorGUI.Foldout(labelRect, this.isVisible.Value, label);
                GUIHelper.PushHierarchyMode(true);
            }
        }
    }
}