1 files changed, 103 insertions, 0 deletions
diff --git a/scraper/themoviedbscraper/themoviedbscraper.c b/scraper/themoviedbscraper/themoviedbscraper.c
new file mode 100644
index 0000000..62263a9
--- /dev/null
+++ b/scraper/themoviedbscraper/themoviedbscraper.c
@@ -0,0 +1,103 @@
+
+#include <string>
+#include <sstream>
+#include <vector>
+#include <map>
+#include <algorithm>
+#include <iostream>
+#include <jansson.h>
+
+#include "../../lib/curl.h"
+#include "../../tools/stringhelpers.h"
+#include "themoviedbscraper.h"
+
+using namespace std;
+
+cMovieDBScraper::cMovieDBScraper(string language) {
+    apiKey = "abb01b5a277b9c2c60ec0302d83c5ee9";
+    this->language = language;
+    baseURL = "api.themoviedb.org/3";
+    posterSize = "w500";
+    backdropSize = "w1280";
+    actorthumbSize = "h632";
+}
+
+cMovieDBScraper::~cMovieDBScraper() {
+}
+
+cMovieDbMovie *cMovieDBScraper::Scrap(string movieName, string year) {
+    int movieID = SearchMovie(movieName, year);
+    if (movieID < 1) {
+        return NULL;
+    }
+    cMovieDbMovie *movie = ReadMovie(movieID);
+    if (!movie)
+        return NULL;
+    return movie;
+}
+
+bool cMovieDBScraper::Connect(void) {
+    stringstream url;
+    url << baseURL << "/configuration?api_key=" << apiKey;
+    string configJSON;
+    if (curl.GetUrl(url.str().c_str(), &configJSON)) {
+        return parseJSON(configJSON);
+    }
+    return false;
+}
+
+bool cMovieDBScraper::parseJSON(string jsonString) {
+    cJsonLoader root(jsonString.c_str());
+    if (!root.isObject()) {
+        return false;
+    }
+    json_t *images = root.objectByName("images");
+    if(!json_is_object(images)) {
+        return false;
+    }
+    
+    json_t *imgUrl;
+    imgUrl = json_object_get(images, "base_url");
+    if(!json_is_string(imgUrl)) {
+        return false;
+    }
+    imageUrl = json_string_value(imgUrl);
+    return true;
+}
+
+int cMovieDBScraper::SearchMovie(string movieName, string year) {
+    stringstream url;
+    string movieJSON;
+    int movieID = -1;
+    char* escUrl = curl.EscapeUrl(movieName.c_str());
+
+    url << baseURL << "/search/movie?api_key=" << apiKey << "&query=" << escUrl << "&language=" << language.c_str();
+    if (year.size() > 0)
+        url << "&year=" << year.c_str();
+    curl.Free(escUrl);
+
+    if (curl.GetUrl(url.str().c_str(), &movieJSON)) {
+        cMovieDbMovie movie(movieJSON);
+        movieID = movie.ParseJSONForMovieId(movieName);
+    }
+
+    return movieID;
+}
+
+cMovieDbMovie *cMovieDBScraper::ReadMovie(int movieID) {
+    stringstream url;
+    url << baseURL << "/movie/" << movieID << "?api_key=" << apiKey << "&language=" << language.c_str();
+    string movieJSON;
+    cMovieDbMovie *movie = NULL;
+    if (curl.GetUrl(url.str().c_str(), &movieJSON)) {
+        movie = new cMovieDbMovie(movieJSON);
+        movie->id = movieID;
+        movie->SetBaseUrl(baseURL);
+        movie->SetApiKey(apiKey);
+        movie->SetPosterBaseUrl(imageUrl + posterSize);
+        movie->SetBackdropBaseUrl(imageUrl + backdropSize);
+        movie->SetActorBaseUrl(imageUrl + actorthumbSize);
+        movie->ParseJSON();
+    }
+    return movie;
+}