Merge branch 'master' of https://github.com/ctrlcctrlv/Tinyboard

3 files changed, 90 insertions, 53 deletions

diff --git a/inc/api.php b/inc/api.php
index 5592d8af..69a58a0c 100644
--- a/inc/api.php
+++ b/inc/api.php
@@ -1,49 +1,47 @@
 <?php
-
 /*
  *  Copyright (c) 2010-2013 Tinyboard Development Group
  */
 
+
 /**
  * Class for generating json API compatible with 4chan API
  */
 class Api {
+	function __construct($config){
+		/**
+		 * Translation from local fields to fields in 4chan-style API
+		 */
+		$this->config = $config;
 
-	/**
-	 * Translation from local fields to fields in 4chan-style API
-	 */
-	public static $postFields = array(
-		'id' => 'no',
-		'thread' => 'resto',
-		'subject' => 'sub',
-		'email' => 'email',
-		'name' => 'name',
-		'trip' => 'trip',
-		'capcode' => 'capcode',
-		'body' => 'com',
-		'time' => 'time',
-		'thumb' => 'thumb', // non-compatible field
-		'thumbx' => 'tn_w',
-		'thumby' => 'tn_h',
-		'file' => 'file', // non-compatible field
-		'filex' => 'w',
-		'filey' => 'h',
-		'filesize' => 'fsize',
-		//'filename' => 'filename',
-		'omitted' => 'omitted_posts',
-		'omitted_images' => 'omitted_images',
-		//'posts' => 'replies',
-		//'ip' => '',
-		'sticky' => 'sticky',
-		'locked' => 'locked',
-		//'bumplocked' => '',
-		//'embed' => '',
-		//'root' => '',
-		//'mod' => '',
-		//'hr' => '',
-	);
+		$this->postFields = array(
+			'id' => 'no',
+			'thread' => 'resto',
+			'subject' => 'sub',
+			'body' => 'com',
+			'email' => 'email',
+			'name' => 'name',
+			'trip' => 'trip',
+			'capcode' => 'capcode',
+			'time' => 'time',
+			'thumbx' => 'tn_w',
+			'thumby' => 'tn_h',
+			'filex' => 'w',
+			'filey' => 'h',
+			'filesize' => 'fsize',
+			'filename' => 'filename',
+			'omitted' => 'omitted_posts',
+			'omitted_images' => 'omitted_images',
+			'sticky' => 'sticky',
+			'locked' => 'locked',
+		);
+
+		if (isset($config['api']['extra_fields']) && gettype($config['api']['extra_fields']) == 'array'){
+			$this->postFields = array_merge($this->postFields, $config['api']['extra_fields']);
+		}
+	}
 
-	static $ints = array(
+	private static $ints = array(
 		'no' => 1,
 		'resto' => 1,
 		'time' => 1,
@@ -60,7 +58,7 @@ class Api {
 
 	private function translatePost($post) {
 		$apiPost = array();
-		foreach (self::$postFields as $local => $translated) {
+		foreach ($this->postFields as $local => $translated) {
 			if (!isset($post->$local))
 				continue;
 
@@ -69,6 +67,7 @@ class Api {
 			if ($val !== null && $val !== '') {
 				$apiPost[$translated] = $toInt ? (int) $val : $val;
 			}
+
 		}
 
 		if (isset($post->filename)) {
@@ -77,6 +76,18 @@ class Api {
 			$apiPost['ext'] = substr($post->filename, $dotPos);
 		}
 
+		// Handle country field
+		if (isset($post->body_nomarkup) && $this->config['country_flags']) {
+			$modifiers = extract_modifiers($post->body_nomarkup);
+			if (isset($modifiers['flag']) && isset($modifiers['flag alt'])) {
+				$country = mb_convert_case($modifiers['flag'], MB_CASE_UPPER);
+				if ($country) {
+					$apiPost['country'] = $country;
+					$apiPost['country_name'] = $modifiers['flag alt'];
+				}
+			}
+		}
+
 		return $apiPost;
 	}
 
diff --git a/inc/config.php b/inc/config.php
index ae3ce0c0..956851f2 100644
--- a/inc/config.php
+++ b/inc/config.php
@@ -1343,6 +1343,21 @@
 	// });
 
 /*
+ * =============
+ *  API settings
+ * =============
+ */
+
+	// Whether or not to use the API, disabled by default.
+	$config['api']['enabled'] = false;
+
+	// Extra fields in to be shown in the array that are not 4chan API compatible.
+	// You canget these by taking a look at the schema for posts_ tables. The array should be formatted as $db_name => $translated_name.
+	// For example:
+	
+	// $config['api']['extra_fields'] = array('body_nomarkup'=>'com_nomarkup');
+
+/*
  * ====================
  *  Other/uncategorized
  * ====================
diff --git a/inc/functions.php b/inc/functions.php
index a75ebb8a..02e2a932 100644
--- a/inc/functions.php
+++ b/inc/functions.php
@@ -959,6 +959,7 @@ function deletePost($id, $error_if_doesnt_exist=true, $rebuild_after=true) {
 			// Delete thread HTML page
 			file_unlink($board['dir'] . $config['dir']['res'] . sprintf($config['file_page'], $post['id']));
 			file_unlink($board['dir'] . $config['dir']['res'] . sprintf($config['file_page50'], $post['id']));
+			file_unlink($board['dir'] . $config['dir']['res'] . sprintf('%d.json', $post['id']));
 
 			$antispam_query = prepare('DELETE FROM ``antispam`` WHERE `board` = :board AND `thread` = :thread');
 			$antispam_query->bindValue(':board', $board['uri']);
@@ -1299,8 +1300,10 @@ function buildIndex() {
 	if (!$config['try_smarter'])
 		$antibot = create_antibot($board['uri']);
 
-	$api = new Api();
-	$catalog = array();
+	if ($config['api']['enabled']) {
+		$api = new Api($config);
+		$catalog = array();
+	}
 
 	for ($page = 1; $page <= $config['max_pages']; $page++) {
 		$filename = $board['dir'] . ($page == 1 ? $config['file_index'] : sprintf($config['file_page'], $page));
@@ -1324,27 +1327,33 @@ function buildIndex() {
 		file_write($filename, Element('index.html', $content));
 
 		// json api
-		$threads = $content['threads'];
-		$json = json_encode($api->translatePage($threads));
-		$jsonFilename = $board['dir'] . ($page-1) . ".json"; // pages should start from 0
-		file_write($jsonFilename, $json);
+		if ($config['api']['enabled']) {
+			$threads = $content['threads'];
+			$json = json_encode($api->translatePage($threads));
+			$jsonFilename = $board['dir'] . ($page-1) . ".json"; // pages should start from 0
+			file_write($jsonFilename, $json);
 
-		$catalog[$page-1] = $threads; 
+			$catalog[$page-1] = $threads; 
+		}
 	}
 	if ($page < $config['max_pages']) {
 		for (;$page<=$config['max_pages'];$page++) {
 			$filename = $board['dir'] . ($page==1 ? $config['file_index'] : sprintf($config['file_page'], $page));
 			file_unlink($filename);
-
-			$jsonFilename = $board['dir'] . ($page-1) . ".json";
-			file_unlink($jsonFilename);
+			
+			if ($config['api']['enabled']) {
+				$jsonFilename = $board['dir'] . ($page-1) . ".json";
+				file_unlink($jsonFilename);
+			}
 		}
 	}
 
 	// json api catalog
-	$json = json_encode($api->translateCatalog($catalog));
-	$jsonFilename = $board['dir'] . "catalog.json";
-	file_write($jsonFilename, $json);
+	if ($config['api']['enabled']) {
+		$json = json_encode($api->translateCatalog($catalog));
+		$jsonFilename = $board['dir'] . "catalog.json";
+		file_write($jsonFilename, $json);
+	}
 }
 
 function buildJavascript() {
@@ -1765,10 +1774,12 @@ function buildThread($id, $return = false, $mod = false) {
 		$build_pages[] = thread_find_page($id);
 
 	// json api
-	$api = new Api();
-	$json = json_encode($api->translateThread($thread));
-	$jsonFilename = $board['dir'] . $config['dir']['res'] . $id . ".json";
-	file_write($jsonFilename, $json);
+	if ($config['api']['enabled']) {
+		$api = new Api($config);
+		$json = json_encode($api->translateThread($thread));
+		$jsonFilename = $board['dir'] . $config['dir']['res'] . $id . ".json";
+		file_write($jsonFilename, $json);
+	}
 
 	if ($return) {
 		return $body;