84 lines
2.2 KiB
C#
84 lines
2.2 KiB
C#
|
using System;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Linq;
|
||
|
|
|
||
|
|
using UnityEditor;
|
||
|
|
using UnityEditor.IMGUI.Controls;
|
||
|
|
#if UNITY_6000_2_OR_NEWER
|
||
|
|
using TreeView = UnityEditor.IMGUI.Controls.TreeView<int>;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
using PlasticGui;
|
||
|
|
|
||
|
|
namespace Unity.PlasticSCM.Editor.UI.Tree
|
||
|
|
{
|
||
|
|
internal class TreeViewSessionState
|
||
|
|
{
|
||
|
|
internal static void Save(
|
||
|
|
TreeView treeView,
|
||
|
|
string uncheckedKey)
|
||
|
|
{
|
||
|
|
var rows = treeView.GetRows();
|
||
|
|
|
||
|
|
if (!rows.Any())
|
||
|
|
return;
|
||
|
|
|
||
|
|
List<string> uncheckedRows = new List<string>();
|
||
|
|
|
||
|
|
for (int i = 0; i < rows.Count; i++)
|
||
|
|
{
|
||
|
|
bool? isChecked = CheckableItems.GetIsCheckedValue(
|
||
|
|
rows[i].GetPlasticTreeNode());
|
||
|
|
|
||
|
|
if (!isChecked.HasValue)
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (string.IsNullOrEmpty(rows[i].displayName))
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (!isChecked.Value)
|
||
|
|
uncheckedRows.Add(rows[i].displayName);
|
||
|
|
}
|
||
|
|
|
||
|
|
SessionState.SetString(uncheckedKey, string.Join(":", uncheckedRows));
|
||
|
|
}
|
||
|
|
|
||
|
|
internal static void Restore(
|
||
|
|
TreeView treeView,
|
||
|
|
string uncheckedKey)
|
||
|
|
{
|
||
|
|
var rows = treeView.GetRows();
|
||
|
|
|
||
|
|
if (!rows.Any())
|
||
|
|
return;
|
||
|
|
|
||
|
|
string uncheckedRows = SessionState.GetString(
|
||
|
|
uncheckedKey, string.Empty);
|
||
|
|
|
||
|
|
if (string.IsNullOrEmpty(uncheckedRows))
|
||
|
|
return;
|
||
|
|
|
||
|
|
string[] uncheckedArray = uncheckedRows.Split(':');
|
||
|
|
|
||
|
|
for (int i = 0; i < rows.Count; i++)
|
||
|
|
{
|
||
|
|
if (string.IsNullOrEmpty(rows[i].displayName))
|
||
|
|
continue;
|
||
|
|
|
||
|
|
if (uncheckedArray.Contains(rows[i].displayName))
|
||
|
|
{
|
||
|
|
CheckableItems.SetCheckedValue(
|
||
|
|
rows[i].GetPlasticTreeNode(), false);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
CheckableItems.SetCheckedValue(
|
||
|
|
rows[i].GetPlasticTreeNode(), true);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Clear session state after the every update
|
||
|
|
SessionState.EraseString(uncheckedKey);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|