Request.cs 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175
  1. using System;
  2. using System.Collections.Generic;
  3. using System.IO;
  4. using System.Runtime.InteropServices.ComTypes;
  5. using System.Text.Json;
  6. using System.Text.Json.Serialization;
  7. using InABox.Core;
  8. namespace InABox.Clients
  9. {
  10. public class Credentials : ISerializeBinary
  11. {
  12. [Obsolete]
  13. public string? UserID { get; set; }
  14. [Obsolete]
  15. public string? Password { get; set; }
  16. public Platform Platform { get; set; }
  17. public string? Version { get; set; }
  18. public Guid Session { get; set; }
  19. public void SerializeBinary(CoreBinaryWriter writer)
  20. {
  21. writer.Write(UserID ?? string.Empty);
  22. writer.Write(Password ?? string.Empty);
  23. writer.Write(Platform.ToString());
  24. writer.Write(Version ?? string.Empty);
  25. writer.Write(Session);
  26. }
  27. public void DeserializeBinary(CoreBinaryReader reader)
  28. {
  29. UserID = reader.ReadString();
  30. Password = reader.ReadString();
  31. if(Enum.TryParse<Platform>(reader.ReadString(), out var platform))
  32. {
  33. Platform = platform;
  34. }
  35. Version = reader.ReadString();
  36. Session = reader.ReadGuid();
  37. }
  38. }
  39. public abstract class Request
  40. {
  41. public Request()
  42. {
  43. Credentials = new Credentials();
  44. }
  45. public static Action<Request>? BeforeRequest { get; set; }
  46. public Credentials Credentials { get; set; }
  47. public abstract RequestMethod GetMethod();
  48. public virtual void SerializeBinary(CoreBinaryWriter writer)
  49. {
  50. Credentials.SerializeBinary(writer);
  51. }
  52. public virtual void DeserializeBinary(CoreBinaryReader reader)
  53. {
  54. Credentials.DeserializeBinary(reader);
  55. }
  56. }
  57. public enum StatusCode
  58. {
  59. OK,
  60. Incomplete,
  61. Error,
  62. Unauthenticated,
  63. BadServer
  64. }
  65. public enum RequestMethod
  66. {
  67. Info,
  68. Ping,
  69. Check2FA,
  70. Validate,
  71. MultiQuery,
  72. MultiDelete,
  73. Delete,
  74. Save,
  75. MultiSave,
  76. Query,
  77. Push,
  78. Version,
  79. ReleaseNotes,
  80. Installer
  81. }
  82. public abstract class Response
  83. {
  84. public Response()
  85. {
  86. Status = StatusCode.Incomplete;
  87. Messages = new List<string>();
  88. }
  89. public StatusCode Status { get; set; }
  90. public List<string> Messages { get; }
  91. public virtual void DeserializeBinary(CoreBinaryReader reader)
  92. {
  93. Status = (StatusCode)Enum.ToObject(typeof(StatusCode), reader.ReadInt32());
  94. Messages.Clear();
  95. var nMessages = reader.ReadInt32();
  96. for (int i = 0; i < nMessages; ++i)
  97. {
  98. Messages.Add(reader.ReadString());
  99. }
  100. }
  101. public virtual void SerializeBinary(CoreBinaryWriter writer)
  102. {
  103. writer.Write((int)Status);
  104. writer.Write(Messages.Count);
  105. foreach(var message in Messages)
  106. {
  107. writer.Write(message);
  108. }
  109. }
  110. }
  111. public abstract class BaseRequest<TEntity> : Request where TEntity : Entity, new()
  112. {
  113. }
  114. public abstract class BaseResponse<TEntity> : Response where TEntity : Entity, new()
  115. {
  116. }
  117. /*public class ListRequest<TEntity> : BaseRequest<TEntity> where TEntity : Entity, new()
  118. {
  119. public Filter<TEntity>? Filter { get; set; }
  120. public Columns<TEntity>? Columns { get; set; }
  121. public SortOrder<TEntity>? Sort { get; set; }
  122. public override RequestMethod GetMethod() => RequestMethod.Query;
  123. }
  124. public class ListResponse<TEntity> : BaseResponse<TEntity> where TEntity : Entity, new()
  125. {
  126. public IEnumerable<object[]> Data { get; set; }
  127. }*/
  128. public interface IQueryRequest : ISerializeBinary
  129. {
  130. public IFilter? Filter { get; set; }
  131. public IColumns? Columns { get; set; }
  132. public ISortOrder? Sort { get; set; }
  133. public CoreRange? Range { get; set; }
  134. }
  135. public class QueryRequest<TEntity> : BaseRequest<TEntity>, IQueryRequest, ISerializeBinary where TEntity : Entity, new()
  136. {
  137. IFilter? IQueryRequest.Filter
  138. {
  139. get => Filter;
  140. set => Filter = value as Filter<TEntity>;
  141. }
  142. public Filter<TEntity>? Filter { get; set; }
  143. IColumns? IQueryRequest.Columns
  144. {
  145. get => Columns;
  146. set => Columns = value as Columns<TEntity>;
  147. }
  148. public Columns<TEntity>? Columns { get; set; }
  149. ISortOrder? IQueryRequest.Sort
  150. {
  151. get => Sort;
  152. set => Sort = value as SortOrder<TEntity>;
  153. }
  154. public SortOrder<TEntity>? Sort { get; set; }
  155. public CoreRange? Range { get; set; }
  156. [JsonConstructor]
  157. public QueryRequest()
  158. {
  159. Filter = null;
  160. Columns = null;
  161. Sort = null;
  162. Range = null;
  163. }
  164. public QueryRequest(Filter<TEntity>? filter, Columns<TEntity>? columns, SortOrder<TEntity>? sort, CoreRange? range)
  165. {
  166. Filter = filter;
  167. Columns = columns;
  168. Sort = sort;
  169. Range = range;
  170. }
  171. public override RequestMethod GetMethod() => RequestMethod.Query;
  172. public override void SerializeBinary(CoreBinaryWriter writer)
  173. {
  174. base.SerializeBinary(writer);
  175. writer.Write(Filter);
  176. writer.Write(Columns);
  177. writer.Write(Sort);
  178. writer.WriteBinaryValue(Range);
  179. }
  180. public override void DeserializeBinary(CoreBinaryReader reader)
  181. {
  182. base.DeserializeBinary(reader);
  183. Filter = reader.ReadFilter<TEntity>();
  184. Columns = reader.ReadColumns<TEntity>();
  185. Sort = reader.ReadSortOrder<TEntity>();
  186. Range = reader.ReadBinaryValue<CoreRange>();
  187. }
  188. }
  189. public class QueryResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  190. {
  191. public CoreTable Items { get; set; } = new CoreTable();
  192. public override void DeserializeBinary(CoreBinaryReader reader)
  193. {
  194. base.DeserializeBinary(reader);
  195. Items = new CoreTable();
  196. Items.DeserializeBinary(reader);
  197. }
  198. public override void SerializeBinary(CoreBinaryWriter writer)
  199. {
  200. base.SerializeBinary(writer);
  201. Items.SerializeBinary(writer);
  202. }
  203. }
  204. /*public class LoadRequest<TEntity> : BaseRequest<TEntity> where TEntity : Entity, new()
  205. {
  206. public Filter<TEntity>? Filter { get; set; }
  207. public SortOrder<TEntity>? Sort { get; set; }
  208. }
  209. public class LoadResponse<TEntity> : BaseResponse<TEntity> where TEntity : Entity, new()
  210. {
  211. public TEntity[] Items { get; set; }
  212. }*/
  213. public class MultiSaveRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  214. {
  215. public TEntity[] Items { get; set; }
  216. public string AuditNote { get; set; }
  217. [Obsolete("We don't like this; it should always be true.")]
  218. // Added 11/04/23 to address incompatibility, remove as soon as possible; the relevant code to update is in RestService; update assuming that ReturnOnlyChanged is always true.
  219. public bool ReturnOnlyChanged { get; set; } = false;
  220. [JsonConstructor]
  221. public MultiSaveRequest()
  222. {
  223. Items = Array.Empty<TEntity>();
  224. AuditNote = "";
  225. }
  226. public MultiSaveRequest(TEntity[] items, string auditNode)
  227. {
  228. Items = items;
  229. AuditNote = auditNode;
  230. }
  231. public override RequestMethod GetMethod() => RequestMethod.MultiSave;
  232. public override void SerializeBinary(CoreBinaryWriter writer)
  233. {
  234. base.SerializeBinary(writer);
  235. writer.WriteObjects(Items);
  236. writer.Write(AuditNote);
  237. writer.Write(ReturnOnlyChanged);
  238. }
  239. public override void DeserializeBinary(CoreBinaryReader reader)
  240. {
  241. base.DeserializeBinary(reader);
  242. Items = reader.ReadObjects<TEntity>().ToArray();
  243. AuditNote = reader.ReadString();
  244. ReturnOnlyChanged = reader.ReadBoolean();
  245. }
  246. }
  247. public class MultiSaveResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  248. {
  249. //public Guid[] IDs { get; set; }
  250. public TEntity[]? Items { get; set; }
  251. public List<Dictionary<string, object?>> ChangedValues { get; set; } = new List<Dictionary<string, object?>>();
  252. public override void SerializeBinary(CoreBinaryWriter writer)
  253. {
  254. base.SerializeBinary(writer);
  255. if(Items is null)
  256. {
  257. writer.Write(false);
  258. }
  259. else
  260. {
  261. writer.Write(true);
  262. writer.WriteObjects(Items);
  263. }
  264. writer.Write(ChangedValues.Count);
  265. foreach(var changed in ChangedValues)
  266. {
  267. SaveResponse<TEntity>.SerializeChangedValues(writer, changed);
  268. }
  269. }
  270. public override void DeserializeBinary(CoreBinaryReader reader)
  271. {
  272. base.DeserializeBinary(reader);
  273. if (reader.ReadBoolean())
  274. {
  275. Items = reader.ReadObjects<TEntity>().ToArray();
  276. }
  277. else
  278. {
  279. Items = null;
  280. }
  281. ChangedValues.Clear();
  282. var nChangedValues = reader.ReadInt32();
  283. for(int i = 0; i < nChangedValues; ++i)
  284. {
  285. ChangedValues.Add(SaveResponse<TEntity>.DeserializeChangedValues(reader));
  286. }
  287. }
  288. }
  289. public class SaveRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  290. {
  291. public TEntity Item { get; set; }
  292. public string AuditNote { get; set; }
  293. public bool ReturnOnlyChanged { get; set; } = false;
  294. [JsonConstructor]
  295. public SaveRequest()
  296. {
  297. }
  298. public SaveRequest(TEntity item, string auditNote)
  299. {
  300. Item = item;
  301. AuditNote = auditNote;
  302. }
  303. public override RequestMethod GetMethod() => RequestMethod.Save;
  304. public override void SerializeBinary(CoreBinaryWriter writer)
  305. {
  306. base.SerializeBinary(writer);
  307. writer.WriteObject(Item);
  308. writer.Write(AuditNote);
  309. writer.Write(ReturnOnlyChanged);
  310. }
  311. public override void DeserializeBinary(CoreBinaryReader reader)
  312. {
  313. base.DeserializeBinary(reader);
  314. Item = reader.ReadObject<TEntity>();
  315. AuditNote = reader.ReadString();
  316. ReturnOnlyChanged = reader.ReadBoolean();
  317. }
  318. }
  319. public class SaveResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  320. {
  321. //public Guid ID { get; set; }
  322. public TEntity? Item { get; set; }
  323. public Dictionary<string, object?> ChangedValues { get; set; } = new Dictionary<string, object?>();
  324. public override void SerializeBinary(CoreBinaryWriter writer)
  325. {
  326. base.SerializeBinary(writer);
  327. if (Item is null)
  328. {
  329. writer.Write(false);
  330. }
  331. else
  332. {
  333. writer.Write(true);
  334. writer.WriteObject(Item);
  335. }
  336. SerializeChangedValues(writer, ChangedValues);
  337. }
  338. public override void DeserializeBinary(CoreBinaryReader reader)
  339. {
  340. base.DeserializeBinary(reader);
  341. if (reader.ReadBoolean())
  342. {
  343. Item = reader.ReadObject<TEntity>();
  344. }
  345. ChangedValues = DeserializeChangedValues(reader);
  346. }
  347. public static void SerializeChangedValues(CoreBinaryWriter writer, Dictionary<string, object?> changedValues)
  348. {
  349. var props = new List<Tuple<string, Type, object?>>();
  350. writer.Write(changedValues.Count);
  351. foreach (var (key, value) in changedValues)
  352. {
  353. var property = DatabaseSchema.Property(typeof(TEntity), key);
  354. if (property != null)
  355. {
  356. props.Add(new Tuple<string, Type, object?>(key, property.PropertyType, value));
  357. }
  358. else
  359. {
  360. Logger.Send(LogType.Error, "", $"Error serializing changedValues: Property {key} does not exist");
  361. }
  362. }
  363. foreach (var (key, type, value) in props)
  364. {
  365. writer.Write(key);
  366. writer.WriteBinaryValue(type, value);
  367. }
  368. }
  369. public static Dictionary<string, object?> DeserializeChangedValues(CoreBinaryReader reader)
  370. {
  371. var changedDict = new Dictionary<string, object?>();
  372. var nChanged = reader.ReadInt32();
  373. for (int j = 0; j < nChanged; ++j)
  374. {
  375. var key = reader.ReadString();
  376. var property = DatabaseSchema.Property(typeof(TEntity), key);
  377. if (property != null)
  378. {
  379. changedDict.Add(key, reader.ReadBinaryValue(property.PropertyType));
  380. }
  381. else
  382. {
  383. throw new Exception($"Property {key} does not exist");
  384. }
  385. }
  386. return changedDict;
  387. }
  388. }
  389. public class DeleteRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  390. {
  391. public TEntity Item { get; set; }
  392. public string AuditNote { get; set; }
  393. [JsonConstructor]
  394. public DeleteRequest()
  395. {
  396. }
  397. public DeleteRequest(TEntity item, string auditNote)
  398. {
  399. Item = item;
  400. AuditNote = auditNote;
  401. }
  402. public override RequestMethod GetMethod() => RequestMethod.Delete;
  403. public override void SerializeBinary(CoreBinaryWriter writer)
  404. {
  405. base.SerializeBinary(writer);
  406. writer.WriteObject(Item);
  407. writer.Write(AuditNote);
  408. }
  409. public override void DeserializeBinary(CoreBinaryReader reader)
  410. {
  411. base.DeserializeBinary(reader);
  412. Item = reader.ReadObject<TEntity>();
  413. AuditNote = reader.ReadString();
  414. }
  415. }
  416. public class DeleteResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  417. {
  418. }
  419. public class MultiDeleteRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  420. {
  421. public TEntity[] Items { get; set; }
  422. public string AuditNote { get; set; }
  423. [JsonConstructor]
  424. public MultiDeleteRequest()
  425. {
  426. }
  427. public MultiDeleteRequest(TEntity[] items, string auditNote)
  428. {
  429. Items = items;
  430. AuditNote = auditNote;
  431. }
  432. public override RequestMethod GetMethod() => RequestMethod.MultiDelete;
  433. public override void SerializeBinary(CoreBinaryWriter writer)
  434. {
  435. base.SerializeBinary(writer);
  436. writer.WriteObjects(Items);
  437. writer.Write(AuditNote);
  438. }
  439. public override void DeserializeBinary(CoreBinaryReader reader)
  440. {
  441. base.DeserializeBinary(reader);
  442. Items = reader.ReadObjects<TEntity>().ToArray();
  443. AuditNote = reader.ReadString();
  444. }
  445. }
  446. public class MultiDeleteResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  447. {
  448. }
  449. public interface IMultiQueryTableQuery
  450. {
  451. [JsonPropertyOrder(0)]
  452. string Type { get; }
  453. IFilter? Filter { get; set; }
  454. IColumns? Columns { get; set; }
  455. ISortOrder? Sort { get; set; }
  456. void SerializeBinary(CoreBinaryWriter writer) => SerializeBinary(this, writer);
  457. public static void SerializeBinary(IMultiQueryTableQuery query, CoreBinaryWriter writer)
  458. {
  459. writer.Write(query.Type);
  460. }
  461. public static IMultiQueryTableQuery DeserializeBinary(CoreBinaryReader reader)
  462. {
  463. var typeString = reader.ReadString();
  464. var type = CoreUtils.GetEntity(typeString);
  465. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(type)) as ISerializeBinary)!;
  466. query.DeserializeBinary(reader);
  467. return (query as IMultiQueryTableQuery)!;
  468. }
  469. }
  470. public class MultiQueryTableQuery<TEntity> : IMultiQueryTableQuery, ISerializeBinary
  471. where TEntity : Entity, new()
  472. {
  473. public string Type => typeof(TEntity).EntityName();
  474. public Filter<TEntity>? Filter { get; set; }
  475. public Columns<TEntity>? Columns { get; set; }
  476. public SortOrder<TEntity>? Sort { get; set; }
  477. IFilter? IMultiQueryTableQuery.Filter { get => Filter; set => Filter = value as Filter<TEntity>; }
  478. IColumns? IMultiQueryTableQuery.Columns { get => Columns; set => Columns = value as Columns<TEntity>; }
  479. ISortOrder? IMultiQueryTableQuery.Sort { get => Sort; set => Sort = value as SortOrder<TEntity>; }
  480. public void SerializeBinary(CoreBinaryWriter writer)
  481. {
  482. IMultiQueryTableQuery.SerializeBinary(this, writer);
  483. writer.Write(Filter);
  484. writer.Write(Columns);
  485. writer.Write(Sort);
  486. }
  487. public void DeserializeBinary(CoreBinaryReader reader)
  488. {
  489. Filter = reader.ReadFilter<TEntity>();
  490. Columns = reader.ReadColumns<TEntity>();
  491. Sort = reader.ReadSortOrder<TEntity>();
  492. }
  493. }
  494. public class MultiQueryRequest : Request, ISerializeBinary
  495. {
  496. public Dictionary<string, IMultiQueryTableQuery> Queries { get; set; } = new Dictionary<string, IMultiQueryTableQuery>();
  497. public void AddQuery(string key, IQueryDef queryDef)
  498. {
  499. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(queryDef.Type)) as IMultiQueryTableQuery)!;
  500. query.Filter = queryDef.Filter;
  501. query.Columns = queryDef.Columns;
  502. query.Sort = queryDef.SortOrder;
  503. Queries.Add(key, query);
  504. }
  505. public override RequestMethod GetMethod() => RequestMethod.MultiQuery;
  506. public override void SerializeBinary(CoreBinaryWriter writer)
  507. {
  508. base.SerializeBinary(writer);
  509. writer.Write(Queries.Count);
  510. foreach(var (key, query) in Queries)
  511. {
  512. writer.Write(key);
  513. query.SerializeBinary(writer);
  514. }
  515. }
  516. public override void DeserializeBinary(CoreBinaryReader reader)
  517. {
  518. base.DeserializeBinary(reader);
  519. var nQueries = reader.ReadInt32();
  520. for(int i = 0; i < nQueries; ++i)
  521. {
  522. var key = reader.ReadString();
  523. var query = IMultiQueryTableQuery.DeserializeBinary(reader);
  524. Queries[key] = query;
  525. }
  526. }
  527. }
  528. public class MultiQueryRequestConverter : CustomJsonConverter<MultiQueryRequest>
  529. {
  530. private IMultiQueryTableQuery ReadQuery(ref Utf8JsonReader reader, JsonSerializerOptions options)
  531. {
  532. Type? type = null;
  533. IFilter? filter = null;
  534. IColumns? columns = null;
  535. ISortOrder? sort = null;
  536. ReadObject(ref reader, (ref Utf8JsonReader reader, string property) =>
  537. {
  538. if(property == "Type")
  539. {
  540. var typeName = reader.GetString() ?? "";
  541. type = CoreUtils.GetEntity(typeName);
  542. if(type is null)
  543. {
  544. throw new JsonException($"Invalid type name {typeName}");
  545. }
  546. }
  547. else if(type is null)
  548. {
  549. throw new JsonException($"'Type' expected as first field, got {property}");
  550. }
  551. else if(property == "Filter")
  552. {
  553. filter = JsonSerializer.Deserialize(ref reader, typeof(Filter<>).MakeGenericType(type), options) as IFilter;
  554. }
  555. else if(property == "Columns")
  556. {
  557. columns = JsonSerializer.Deserialize(ref reader, typeof(Columns<>).MakeGenericType(type), options) as IColumns;
  558. }
  559. else if(property == "Sort")
  560. {
  561. sort = JsonSerializer.Deserialize(ref reader, typeof(SortOrder<>).MakeGenericType(type), options) as ISortOrder;
  562. }
  563. else
  564. {
  565. reader.Skip();
  566. }
  567. });
  568. if(type is null)
  569. {
  570. throw new JsonException($"'Type' field expected");
  571. }
  572. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(type)) as IMultiQueryTableQuery)!;
  573. query.Filter = filter;
  574. query.Columns = columns;
  575. query.Sort = sort;
  576. return query;
  577. }
  578. public override MultiQueryRequest? Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  579. {
  580. if (reader.TokenType != JsonTokenType.StartObject)
  581. {
  582. throw new JsonException();
  583. }
  584. var multiQuery = new MultiQueryRequest();
  585. ReadObject(ref reader, (ref Utf8JsonReader reader, string property) =>
  586. {
  587. if (property == "Queries")
  588. {
  589. ReadObject(ref reader, (ref Utf8JsonReader reader, string key) =>
  590. {
  591. multiQuery.Queries[key] = ReadQuery(ref reader, options);
  592. });
  593. }
  594. else if (property == "Credentials")
  595. {
  596. multiQuery.Credentials = JsonSerializer.Deserialize<Credentials>(ref reader, options) ?? new Credentials();
  597. }
  598. else
  599. {
  600. reader.Skip();
  601. }
  602. });
  603. return multiQuery;
  604. }
  605. public override void Write(Utf8JsonWriter writer, MultiQueryRequest value, JsonSerializerOptions options)
  606. {
  607. writer.WriteStartObject();
  608. writer.WritePropertyName("Credentials");
  609. JsonSerializer.Serialize(writer, value.Credentials, options);
  610. writer.WriteStartObject("Queries");
  611. foreach(var (key, query) in value.Queries)
  612. {
  613. writer.WritePropertyName(key);
  614. JsonSerializer.Serialize(writer, query, options);
  615. }
  616. writer.WriteEndObject();
  617. writer.WriteEndObject();
  618. }
  619. }
  620. public class MultiQueryResponse : Response, ISerializeBinary
  621. {
  622. public MultiQueryResponse()
  623. {
  624. Tables = new Dictionary<string, CoreTable>();
  625. }
  626. public Dictionary<string, CoreTable> Tables { get; set; }
  627. public override void DeserializeBinary(CoreBinaryReader reader)
  628. {
  629. base.DeserializeBinary(reader);
  630. Tables.Clear();
  631. var nTables = reader.ReadInt32();
  632. for(int i = 0; i < nTables; ++i)
  633. {
  634. var name = reader.ReadString();
  635. var table = new CoreTable();
  636. table.DeserializeBinary(reader);
  637. Tables[name] = table;
  638. }
  639. }
  640. public override void SerializeBinary(CoreBinaryWriter writer)
  641. {
  642. base.SerializeBinary(writer);
  643. writer.Write(Tables.Count);
  644. foreach(var (name, table) in Tables)
  645. {
  646. writer.Write(name);
  647. table.SerializeBinary(writer);
  648. }
  649. }
  650. }
  651. public enum ValidationStatus
  652. {
  653. VALID,
  654. INVALID,
  655. REQUIRE_2FA,
  656. PASSWORD_EXPIRED
  657. }
  658. public class ValidateRequest : Request, ISerializeBinary
  659. {
  660. public string? UserID { get; set; }
  661. public string? Password { get; set; }
  662. public string? PIN { get; set; }
  663. public bool UsePIN { get; set; }
  664. public override RequestMethod GetMethod() => RequestMethod.Validate;
  665. public override void SerializeBinary(CoreBinaryWriter writer)
  666. {
  667. base.SerializeBinary(writer);
  668. writer.Write(UserID ?? string.Empty);
  669. writer.Write(Password ?? string.Empty);
  670. writer.Write(PIN ?? string.Empty);
  671. writer.Write(UsePIN);
  672. }
  673. public override void DeserializeBinary(CoreBinaryReader reader)
  674. {
  675. base.DeserializeBinary(reader);
  676. UserID = reader.ReadString();
  677. Password = reader.ReadString();
  678. PIN = reader.ReadString();
  679. UsePIN = reader.ReadBoolean();
  680. }
  681. }
  682. public class ValidateResponse : Response, ISerializeBinary
  683. {
  684. public ValidationStatus ValidationStatus { get; set; }
  685. public Guid UserGuid { get; set; }
  686. public string? UserID { get; set; }
  687. public Guid SecurityID { get; set; }
  688. public Guid Session { get; set; }
  689. public string? Recipient2FA { get; set; }
  690. public DateTime PasswordExpiration { get; set; }
  691. public override void SerializeBinary(CoreBinaryWriter writer)
  692. {
  693. base.SerializeBinary(writer);
  694. writer.Write((int)ValidationStatus);
  695. writer.Write(UserGuid);
  696. writer.Write(UserID ?? string.Empty);
  697. writer.Write(SecurityID);
  698. writer.Write(Session);
  699. writer.Write(Recipient2FA ?? string.Empty);
  700. writer.Write(PasswordExpiration.ToUniversalTime().Ticks);
  701. }
  702. public override void DeserializeBinary(CoreBinaryReader reader)
  703. {
  704. base.DeserializeBinary(reader);
  705. ValidationStatus = (ValidationStatus)reader.ReadInt32();
  706. UserGuid = reader.ReadGuid();
  707. UserID = reader.ReadString();
  708. SecurityID = reader.ReadGuid();
  709. Session = reader.ReadGuid();
  710. Recipient2FA = reader.ReadString();
  711. PasswordExpiration = new DateTime(reader.ReadInt64(), DateTimeKind.Utc).ToLocalTime();
  712. }
  713. }
  714. public class Check2FARequest : Request, ISerializeBinary
  715. {
  716. public string Code { get; set; }
  717. [JsonConstructor]
  718. public Check2FARequest()
  719. {
  720. Code = "";
  721. }
  722. public Check2FARequest(string code)
  723. {
  724. Code = code;
  725. }
  726. public override RequestMethod GetMethod() => RequestMethod.Check2FA;
  727. public override void SerializeBinary(CoreBinaryWriter writer)
  728. {
  729. base.SerializeBinary(writer);
  730. writer.Write(Code);
  731. }
  732. public override void DeserializeBinary(CoreBinaryReader reader)
  733. {
  734. base.DeserializeBinary(reader);
  735. Code = reader.ReadString();
  736. }
  737. }
  738. public class Check2FAResponse : Response, ISerializeBinary
  739. {
  740. public bool Valid { get; set; }
  741. [JsonConstructor]
  742. public Check2FAResponse()
  743. {
  744. }
  745. public Check2FAResponse(bool valid)
  746. {
  747. Valid = valid;
  748. }
  749. public override void SerializeBinary(CoreBinaryWriter writer)
  750. {
  751. base.SerializeBinary(writer);
  752. writer.Write(Valid);
  753. }
  754. public override void DeserializeBinary(CoreBinaryReader reader)
  755. {
  756. base.DeserializeBinary(reader);
  757. Valid = reader.ReadBoolean();
  758. }
  759. }
  760. public class PingRequest : Request, ISerializeBinary
  761. {
  762. public override RequestMethod GetMethod() => RequestMethod.Ping;
  763. }
  764. public class PingResponse : Response, ISerializeBinary { }
  765. public class InfoRequest : Request, ISerializeBinary
  766. {
  767. public override RequestMethod GetMethod() => RequestMethod.Info;
  768. }
  769. public class DatabaseInfo : ISerializeBinary
  770. {
  771. public string? ColorScheme { get; set; }
  772. public byte[]? Logo { get; set; }
  773. public string Version { get; set; }
  774. public bool IsHTTPS { get; set; }
  775. // Ideally, we should not need these. However, there is currently a bug in RPCServer
  776. // that crashes the server when uploading large files (ie photos)
  777. // So we need to be a able to maintain both a Rest and RPC-style connection
  778. // to the server to try and have both session (RPC) capabilities
  779. // as well as having stable upload performance :-(
  780. // Once the RPC listeners are stable, then this needs to be removed.
  781. public int RestPort { get; set; }
  782. public int RPCPort { get; set; }
  783. public Guid DatabaseID { get; set; }
  784. [JsonConstructor]
  785. public DatabaseInfo()
  786. {
  787. Version = "";
  788. }
  789. public DatabaseInfo(string? colorScheme, byte[]? logo, string version, bool isHTTTPS, int restPort, int rpcPort, Guid databaseID)
  790. {
  791. ColorScheme = colorScheme;
  792. Logo = logo;
  793. Version = version;
  794. IsHTTPS = isHTTTPS;
  795. RestPort = restPort;
  796. RPCPort = rpcPort;
  797. DatabaseID = databaseID;
  798. }
  799. public void SerializeBinary(CoreBinaryWriter writer)
  800. {
  801. writer.Write(ColorScheme ?? string.Empty);
  802. if (Logo is null)
  803. {
  804. writer.Write(0);
  805. }
  806. else
  807. {
  808. writer.Write(Logo.Length);
  809. writer.Write(Logo);
  810. }
  811. writer.Write(Version);
  812. writer.Write(IsHTTPS);
  813. if (String.CompareOrdinal(writer.Settings.Version, "1.1") > 0)
  814. {
  815. writer.Write(RestPort);
  816. writer.Write(RPCPort);
  817. }
  818. writer.Write(DatabaseID);
  819. }
  820. public void DeserializeBinary(CoreBinaryReader reader)
  821. {
  822. ColorScheme = reader.ReadString();
  823. var nLogoBytes = reader.ReadInt32();
  824. Logo = reader.ReadBytes(nLogoBytes);
  825. Version = reader.ReadString();
  826. IsHTTPS = reader.ReadBoolean();
  827. if (String.CompareOrdinal(reader.Settings.Version, "1.1") > 0)
  828. {
  829. try
  830. {
  831. RestPort = reader.ReadInt32();
  832. RPCPort = reader.ReadInt32();
  833. }
  834. catch
  835. {
  836. Logger.Send(LogType.Error,"","Unable to read RestPort and RPCPort Values");
  837. }
  838. }
  839. DatabaseID = reader.ReadGuid();
  840. }
  841. }
  842. public class InfoResponse : Response, ISerializeBinary
  843. {
  844. public DatabaseInfo Info { get; set; }
  845. [JsonConstructor]
  846. public InfoResponse() : base()
  847. {
  848. Info = new DatabaseInfo();
  849. }
  850. public InfoResponse(DatabaseInfo info)
  851. {
  852. Info = info;
  853. }
  854. public override void SerializeBinary(CoreBinaryWriter writer)
  855. {
  856. base.SerializeBinary(writer);
  857. Info.SerializeBinary(writer);
  858. }
  859. public override void DeserializeBinary(CoreBinaryReader reader)
  860. {
  861. base.DeserializeBinary(reader);
  862. Info.DeserializeBinary(reader);
  863. }
  864. }
  865. public class VersionRequest : Request, ISerializeBinary
  866. {
  867. public override RequestMethod GetMethod() => RequestMethod.Version;
  868. }
  869. public class VersionResponse : Response, ISerializeBinary
  870. {
  871. public string Version { get; set; } = "";
  872. public override void SerializeBinary(CoreBinaryWriter writer)
  873. {
  874. base.SerializeBinary(writer);
  875. writer.Write(Version);
  876. }
  877. public override void DeserializeBinary(CoreBinaryReader reader)
  878. {
  879. base.DeserializeBinary(reader);
  880. Version = reader.ReadString();
  881. }
  882. }
  883. public class ReleaseNotesRequest : Request, ISerializeBinary
  884. {
  885. public override RequestMethod GetMethod() => RequestMethod.ReleaseNotes;
  886. }
  887. public class ReleaseNotesResponse : Response, ISerializeBinary
  888. {
  889. public string ReleaseNotes { get; set; } = "";
  890. public override void SerializeBinary(CoreBinaryWriter writer)
  891. {
  892. base.SerializeBinary(writer);
  893. writer.Write(ReleaseNotes);
  894. }
  895. public override void DeserializeBinary(CoreBinaryReader reader)
  896. {
  897. base.DeserializeBinary(reader);
  898. ReleaseNotes = reader.ReadString();
  899. }
  900. }
  901. public class InstallerRequest : Request, ISerializeBinary
  902. {
  903. public override RequestMethod GetMethod() => RequestMethod.Installer;
  904. }
  905. public class InstallerResponse : Response, ISerializeBinary
  906. {
  907. public byte[]? Installer { get; set; }
  908. public override void SerializeBinary(CoreBinaryWriter writer)
  909. {
  910. base.SerializeBinary(writer);
  911. if(Installer != null)
  912. {
  913. writer.Write(Installer.Length);
  914. writer.Write(Installer);
  915. }
  916. else
  917. {
  918. writer.Write(0);
  919. }
  920. }
  921. public override void DeserializeBinary(CoreBinaryReader reader)
  922. {
  923. base.DeserializeBinary(reader);
  924. Installer = reader.ReadBytes(reader.ReadInt32());
  925. if(Installer.Length == 0)
  926. {
  927. Installer = null;
  928. }
  929. }
  930. }
  931. public static class Extensions
  932. {
  933. public static T Status<T>(this T response, StatusCode status) where T : Response
  934. {
  935. response.Status = status;
  936. return response;
  937. }
  938. }
  939. public class RequestException : Exception
  940. {
  941. public StatusCode Status { get; set; }
  942. public RequestMethod Method { get; set; }
  943. public RequestException(string message, StatusCode status, Request request): base(message)
  944. {
  945. Status = status;
  946. Method = request.GetMethod();
  947. }
  948. }
  949. }