123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443 |
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Net;
- using System.Net.Http;
- using InABox.Client.WebSocket;
- using InABox.Core;
- using InABox.WebSocket.Shared;
- using RestSharp;
- using RestSharp.Extensions;
- namespace InABox.Clients
- {
- internal static class LocalCache
- {
- public static string Password { get; set; }
- }
- public static class URLCache
- {
- private static string? Domain { get; set; }
- private static int? Port { get; set; }
- private static string URL { get; set; }
- private static bool isHTTPS;
- public static bool IsHTTPS { get => isHTTPS; }
- public static void Clear()
- {
- Domain = null;
- Port = null;
- URL = null;
- isHTTPS = false;
- }
- public static string GetURL(string domain, int port)
- {
- domain = domain.Split(new[] { "://" }, StringSplitOptions.RemoveEmptyEntries).Last();
- if (domain != Domain || port != Port)
- {
- var client = new HttpClient { BaseAddress = new Uri($"https://{domain}:{port}") };
- try
- {
- client.GetAsync("operations").Wait();
- URL = $"https://{domain}";
- isHTTPS = true;
- }
- catch (Exception)
- {
- URL = $"http://{domain}";
- isHTTPS = false;
- }
- Port = port;
- Domain = domain;
- }
- return URL;
- }
- }
- public static class WebSocketFactory
- {
- private static Dictionary<string, WebSocketClient> Clients = new Dictionary<string, WebSocketClient>();
- public static void StartWebSocket(string url, int port, Guid session)
- {
- url = url.Split(new[] { "://" }, StringSplitOptions.RemoveEmptyEntries).Last();
- var key = $"{url}:{port}${session}";
- if (!Clients.ContainsKey(key))
- {
- Clients[key] = new WebSocketClient(url, port, session);
- }
- }
- }
- public abstract class RemoteClient<TEntity> : BaseClient<TEntity> where TEntity : Entity, new()
- {
- public RemoteClient(string url, int port, bool useSimpleEncryption = false)
- {
- URL = URLCache.GetURL(url, port);
- Port = port;
- UseSimpleEncryption = useSimpleEncryption;
- }
- public string URL { get; set; }
- public int Port { get; set; }
- public bool UseSimpleEncryption { get; set; }
- private void PrepareRequest(Request request)
- {
- request.Credentials.Platform = ClientFactory.Platform;
- request.Credentials.Version = ClientFactory.Version;
- request.Credentials.Session = ClientFactory.SessionID;
- Request.BeforeRequest?.Invoke(request);
- }
- protected override ValidationData DoValidate(Guid session)
- {
- return Validate(
- null, null, false, session);
- }
- protected override ValidationData DoValidate(string pin, Guid session)
- {
- return Validate(
- null, pin, true, session);
- }
- protected override ValidationData DoValidate(string userid, string password, Guid session)
- {
- return Validate(
- userid, password, false, session);
- }
- private ValidationData Validate(string? userid, string? password, bool usePin, Guid session = default)
- {
- var ticks = DateTime.Now.ToUniversalTime().Ticks.ToString();
- var request = new ValidateRequest();
- request.UsePIN = usePin;
- if (usePin)
- {
- request.UserID = Encryption.Encrypt(ticks, "wCq9rryEJEuHIifYrxRjxg", UseSimpleEncryption);
- request.Password = Encryption.Encrypt(ticks, "7mhvLnqMwkCAzN+zNGlyyg", UseSimpleEncryption);
- request.PIN = password;
- }
- else
- {
- request.UserID = userid;
- request.Password = password;
- }
- PrepareRequest(request);
- if (session != Guid.Empty)
- {
- request.Credentials.Session = session;
- }
- var response = SendRequest<ValidateRequest, ValidateResponse>(request, "validate", false);
- if (response != null)
- if (response.Status.Equals(StatusCode.OK))
- {
- if(response.Session != Guid.Empty)
- {
- var notifyRequest = new NotifyRequest();
- // Session is required so that the server can exclude any requests from bad actors
- notifyRequest.Credentials.Session = response.Session;
- var notifyResponse = SendRequest<NotifyRequest, NotifyResponse>(notifyRequest, "notify", false);
- if(notifyResponse != null && notifyResponse.Status.Equals(StatusCode.OK))
- {
- if (notifyResponse.SocketPort.HasValue)
- {
- WebSocketFactory.StartWebSocket(URL, notifyResponse.SocketPort.Value, response.Session);
- }
- }
- }
- LocalCache.Password = password;
- return new ValidationData(
- response.ValidationResult,
- response.UserID,
- response.UserGuid,
- response.SecurityID,
- response.Session,
- response.Recipient2FA,
- response.PasswordExpiration
- );
- }
- else if(response.Status == StatusCode.BadServer)
- {
- throw new RemoteException(response.Messages);
- }
- return new ValidationData(
- ValidationResult.INVALID,
- "",
- Guid.Empty,
- Guid.Empty,
- Guid.Empty,
- null,
- DateTime.MinValue
- );
- }
- protected abstract TResponse SendRequest<TRequest, TResponse>(TRequest request, string Action, bool includeEntity = true)
- where TRequest : Request, new() where TResponse : Response, new();
- #region Query Data
- protected override CoreTable DoQuery(Filter<TEntity> filter, Columns<TEntity> columns, SortOrder<TEntity> sort = null)
- {
- var request = new QueryRequest<TEntity>
- {
- Columns = columns,
- Filter = filter,
- Sort = sort
- };
- PrepareRequest(request);
- var response = SendRequest<QueryRequest<TEntity>, QueryResponse<TEntity>>(request, "List");
- if (response != null)
- {
- return response.Status switch
- {
- StatusCode.OK => response.Items,
- StatusCode.Unauthenticated => throw new RemoteException("Client not authenticated", StatusCode.Unauthenticated),
- _ => throw new RemoteException(response.Messages),
- };
- }
- return null;
- //throw new Exception("Response is null");
- }
- #endregion
- #region Load
- protected override TEntity[] DoLoad(Filter<TEntity>? filter = null, SortOrder<TEntity>? sort = null)
- {
- var result = new List<TEntity>();
- var request = new QueryRequest<TEntity>
- {
- Filter = filter,
- Sort = sort
- };
- PrepareRequest(request);
- var response = SendRequest<QueryRequest<TEntity>, QueryResponse<TEntity>>(request, "List");
- if (response.Items != null)
- foreach (var row in response.Items.Rows)
- result.Add(row.ToObject<TEntity>());
- return result.ToArray();
- }
- #endregion
- #region MultipleTables
- protected override Dictionary<string, CoreTable> DoQueryMultiple(Dictionary<string, IQueryDef> queries)
- {
- var request = new MultiQueryRequest();
- request.TableTypes = new Dictionary<string, string>();
- request.Filters = new Dictionary<string, string>();
- request.Columns = new Dictionary<string, string>();
- request.Sorts = new Dictionary<string, string>();
- foreach (var item in queries)
- {
- request.TableTypes[item.Key] = item.Value.Type.EntityName();
- request.Filters[item.Key] = Serialization.Serialize(item.Value.Filter);
- request.Columns[item.Key] = Serialization.Serialize(item.Value.Columns);
- request.Sorts[item.Key] = Serialization.Serialize(item.Value.SortOrder);
- }
- PrepareRequest(request);
- var response = SendRequest<MultiQueryRequest, MultiQueryResponse>(request, "QueryMultiple", false);
- if (response != null)
- {
- return response.Status switch
- {
- StatusCode.OK => response.Tables,
- StatusCode.Unauthenticated => throw new RemoteException("Client not authenticated"),
- _ => throw new RemoteException(response.Messages),
- };
- }
- return null;
- //throw new Exception("Response is null");
- }
- #endregion
- #region Save
- protected override void DoSave(TEntity entity, string auditnote)
- {
- var request = new SaveRequest<TEntity>();
- request.Item = entity;
- request.AuditNote = auditnote;
- PrepareRequest(request);
- var response = SendRequest<SaveRequest<TEntity>, SaveResponse<TEntity>>(request, "Save");
- switch (response.Status)
- {
- case StatusCode.OK:
- var props = CoreUtils.PropertyList(typeof(TEntity), x => true, true);
- entity.SetObserving(false);
- foreach (var prop in props.Keys)
- {
- var value = CoreUtils.GetPropertyValue(response.Item, prop);
- CoreUtils.SetPropertyValue(entity, prop, value);
- }
- entity.CommitChanges();
- entity.SetObserving(true);
- break;
- case StatusCode.Unauthenticated:
- throw new RemoteException("Client not authenticated");
- default:
- throw new RemoteException(response.Messages);
- }
- }
- protected override void DoSave(IEnumerable<TEntity> entities, string auditnote)
- {
- var items = entities.ToArray();
- var request = new MultiSaveRequest<TEntity>();
- request.Items = items;
- request.AuditNote = auditnote;
- PrepareRequest(request);
- var response = SendRequest<MultiSaveRequest<TEntity>, MultiSaveResponse<TEntity>>(request, "MultiSave");
- switch (response.Status)
- {
- case StatusCode.OK:
- var props = CoreUtils.PropertyList(typeof(TEntity), x => true, true);
- for (var i = 0; i < items.Length; i++)
- {
- items[i].SetObserving(false);
- foreach (var prop in props.Keys)
- {
- var value = CoreUtils.GetPropertyValue(response.Items[i], prop);
- CoreUtils.SetPropertyValue(items[i], prop, value);
- }
- //CoreUtils.DeepClone<TEntity>(response.Items[i], items[i]);
- items[i].CommitChanges();
- items[i].SetObserving(true);
- }
- break;
- case StatusCode.Unauthenticated:
- throw new RemoteException("Client not authenticated");
- default:
- throw new RemoteException(response.Messages);
- }
- }
- #endregion
- #region Delete
- protected override void DoDelete(TEntity entity, string auditnote)
- {
- var request = new DeleteRequest<TEntity>();
- request.Item = entity;
- PrepareRequest(request);
- var response = SendRequest<DeleteRequest<TEntity>, DeleteResponse<TEntity>>(request, "Delete");
- switch (response.Status)
- {
- case StatusCode.OK:
- break;
- case StatusCode.Unauthenticated:
- throw new RemoteException("Client not authenticated");
- default:
- throw new RemoteException(response.Messages);
- }
- }
- protected override void DoDelete(IList<TEntity> entities, string auditnote)
- {
- var items = entities.ToArray();
- var request = new MultiDeleteRequest<TEntity>();
- request.Items = items;
- request.AuditNote = auditnote;
- PrepareRequest(request);
- var response = SendRequest<MultiDeleteRequest<TEntity>, MultiDeleteResponse<TEntity>>(request, "MultiDelete");
- switch (response.Status)
- {
- case StatusCode.OK:
- break;
- case StatusCode.Unauthenticated:
- throw new RemoteException("Client not authenticated");
- default:
- throw new RemoteException(response.Messages);
- }
- }
- #endregion
- #region 2FA
- protected override bool DoCheck2FA(string code, Guid? session)
- {
- var request = new Check2FARequest { Code = code };
- PrepareRequest(request);
- var response = SendRequest<Check2FARequest, Check2FAResponse>(request, "check_2fa", false);
- if (response != null)
- {
- return response.Status switch
- {
- StatusCode.OK => response.Valid,
- StatusCode.Unauthenticated => false,
- _ => throw new RemoteException(response.Messages),
- };
- }
- return false;
- }
- #endregion
- #region Ping
- protected override bool DoPing()
- {
- var uri = new Uri(string.Format("{0}:{1}", URL, Port));
- var cli = new RestClient(uri);
- var req = new RestRequest("/classes", Method.GET) { Timeout = 20000 };
- try
- {
- var res = cli.Execute(req);
- if (res.StatusCode != HttpStatusCode.OK || res.ErrorException != null)
- return false;
- return true;
- }
- catch (Exception e)
- {
- }
- return false;
- }
- #endregion
- }
- }
|