1 files changed, 101 insertions, 0 deletions
diff --git a/www/wiki/resources/src/mediawiki/api/category.js b/www/wiki/resources/src/mediawiki/api/category.js
new file mode 100644
index 00000000..85df90e9
--- /dev/null
+++ b/www/wiki/resources/src/mediawiki/api/category.js
@@ -0,0 +1,101 @@
+/**
+ * @class mw.Api.plugin.category
+ */
+( function ( mw, $ ) {
+
+	$.extend( mw.Api.prototype, {
+		/**
+		 * Determine if a category exists.
+		 *
+		 * @param {mw.Title|string} title
+		 * @return {jQuery.Promise}
+		 * @return {Function} return.done
+		 * @return {boolean} return.done.isCategory Whether the category exists.
+		 */
+		isCategory: function ( title ) {
+			var apiPromise = this.get( {
+				formatversion: 2,
+				prop: 'categoryinfo',
+				titles: [ String( title ) ]
+			} );
+
+			return apiPromise
+				.then( function ( data ) {
+					return !!(
+						data.query && // query is missing on title=""
+						data.query.pages && // query.pages is missing on title="#" or title="mw:"
+						data.query.pages[ 0 ].categoryinfo
+					);
+				} )
+				.promise( { abort: apiPromise.abort } );
+		},
+
+		/**
+		 * Get a list of categories that match a certain prefix.
+		 *
+		 * E.g. given "Foo", return "Food", "Foolish people", "Foosball tables"...
+		 *
+		 * @param {string} prefix Prefix to match.
+		 * @return {jQuery.Promise}
+		 * @return {Function} return.done
+		 * @return {string[]} return.done.categories Matched categories
+		 */
+		getCategoriesByPrefix: function ( prefix ) {
+			// Fetch with allpages to only get categories that have a corresponding description page.
+			var apiPromise = this.get( {
+				formatversion: 2,
+				list: 'allpages',
+				apprefix: prefix,
+				apnamespace: mw.config.get( 'wgNamespaceIds' ).category
+			} );
+
+			return apiPromise
+				.then( function ( data ) {
+					return data.query.allpages.map( function ( category ) {
+						return new mw.Title( category.title ).getMainText();
+					} );
+				} )
+				.promise( { abort: apiPromise.abort } );
+		},
+
+		/**
+		 * Get the categories that a particular page on the wiki belongs to.
+		 *
+		 * @param {mw.Title|string} title
+		 * @return {jQuery.Promise}
+		 * @return {Function} return.done
+		 * @return {boolean|mw.Title[]} return.done.categories List of category titles or false
+		 *  if title was not found.
+		 */
+		getCategories: function ( title ) {
+			var apiPromise = this.get( {
+				formatversion: 2,
+				prop: 'categories',
+				titles: [ String( title ) ]
+			} );
+
+			return apiPromise
+				.then( function ( data ) {
+					var page;
+
+					if ( !data.query || !data.query.pages ) {
+						return false;
+					}
+					page = data.query.pages[ 0 ];
+					if ( !page.categories ) {
+						return false;
+					}
+					return page.categories.map( function ( cat ) {
+						return new mw.Title( cat.title );
+					} );
+				} )
+				.promise( { abort: apiPromise.abort } );
+		}
+	} );
+
+	/**
+	 * @class mw.Api
+	 * @mixins mw.Api.plugin.category
+	 */
+
+}( mediaWiki, jQuery ) );