| | | 1 | | using System; |
| | | 2 | | using UnityEngine; |
| | | 3 | | |
| | | 4 | | namespace DCL.Helpers |
| | | 5 | | { |
| | | 6 | | public class Promise<T> : CustomYieldInstruction, IDisposable |
| | | 7 | | { |
| | | 8 | | private bool resolved = false; |
| | | 9 | | private bool failed = false; |
| | | 10 | | |
| | | 11 | | private Action<T> onSuccess; |
| | | 12 | | private Action<string> onError; |
| | | 13 | | |
| | 116 | 14 | | public override bool keepWaiting => !resolved; |
| | 0 | 15 | | public T value { private set; get; } |
| | 0 | 16 | | public string error { private set; get; } |
| | | 17 | | |
| | | 18 | | public void Resolve(T result) |
| | | 19 | | { |
| | 283 | 20 | | failed = false; |
| | 283 | 21 | | resolved = true; |
| | 283 | 22 | | value = result; |
| | 283 | 23 | | error = null; |
| | 283 | 24 | | onSuccess?.Invoke(result); |
| | 26 | 25 | | } |
| | | 26 | | |
| | | 27 | | public void Reject(string errorMessage) |
| | | 28 | | { |
| | 5 | 29 | | failed = true; |
| | 5 | 30 | | resolved = true; |
| | 5 | 31 | | error = errorMessage; |
| | 5 | 32 | | onError?.Invoke(error); |
| | 2 | 33 | | } |
| | | 34 | | |
| | | 35 | | public Promise<T> Then(Action<T> successCallback) |
| | | 36 | | { |
| | 574 | 37 | | if (!resolved) |
| | | 38 | | { |
| | 317 | 39 | | onSuccess = successCallback; |
| | 317 | 40 | | } |
| | 257 | 41 | | else if (!failed) |
| | | 42 | | { |
| | 256 | 43 | | successCallback?.Invoke(value); |
| | | 44 | | } |
| | | 45 | | |
| | 574 | 46 | | return this; |
| | | 47 | | } |
| | | 48 | | |
| | | 49 | | public void Catch(Action<string> errorCallback) |
| | | 50 | | { |
| | 167 | 51 | | if (!resolved) |
| | | 52 | | { |
| | 104 | 53 | | onError = errorCallback; |
| | 104 | 54 | | } |
| | 63 | 55 | | else if (failed) |
| | | 56 | | { |
| | 1 | 57 | | errorCallback?.Invoke(this.error); |
| | | 58 | | } |
| | 63 | 59 | | } |
| | | 60 | | |
| | | 61 | | public void Dispose() |
| | | 62 | | { |
| | 139 | 63 | | onSuccess = null; |
| | 139 | 64 | | onError = null; |
| | 139 | 65 | | resolved = true; |
| | 139 | 66 | | } |
| | | 67 | | } |
| | | 68 | | } |