<?php

/*

IMDb Scraper v. 1.0 - 14th of September, 2011

Scrapes information about movie and tv show titles from IMDb (imdb.com).

By Aram Kocharyan
http://ak.net84.net/php/imdb-scraper/
hide@address.com
twitter.com/akarmenia

*/

// Utility functions
require_once('util.php');

// Prevent timeout
set_time_limit(0);
ini_set('max_execution_time', 0);

Class IMDbScraper {
	
	// Performs an IMDb search and returns the info for the best match using the given query title and year
	public static function get($title, $year = NULL) {
		if ( ($result = self::find($title, $year = NULL)) !== FALSE ) {
			return self::info($result['id']);
		} else {
			return FALSE;
		}
	}
	
	// Return array of info for a given IMDb id string. eg. 'tt0206512'
	public static function info($id) {
		if (!is_string($id)) {
			return FALSE;
		} else {
			$id = preg_replace('#[^t\d]#', '', $id);
		}
		
		$url = 'http://www.imdb.com/title/' . $id . '/';
		if ( ($html = curl_get_html($url)) !== FALSE ) {
			$info = self::scrape_info($html);
			$info['id'] = $id;
			$info['url'] = $url;
			return $info;
		} else {
			return FALSE;
		}
		
	}
	
	// Returns the list of IMDb search results for the given title query.
	function search($title) {
		if ( !is_string($title) ) {
			return FALSE;
		}
		$url = 'http://www.imdb.com/find?s=tt&q=' . urlencode($title);
		$html = curl_get_html($url);
		
		return self::scrape_search($html);
	}
	
	// Performs an IMDb search and finds the best match to the given title and year.
	function find($title, $year = NULL) {
		if ( !is_string($title) || empty($title) ) {
			return FALSE;
		}
		$query = $title;
		if ( is_string($year) ) {
			$year = intval($year);
		}
		if ( is_int($year) ) {
			$query .= ' ' . $year;
		}
		
		// Get results for the search query
		$results = self::search($query);
		if ( empty($results) ) {
			return FALSE;
		}
		
		// Remove any queries that don't match the year
		if ($year !== NULL) {
			$subset = array();
			foreach ($results as $r) {
				if ( intval($r[2]) == $year ) {
					// Add result into subset, year matches
					$subset[] = $r;
				}
			}
		}
		// If no year is provided, or it was and we were left with no results, use the original results
		if ($year === NULL || empty($subset))  {
			$subset = $results;
		}
		
		// Break title query into words
		$query_bits = explode(' ', $title);
		// Get the search result titles
		$titles = array();
		foreach ($results as $r) {
			$titles[] = $r[1];
		}
		// Run a search using the words and see how many matches each search result gets
		$counts = substr_count_arrays($titles, $query_bits);
		
		// TODO check the results and see if the counts are equal (no good matches)
		
		// Get the highest count, or if they are all equal use the first result
		$highest_index = 0;
		$highest_count = $counts[0];
		for ($i = 1; $i < count($counts); $i++) {
			if ($counts[$i] > $highest_count) {
				$highest_index = $i;
			}
		}
		
		// Create an associative array, now that we have our result
		$result['id'] = $subset[$highest_index][0];
		$result['title'] = $subset[$highest_index][1];
		$result['year'] = $subset[$highest_index][2];
		
		return $result;
	}
	
	// Returns an associative array of IMDb information scrapped from an HTML string.
	public static function scrape_info($html) {
		$result = array();
		
		$result['name'] = regex_get('#<h1.*?>(.*?)<span#msi', $html, 1);
		$result['desc'] = regex_get('#"description">(.*?)</p>#msi', $html, 1);
		$date = regex_get('#datetime="(\d+)#msi', $html, 1, 'num');
		if (empty($date)) {
			$date = clean_num(regex_get('#<title>[^\(]*\(([^\)]+)\)#msi', $html, 1, 'num'));
		}
		$result['date'] = $date;
		$result['duration'] = regex_get('#class="absmiddle"[^<]*?(\d+\s*min)#msi', $html, 1);
		
		// Only for Movies
		$result['director'] = regex_get('#writer.*?([\s\w]*)</a#msi', $html, 1);
		$result['writer'] = regex_get('#writer.*?([\s\w]*)</a#msi', $html, 1);
		// Only for TV shows
		$result['creator'] = regex_get('#creator.*?([\s\w]*)</a#msi', $html, 1);
		
		$result['cast'] = array();
		if (preg_match_all('#class="name".*?>([^<]*)</a>#msi', $html, $cast)) {
			$result['cast'] = $cast[1];
		}
		
		$result['genres'] = array();
		if (preg_match_all('#/genre/([^"]*)"\s*>\1#msi', $html, $genre)) {
			$result['genres'] = $genre[1];
		}
		
		$result['plot'] = regex_get('#storyline</h2>\s*<p>(.*?)<#msi', $html, 1);
		
		$result['rating'] = regex_get('#"ratingValue">(.*?)<#msi', $html, 1, 'num');
		$result['max-rating'] = regex_get('#"bestRating">(.*?)<#msi', $html, 1, 'num');
		$result['voter-count'] = regex_get('#"ratingCount">(.*?)<#msi', $html, 1, 'num');
		$result['user-review-count'] = regex_get('#"reviewCount">(.*?)<#msi', $html, 1, 'num');
		$result['critic-review-count'] = regex_get('#(\d+) external critic#msi', $html, 1, 'num');
		
		return $result;
	}
	
	// Returns an array of search results for the given HTML string of an IMDB search page.
	// Each result is an array: (title ID, title, year)
	public static function scrape_search($html) {
		$results = array();
		if (preg_match_all('#<a\s*href\s*=\s*"([^)]*?)"[^>]*?>([^<]*)</a>\s*\((\d*)\)#msi', $html, $matches)) {
			for ($i = 0; $i < count($matches[0]); $i++) {
				$results[$i] = array( imdb_url_id($matches[1][$i]),
									  clean_str($matches[2][$i]),
									  clean_str($matches[3][$i]) );
			}
		}
		return $results;
	}
	 
}

?>