An unfinished system to manage all your paper documentation in an easy way.
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

autodoc.fcgi 10KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467
  1. #!/usr/bin/perl
  2. use strict;
  3. use FCGI;
  4. use JSON;
  5. use DBI;
  6. use GD::Simple;
  7. use Data::Dumper;
  8. use warnings;
  9. $Data::Dumper::Sortkeys = 1;
  10. my $conf = load_conf("../etc/autodoc.json");
  11. my $dbh = sqlconnect($conf->{sql});
  12. my %map = (
  13. api_v1_POST_documents => \&api_v1_POST_documents,
  14. api_v1_POST_documents_id_data => \&api_v1_POST_documents_id_data,
  15. api_v1_GET_documents_id_image => \&api_v1_GET_documents_id_image,
  16. api_v1_GET_pages_image => \&api_v1_GET_pages_image,
  17. api_v1_GET_documents => \&api_v1_GET_documents,
  18. api_v1_GET_documents_id => \&api_v1_GET_documents_id,
  19. api_v1_GET_pages_id => \&api_v1_GET_pages_id,
  20. api_v1_PATCH_documents_id => \&api_v1_PATCH_documents_id,
  21. );
  22. my $request = FCGI::Request();
  23. while($request->Accept() >= 0) {
  24. my $user = $ENV{REMOTE_USER} || 'undefined';
  25. my $qs = parse_querystring($ENV{QUERY_STRING});
  26. my $method = $ENV{REQUEST_METHOD};
  27. # QS sanity check
  28. if ( exists $qs->{id} && $qs->{id} !~ /^\d+$/ ) { fatal_api_error(400,"invalid id"); }
  29. if ( exists $qs->{maxWidth} && $qs->{maxWidth} !~ /^\d+$/ ) { fatal_api_error(400,"invalid maxWidth"); }
  30. if ( exists $qs->{maxHeight} && $qs->{maxHeight} !~ /^\d+$/ ) { fatal_api_error(400,"invalid maxHeight"); }
  31. if ( exists $qs->{pageSize} && $qs->{pageSize} !~ /^\d+$/ ) { fatal_api_error(400,"invalid pageSize"); }
  32. if ( exists $qs->{pageIndex} && $qs->{pageIndex} !~ /^\d+$/ ) { fatal_api_error(400,"invalid pageIndex"); }
  33. my $path = [ split(/\//,$ENV{SCRIPT_NAME}) ] if exists $ENV{SCRIPT_NAME};
  34. shift(@{$path});
  35. my $post = parse_post(
  36. \*STDIN,
  37. exists $ENV{CONTENT_LENGTH} ? $ENV{CONTENT_LENGTH} : 0,
  38. exists $ENV{CONTENT_TYPE} ? $ENV{CONTENT_TYPE} : 0
  39. );
  40. my($code, $type, $data) = process_query($method, $path, $qs, $post, $user);
  41. if ( defined $type ) {
  42. if ( $type eq 'application/json' ) {
  43. $data = to_json($data, { utf8 => 1, pretty => 1 });
  44. }
  45. }
  46. send_response($code, $type, $data);
  47. }
  48. sub send_response {
  49. my ($code, $type, $data) = @_;
  50. printf("Status: %s\r\n", $code);
  51. printf("Content-type: %s\r\n", $type) if defined $type;
  52. printf("Content-length: %i\r\n", length($data)) if defined $data;
  53. printf("\r\n");
  54. print $data if defined $data;
  55. }
  56. sub fatal_api_error {
  57. my($code,$type,$body)=api_error(@_);
  58. send_response($code, $type, $body);
  59. exit;
  60. }
  61. sub load_conf {
  62. my($file) = @_;
  63. my $x='';
  64. open(F,"$file") || fatal_api_error(500,"Failed to load configuration file");
  65. while(<F>) { $x.=$_; }
  66. close(F);
  67. return from_json($x);
  68. }
  69. sub process_query {
  70. my($method, $path, $qs, $post, $user) = @_;
  71. my ($api_version, $path_main, $path_id, $path_sub) = @{$path};
  72. return api_error(404, "Unknown API version") if !defined $api_version;
  73. return api_error(405, "Unknown METHOD") if !defined $method;
  74. return api_error(404, "Unknown API function") if !defined $path_main;
  75. my $func = 'api_' . $api_version . '_' . $method . '_' . $path_main;
  76. $func .= '_id' if defined $path_id;
  77. $func .= '_'.$path_sub if defined $path_sub;
  78. return $map{$func}->($path_id, $qs, $post, $user) if exists $map{$func};
  79. return api_error(404, "Invalid path/method combination");
  80. }
  81. sub db_get_document_object {
  82. my($id) = @_;
  83. my $document;
  84. my @pages;
  85. my @pageids;
  86. my @tags;
  87. my $q = sqlquery($dbh, "SELECT * FROM documents WHERE id = ?", $id);
  88. while(my $hash = $q->fetchrow_hashref()) { $document = $hash; }
  89. $q = sqlquery($dbh, "SELECT * FROM pages WHERE documentId = ?", $id);
  90. while(my $hash = $q->fetchrow_hashref()) { push @pages, $hash; push @pageids, $hash->{id}; }
  91. $q = sqlquery($dbh, "
  92. SELECT tags.tag AS tag
  93. FROM documents_tags LEFT JOIN tags ON documents_tags.tagId = tags.id
  94. WHERE documentId = ? ORDER BY tag", $id);
  95. while(my ($tag) = $q->fetchrow_array()) { push @tags, $tag; }
  96. my %lang;
  97. foreach my $page ( @pages ) {
  98. $q = sqlquery($dbh, "SELECT * FROM pages_lang WHERE pageId = ?", $page->{id});
  99. while(my $hash = $q->fetchrow_hashref()) {
  100. $lang{$hash->{language}}++;
  101. }
  102. }
  103. if ( defined $document ) {
  104. my $out = {
  105. id => $document->{id},
  106. pageId => [ @pageids ],
  107. name => $document->{name},
  108. created => $document->{created},
  109. owner => $document->{owner},
  110. status => $document->{status},
  111. languages => [ keys %lang ],
  112. tags => [ @tags ],
  113. };
  114. return (200, "application/json", $out);
  115. }
  116. return (404, "application/json", "");
  117. }
  118. sub db_get_page_object {
  119. my($id) = @_;
  120. my $out;
  121. my %lang;
  122. my $q = sqlquery($dbh, "SELECT * FROM pages_lang WHERE pageId = ?", $id);
  123. while(my $hash = $q->fetchrow_hashref()) {
  124. $lang{$hash->{language}}++;
  125. }
  126. $q = sqlquery($dbh, "SELECT * FROM pages WHERE documentId = ?", $id);
  127. while(my $hash = $q->fetchrow_hashref()) {
  128. $out = {
  129. id => $id,
  130. documentId => $hash->{documentId},
  131. name => $hash->{name},
  132. created => $hash->{created},
  133. owner => $hash->{owner},
  134. status => $hash->{status},
  135. language => [ keys %lang ],
  136. };
  137. }
  138. if ( exists $out->{id} ) {
  139. return (200, "application/json", $out);
  140. }
  141. return (404, "application/json", "");
  142. }
  143. sub get_page_image {
  144. my($id, $qs) = @_;
  145. my $wh;
  146. my $size;
  147. if ( exists $qs->{maxWidth} ) {
  148. $wh = 'w';
  149. $size = $qs->{maxWidth};
  150. }
  151. elsif ( exists $qs->{maxHeight} ) {
  152. $wh = 'h';
  153. $size = $qs->{maxWidth};
  154. }
  155. my $src;
  156. my $dst;
  157. if ( !defined $id ) {
  158. $src = sprintf("%s/%s",
  159. $conf->{path}{global}, $conf->{path}{error_img});
  160. }
  161. else {
  162. $src = sprintf("%s/%s/%s.jpeg",
  163. $conf->{path}{global}, $conf->{path}{original}, $id);
  164. }
  165. if ( defined $wh && defined $size ) {
  166. if ( defined $id ) {
  167. $dst = sprintf("%s/%s/%s-%s-%s.jpeg",
  168. $conf->{path}{global}, $conf->{path}{cache}, $id, $wh, $size);
  169. }
  170. else {
  171. $dst = sprintf("%s/%s/error_img-%s-%s.jpeg",
  172. $conf->{path}{global}, $conf->{path}{cache}, $wh, $size);
  173. }
  174. }
  175. my $imgfile;
  176. if ( !defined $dst ) {
  177. $imgfile = $src;
  178. }
  179. else {
  180. if ( ! -r $dst ) {
  181. my $cmd = sprintf("convert %s -resize %s%s %s",
  182. $src,
  183. $wh eq 'h' ? 'x' : '',
  184. $size,
  185. $dst
  186. );
  187. print STDERR "CMD=$cmd\n";
  188. system($cmd);
  189. }
  190. $imgfile = $dst;
  191. }
  192. my $img = '';
  193. open(IMG, $imgfile) || fatal_api_error(500,"Failed to generate image");
  194. binmode(IMG);
  195. while(<IMG>) { $img.=$_; }
  196. close(IMG);
  197. return $img;
  198. }
  199. # create an empty document object.
  200. sub api_v1_POST_documents {
  201. my($id, $qs, $post, $user) = @_;
  202. my $q = sqlquery($dbh, "
  203. INSERT INTO documents
  204. SET
  205. owner = ?,
  206. status = 'nodata'
  207. ", $user);
  208. $q = sqlquery($dbh, "SELECT LAST_INSERT_ID()");
  209. while(my($lastid) = $q->fetchrow_array()) {
  210. $id = $lastid;
  211. }
  212. return db_get_document_object($id);
  213. }
  214. sub api_v1_POST_documents_id_data {
  215. my($id, $qs, $post, $user) = @_;
  216. open(TMP,">/tmp/upload.meta");
  217. print TMP $post->{ctype} . "\n";
  218. print TMP $post->{len} . "\n";
  219. close(TMP);
  220. open(TMP,">/tmp/upload.data");
  221. print TMP $post->{data} . "\n";
  222. close(TMP);
  223. return (200, "application/json", { })
  224. }
  225. # get document image/thumbnail
  226. sub api_v1_GET_documents_id_image {
  227. my($id, $qs, $post, $user) = @_;
  228. my $pageid;
  229. my $q = sqlquery($dbh, "SELECT id FROM pages WHERE documentId = ? ORDER BY id DESC LIMIT 1", $id);
  230. while(my($myid)=$q->fetchrow_array()) {
  231. $pageid = $myid;
  232. }
  233. return api_v1_GET_pages_id_image($pageid, $qs, $post, $user);
  234. }
  235. # get page image/thumbnail
  236. sub api_v1_GET_pages_id_image {
  237. my($id, $qs, $post, $user) = @_;
  238. return (200, "image/jpeg", get_page_image($id, $qs));
  239. }
  240. # get a list of document objects
  241. sub api_v1_GET_documents {
  242. my($id, $qs, $post, $user) = @_;
  243. $qs->{pageSize} = $conf->{query}{pageSize} if !exists $qs->{pageSize};
  244. $qs->{pageIndex} = $conf->{query}{pageIndex} if !exists $qs->{pageIndex};
  245. my @out;
  246. my $q = sqlquery($dbh, "SELECT id FROM documents LIMIT ?,?",
  247. $qs->{pageSize} * $qs->{pageIndex},
  248. $qs->{pageSize});
  249. while(my ($id) = $q->fetchrow_array()) {
  250. my ($code, $ct, $body) = db_get_document_object($id);
  251. push @out, $body;
  252. }
  253. return (
  254. 200,
  255. "application/json",
  256. \@out
  257. );
  258. }
  259. # get a single document object
  260. sub api_v1_GET_documents_id {
  261. my($id, $qs, $post, $user) = @_;
  262. return db_get_document_object($id);
  263. }
  264. # get a specific page
  265. sub api_v1_GET_pages_id {
  266. my($id, $qs, $post, $user) = @_;
  267. return db_get_page_object($id);
  268. }
  269. # change document properties
  270. sub api_v1_PATCH_documents_id {
  271. my($id, $qs, $post, $user) = @_;
  272. if ( exists $qs->{addTags} ) {
  273. my $tags = get_array($qs->{addTags});
  274. foreach my $tag ( @{$tags} ) {
  275. sqlquery($dbh, "CALL tag_add(?,?)", $id, $tag);
  276. }
  277. }
  278. if ( exists $qs->{deleteTags} ) {
  279. my $tags = get_array($qs->{deleteTags});
  280. foreach my $tag ( @{$tags} ) {
  281. sqlquery($dbh, "CALL tag_del(?,?)", $id, $tag);
  282. }
  283. }
  284. if ( exists $qs->{name} ) {
  285. sqlquery($dbh, "UPDATE documents SET name = ? WHERE id = ?",
  286. $qs->{name}, $id);
  287. }
  288. return (200);
  289. }
  290. sub get_array {
  291. my($x) = @_;
  292. my @arr;
  293. if ( ref $x eq 'ARRAY' ) {
  294. @arr = @{$x};
  295. }
  296. else {
  297. @arr = [ $x ];
  298. }
  299. return \@arr;
  300. }
  301. sub api_error {
  302. my($code, $text)=@_;
  303. my %deftext = (
  304. "000" => "An error produced an internal error, cascading errors over errors",
  305. "404" => "No such API path"
  306. );
  307. $code = "000" if !defined $code;
  308. $text = $deftext{$code} if ( !defined $text || $code eq "000" );
  309. return ( $code, "text/plain", $text . "\r\n" );
  310. }
  311. sub parse_querystring {
  312. my($in) = @_;
  313. my %out;
  314. if ( defined $in && length $in ) {
  315. foreach(split(/&/,$in)) {
  316. my($name,$value) = split(/=/);
  317. $value =~ s/%([a-fA-F0-9][a-fA-F0-9])/pack("C", hex($1))/eg;
  318. # handle arrays
  319. if ( exists $out{$name} ) {
  320. if ( ! ref $out{$name} ) {
  321. my $old = $out{$name};
  322. $out{$name} = [ $old ];
  323. }
  324. push @{$out{$name}}, $value;
  325. }
  326. else {
  327. $out{$name}=$value;
  328. }
  329. }
  330. }
  331. return \%out;
  332. }
  333. sub parse_post {
  334. my($fd, $len, $ct) = @_;
  335. my $data = '';
  336. while ( $len > 0 ) {
  337. my $buf;
  338. my $read = read($fd, $buf, $len);
  339. $len -= $read;
  340. $data .= $buf;
  341. }
  342. if ( $ct eq 'application/json' ) {
  343. my $tmp = from_json($data);
  344. $data = $tmp;
  345. }
  346. return { ctype => $ct, len => $len, data => $data);
  347. }
  348. sub sqlconnect {
  349. my($sql) = @_;
  350. my $dsn = "DBI:mysql:database=$sql->{base};host=$sql->{host}";
  351. my $dbh = DBI->connect($dsn, $sql->{user}, $sql->{pass}) || \\
  352. fatal_api_error(500,"Failed to connect to database");
  353. return $dbh;
  354. }
  355. sub sqlquery {
  356. my $dbh = shift;
  357. my $query = shift;
  358. my @args = @_;
  359. my $sth = $dbh->prepare($query) || fatal_api_error(500,"Failed to execute SQL query");
  360. $sth->execute(@args) || fatal_api_error(500,"Failed to execute SQL query");
  361. return $sth;
  362. }