暂无描述
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

AssetMenuOperations.cs 7.4KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224
  1. using System;
  2. using UnityEditor.VersionControl;
  3. using Codice;
  4. using Codice.Client.Commands.WkTree;
  5. using Unity.PlasticSCM.Editor.AssetsOverlays.Cache;
  6. using Unity.PlasticSCM.Editor.AssetsOverlays;
  7. using Unity.PlasticSCM.Editor.AssetUtils;
  8. namespace Unity.PlasticSCM.Editor.AssetMenu
  9. {
  10. [Flags]
  11. internal enum AssetMenuOperations : byte
  12. {
  13. None = 0,
  14. Checkout = 1 << 0,
  15. Diff = 1 << 1,
  16. History = 1 << 2,
  17. Add = 1 << 3,
  18. Checkin = 1 << 4,
  19. Undo = 1 << 5,
  20. }
  21. internal class SelectedAssetGroupInfo
  22. {
  23. internal int SelectedCount;
  24. internal bool IsControlledSelection;
  25. internal bool IsCheckedInSelection;
  26. internal bool IsCheckedOutSelection;
  27. internal bool IsPrivateSelection;
  28. internal bool IsAddedSelection;
  29. internal bool IsFileSelection;
  30. internal bool HasAnyAddedInSelection;
  31. internal bool HasAnyRemoteLockedInSelection;
  32. internal static SelectedAssetGroupInfo BuildFromAssetList(
  33. string wkPath,
  34. AssetList assetList,
  35. IAssetStatusCache statusCache)
  36. {
  37. bool isCheckedInSelection = true;
  38. bool isControlledSelection = true;
  39. bool isCheckedOutSelection = true;
  40. bool isPrivateSelection = true;
  41. bool isAddedSelection = true;
  42. bool isFileSelection = true;
  43. bool hasAnyAddedInSelection = false;
  44. bool hasAnyRemoteLockedInSelection = false;
  45. int selectedCount = 0;
  46. foreach (Asset asset in assetList)
  47. {
  48. string fullPath = AssetsPath.GetFullPathUnderWorkspace.
  49. ForAsset(wkPath, asset.path);
  50. if (fullPath == null)
  51. continue;
  52. SelectedAssetGroupInfo singleFileGroupInfo = BuildFromSingleFile(
  53. fullPath, asset.isFolder, statusCache);
  54. if (!singleFileGroupInfo.IsCheckedInSelection)
  55. isCheckedInSelection = false;
  56. if (!singleFileGroupInfo.IsControlledSelection)
  57. isControlledSelection = false;
  58. if (!singleFileGroupInfo.IsCheckedOutSelection)
  59. isCheckedOutSelection = false;
  60. if (!singleFileGroupInfo.IsPrivateSelection)
  61. isPrivateSelection = false;
  62. if (!singleFileGroupInfo.IsAddedSelection)
  63. isAddedSelection = false;
  64. if (!singleFileGroupInfo.IsFileSelection)
  65. isFileSelection = false;
  66. if (singleFileGroupInfo.HasAnyAddedInSelection)
  67. hasAnyAddedInSelection = true;
  68. if (singleFileGroupInfo.HasAnyRemoteLockedInSelection)
  69. hasAnyRemoteLockedInSelection = true;
  70. selectedCount++;
  71. }
  72. return new SelectedAssetGroupInfo()
  73. {
  74. IsCheckedInSelection = isCheckedInSelection,
  75. IsCheckedOutSelection = isCheckedOutSelection,
  76. IsControlledSelection = isControlledSelection,
  77. IsPrivateSelection = isPrivateSelection,
  78. IsAddedSelection = isAddedSelection,
  79. IsFileSelection = isFileSelection,
  80. HasAnyAddedInSelection = hasAnyAddedInSelection,
  81. HasAnyRemoteLockedInSelection = hasAnyRemoteLockedInSelection,
  82. SelectedCount = selectedCount,
  83. };
  84. }
  85. internal static SelectedAssetGroupInfo BuildFromSingleFile(
  86. string fullPath,
  87. bool isDirectory,
  88. IAssetStatusCache statusCache)
  89. {
  90. bool isCheckedInSelection = true;
  91. bool isControlledSelection = true;
  92. bool isCheckedOutSelection = true;
  93. bool isPrivateSelection = true;
  94. bool isAddedSelection = true;
  95. bool isFileSelection = true;
  96. bool hasAnyAddedInSelection = false;
  97. bool hasAnyRemoteLockedInSelection = false;
  98. WorkspaceTreeNode wkTreeNode =
  99. PlasticGui.Plastic.API.GetWorkspaceTreeNode(fullPath);
  100. if (isDirectory)
  101. isFileSelection = false;
  102. if (CheckWorkspaceTreeNodeStatus.IsPrivate(wkTreeNode))
  103. isControlledSelection = false;
  104. else
  105. isPrivateSelection = false;
  106. if (CheckWorkspaceTreeNodeStatus.IsCheckedOut(wkTreeNode))
  107. isCheckedInSelection = false;
  108. else
  109. isCheckedOutSelection = false;
  110. if (CheckWorkspaceTreeNodeStatus.IsAdded(wkTreeNode))
  111. hasAnyAddedInSelection = true;
  112. else
  113. isAddedSelection = false;
  114. AssetStatus assetStatus = statusCache.GetStatus(fullPath);
  115. if (ClassifyAssetStatus.IsLockedRemote(assetStatus))
  116. hasAnyRemoteLockedInSelection = true;
  117. return new SelectedAssetGroupInfo()
  118. {
  119. IsCheckedInSelection = isCheckedInSelection,
  120. IsCheckedOutSelection = isCheckedOutSelection,
  121. IsControlledSelection = isControlledSelection,
  122. IsPrivateSelection = isPrivateSelection,
  123. IsAddedSelection = isAddedSelection,
  124. IsFileSelection = isFileSelection,
  125. HasAnyAddedInSelection = hasAnyAddedInSelection,
  126. HasAnyRemoteLockedInSelection = hasAnyRemoteLockedInSelection,
  127. SelectedCount = 1,
  128. };
  129. }
  130. }
  131. internal interface IAssetMenuOperations
  132. {
  133. void ShowPendingChanges();
  134. void Add();
  135. void Checkout();
  136. void Checkin();
  137. void Undo();
  138. void ShowDiff();
  139. void ShowHistory();
  140. }
  141. internal static class AssetMenuUpdater
  142. {
  143. internal static AssetMenuOperations GetAvailableMenuOperations(
  144. SelectedAssetGroupInfo info)
  145. {
  146. AssetMenuOperations result = AssetMenuOperations.None;
  147. if (info.SelectedCount == 0)
  148. {
  149. return result;
  150. }
  151. if (info.IsControlledSelection &&
  152. info.IsCheckedInSelection &&
  153. info.IsFileSelection &&
  154. !info.HasAnyRemoteLockedInSelection)
  155. {
  156. result |= AssetMenuOperations.Checkout;
  157. }
  158. if (info.IsFileSelection &&
  159. info.IsPrivateSelection)
  160. {
  161. result |= AssetMenuOperations.Add;
  162. }
  163. if (info.IsFileSelection &&
  164. info.IsControlledSelection &&
  165. info.IsCheckedOutSelection)
  166. {
  167. result |= AssetMenuOperations.Checkin;
  168. result |= AssetMenuOperations.Undo;
  169. }
  170. if (info.SelectedCount == 1 &&
  171. info.IsControlledSelection &&
  172. !info.HasAnyAddedInSelection &&
  173. info.IsFileSelection)
  174. {
  175. result |= AssetMenuOperations.Diff;
  176. }
  177. if (info.SelectedCount == 1 &&
  178. info.IsControlledSelection &&
  179. !info.HasAnyAddedInSelection)
  180. {
  181. result |= AssetMenuOperations.History;
  182. }
  183. return result;
  184. }
  185. }
  186. }