aboutsummaryrefslogtreecommitdiffstats
path: root/Software/Visual_Studio/Tango.Web/TangoWebClientv2.cs
blob: 3a4ff7756b0138901d61d40ab33c0115cb19570b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Authentication;
using System.Text;
using System.Threading.Tasks;
using Tango.Transport.Web;
using Tango.Web.Security;

namespace Tango.Web
{
    public abstract class TangoWebClientV2<TLoginRequest, TLoginResponse> : IDisposable where TLoginRequest : WebRequestMessage where TLoginResponse : WebTokenResponse
    {
        private WebTransportClient _client;
        private bool _disposed;
        private TLoginRequest _lastLoginRequest;

        public String Address { get; set; }
        public String Controller { get; private set; }
        public String Token { get; private set; }
        public WebToken WebToken { get; private set; }
        public bool IsAuthenticated { get; private set; }

        public TimeSpan RequestTimeout
        {
            get { return _client.RequestTimeout; }
            set { _client.RequestTimeout = value; }
        }

        public TangoWebClientV2(String address, String controller)
        {
            _client = new WebTransportClient();
            Controller = controller;
            Address = address;
        }

        public TangoWebClientV2(String address, String controller, String token) : this(address, controller)
        {
            Token = token;
        }

        public TangoWebClientV2(TangoWebClientV2<TLoginRequest, TLoginResponse> cloned)
        {
            _client = new WebTransportClient();
            Address = cloned.Address;
            Controller = cloned.Controller;
            IsAuthenticated = cloned.IsAuthenticated;
            RequestTimeout = cloned.RequestTimeout;
            WebToken = cloned.WebToken;
            _lastLoginRequest = cloned._lastLoginRequest;
            Token = cloned.Token;
        }

        public T Clone<T>() where T : TangoWebClientV2<TLoginRequest, TLoginResponse>
        {
            return Activator.CreateInstance(typeof(T), new object[] { this }) as T;
        }

        public async Task<TLoginResponse> Login(TLoginRequest request)
        {
            var response = await _client.PostJson<TLoginRequest, TLoginResponse>(GetActionAddress("Login"), request);
            Token = response.AccessToken;
            _client.AuthenticationToken = Token;

            WebToken = WebToken.FromToken(Token);

            _lastLoginRequest = request;
            IsAuthenticated = true;
            return response;
        }

        protected virtual async Task<TResponse> Post<TRequest, TResponse>(String action, TRequest request) where TRequest : class, IWebRequestMessage where TResponse : class, IWebResponseMessage
        {
            if (IsAuthenticated)
            {
                if (DateTime.UtcNow >= WebToken.Expiration)
                {
                    await Login(_lastLoginRequest);
                }
            }

            try
            {
                var response = await _client.PostJson<TRequest, TResponse>(GetActionAddress(action), request);
                return response;
            }
            catch (TokenExpiredException ex)
            {
                try
                {
                    await Login(_lastLoginRequest);
                    var response = await _client.PostJson<TRequest, TResponse>(GetActionAddress(action), request);
                    return response;
                }
                catch
                {
                    throw;
                }
            }
            catch (Exception ex)
            {
                throw;
            }
        }

        private String GetActionAddress(String action)
        {
            return GetServiceAddress() + action;
        }

        protected virtual String GetServiceAddress()
        {
            return Address + $"/api/{Controller}/";
        }

        public virtual void Dispose()
        {
            if (!_disposed)
            {
                _disposed = true;
                _client.Dispose();
            }
        }
    }
}