summaryrefslogtreecommitdiff
path: root/tvscraper.c
blob: ae6e43abc550cd6d74bab4a4402f8b981627e529 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
/*
 * tvscraper.c: A plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 *
 * $Id$
 */
#include <getopt.h>
#include <vdr/plugin.h>
#include "tools/curlfuncs.cpp"
#include "tools/filesystem.c"
#include "tools/fuzzy.c"
#include "tools/splitstring.c"
#include "config.c"
cTVScraperConfig config;
#include "tvscraperdb.c"
#include "thetvdbscraper/tvdbmirrors.c"
#include "thetvdbscraper/tvdbseries.c"
#include "thetvdbscraper/tvdbmedia.c"
#include "thetvdbscraper/tvdbactors.c"
#include "thetvdbscraper/thetvdbscraper.c"
#include "themoviedbscraper/moviedbmovie.c"
#include "themoviedbscraper/moviedbactors.c"
#include "themoviedbscraper/themoviedbscraper.c"
#include "worker.c"
#include "services.h"
#include "imageserver.c"
#include "setup.c"

static const char *VERSION        = "0.0.2";
static const char *DESCRIPTION    = "Scraping movie and series info";
static const char *MAINMENUENTRY  = "TV Scraper";

class cPluginTvscraper : public cPlugin {
private:
    bool cacheDirSet;
    cTVScraperDB *db;
    cTVScraperWorker *workerThread;
    cImageServer *imageServer;
public:
    cPluginTvscraper(void);
    virtual ~cPluginTvscraper();
    virtual const char *Version(void) { return VERSION; }
    virtual const char *Description(void) { return DESCRIPTION; }
    virtual const char *CommandLineHelp(void);
    virtual bool ProcessArgs(int argc, char *argv[]);
    virtual bool Initialize(void);
    virtual bool Start(void);
    virtual void Stop(void);
    virtual void Housekeeping(void);
    virtual void MainThreadHook(void);
    virtual cString Active(void);
    virtual time_t WakeupTime(void);
    virtual const char *MainMenuEntry(void) { return NULL; }
    virtual cOsdObject *MainMenuAction(void);
    virtual cMenuSetupPage *SetupMenu(void);
    virtual bool SetupParse(const char *Name, const char *Value);
    virtual bool Service(const char *Id, void *Data = NULL);
    virtual const char **SVDRPHelpPages(void);
    virtual cString SVDRPCommand(const char *Command, const char *Option, int &ReplyCode);
};

cPluginTvscraper::cPluginTvscraper(void) {
}

cPluginTvscraper::~cPluginTvscraper() {
}

const char *cPluginTvscraper::CommandLineHelp(void) {
    return "  -d <CACHEDIR>, --dir=<CACHEDIR> Set directory where database and images are stored\n";
}

bool cPluginTvscraper::ProcessArgs(int argc, char *argv[]) {
    static const struct option long_options[] = {
        { "dir", required_argument, NULL, 'd' },
        { 0, 0, 0, 0 }
    };

    int c;
    cacheDirSet = false;
    while ((c = getopt_long(argc, argv, "d:", long_options, NULL)) != -1) {
        switch (c) {
            case 'd':
                cacheDirSet = true;
                config.SetBaseDir(optarg);
                break;
            default:
                return false;
        }
    }
    return true;
}

bool cPluginTvscraper::Initialize(void) {
    return true;
}

bool cPluginTvscraper::Start(void) {
    if (!cacheDirSet) {
        config.SetBaseDir(cPlugin::CacheDirectory(PLUGIN_NAME_I18N));
    }
    db = new cTVScraperDB();
    if (!db->Connect()) {
        esyslog("tvscraper: could not connect to Database. Aborting!");
        return false;
    };
    imageServer = new cImageServer(db);
    workerThread = new cTVScraperWorker(db);
    workerThread->SetDirectories();
    workerThread->SetLanguage();
    workerThread->Start();
    return true;
}

void cPluginTvscraper::Stop(void) {
    while (workerThread->Active()) {
        workerThread->Stop();
    }   
    delete workerThread;
    delete imageServer;
    delete db;
}

void cPluginTvscraper::Housekeeping(void) {
}

void cPluginTvscraper::MainThreadHook(void) {
}

cString cPluginTvscraper::Active(void) {
    return NULL;
}

time_t cPluginTvscraper::WakeupTime(void) {
    return 0;
}

cOsdObject *cPluginTvscraper::MainMenuAction(void) {
    return NULL;
}

cMenuSetupPage *cPluginTvscraper::SetupMenu(void) {
    return new cTVScraperSetup(workerThread);
}

bool cPluginTvscraper::SetupParse(const char *Name, const char *Value) {
    return config.SetupParse(Name, Value);
}

bool cPluginTvscraper::Service(const char *Id, void *Data) {
    if (strcmp(Id, "TVScraperGetPosterOrBanner") == 0) {
        if (Data == NULL)
            return false;
        TVScraperGetPosterOrBanner* call = (TVScraperGetPosterOrBanner*) Data;
        if (!call->event)
            return false;
        scrapType type = imageServer->GetScrapType(call->event);
        if (type == scrapSeries)
            call->type = typeSeries;
        else if (type == scrapMovie)
            call->type = typeMovie;
        else
            call->type = typeNone;
        int id = imageServer->GetID(call->event->EventID(), type, false);
        if (id > 0) {
            call->media = imageServer->GetPosterOrBanner(id, type);
            return true;
        }
        return false;
    }
    if (strcmp(Id, "TVScraperGetPoster") == 0) {
        if (Data == NULL)
            return false;
        TVScraperGetPoster* call = (TVScraperGetPoster*) Data;
        if (!call->event)
            return false;
        scrapType type = imageServer->GetScrapType(call->event);
        int id = imageServer->GetID(call->event->EventID(), type, call->isRecording);
        if (id > 0) {
            call->media = imageServer->GetPoster(id, type);
            return true;
        }
        return false;
    }
    if (strcmp(Id, "TVScraperGetFullInformation") == 0) {
        if (Data == NULL)
            return false;
        TVScraperGetFullInformation* call = (TVScraperGetFullInformation*) Data;
        if (!call->event)
            return false;
        
        scrapType type = imageServer->GetScrapType(call->event);
        int id = imageServer->GetID(call->event->EventID(), type, call->isRecording);

        if (id == 0)
            return false;
        
        if (type == scrapSeries) {
            call->type = typeSeries;
            call->banner = imageServer->GetBanner(id);
        } else if (type == scrapMovie) {
            call->type = typeMovie;
        } else {
            call->type = typeNone;
        }
        call->posters = imageServer->GetPosters(id, type);
        call->fanart = imageServer->GetFanart(id, type);
        call->actors = imageServer->GetActors(id, type);
        call->description = imageServer->GetDescription(id, type);
        return true;
    }
    return false;
}

const char **cPluginTvscraper::SVDRPHelpPages(void) {
    return NULL;
}

cString cPluginTvscraper::SVDRPCommand(const char *Command, const char *Option, int &ReplyCode) {
    return NULL;
}

VDRPLUGINCREATOR(cPluginTvscraper);