1 files changed, 108 insertions, 0 deletions
diff --git a/tvdbmanager.h b/tvdbmanager.h
new file mode 100644
index 0000000..1045432
--- /dev/null
+++ b/tvdbmanager.h
@@ -0,0 +1,108 @@
+
+#ifndef __TVSCRAPER_TVDBMANAGER_H
+#define __TVSCRAPER_TVDBMANAGER_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 "epgdconfig.h"
+
+#include "scraper/thetvdbscraper/thetvdbscraper.h"
+#include "scraper/thetvdbscraper/tvdbseries.h"
+#include "scraper/thetvdbscraper/tvdbepisode.h"
+#include "scraper/thetvdbscraper/tvdbmedia.h"
+#include "scraper/thetvdbscraper/tvdbactor.h"
+
+struct sSeriesResult {
+    tEventId eventId;
+    string title;
+    int lastScraped;
+    int season;
+    int part;
+    int number;
+    string episodeName;  
+};
+
+// --- cTVDBManager -------------------------------------------------------------
+
+class cTVDBManager {
+private:
+    int bytesDownloaded;
+    int withutf8;
+    string language;
+    int serverTime;
+    map<string, int> alreadyScraped;
+    cTVDBScraper *tvdbScraper;
+    cDbConnection *connection;
+    cDbTable *tSeries;
+    cDbTable *tSeriesEpsiode;
+    cDbTable *tSeriesMedia;
+    cDbTable *tSeriesActor;
+    cDbTable *tEvents;
+    cDbTable *tEpisodes;
+    cDbTable *tRecordingList;
+
+    enum mediaType {
+         mtBanner1,
+         mtBanner2,
+         mtBanner3,
+         mtPoster1,
+         mtPoster2,
+         mtPoster3,
+         mtSeasonPoster,
+         mtFanart1,
+         mtFanart2,
+         mtFanart3,
+         mtEpisodePic,
+         mtActorThumb
+    };
+    int GetLastScrap(void);
+    void UpdateScrapTS(void);
+    bool GetAllIDs(set<int> *IDs, cDbTable *table, const char* fname);
+    cTVDBSeries *ScrapSeries(string search);
+    cTVDBSeries *ScrapSeries(int seriesID);
+    void SaveSeries(cTVDBSeries *series);
+    void SaveSeriesBasics(cTVDBSeries *series);
+    void SaveSeriesMedia(cTVDBSeries *series);
+    bool SavePosterBannerFanart(int mediaType, cTVDBMedia *media, int seriesID);
+    void SaveSeriesEpisodes(cTVDBSeries *series);
+    void SaveSeriesEpisode(cTVDBEpisode *episode, int seriesID = 0);
+    void SaveSeriesActors(cTVDBSeries *series);
+    bool LoadMedia(int seriesId, int seasonNumber, int episodeId, int actorId, int mediaType);
+    bool ReloadImage(string url);
+    int GetPicture(const char* url, MemoryStruct* data);
+    int LoadSeriesFromDB(string name);
+    void GetSeasonEpisodeFromEpisodename(int seriesID, int &season, int &part, string episodeName);
+    void LoadSeasonPoster(int seriesID, int season);
+    int LoadEpisodePicture(int seriesID, int season, int part);
+    void UpdateEvent(tEventId eventID, int seriesID, int episodeID);
+    void DeleteSeries(int seriesId);
+    int LoadEpisode(string name, int seriesId);
+public:
+    cTVDBManager(void);
+    virtual ~cTVDBManager(void);
+    int ConnectDatabase(cDbConnection *conn);
+    bool ConnectScraper(void);
+    void SetServerTime(void);
+    void ResetBytesDownloaded(void) { bytesDownloaded = 0; };
+    int GetBytesDownloaded(void) { return bytesDownloaded; };
+    void UpdateSeries(void);
+    bool GetSeriesWithEpisodesFromEPG(vector<sSeriesResult> *result);
+    void ProcessSeries(sSeriesResult ser);
+    int CleanupSeries(void);
+    bool SearchRecordingDB(string name, string episode, int &seriesId, int &episodeId);
+    bool SearchRecordingOnline(string name, string episode, int &seriesId, int &episodeId);
+    bool CheckScrapInfoDB(int scrapSeriesId, int scrapEpisodeId);
+    bool CheckScrapInfoOnline(int scrapSeriesId, int scrapEpisodeId);
+};
+
+#endif //__TVSCRAPER_TVDBMANAGER_H