sources: add support for showing followed results on Pixiv

Bionus · Feb 13, 2023 · eff25a0 · eff25a0
1 parent cad50e9
commit eff25a0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/docs/_sources/pixiv.md b/docs/_sources/pixiv.md
@@ -15,6 +15,7 @@ Whatever URL is entered when adding a site with this source, it will be override
 * `mode:{partial,full,tc}`: tag search type (default: `partial`)
 * `bookmarks:USER_ID`: fetch a given user's bookmarks
 * `user:USER_ID`: fetch art from a given user
+* `is:followed`: show illustrations from artists you follow
 * `type:{illust,manga}`: only load works of a given type (only applies to `user:` and empty searches)
 
 ### Date

diff --git a/src/sites/Pixiv/model.ts b/src/sites/Pixiv/model.ts
@@ -26,7 +26,7 @@ const meta: ISource["meta"] = {
     },
 };
 
-function parseSearch(search: string): { mode: string, tags: string[], bookmarks?: number, user?: number, startDate?: string, endDate?: string, type?: string } {
+function parseSearch(search: string): { mode: string, tags: string[], bookmarks?: number, user?: number, followed: boolean, startDate?: string, endDate?: string, type?: string } {
     const modes: any = {
         "partial": "partial_match_for_tags",
         "full": "exact_match_for_tags",
@@ -40,6 +40,7 @@ function parseSearch(search: string): { mode: string, tags: string[], bookmarks?
     const type = parsed.type;
 
     const tags = [];
+    let followed = false;
     let startDate = undefined;
     let endDate = undefined;
 
@@ -66,11 +67,15 @@ function parseSearch(search: string): { mode: string, tags: string[], bookmarks?
             }
             continue;
         }
+        if (part === "is:followed") {
+            followed = true;
+            continue;
+        }
 
         tags.push(part);
     }
 
-    return { mode, tags, bookmarks, user, startDate, endDate, type };
+    return { mode, tags, bookmarks, user, followed, startDate, endDate, type };
 }
 
 function parseImage(image: any, fromGallery: boolean): IImage {
@@ -134,7 +139,7 @@ function parseImage(image: any, fromGallery: boolean): IImage {
 
 export const source: ISource = {
     name: "Pixiv",
-    modifiers: ["mode:partial", "mode:full", "mode:tc", "bookmarks:", "user:", "date:"],
+    modifiers: ["mode:partial", "mode:full", "mode:tc", "bookmarks:", "user:", "date:", "is:followed"],
     forcedTokens: [],
     searchFormat: {
         and: " ",
@@ -203,6 +208,12 @@ export const source: ISource = {
                         return "https://app-api.pixiv.net/v1/user/illusts?" + illustParams.join("&");
                     }
 
+                    // User's follows
+                    if (search.followed) {
+                        illustParams.push("restrict=public");
+                        return "https://app-api.pixiv.net/v2/illust/follow?" + illustParams.join("&");
+                    }
+
                     // Newest (when no tag is provided)
                     if (search.tags.length === 0) {
                         if (search.type) {