james/fembooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

api safety

Browse Source
This commit is contained in:
Shish 2010-07-09 13:50:26 +01:00
parent 06b9eebe49
commit 6eec2a369d
1 changed files with 31 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
contrib/shimmie_api/main.php
View File

@ -8,6 +8,24 @@
* <p><b>Developer Warning:</b> the API is unstable; notably, private data may get hidden
*/
class _SafeImage {
#{"id":"2","height":"768","width":"1024","hash":"71cdfaabbcdad3f777e0b60418532e94","filesize":"439561","filename":"HeilAmu.png","ext":"png","owner_ip":"0.0.0.0","posted":"0000-00-00 00:00:00","source":null,"locked":"N","owner_id":"0","rating":"u","numeric_score":"0","text_score":"0","notes":"0","favorites":"0","posted_timestamp":-62169955200,"tag_array":["cat","kunimitsu"]}
function __construct(Image $img) {
$this->id = $img->id;
$this->height = $img->height;
$this->width = $img->width;
$this->hash = $img->hash;
$this->filesize = $img->filesize;
$this->ext = $img->ext;
$this->posted = $img->posted_timestamp;
$this->source = $img->source;
$this->owner_id = $img->owner_id;
$this->tags = $img->tag_array;
}
}
class ShimmieApi extends SimpleExtension {
public function onPageRequest(PageRequestEvent $event) {
global $database, $page;
@ -16,11 +34,11 @@ class ShimmieApi extends SimpleExtension {
$page->set_mode("data");
$page->set_type("text/plain");
if($event->get_arg(0) == "get_tags") {
if($event->page_matches("api/shimmie/get_tags")) {
if($event->count_args() == 2) {
$all = $database->get_all(
"SELECT tag FROM tags WHERE tag LIKE ?",
array($event->get_arg(1)."%"));
array($event->get_arg(0)."%"));
}
else {
$all = $database->get_all("SELECT tag FROM tags");
@ -30,19 +48,24 @@ class ShimmieApi extends SimpleExtension {
$page->set_data(json_encode($res));
}
if($event->get_arg(0) == "get_image") {
$image = Image::by_id(int_escape($event->get_arg(1)));
if($event->page_matches("api/shimmie/get_image")) {
$image = Image::by_id(int_escape($event->get_arg(0)));
$image->get_tag_array(); // tag data isn't loaded into the object until necessary
$page->set_data(json_encode($image));
$safe_image = new _SafeImage($image);
$page->set_data(json_encode($safe_image));
}
if($event->get_arg(0) == "find_images") {
if($event->page_matches("api/shimmie/find_images")) {
$search_terms = $event->get_search_terms();
$page_number = $event->get_page_number();
$page_size = $event->get_page_size();
$images = Image::find_images(($page_number-1)*$page_size, $page_size, $search_terms);
foreach($images as $image) $image->get_tag_array();
$page->set_data(json_encode($images));
$safe_images = array();
foreach($images as $image) {
$image->get_tag_array();
$safe_images[] = new _SafeImage($image);
}
$page->set_data(json_encode($safe_images));
}
}
}