Request.cs 34 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169
  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. [JsonConstructor]
  218. public MultiSaveRequest()
  219. {
  220. Items = Array.Empty<TEntity>();
  221. AuditNote = "";
  222. }
  223. public MultiSaveRequest(TEntity[] items, string auditNode)
  224. {
  225. Items = items;
  226. AuditNote = auditNode;
  227. }
  228. public override RequestMethod GetMethod() => RequestMethod.MultiSave;
  229. public override void SerializeBinary(CoreBinaryWriter writer)
  230. {
  231. base.SerializeBinary(writer);
  232. writer.WriteObjects(Items);
  233. writer.Write(AuditNote);
  234. }
  235. public override void DeserializeBinary(CoreBinaryReader reader)
  236. {
  237. base.DeserializeBinary(reader);
  238. Items = reader.ReadObjects<TEntity>().ToArray();
  239. AuditNote = reader.ReadString();
  240. }
  241. }
  242. public class MultiSaveResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  243. {
  244. //public Guid[] IDs { get; set; }
  245. public TEntity[]? Items { get; set; }
  246. public List<Dictionary<string, object?>> ChangedValues { get; set; } = new List<Dictionary<string, object?>>();
  247. public override void SerializeBinary(CoreBinaryWriter writer)
  248. {
  249. base.SerializeBinary(writer);
  250. if(Items is null)
  251. {
  252. writer.Write(false);
  253. }
  254. else
  255. {
  256. writer.Write(true);
  257. writer.WriteObjects(Items);
  258. }
  259. writer.Write(ChangedValues.Count);
  260. foreach(var changed in ChangedValues)
  261. {
  262. SaveResponse<TEntity>.SerializeChangedValues(writer, changed);
  263. }
  264. }
  265. public override void DeserializeBinary(CoreBinaryReader reader)
  266. {
  267. base.DeserializeBinary(reader);
  268. if (reader.ReadBoolean())
  269. {
  270. Items = reader.ReadObjects<TEntity>().ToArray();
  271. }
  272. else
  273. {
  274. Items = null;
  275. }
  276. ChangedValues.Clear();
  277. var nChangedValues = reader.ReadInt32();
  278. for(int i = 0; i < nChangedValues; ++i)
  279. {
  280. ChangedValues.Add(SaveResponse<TEntity>.DeserializeChangedValues(reader));
  281. }
  282. }
  283. }
  284. public class SaveRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  285. {
  286. public TEntity Item { get; set; }
  287. public string AuditNote { get; set; }
  288. public bool ReturnOnlyChanged { get; set; } = false;
  289. [JsonConstructor]
  290. public SaveRequest()
  291. {
  292. }
  293. public SaveRequest(TEntity item, string auditNote)
  294. {
  295. Item = item;
  296. AuditNote = auditNote;
  297. }
  298. public override RequestMethod GetMethod() => RequestMethod.Save;
  299. public override void SerializeBinary(CoreBinaryWriter writer)
  300. {
  301. base.SerializeBinary(writer);
  302. writer.WriteObject(Item);
  303. writer.Write(AuditNote);
  304. writer.Write(ReturnOnlyChanged);
  305. }
  306. public override void DeserializeBinary(CoreBinaryReader reader)
  307. {
  308. base.DeserializeBinary(reader);
  309. Item = reader.ReadObject<TEntity>();
  310. AuditNote = reader.ReadString();
  311. ReturnOnlyChanged = reader.ReadBoolean();
  312. }
  313. }
  314. public class SaveResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  315. {
  316. //public Guid ID { get; set; }
  317. public TEntity? Item { get; set; }
  318. public Dictionary<string, object?> ChangedValues { get; set; } = new Dictionary<string, object?>();
  319. public override void SerializeBinary(CoreBinaryWriter writer)
  320. {
  321. base.SerializeBinary(writer);
  322. if (Item is null)
  323. {
  324. writer.Write(false);
  325. }
  326. else
  327. {
  328. writer.Write(true);
  329. writer.WriteObject(Item);
  330. }
  331. SerializeChangedValues(writer, ChangedValues);
  332. }
  333. public override void DeserializeBinary(CoreBinaryReader reader)
  334. {
  335. base.DeserializeBinary(reader);
  336. if (reader.ReadBoolean())
  337. {
  338. Item = reader.ReadObject<TEntity>();
  339. }
  340. ChangedValues = DeserializeChangedValues(reader);
  341. }
  342. public static void SerializeChangedValues(CoreBinaryWriter writer, Dictionary<string, object?> changedValues)
  343. {
  344. var props = new List<Tuple<string, Type, object?>>();
  345. writer.Write(changedValues.Count);
  346. foreach (var (key, value) in changedValues)
  347. {
  348. var property = DatabaseSchema.Property(typeof(TEntity), key);
  349. if (property != null)
  350. {
  351. props.Add(new Tuple<string, Type, object?>(key, property.PropertyType, value));
  352. }
  353. else
  354. {
  355. Logger.Send(LogType.Error, "", $"Error serializing changedValues: Property {key} does not exist");
  356. }
  357. }
  358. foreach (var (key, type, value) in props)
  359. {
  360. writer.Write(key);
  361. writer.WriteBinaryValue(type, value);
  362. }
  363. }
  364. public static Dictionary<string, object?> DeserializeChangedValues(CoreBinaryReader reader)
  365. {
  366. var changedDict = new Dictionary<string, object?>();
  367. var nChanged = reader.ReadInt32();
  368. for (int j = 0; j < nChanged; ++j)
  369. {
  370. var key = reader.ReadString();
  371. var property = DatabaseSchema.Property(typeof(TEntity), key);
  372. if (property != null)
  373. {
  374. changedDict.Add(key, reader.ReadBinaryValue(property.PropertyType));
  375. }
  376. else
  377. {
  378. throw new Exception($"Property {key} does not exist");
  379. }
  380. }
  381. return changedDict;
  382. }
  383. }
  384. public class DeleteRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  385. {
  386. public TEntity Item { get; set; }
  387. public string AuditNote { get; set; }
  388. [JsonConstructor]
  389. public DeleteRequest()
  390. {
  391. }
  392. public DeleteRequest(TEntity item, string auditNote)
  393. {
  394. Item = item;
  395. AuditNote = auditNote;
  396. }
  397. public override RequestMethod GetMethod() => RequestMethod.Delete;
  398. public override void SerializeBinary(CoreBinaryWriter writer)
  399. {
  400. base.SerializeBinary(writer);
  401. writer.WriteObject(Item);
  402. writer.Write(AuditNote);
  403. }
  404. public override void DeserializeBinary(CoreBinaryReader reader)
  405. {
  406. base.DeserializeBinary(reader);
  407. Item = reader.ReadObject<TEntity>();
  408. AuditNote = reader.ReadString();
  409. }
  410. }
  411. public class DeleteResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  412. {
  413. }
  414. public class MultiDeleteRequest<TEntity> : BaseRequest<TEntity>, ISerializeBinary where TEntity : Entity, new()
  415. {
  416. public TEntity[] Items { get; set; }
  417. public string AuditNote { get; set; }
  418. [JsonConstructor]
  419. public MultiDeleteRequest()
  420. {
  421. }
  422. public MultiDeleteRequest(TEntity[] items, string auditNote)
  423. {
  424. Items = items;
  425. AuditNote = auditNote;
  426. }
  427. public override RequestMethod GetMethod() => RequestMethod.MultiDelete;
  428. public override void SerializeBinary(CoreBinaryWriter writer)
  429. {
  430. base.SerializeBinary(writer);
  431. writer.WriteObjects(Items);
  432. writer.Write(AuditNote);
  433. }
  434. public override void DeserializeBinary(CoreBinaryReader reader)
  435. {
  436. base.DeserializeBinary(reader);
  437. Items = reader.ReadObjects<TEntity>().ToArray();
  438. AuditNote = reader.ReadString();
  439. }
  440. }
  441. public class MultiDeleteResponse<TEntity> : BaseResponse<TEntity>, ISerializeBinary where TEntity : Entity, new()
  442. {
  443. }
  444. public interface IMultiQueryTableQuery
  445. {
  446. [JsonPropertyOrder(0)]
  447. string Type { get; }
  448. IFilter? Filter { get; set; }
  449. IColumns? Columns { get; set; }
  450. ISortOrder? Sort { get; set; }
  451. void SerializeBinary(CoreBinaryWriter writer) => SerializeBinary(this, writer);
  452. public static void SerializeBinary(IMultiQueryTableQuery query, CoreBinaryWriter writer)
  453. {
  454. writer.Write(query.Type);
  455. }
  456. public static IMultiQueryTableQuery DeserializeBinary(CoreBinaryReader reader)
  457. {
  458. var typeString = reader.ReadString();
  459. var type = CoreUtils.GetEntity(typeString);
  460. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(type)) as ISerializeBinary)!;
  461. query.DeserializeBinary(reader);
  462. return (query as IMultiQueryTableQuery)!;
  463. }
  464. }
  465. public class MultiQueryTableQuery<TEntity> : IMultiQueryTableQuery, ISerializeBinary
  466. where TEntity : Entity, new()
  467. {
  468. public string Type => typeof(TEntity).EntityName();
  469. public Filter<TEntity>? Filter { get; set; }
  470. public Columns<TEntity>? Columns { get; set; }
  471. public SortOrder<TEntity>? Sort { get; set; }
  472. IFilter? IMultiQueryTableQuery.Filter { get => Filter; set => Filter = value as Filter<TEntity>; }
  473. IColumns? IMultiQueryTableQuery.Columns { get => Columns; set => Columns = value as Columns<TEntity>; }
  474. ISortOrder? IMultiQueryTableQuery.Sort { get => Sort; set => Sort = value as SortOrder<TEntity>; }
  475. public void SerializeBinary(CoreBinaryWriter writer)
  476. {
  477. IMultiQueryTableQuery.SerializeBinary(this, writer);
  478. writer.Write(Filter);
  479. writer.Write(Columns);
  480. writer.Write(Sort);
  481. }
  482. public void DeserializeBinary(CoreBinaryReader reader)
  483. {
  484. Filter = reader.ReadFilter<TEntity>();
  485. Columns = reader.ReadColumns<TEntity>();
  486. Sort = reader.ReadSortOrder<TEntity>();
  487. }
  488. }
  489. public class MultiQueryRequest : Request, ISerializeBinary
  490. {
  491. public Dictionary<string, IMultiQueryTableQuery> Queries { get; set; } = new Dictionary<string, IMultiQueryTableQuery>();
  492. public void AddQuery(string key, IQueryDef queryDef)
  493. {
  494. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(queryDef.Type)) as IMultiQueryTableQuery)!;
  495. query.Filter = queryDef.Filter;
  496. query.Columns = queryDef.Columns;
  497. query.Sort = queryDef.SortOrder;
  498. Queries.Add(key, query);
  499. }
  500. public override RequestMethod GetMethod() => RequestMethod.MultiQuery;
  501. public override void SerializeBinary(CoreBinaryWriter writer)
  502. {
  503. base.SerializeBinary(writer);
  504. writer.Write(Queries.Count);
  505. foreach(var (key, query) in Queries)
  506. {
  507. writer.Write(key);
  508. query.SerializeBinary(writer);
  509. }
  510. }
  511. public override void DeserializeBinary(CoreBinaryReader reader)
  512. {
  513. base.DeserializeBinary(reader);
  514. var nQueries = reader.ReadInt32();
  515. for(int i = 0; i < nQueries; ++i)
  516. {
  517. var key = reader.ReadString();
  518. var query = IMultiQueryTableQuery.DeserializeBinary(reader);
  519. Queries[key] = query;
  520. }
  521. }
  522. }
  523. public class MultiQueryRequestConverter : CustomJsonConverter<MultiQueryRequest>
  524. {
  525. private IMultiQueryTableQuery ReadQuery(ref Utf8JsonReader reader, JsonSerializerOptions options)
  526. {
  527. Type? type = null;
  528. IFilter? filter = null;
  529. IColumns? columns = null;
  530. ISortOrder? sort = null;
  531. ReadObject(ref reader, (ref Utf8JsonReader reader, string property) =>
  532. {
  533. if(property == "Type")
  534. {
  535. var typeName = reader.GetString() ?? "";
  536. type = CoreUtils.GetEntity(typeName);
  537. if(type is null)
  538. {
  539. throw new JsonException($"Invalid type name {typeName}");
  540. }
  541. }
  542. else if(type is null)
  543. {
  544. throw new JsonException($"'Type' expected as first field, got {property}");
  545. }
  546. else if(property == "Filter")
  547. {
  548. filter = JsonSerializer.Deserialize(ref reader, typeof(Filter<>).MakeGenericType(type), options) as IFilter;
  549. }
  550. else if(property == "Columns")
  551. {
  552. columns = JsonSerializer.Deserialize(ref reader, typeof(Columns<>).MakeGenericType(type), options) as IColumns;
  553. }
  554. else if(property == "Sort")
  555. {
  556. sort = JsonSerializer.Deserialize(ref reader, typeof(SortOrder<>).MakeGenericType(type), options) as ISortOrder;
  557. }
  558. else
  559. {
  560. reader.Skip();
  561. }
  562. });
  563. if(type is null)
  564. {
  565. throw new JsonException($"'Type' field expected");
  566. }
  567. var query = (Activator.CreateInstance(typeof(MultiQueryTableQuery<>).MakeGenericType(type)) as IMultiQueryTableQuery)!;
  568. query.Filter = filter;
  569. query.Columns = columns;
  570. query.Sort = sort;
  571. return query;
  572. }
  573. public override MultiQueryRequest? Read(ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options)
  574. {
  575. if (reader.TokenType != JsonTokenType.StartObject)
  576. {
  577. throw new JsonException();
  578. }
  579. var multiQuery = new MultiQueryRequest();
  580. ReadObject(ref reader, (ref Utf8JsonReader reader, string property) =>
  581. {
  582. if (property == "Queries")
  583. {
  584. ReadObject(ref reader, (ref Utf8JsonReader reader, string key) =>
  585. {
  586. multiQuery.Queries[key] = ReadQuery(ref reader, options);
  587. });
  588. }
  589. else if (property == "Credentials")
  590. {
  591. multiQuery.Credentials = JsonSerializer.Deserialize<Credentials>(ref reader, options) ?? new Credentials();
  592. }
  593. else
  594. {
  595. reader.Skip();
  596. }
  597. });
  598. return multiQuery;
  599. }
  600. public override void Write(Utf8JsonWriter writer, MultiQueryRequest value, JsonSerializerOptions options)
  601. {
  602. writer.WriteStartObject();
  603. writer.WritePropertyName("Credentials");
  604. JsonSerializer.Serialize(writer, value.Credentials, options);
  605. writer.WriteStartObject("Queries");
  606. foreach(var (key, query) in value.Queries)
  607. {
  608. writer.WritePropertyName(key);
  609. JsonSerializer.Serialize(writer, query, options);
  610. }
  611. writer.WriteEndObject();
  612. writer.WriteEndObject();
  613. }
  614. }
  615. public class MultiQueryResponse : Response, ISerializeBinary
  616. {
  617. public MultiQueryResponse()
  618. {
  619. Tables = new Dictionary<string, CoreTable>();
  620. }
  621. public Dictionary<string, CoreTable> Tables { get; set; }
  622. public override void DeserializeBinary(CoreBinaryReader reader)
  623. {
  624. base.DeserializeBinary(reader);
  625. Tables.Clear();
  626. var nTables = reader.ReadInt32();
  627. for(int i = 0; i < nTables; ++i)
  628. {
  629. var name = reader.ReadString();
  630. var table = new CoreTable();
  631. table.DeserializeBinary(reader);
  632. Tables[name] = table;
  633. }
  634. }
  635. public override void SerializeBinary(CoreBinaryWriter writer)
  636. {
  637. base.SerializeBinary(writer);
  638. writer.Write(Tables.Count);
  639. foreach(var (name, table) in Tables)
  640. {
  641. writer.Write(name);
  642. table.SerializeBinary(writer);
  643. }
  644. }
  645. }
  646. public enum ValidationStatus
  647. {
  648. VALID,
  649. INVALID,
  650. REQUIRE_2FA,
  651. PASSWORD_EXPIRED
  652. }
  653. public class ValidateRequest : Request, ISerializeBinary
  654. {
  655. public string? UserID { get; set; }
  656. public string? Password { get; set; }
  657. public string? PIN { get; set; }
  658. public bool UsePIN { get; set; }
  659. public override RequestMethod GetMethod() => RequestMethod.Validate;
  660. public override void SerializeBinary(CoreBinaryWriter writer)
  661. {
  662. base.SerializeBinary(writer);
  663. writer.Write(UserID ?? string.Empty);
  664. writer.Write(Password ?? string.Empty);
  665. writer.Write(PIN ?? string.Empty);
  666. writer.Write(UsePIN);
  667. }
  668. public override void DeserializeBinary(CoreBinaryReader reader)
  669. {
  670. base.DeserializeBinary(reader);
  671. UserID = reader.ReadString();
  672. Password = reader.ReadString();
  673. PIN = reader.ReadString();
  674. UsePIN = reader.ReadBoolean();
  675. }
  676. }
  677. public class ValidateResponse : Response, ISerializeBinary
  678. {
  679. public ValidationStatus ValidationStatus { get; set; }
  680. public Guid UserGuid { get; set; }
  681. public string? UserID { get; set; }
  682. public Guid SecurityID { get; set; }
  683. public Guid Session { get; set; }
  684. public string? Recipient2FA { get; set; }
  685. public DateTime PasswordExpiration { get; set; }
  686. public override void SerializeBinary(CoreBinaryWriter writer)
  687. {
  688. base.SerializeBinary(writer);
  689. writer.Write((int)ValidationStatus);
  690. writer.Write(UserGuid);
  691. writer.Write(UserID ?? string.Empty);
  692. writer.Write(SecurityID);
  693. writer.Write(Session);
  694. writer.Write(Recipient2FA ?? string.Empty);
  695. writer.Write(PasswordExpiration.ToUniversalTime().Ticks);
  696. }
  697. public override void DeserializeBinary(CoreBinaryReader reader)
  698. {
  699. base.DeserializeBinary(reader);
  700. ValidationStatus = (ValidationStatus)reader.ReadInt32();
  701. UserGuid = reader.ReadGuid();
  702. UserID = reader.ReadString();
  703. SecurityID = reader.ReadGuid();
  704. Session = reader.ReadGuid();
  705. Recipient2FA = reader.ReadString();
  706. PasswordExpiration = new DateTime(reader.ReadInt64(), DateTimeKind.Utc).ToLocalTime();
  707. }
  708. }
  709. public class Check2FARequest : Request, ISerializeBinary
  710. {
  711. public string Code { get; set; }
  712. [JsonConstructor]
  713. public Check2FARequest()
  714. {
  715. Code = "";
  716. }
  717. public Check2FARequest(string code)
  718. {
  719. Code = code;
  720. }
  721. public override RequestMethod GetMethod() => RequestMethod.Check2FA;
  722. public override void SerializeBinary(CoreBinaryWriter writer)
  723. {
  724. base.SerializeBinary(writer);
  725. writer.Write(Code);
  726. }
  727. public override void DeserializeBinary(CoreBinaryReader reader)
  728. {
  729. base.DeserializeBinary(reader);
  730. Code = reader.ReadString();
  731. }
  732. }
  733. public class Check2FAResponse : Response, ISerializeBinary
  734. {
  735. public bool Valid { get; set; }
  736. [JsonConstructor]
  737. public Check2FAResponse()
  738. {
  739. }
  740. public Check2FAResponse(bool valid)
  741. {
  742. Valid = valid;
  743. }
  744. public override void SerializeBinary(CoreBinaryWriter writer)
  745. {
  746. base.SerializeBinary(writer);
  747. writer.Write(Valid);
  748. }
  749. public override void DeserializeBinary(CoreBinaryReader reader)
  750. {
  751. base.DeserializeBinary(reader);
  752. Valid = reader.ReadBoolean();
  753. }
  754. }
  755. public class PingRequest : Request, ISerializeBinary
  756. {
  757. public override RequestMethod GetMethod() => RequestMethod.Ping;
  758. }
  759. public class PingResponse : Response, ISerializeBinary { }
  760. public class InfoRequest : Request, ISerializeBinary
  761. {
  762. public override RequestMethod GetMethod() => RequestMethod.Info;
  763. }
  764. public class DatabaseInfo : ISerializeBinary
  765. {
  766. public string? ColorScheme { get; set; }
  767. public byte[]? Logo { get; set; }
  768. public string Version { get; set; }
  769. public bool IsHTTPS { get; set; }
  770. // Ideally, we should not need these. However, there is currently a bug in RPCServer
  771. // that crashes the server when uploading large files (ie photos)
  772. // So we need to be a able to maintain both a Rest and RPC-style connection
  773. // to the server to try and have both session (RPC) capabilities
  774. // as well as having stable upload performance :-(
  775. // Once the RPC listeners are stable, then this needs to be removed.
  776. public int RestPort { get; set; }
  777. public int RPCPort { get; set; }
  778. public Guid DatabaseID { get; set; }
  779. [JsonConstructor]
  780. public DatabaseInfo()
  781. {
  782. Version = "";
  783. }
  784. public DatabaseInfo(string? colorScheme, byte[]? logo, string version, bool isHTTTPS, int restPort, int rpcPort, Guid databaseID)
  785. {
  786. ColorScheme = colorScheme;
  787. Logo = logo;
  788. Version = version;
  789. IsHTTPS = isHTTTPS;
  790. RestPort = restPort;
  791. RPCPort = rpcPort;
  792. DatabaseID = databaseID;
  793. }
  794. public void SerializeBinary(CoreBinaryWriter writer)
  795. {
  796. writer.Write(ColorScheme ?? string.Empty);
  797. if (Logo is null)
  798. {
  799. writer.Write(0);
  800. }
  801. else
  802. {
  803. writer.Write(Logo.Length);
  804. writer.Write(Logo);
  805. }
  806. writer.Write(Version);
  807. writer.Write(IsHTTPS);
  808. if (String.CompareOrdinal(writer.Settings.Version, "1.1") > 0)
  809. {
  810. writer.Write(RestPort);
  811. writer.Write(RPCPort);
  812. }
  813. writer.Write(DatabaseID);
  814. }
  815. public void DeserializeBinary(CoreBinaryReader reader)
  816. {
  817. ColorScheme = reader.ReadString();
  818. var nLogoBytes = reader.ReadInt32();
  819. Logo = reader.ReadBytes(nLogoBytes);
  820. Version = reader.ReadString();
  821. IsHTTPS = reader.ReadBoolean();
  822. if (String.CompareOrdinal(reader.Settings.Version, "1.1") > 0)
  823. {
  824. try
  825. {
  826. RestPort = reader.ReadInt32();
  827. RPCPort = reader.ReadInt32();
  828. }
  829. catch
  830. {
  831. Logger.Send(LogType.Error,"","Unable to read RestPort and RPCPort Values");
  832. }
  833. }
  834. DatabaseID = reader.ReadGuid();
  835. }
  836. }
  837. public class InfoResponse : Response, ISerializeBinary
  838. {
  839. public DatabaseInfo Info { get; set; }
  840. [JsonConstructor]
  841. public InfoResponse() : base()
  842. {
  843. Info = new DatabaseInfo();
  844. }
  845. public InfoResponse(DatabaseInfo info)
  846. {
  847. Info = info;
  848. }
  849. public override void SerializeBinary(CoreBinaryWriter writer)
  850. {
  851. base.SerializeBinary(writer);
  852. Info.SerializeBinary(writer);
  853. }
  854. public override void DeserializeBinary(CoreBinaryReader reader)
  855. {
  856. base.DeserializeBinary(reader);
  857. Info.DeserializeBinary(reader);
  858. }
  859. }
  860. public class VersionRequest : Request, ISerializeBinary
  861. {
  862. public override RequestMethod GetMethod() => RequestMethod.Version;
  863. }
  864. public class VersionResponse : Response, ISerializeBinary
  865. {
  866. public string Version { get; set; } = "";
  867. public override void SerializeBinary(CoreBinaryWriter writer)
  868. {
  869. base.SerializeBinary(writer);
  870. writer.Write(Version);
  871. }
  872. public override void DeserializeBinary(CoreBinaryReader reader)
  873. {
  874. base.DeserializeBinary(reader);
  875. Version = reader.ReadString();
  876. }
  877. }
  878. public class ReleaseNotesRequest : Request, ISerializeBinary
  879. {
  880. public override RequestMethod GetMethod() => RequestMethod.ReleaseNotes;
  881. }
  882. public class ReleaseNotesResponse : Response, ISerializeBinary
  883. {
  884. public string ReleaseNotes { get; set; } = "";
  885. public override void SerializeBinary(CoreBinaryWriter writer)
  886. {
  887. base.SerializeBinary(writer);
  888. writer.Write(ReleaseNotes);
  889. }
  890. public override void DeserializeBinary(CoreBinaryReader reader)
  891. {
  892. base.DeserializeBinary(reader);
  893. ReleaseNotes = reader.ReadString();
  894. }
  895. }
  896. public class InstallerRequest : Request, ISerializeBinary
  897. {
  898. public override RequestMethod GetMethod() => RequestMethod.Installer;
  899. }
  900. public class InstallerResponse : Response, ISerializeBinary
  901. {
  902. public byte[]? Installer { get; set; }
  903. public override void SerializeBinary(CoreBinaryWriter writer)
  904. {
  905. base.SerializeBinary(writer);
  906. if(Installer != null)
  907. {
  908. writer.Write(Installer.Length);
  909. writer.Write(Installer);
  910. }
  911. else
  912. {
  913. writer.Write(0);
  914. }
  915. }
  916. public override void DeserializeBinary(CoreBinaryReader reader)
  917. {
  918. base.DeserializeBinary(reader);
  919. Installer = reader.ReadBytes(reader.ReadInt32());
  920. if(Installer.Length == 0)
  921. {
  922. Installer = null;
  923. }
  924. }
  925. }
  926. public static class Extensions
  927. {
  928. public static T Status<T>(this T response, StatusCode status) where T : Response
  929. {
  930. response.Status = status;
  931. return response;
  932. }
  933. }
  934. public class RequestException : Exception
  935. {
  936. public StatusCode Status { get; set; }
  937. public RequestMethod Method { get; set; }
  938. public RequestException(string message, StatusCode status, Request request): base(message)
  939. {
  940. Status = status;
  941. Method = request.GetMethod();
  942. }
  943. }
  944. }