From 5ed245317df5e3edb644d0aab4bb35a759d5eaee Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= <mike_faehrmann@web.de>
Date: Wed, 23 Aug 2023 12:54:45 +0200
Subject: [PATCH] [exhentai] add 'fav' option (#4409)

The name 'favorite' is already taken as extractor subcategory
---
 docs/configuration.rst           | 16 ++++++++++++++++
 gallery_dl/extractor/exhentai.py | 18 ++++++++++++++++++
 2 files changed, 34 insertions(+)

diff --git a/docs/configuration.rst b/docs/configuration.rst
index 8c7f3712..5f8ac6eb 100644
--- a/docs/configuration.rst
+++ b/docs/configuration.rst
@@ -1475,6 +1475,22 @@ Description
     * ``"exhentai.org"``: Use ``exhentai.org`` for all URLs
 
 
+extractor.exhentai.fav
+----------------------
+Type
+    ``string``
+Example
+    ``"4"``
+Description
+    After downloading a gallery,
+    add it to your account's favorites as the given category number.
+
+    Note: Set this to `"favdel"` to remove galleries from your favorites.
+
+    Note: This will remove any Favorite Notes when applied
+    to already favorited galleries.
+
+
 extractor.exhentai.limits
 -------------------------
 Type
diff --git a/gallery_dl/extractor/exhentai.py b/gallery_dl/extractor/exhentai.py
index 5ee008d1..0a60ed82 100644
--- a/gallery_dl/extractor/exhentai.py
+++ b/gallery_dl/extractor/exhentai.py
@@ -179,6 +179,20 @@ class ExhentaiGalleryExtractor(ExhentaiExtractor):
         if source == "hitomi":
             self.items = self._items_hitomi
 
+    def favorite(self, slot="0"):
+        url = self.root + "/gallerypopups.php"
+        params = {
+            "gid": self.gallery_id,
+            "t"  : self.gallery_token,
+            "act": "addfav",
+        }
+        data = {
+            "favcat" : slot,
+            "apply"  : "Apply Changes",
+            "update" : "1",
+        }
+        self.request(url, method="POST", params=params, data=data)
+
     def items(self):
         self.login()
 
@@ -223,6 +237,10 @@ class ExhentaiGalleryExtractor(ExhentaiExtractor):
                 data["_http_validate"] = None
             yield Message.Url, url, data
 
+        fav = self.config("fav")
+        if fav is not None:
+            self.favorite(fav)
+
     def _items_hitomi(self):
         if self.config("metadata", False):
             data = self.metadata_from_api()