restdata.pp 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334
  1. unit restdata;
  2. {$mode objfpc}{$H+}
  3. interface
  4. uses
  5. Classes, SysUtils, httpdefs;
  6. Procedure HandleRest(aRequest : TRequest; AResponse : TResponse);
  7. implementation
  8. uses jsonparser,fpjson;
  9. Var
  10. Packet : TJSONObject;
  11. LastID : integer;
  12. Function Data : TJSONArray;
  13. begin
  14. Result:=Packet.Arrays['Data'];
  15. end;
  16. Function MetaData : TJSONObject;
  17. begin
  18. Result:=Packet.Objects['metaData'];
  19. end;
  20. Procedure LoadRestData;
  21. Var
  22. F : TFileStream;
  23. D : TJSONData;
  24. I : Integer;
  25. begin
  26. F:=TFileStream.Create('countries.json',fmOpenRead or fmShareDenyWrite);
  27. try
  28. D:=GetJSON(F);
  29. if not (D is TJSONObject) then
  30. D.Free
  31. else
  32. Packet:=D as TJSONObject;
  33. MetaData.Arrays['fields'].Add(TJSONObject.Create(['name','id','type','int']));
  34. For I:=0 to Data.Count-1 do
  35. Data.Objects[i].Add('id',I+1);
  36. LastID:=Data.Count;
  37. finally
  38. F.Free;
  39. end;
  40. end;
  41. Procedure DoneRestData;
  42. begin
  43. Packet.Free;
  44. end;
  45. Procedure SendRecords(aFirst,aLast : integer; AResponse : TResponse);
  46. Var
  47. J: TJSONObject;
  48. A : TJSONArray;
  49. I : Integer;
  50. begin
  51. A:=TJSONArray.Create;
  52. J:=TJSONObject.Create(['metaData',MetaData.Clone,'Data',A]);
  53. try
  54. I:=AFirst;
  55. While (I<=ALast) and (I<Data.Count) do
  56. begin
  57. A.Add(Data[i].Clone);
  58. Inc(I);
  59. end;
  60. AResponse.Content:=J.FormatJSON();
  61. AResponse.ContentLength:=Length(AResponse.Content);
  62. AResponse.ContentType:='application/json';
  63. AResponse.SendContent;
  64. finally
  65. J.Free;
  66. end;
  67. end;
  68. Function IndexOfID(ID : Integer) : Integer;
  69. begin
  70. Result:=Data.Count-1;
  71. While (Result>=0) and (Data.Objects[Result].Integers['id']<>ID) do
  72. Dec(Result);
  73. end;
  74. Procedure NotFound(AResponse : TResponse);
  75. begin
  76. AResponse.Code:=404;
  77. AResponse.CodeText:='Not found';
  78. AResponse.SendResponse;
  79. end;
  80. Procedure InvalidParam(AResponse : TResponse; AContent : String = '');
  81. begin
  82. AResponse.Code:=400;
  83. AResponse.CodeText:='Bad request';
  84. AResponse.COntent:=AContent;
  85. AResponse.ContentLength:=Length(AContent);
  86. AResponse.SendResponse;
  87. end;
  88. Function GetRequestID(aRequest : TRequest) : Integer;
  89. Var
  90. P : String;
  91. I : integer;
  92. begin
  93. P:=ARequest.PathInfo;
  94. if (P<>'') and (P[1]='/') then
  95. Delete(P,1,1);
  96. I:=Pos('/',P);
  97. if I=0 then
  98. I:=Length(P);
  99. Delete(P,1,I);
  100. if (P<>'') then
  101. Result:=StrToIntDef(P,-1)
  102. else
  103. Result:=-2;
  104. end;
  105. Procedure GetRecords(aRequest : TRequest; AResponse : TResponse);
  106. Var
  107. P : String;
  108. I,ID,First,Last : Integer;
  109. begin
  110. First:=0;
  111. Last:=Data.Count-1;
  112. ID:=GetRequestID(ARequest);
  113. if ID=-1 then
  114. begin
  115. Writeln('Bad GET request: ',aRequest.PATHINFO);
  116. InvalidParam(AResponse);
  117. exit;
  118. end;
  119. if (ID<>-2) then
  120. begin
  121. First:=IndexOfID(ID);
  122. if First=-1 then
  123. begin
  124. NotFound(AResponse);
  125. exit;
  126. end;
  127. Last:=First;
  128. SendRecords(First,Last,AResponse);
  129. end
  130. else
  131. begin
  132. First:=StrToIntDef(ARequest.QueryFields.Values['Offset'],First);
  133. if (ARequest.QueryFields.Values['Limit']<>'') then
  134. Last:=First+StrToIntDef(ARequest.QueryFields.Values['Limit'],20)-1; // Default page size of 20
  135. Writeln('Count ',Data.Count,', Offset: ',First,', Last: ',Last);
  136. SendRecords(First,last,AResponse);
  137. end;
  138. end;
  139. Function GetObject(aRequest : TRequest; Out AReason : String) : TJSONObject;
  140. Var
  141. D : TJSONData;
  142. C,I : Integer;
  143. begin
  144. Result:=Nil;
  145. if Not SameText(aRequest.ContentType,'application/json') then
  146. Exit;
  147. try
  148. D:=GetJSON(aRequest.Content,true);
  149. if (D is TJSONObject) then
  150. begin
  151. Result:=D as TJSONObject;
  152. C:=Result.Count-1;
  153. D:=Nil;
  154. end
  155. else
  156. begin
  157. FreeAndNil(D);
  158. end;
  159. I:=C;
  160. While Assigned(Result) and (I>=0) do
  161. begin
  162. if (Pos(Result.Names[I]+';','id;Name;Population;')=0) then
  163. begin
  164. AReason:='Invalid property: '+Result.Names[I];
  165. FreeAndNil(Result);
  166. end;
  167. Dec(I);
  168. end;
  169. except
  170. On E : Exception do
  171. AReason:=E.ClassName+': '+E.Message;
  172. end;
  173. end;
  174. Procedure CreateRecord(aRequest : TRequest; AResponse : TResponse);
  175. Var
  176. O,D,F : TJSONObject;
  177. I : integer;
  178. R : String;
  179. begin
  180. I:=GetRequestID(ARequest);
  181. if I<>-2 then
  182. begin
  183. Writeln('Bad POST request: ',aRequest.PathInfo);
  184. R:='No ID allowed for POST';
  185. InvalidParam(AResponse,R);
  186. exit;
  187. end;
  188. O:=GetObject(ARequest,R);
  189. if (O=Nil) then
  190. begin
  191. Writeln('Bad POST request: ',aRequest.Content,' : ',R);
  192. InvalidParam(AResponse,R);
  193. exit;
  194. end;
  195. try
  196. D:=TJSONObject.Create;
  197. I:=Data.Add(D);
  198. D.Add('id',LastID);
  199. D.add('Name',O.Strings['Name']);
  200. D.add('Population',O.Int64s['Population']);
  201. Inc(LastID);
  202. SendRecords(I,I,AResponse);
  203. finally
  204. O.Free;
  205. end;
  206. end;
  207. Procedure UpdateRecord(aRequest : TRequest; AResponse : TResponse);
  208. Var
  209. ID,Idx : Integer;
  210. O,D : TJSONObject;
  211. R : String;
  212. begin
  213. ID:=GetRequestID(ARequest);
  214. if ID<0 then
  215. begin
  216. R:='Need ID for PUT';
  217. InvalidParam(AResponse,R);
  218. exit;
  219. end;
  220. IDx:=IndexOfID(ID);
  221. if IDX=-1 then
  222. begin
  223. NotFound(AResponse);
  224. exit;
  225. end;
  226. O:=GetObject(ARequest,R);
  227. if (O=Nil) then
  228. begin
  229. Writeln('Bad PUT request: ',aRequest.Content,' : ',R);
  230. InvalidParam(AResponse,R);
  231. exit;
  232. end;
  233. D:=Data.Objects[Idx];
  234. if O.IndexOfName('Name')<>-1 then
  235. D.Strings['Name']:=O.Strings['Name'];
  236. if O.IndexOfName('Population')<>-1 then
  237. D.Int64s['Population']:=O.Int64s['Population'];
  238. SendRecords(Idx,Idx,AResponse);
  239. end;
  240. Procedure DeleteRecord(aRequest : TRequest; AResponse : TResponse);
  241. Var
  242. ID,Idx : Integer;
  243. R : String;
  244. begin
  245. ID:=GetRequestID(ARequest);
  246. if ID<0 then
  247. begin
  248. R:='Need ID for DELETE';
  249. InvalidParam(AResponse,R);
  250. exit;
  251. end;
  252. ID:=GetRequestID(ARequest);
  253. IDx:=IndexOfID(ID);
  254. if IDX=-1 then
  255. begin
  256. NotFound(AResponse);
  257. exit;
  258. end;
  259. Data.Delete(Idx);
  260. AResponse.Code:=204;
  261. AResponse.CodeText:='No content';
  262. AResponse.SendResponse;
  263. end;
  264. Procedure HandleRest(aRequest : TRequest; AResponse : TResponse);
  265. begin
  266. Writeln('Method : ',ARequest.Method);
  267. Case UpperCase(ARequest.Method) of
  268. 'GET': GetRecords(ARequest,AResponse);
  269. 'POST': CreateRecord(ARequest,AResponse);
  270. 'PUT': UpdateRecord(ARequest,AResponse);
  271. 'DELETE': DeleteRecord(ARequest,AResponse);
  272. else
  273. Raise EHTTP.CreateHelp('Method not allowed',405);
  274. end;
  275. end;
  276. initialization
  277. LoadRestData;
  278. finalization
  279. DoneRestData;
  280. end.