1 files changed, 97 insertions, 0 deletions
diff --git a/thetvdbscraper/tvdbmirrors.c b/thetvdbscraper/tvdbmirrors.c
new file mode 100644
index 0000000..d0849c2
--- /dev/null
+++ b/thetvdbscraper/tvdbmirrors.c
@@ -0,0 +1,97 @@
+#include <string>
+#include <sstream>
+#include <vector>
+#include <libxml/parser.h>
+#include <libxml/tree.h>
+#include "tvdbmirrors.h"
+
+using namespace std;
+
+cTVDBMirrors::cTVDBMirrors(string xml) {
+    doc = NULL;
+    SetXMLDoc(xml);
+}
+
+cTVDBMirrors::~cTVDBMirrors() {
+    xmlFreeDoc(doc);
+}
+
+void cTVDBMirrors::SetXMLDoc(string xml) {
+    xmlInitParser();
+    doc = xmlReadMemory(xml.c_str(), strlen(xml.c_str()), "noname.xml", NULL, 0);
+}
+
+string cTVDBMirrors::GetMirrorXML(void) {
+    if (xmlmirrors.size() == 0)
+        return "";
+    int randMirror = rand() % xmlmirrors.size();
+    return xmlmirrors[randMirror];
+}
+string cTVDBMirrors::GetMirrorBanner(void) {
+    if (bannermirrors.size() == 0)
+        return "";
+    int randMirror = rand() % bannermirrors.size();
+    return bannermirrors[randMirror];
+}
+
+string cTVDBMirrors::GetMirrorZip(void) {
+    if (zipmirrors.size() == 0)
+        return "";
+    int randMirror = rand() % zipmirrors.size();
+    return zipmirrors[randMirror];
+}
+
+bool cTVDBMirrors::ParseXML(void) {
+    if (doc == NULL)
+        return false;
+    //Root Element has to be <Mirrors>
+    xmlNode *node = NULL;
+    node = xmlDocGetRootElement(doc);
+    if (!(node && !xmlStrcmp(node->name, (const xmlChar *)"Mirrors")))
+        return false;
+    //Loop through <Mirror>
+    node = node->children;
+    xmlNode *cur_node = NULL;
+    bool ok = false;
+    for (cur_node = node; cur_node; cur_node = cur_node->next) {
+        if ((cur_node->type == XML_ELEMENT_NODE) && !xmlStrcmp(cur_node->name, (const xmlChar *)"Mirror")) {
+            ok = ReadEntry(cur_node->children);
+        }
+    }    
+    return ok;
+}
+
+bool cTVDBMirrors::ReadEntry(xmlNode *node) {
+    xmlNode *cur_node = NULL;
+    xmlChar *node_content;
+    string path = "";
+    int typemask = 0;
+    for (cur_node = node; cur_node; cur_node = cur_node->next) {
+        if (cur_node->type == XML_ELEMENT_NODE) {
+            node_content = xmlNodeListGetString(doc, cur_node->xmlChildrenNode, 1);
+            if (!node_content)
+                continue;
+            if (!xmlStrcmp(cur_node->name, (const xmlChar *)"mirrorpath")) {
+                path = (const char *)node_content;
+            } else if (!xmlStrcmp(cur_node->name, (const xmlChar *)"typemask")) {
+                typemask = atoi((const char *)node_content);
+            }
+            xmlFree(node_content);
+        }
+    }
+    return CreateMirror(path, typemask);
+}
+
+bool cTVDBMirrors::CreateMirror(string path, int typemask) {
+    if (path.size() < 1)
+        return false;
+    if (typemask < 1 || typemask > 7)
+        return false;
+    if (typemask & 1)
+       xmlmirrors.push_back(path); 
+    if (typemask & 2)
+       bannermirrors.push_back(path); 
+    if (typemask & 4)
+       zipmirrors.push_back(path);
+    return true;
+}
+\ No newline at end of file