fembooru/ext/shimmie_api/main.php

<?php
/*
 * Name: [Beta] Shimmie JSON API
 * Author: Shish <webmaster@shishnet.org>
 * Description: A JSON interface to shimmie data [WARNING]
 * Documentation:
 *   <b>Admin Warning -</b> this exposes private data, eg IP addresses
 *   <p><b>Developer Warning -</b> the API is unstable; notably, private data may get hidden
 *   <p><b><u>Usage:</b></u>
 *   <p><b>get_tags</b> - List of all tags. (May contain unused tags)
 *   <br><ul>tags - <i>Optional</i> - Search for more specific tags (Searchs TAG*)</ul>
 *   <p><b>get_image</b> - Get image via id.
 *   <br><ul>id - <i>Required</i> - User id. (Defaults to id=1 if empty)</ul>
 *   <p><b>find_images</b> - List of latest 12(?) images.
 *   <p><b>get_user</b> - Get user info. (Defaults to id=2 if both are empty)
 *   <br><ul>id - <i>Optional</i> - User id.</ul>
 *   <ul>name - <i>Optional</i> - User name.</ul>
 */


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 Extension {
	public function onPageRequest(PageRequestEvent $event) {
		global $database, $page, $user;

		if($event->page_matches("api/shimmie")) {
			$page->set_mode("data");
			$page->set_type("text/plain");
			if(!$event->page_matches("api/shimmie/get_tags") && !$event->page_matches("api/shimmie/get_image") && !$event->page_matches("api/shimmie/find_images") && !$event->page_matches("api/shimmie/get_user")){
				$page->set_mode("redirect");
				$page->set_redirect(make_link("ext_doc/shimmie_api"));
			}

			if($event->page_matches("api/shimmie/get_tags")){
				$arg = $event->get_arg(0);

				if(!empty($arg)){
					$all = $database->get_all(
						"SELECT tag FROM tags WHERE tag LIKE ?",
						array($arg."%"));
				}
				elseif(isset($_GET['tag'])){
					$all = $database->get_all(
						"SELECT tag FROM tags WHERE tag LIKE ?",
						array($_GET['tag']."%"));
				}
				else {
					$all = $database->get_all("SELECT tag FROM tags");
				}
				$res = array();
				foreach($all as $row) {$res[] = $row["tag"];}
				$page->set_data(json_encode($res));
			}

			if($event->page_matches("api/shimmie/get_image")) {
				$arg = $event->get_arg(0);
				if(!empty($arg)){
					$image = Image::by_id(int_escape($event->get_arg(0)));
				}
				elseif(isset($_GET['id'])){
					$image = Image::by_id(int_escape($_GET['id']));
				}
				// FIXME: handle null image
				$image->get_tag_array(); // tag data isn't loaded into the object until necessary
				$safe_image = new _SafeImage($image);
				$page->set_data(json_encode($safe_image));
			}

			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);
				$safe_images = array();
				foreach($images as $image) {
					$image->get_tag_array();
					$safe_images[] = new _SafeImage($image);
				}
				$page->set_data(json_encode($safe_images));
			}

			if($event->page_matches("api/shimmie/get_user")) {
				$query = $user->id;
				$type = "id";
				if($event->count_args() == 1) {
					$query = $event->get_arg(0);
				}
				elseif(isset($_GET['id'])) {
					$query = $_GET['id'];
				}
				elseif(isset($_GET['name'])) {
					$query = $_GET['name'];
					$type = "name";
				}

				$all = $database->get_row(
					"SELECT id,name,joindate,class FROM users WHERE ".$type."=?",
					array($query));

				if(!empty($all)){
					//FIXME?: For some weird reason, get_all seems to return twice. Unsetting second value to make things look nice..
					// - it returns data as eg  array(0=>1234, 'id'=>1234, 1=>'bob', 'name'=>bob, ...);
					for($i=0; $i<4; $i++) unset($all[$i]);
					$all['uploadcount'] = Image::count_images(array("user_id=".$all['id']));
					$all['commentcount'] = $database->get_one(
						"SELECT COUNT(*) AS count FROM comments WHERE owner_id=:owner_id",
						array("owner_id"=>$all['id']));

					if(isset($_GET['recent'])){
						$recent = $database->get_all(
						"SELECT * FROM images WHERE owner_id=? ORDER BY id DESC LIMIT 0, 5",
						array($all['id']));

						$i = 0;
						foreach($recent as $all['recentposts'][$i]){
							unset($all['recentposts'][$i]['owner_id']); //We already know the owners id..
							unset($all['recentposts'][$i]['owner_ip']);

							for($x=0; $x<14; $x++) unset($all['recentposts'][$i][$x]);
							if(empty($all['recentposts'][$i]['author'])) unset($all['recentposts'][$i]['author']);
							if($all['recentposts'][$i]['notes'] > 0) $all['recentposts'][$i]['has_notes'] = "Y";
							else $all['recentposts'][$i]['has_notes'] = "N";
							unset($all['recentposts'][$i]['notes']);
							$i += 1;
						}
					}
				}
				$page->set_data(json_encode($all));
			}
		}
	}
}
?>