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 {
	public $id;
	public $height;
	public $width;
	public $hash;
	public $filesize;
	public $ext;
	public $posted;
	public $source;
	public $owner_id;
	public $tags;

	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   = strtotime($img->posted);
		$this->source   = $img->source;
		$this->owner_id = $img->owner_id;
		$this->tags     = $img->get_tag_array();
	}
}

class ShimmieApi extends Extension {
	public function onPageRequest(PageRequestEvent $event) {
		global $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")){
				$tag = $event->get_arg(0);
				if(empty($tag) && isset($_GET['tag'])) $tag = $_GET['tag'];
				$res = $this->api_get_tags($tag);
				$page->set_data(json_encode($res));
			}

			elseif($event->page_matches("api/shimmie/get_image")) {
				$arg = $event->get_arg(0);
				if(empty($arg) && isset($_GET['id'])) $arg = $_GET['id'];
				$image = Image::by_id(int_escape($arg));
				// 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));
			}

			elseif($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));
			}

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

				$all = $this->api_get_user($type, $query);
				$page->set_data(json_encode($all));
			}

			else {
				$page->set_mode("redirect");
				$page->set_redirect(make_link("ext_doc/shimmie_api"));
			}

		}
	}

	/**
	 * @param string $arg
	 * @return string[]
	 */
	private function api_get_tags($arg) {
		global $database;
		if (!empty($arg)) {
			$all = $database->get_all("SELECT tag FROM tags WHERE tag LIKE ?", array($arg . "%"));
		} else {
			$all = $database->get_all("SELECT tag FROM tags");
		}
		$res = array();
		foreach ($all as $row) {
			$res[] = $row["tag"];
		}
		return $res;
	}

	/**
	 * @param $type
	 * @param $query
	 * @return array
	 */
	private function api_get_user($type, $query) {
		global $database;
		$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;
				}
			}
		}
		return $all;
	}
}