1 files changed, 82 insertions, 0 deletions
diff --git a/moviedbmanager.h b/moviedbmanager.h
new file mode 100644
index 0000000..f4e8f5a
--- /dev/null
+++ b/moviedbmanager.h
@@ -0,0 +1,82 @@
+
+#ifndef __TVSCRAPER_MOVIEDBMANAGER_H
+#define __TVSCRAPER_MOVIEDBMANAGER_H
+
+#include <iostream>
+#include <string>
+#include <sstream>
+#include <vector>
+#include <map>
+#include <set>
+#include <algorithm>
+
+#include "lib/common.h"
+#include "lib/db.h"
+#include "lib/epgservice.h"
+
+#include "tools/stringhelpers.h"
+#include "scraper/themoviedbscraper/themoviedbscraper.h"
+#include "scraper/themoviedbscraper/moviedbmovie.h"
+#include "scraper/themoviedbscraper/moviedbactor.h"
+
+#include "epgdconfig.h"
+
+struct sMovieResult {
+    tEventId eventId;
+    string title;
+    string year;
+    int lastScraped;
+};
+
+// --- cMovieDBManager -------------------------------------------------------------
+
+class cMovieDBManager {
+private:
+    int bytesDownloaded;
+    int withutf8;
+    string language;
+    map<string, int> alreadyScraped;
+    cMovieDBScraper *movieDbScraper;
+    cDbConnection *connection;
+    cDbTable *tMovies;
+    cDbTable *tMoviesActor;
+    cDbTable *tMoviesActors;
+    cDbTable *tMovieMedia;
+    cDbTable *tEvents;
+//    cDbTable *tRecordings;
+    cDbTable *tRecordingList;
+    enum mediaType {
+         mtPoster,
+         mtFanart,
+         mtCollectionPoster,
+         mtCollectionFanart,
+         mtActorThumb
+    };
+    int LoadMovieFromDB(string title);
+    void UpdateEvent(tEventId eventID, int movieID);
+    void SaveMovie(cMovieDbMovie *movie);
+    void SaveMovieBasics(cMovieDbMovie *movie);
+    void SaveMovieMedia(cMovieDbMovie *movie);
+    void SaveMovieActors(cMovieDbMovie *movie);
+    bool LoadMedia(int movieId, int actorId, int mediaType);
+    int GetPicture(const char* url, MemoryStruct* data);
+    void DeleteMovie(int movieId);
+    cMovieDbMovie *SearchRecordingSophisticated(string name);
+public:
+    cMovieDBManager(void);
+    virtual ~cMovieDBManager(void);
+    int ConnectDatabase(cDbConnection *conn);
+    bool ConnectScraper(void);
+    void ResetBytesDownloaded(void) { bytesDownloaded = 0; };
+    int GetBytesDownloaded(void) { return bytesDownloaded; };
+    bool GetMoviesFromEPG(vector<sMovieResult> *result);
+    void ProcessMovie(sMovieResult mov);
+    bool SearchRecordingDB(string name, int &movieId);
+    bool SearchRecordingOnline(string name, int &movieId);
+    bool CheckScrapInfoDB(int scrapMovieId);
+    bool CheckScrapInfoOnline(int scrapMovieId);
+    int CleanupMovies(void);
+};
+
+
+#endif //__TVSCRAPER_MOVIEDBMANAGER_H