WikiNetwork/WikiAPI.cpp at main · isaacfuenmayora/WikiNetwork · GitHub

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
#include "WikiAPI.hpp"
#include <cpr/cpr.h>
using cpr::Get;
using cpr::Url;
using cpr::Response;
using std::to_string;

string WikiAPI::wiki="https://simple.wikipedia.org/";
string WikiAPI::outgoing_url="https://simple.wikipedia.org/w/api.php?format=json&formatversion=2&action=query&generator=prefixsearch&gpslimit=10&prop=pageimages|pageterms&piprop=thumbnail&pithumbsize=50&pilimit=10&pilicense=any&wbptterms=description&redirects=&gpssearch=";
string WikiAPI::search_url="https://simple.wikipedia.org/w/api.php?format=json&formatversion=2&action=query&generator=links&gpllimit=500&gplnamespace=0&redirects=&pageids=";
string WikiAPI::thumbnail_url="api.php?format=json&formatversion=2&action=query&prop=pageimages&piprop=original&pilicense=any&titles=";

string formatTitle(string& title){
    string linkTitle;
    for(int i=0; i<title.size(); i++){
        if(title[i]==' ')
            linkTitle+="%20";
        else if(title[i]=='!')
            linkTitle+="%21";
        else if(title[i]=='\'')
            linkTitle+="%27";
        else if(title[i]=='(')
            linkTitle+="%28";
        else if(title[i]==')')
            linkTitle+="%29";
        else if(title[i]==',')
            linkTitle+="%2C";
        else
            linkTitle.push_back(title[i]);
    }
    return linkTitle;
}

WikiAPI::WikiAPI(const string& wikiEndpoint) {
    setWiki(wikiEndpoint);
}

void WikiAPI::setWiki(const string& wikiEndpoint) {
    wiki=wikiEndpoint;
    search_url=wiki+"w/api.php?format=json&formatversion=2&action=query&generator=prefixsearch&gpslimit=10&prop=pageimages|pageterms&piprop=thumbnail&pithumbsize=50&pilimit=10&pilicense=any&wbptterms=description&redirects=&gpssearch=";
    outgoing_url=wiki+"w/api.php?format=json&formatversion=2&action=query&generator=links&gpllimit=500&gplnamespace=0&redirects=&pageids=";
    thumbnail_url=wiki+"w/api.php?format=json&formatversion=2&action=query&prop=pageimages&piprop=original&pilicense=any&titles=";
}
void WikiAPI::setWiki(string&& wikiEndpoint) {
    wiki=wikiEndpoint;
    search_url=wiki+"w/api.php?format=json&formatversion=2&action=query&generator=prefixsearch&gpslimit=10&prop=pageimages|pageterms&piprop=thumbnail&pithumbsize=50&pilimit=10&pilicense=any&wbptterms=description&redirects=&gpssearch=";
    outgoing_url=wiki+"w/api.php?format=json&formatversion=2&action=query&generator=links&gpllimit=500&gplnamespace=0&redirects=&pageids=";
    thumbnail_url=wiki+"w/api.php?format=json&formatversion=2&action=query&prop=pageimages&piprop=original&pilicense=any&titles=";
}
//TODO: Implement handling errors
string WikiAPI::getSearchList(string& query) {
    Response r = Get(Url{search_url+formatTitle(query)});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}
string WikiAPI::getSearchList(string&& query) {
    Response r = Get(Url{search_url+formatTitle(query)});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}

string WikiAPI::getOutgoingLinks(int pageID) {
    Response r = Get(Url{outgoing_url+to_string(pageID)});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}
string WikiAPI::getOutgoingLinks(int pageID, string&& gplcontinue) {
    Response r = Get(Url{outgoing_url+to_string(pageID)+"&gplcontinue="+gplcontinue});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}

string WikiAPI::getOutgoingLinks(int pageID, string& gplcontinue) {
    Response r = Get(Url{outgoing_url+to_string(pageID)+"&gplcontinue="+gplcontinue});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}

string WikiAPI::getThumbnail(int pageID) {
    Response r = Get(Url{thumbnail_url+to_string(pageID)});
    switch(r.status_code) {
        case 200:
            return r.text;
        default:
            return "error";
    }
}

string WikiAPI::getPageURL(string& pageTitle) {
    pageTitle=formatTitle(pageTitle);
    return wiki+"wiki/"+pageTitle;
}
string WikiAPI::getPageURL(string&& pageTitle) {
    pageTitle=formatTitle(pageTitle);
    return wiki+"wiki/"+pageTitle;
}