aboutsummaryrefslogtreecommitdiff
path: root/plugins/MangaGeneric.hpp
blob: 7d636226744cc9bc38334c58e8346e8e0f1f6b88 (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
#pragma once

#include "Manga.hpp"
#include <functional>

namespace QuickMedia {
    struct SearchQuery {
        const char *search_prefix = nullptr;
        const char *page_prefix = nullptr;
        int page_start = 0;
    };

    struct TextQuery {
        const char *html_query = nullptr;
        const char *title_field = nullptr;
        const char *url_field = nullptr;
        const char *url_contains = nullptr;
    };

    struct ThumbnailQuery {
        const char *html_query = nullptr;
        const char *field_name = nullptr;
        const char *field_contains = nullptr;
    };

    struct ListChaptersQuery {
        const char *html_query = nullptr;
        const char *title_field = nullptr;
        const char *url_field = nullptr;
        const char *url_contains = nullptr;

        const char *uploaded_time_html_query = nullptr;
        const char *uploaded_time_field_name = nullptr;
        const char *uploaded_time_field_contains = nullptr;
    };

    using ListPageImagesQueryPost = std::function<void(std::vector<std::string> &image_urls)>;
    struct ListPageImagesQuery {
        const char *html_query = nullptr;
        const char *field_name = nullptr;
        const char *field_contains = nullptr;
        ListPageImagesQueryPost post_handler = nullptr;
    };

    // Return the actual number of pages
    using ListPagePaginationPagesPost = std::function<int(int num_pages)>;
    struct ListPagePaginationQuery {
        const char *pages_html_query = nullptr;
        const char *pages_field_name = nullptr;
        const char *pages_field_contains = nullptr;
        ListPagePaginationPagesPost pages_post_handler = nullptr;

        const char *image_html_query = nullptr;
        const char *image_field_name = nullptr;
        const char *image_field_contains = nullptr;

        const char *next_page_html_query = nullptr;
        const char *next_page_field_name = nullptr;
        const char *next_page_field_contains = nullptr;
    };

    enum class ListPageQueryType {
        IMAGES,
        PAGINATION
    };

    struct ListPageQuery {
        ListPageQueryType type = ListPageQueryType::IMAGES;
        ListPageImagesQuery images_query;
        ListPagePaginationQuery pagination_query;
    };

    struct MangaIdExtractor {
        const char *prefix = nullptr;
        const char *end = nullptr;
    };

    class MangaGenericSearchPage : public Page {
    public:
        MangaGenericSearchPage(Program *program, const char *service_name, const char *website_url);
        const char* get_title() const override { return "All"; }
        bool search_is_filter() override { return false; }
        SearchResult search(const std::string &str, BodyItems &result_items) override;
        PluginResult get_page(const std::string &str, int page, BodyItems &result_items) override;
        PluginResult submit(const std::string &title, const std::string &url, std::vector<Tab> &result_tabs) override;
        sf::Vector2i get_thumbnail_max_size() override { return sf::Vector2i(101, 141); };

        // Add a %s where the query or page number should be inserted into |search_prefix| and |page_prefix|, for example:
        // search_prefix: example.com/search?q=%s
        // page_prefix:   &page=%s
        // |page_start| is the first page, so the result page is |page_start| + |page| where page is the current page we are navigating on.
        // This is required.
        MangaGenericSearchPage& search_handler(const char *search_prefix, const char *page_prefix, int page_start);
        // If |url_contains| is null, then any matching query is added. If |title_field| is "text", then the inner text is used.
        // This is required.
        MangaGenericSearchPage& text_handler(const char *html_query, const char *title_field, const char *url_field, const char *url_contains);
        // If |field_contains| is null, then any matching query is added. If |field_name| is "text", then the inner text is used.
        // This is optional.
        MangaGenericSearchPage& thumbnail_handler(const char *html_query, const char *field_name, const char *field_contains);

        // If |url_contains| is null, then any matching query is added. If |title_field| is "text", then the inner text is used.
        // This is required.
        MangaGenericSearchPage& list_chapters_handler(const char *html_query, const char *title_field, const char *url_field, const char *url_contains);
        // If |field_contains| is null, then any matching query is added. If |field_name| is "text", then the inner text is used.
        // This is optional.
        MangaGenericSearchPage& list_chapters_uploaded_time_handler(const char *html_query, const char *field_name, const char *field_contains);

        // If |field_contains| is null, then any matching query is added. If |field_name| is "text", then the inner text is used.
        // This or |list_page_images_pagination_handler| is required.
        MangaGenericSearchPage& list_page_images_handler(const char *html_query, const char *field_name, const char *field_contains, ListPageImagesQueryPost post_handler = nullptr);

        // If |pages_field_contains| or |image_field_contains| is null, then any matching query is added. If |pages_field_name| or |image_field_name| is "text", then the inner text is used.
        // This or |list_page_images_handler| is required.
        MangaGenericSearchPage& list_page_images_pagination_handler(
            const char *pages_html_query, const char *pages_field_name, const char *pages_field_contains, ListPagePaginationPagesPost pages_post_handler,
            const char *image_html_query, const char *image_field_name, const char *image_field_contains,
            const char *next_page_html_query, const char *next_page_field_name, const char *next_page_field_contains);

        // For example: mangasite.com/manga/204353&f=23
        // /manga/ here would be the |prefix| and & would be |end|. |end| is optional.
        // The goal is to extract 204353 from the manga chapter page url.
        MangaGenericSearchPage& manga_id_handler(const char *prefix, const char *end);
    private:
        const char *service_name;
        std::string website_url;
        SearchQuery search_query;
        TextQuery text_query;
        ThumbnailQuery thumbnail_query;
        ListChaptersQuery list_chapters_query;
        ListPageQuery list_page_query;
        MangaIdExtractor manga_id_extractor;
    };

    class MangaGenericChaptersPage : public MangaChaptersPage {
    public:
        MangaGenericChaptersPage(Program *program, std::string manga_name, std::string manga_url, const MangaIdExtractor &manga_id_extractor, const char *service_name, const std::string &website_url, const ListPageQuery *list_page_query) :
            MangaChaptersPage(program, std::move(manga_name), std::move(manga_url)), manga_id_extractor(manga_id_extractor), service_name(service_name), website_url(website_url), list_page_query(list_page_query) {}
        PluginResult submit(const std::string &title, const std::string &url, std::vector<Tab> &result_tabs) override;
    protected:
        bool extract_id_from_url(const std::string &url, std::string &manga_id) const override;
        const char* get_service_name() const override { return service_name; }
    private:
        MangaIdExtractor manga_id_extractor;
        const char *service_name;
        std::string website_url;
        const ListPageQuery *list_page_query;
    };

    class MangaGenericImagesPage : public MangaImagesPage {
    public:
        MangaGenericImagesPage(Program *program, std::string manga_name, std::string chapter_name, std::string url, const char *service_name, const std::string &website_url, const ListPageQuery *list_page_query) :
            MangaImagesPage(program, std::move(manga_name), std::move(chapter_name), std::move(url)), service_name(service_name), website_url(website_url), list_page_query(list_page_query) {}
        ImageResult get_number_of_images(int &num_images) override;
        ImageResult for_each_page_in_chapter(PageCallback callback) override;
        const char* get_service_name() const override { return service_name; }
    private:
        ImageResult get_page_image_urls();
    private:
        const char *service_name;
        std::string website_url;
        const ListPageQuery *list_page_query;
        std::string prev_chapter_url;
        std::string current_image_url;
        std::string next_page_url;
    };
}