AutoLogin.cs 8.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. using Codice.Client.Common.Servers;
  2. using Codice.Client.Common.Threading;
  3. using Codice.LogWrapper;
  4. using PlasticGui.Configuration.CloudEdition.Welcome;
  5. using PlasticGui.WebApi;
  6. using System;
  7. using System.Collections;
  8. using System.Collections.Generic;
  9. using Unity.PlasticSCM.Editor.UI;
  10. using Unity.PlasticSCM.Editor.WebApi;
  11. using UnityEditor;
  12. using UnityEngine;
  13. namespace Unity.PlasticSCM.Editor.Configuration.CloudEdition.Welcome
  14. {
  15. internal class AutoLogin : OAuthSignIn.INotify
  16. {
  17. internal enum State : byte
  18. {
  19. Off = 0,
  20. Started = 1,
  21. Running = 2,
  22. ResponseInit = 3,
  23. ResponseEnd = 6,
  24. ResponseSuccess = 7,
  25. OrganizationChoosed = 8,
  26. InitializingPlastic = 9,
  27. ErrorNoToken = 20,
  28. ErrorTokenException = 21,
  29. ErrorResponseNull = 22,
  30. ErrorResponseError = 23,
  31. ErrorTokenEmpty = 24,
  32. ErrorResponseCancel = 25
  33. }
  34. void OAuthSignIn.INotify.SuccessForConfigure(
  35. List<string> organizations,
  36. bool canCreateAnOrganization,
  37. string userName,
  38. string accessToken)
  39. {
  40. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ResponseSuccess;
  41. ChooseOrganization(organizations, canCreateAnOrganization);
  42. }
  43. void OAuthSignIn.INotify.SuccessForSSO(string organization)
  44. {
  45. }
  46. void OAuthSignIn.INotify.SuccessForProfile(string email)
  47. {
  48. }
  49. void OAuthSignIn.INotify.SuccessForCredentials(
  50. string email,
  51. string accessToken)
  52. {
  53. }
  54. void OAuthSignIn.INotify.Cancel(string errorMessage)
  55. {
  56. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorResponseCancel;
  57. }
  58. internal void Run()
  59. {
  60. mPlasticWindow = GetPlasticWindow();
  61. if (!string.IsNullOrEmpty(CloudProjectSettings.accessToken))
  62. {
  63. ExchangeTokensAndJoinOrganization(CloudProjectSettings.accessToken);
  64. }
  65. else
  66. {
  67. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorNoToken;
  68. }
  69. }
  70. void ExchangeTokensAndJoinOrganization(string unityAccessToken)
  71. {
  72. int ini = Environment.TickCount;
  73. TokenExchangeResponse response = null;
  74. IThreadWaiter waiter = ThreadWaiter.GetWaiter(10);
  75. waiter.Execute(
  76. /*threadOperationDelegate*/ delegate
  77. {
  78. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ResponseInit;
  79. response = PlasticScmRestApiClient.TokenExchange(unityAccessToken);
  80. },
  81. /*afterOperationDelegate*/ delegate
  82. {
  83. mLog.DebugFormat(
  84. "TokenExchange time {0} ms",
  85. Environment.TickCount - ini);
  86. if (waiter.Exception != null)
  87. {
  88. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorTokenException;
  89. ExceptionsHandler.LogException(
  90. "TokenExchangeSetting",
  91. waiter.Exception);
  92. return;
  93. }
  94. if (response == null)
  95. {
  96. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorResponseNull;
  97. Debug.Log("response null");
  98. return;
  99. }
  100. if (response.Error != null)
  101. {
  102. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorResponseError;
  103. mLog.ErrorFormat(
  104. "Unable to exchange token: {0} [code {1}]",
  105. response.Error.Message, response.Error.ErrorCode);
  106. return;
  107. }
  108. if (string.IsNullOrEmpty(response.AccessToken))
  109. {
  110. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ErrorTokenEmpty;
  111. mLog.InfoFormat(
  112. "Access token is empty for user: {0}",
  113. response.User);
  114. return;
  115. }
  116. mPlasticWindow.GetWelcomeView().autoLoginState = AutoLogin.State.ResponseEnd;
  117. sAccessToken = response.AccessToken;
  118. sUserName = response.User;
  119. GetOrganizationList();
  120. });
  121. }
  122. internal void ExchangeTokens(string unityAccessToken)
  123. {
  124. int ini = Environment.TickCount;
  125. TokenExchangeResponse response = null;
  126. IThreadWaiter waiter = ThreadWaiter.GetWaiter(10);
  127. waiter.Execute(
  128. /*threadOperationDelegate*/ delegate
  129. {
  130. response = PlasticScmRestApiClient.TokenExchange(unityAccessToken);
  131. },
  132. /*afterOperationDelegate*/ delegate
  133. {
  134. mLog.DebugFormat(
  135. "TokenExchange time {0} ms",
  136. Environment.TickCount - ini);
  137. if (waiter.Exception != null)
  138. {
  139. ExceptionsHandler.LogException(
  140. "TokenExchangeSetting",
  141. waiter.Exception);
  142. return;
  143. }
  144. if (response == null)
  145. {
  146. Debug.Log("response null");
  147. return;
  148. }
  149. if (response.Error != null)
  150. {
  151. mLog.ErrorFormat(
  152. "Unable to exchange token: {0} [code {1}]",
  153. response.Error.Message,
  154. response.Error.ErrorCode);
  155. return;
  156. }
  157. if (string.IsNullOrEmpty(response.AccessToken))
  158. {
  159. mLog.InfoFormat(
  160. "Access token is empty for user: {0}",
  161. response.User);
  162. return;
  163. }
  164. sAccessToken = response.AccessToken;
  165. sUserName = response.User;
  166. });
  167. }
  168. internal void GetOrganizationList()
  169. {
  170. OAuthSignIn.GetOrganizationsFromAccessToken(
  171. mPlasticWindow.PlasticWebRestApiForTesting,
  172. new Editor.UI.Progress.ProgressControlsForDialogs(),
  173. this,
  174. CloudProjectSettings.userName,
  175. sAccessToken
  176. );
  177. }
  178. PlasticWindow GetPlasticWindow()
  179. {
  180. var windows = Resources.FindObjectsOfTypeAll<PlasticWindow>();
  181. PlasticWindow plasticWindow = windows.Length > 0 ? windows[0] : null;
  182. if (plasticWindow == null)
  183. plasticWindow = ShowWindow.Plastic();
  184. return plasticWindow;
  185. }
  186. internal void ChooseOrganization(List<string> organizations,
  187. bool canCreateAnOrganization)
  188. {
  189. mPlasticWindow = GetPlasticWindow();
  190. CloudEditionWelcomeWindow.ShowWindow(
  191. mPlasticWindow.PlasticWebRestApiForTesting,
  192. mPlasticWindow.CmConnectionForTesting,null, true);
  193. mCloudEditionWelcomeWindow = CloudEditionWelcomeWindow.GetWelcomeWindow();
  194. mCloudEditionWelcomeWindow.FillUserAndToken(sUserName, sAccessToken);
  195. if (organizations.Count == 1)
  196. {
  197. mCloudEditionWelcomeWindow.JoinOrganizationAndWelcomePage(organizations[0]);
  198. return;
  199. }
  200. mCloudEditionWelcomeWindow.ShowOrganizationPanelFromAutoLogin(organizations, canCreateAnOrganization);
  201. }
  202. internal static string sAccessToken = string.Empty;
  203. internal static string sUserName = string.Empty;
  204. PlasticWindow mPlasticWindow;
  205. CloudEditionWelcomeWindow mCloudEditionWelcomeWindow;
  206. static readonly ILog mLog = LogManager.GetLogger("TokensExchange");
  207. }
  208. }