Add include/exclude groups config for dex + cb scraper

browningluke · Aug 10, 2024 · ff8f38e · ff8f38e
1 parent 7576ff7
commit ff8f38e
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 0 deletions.
diff --git a/examples/config.yml b/examples/config.yml
@@ -38,5 +38,21 @@ sources:
     ratingFilter: ["safe", "suggestive", "erotica"]
     languageFilter: ["en"] # , "fr"
     dataSaver: false
+    groups:
+      # Only download chapters from <Example A> and <Example B>
+      # include:
+        # - "Example A"
+        # - "Example B"
+      # Do not download chapters from <Example C>
+      exclude:
+        - "Example C"
   cubari:
     filenameTemplate: "{num:4} - Chapter {num:2}{title: - <.>}{groups: {<.>}}" # Set specific for source (overrides global)
+    groups:
+      # Only download chapters from <Example A> and <Example B>
+      # include:
+      # - "Example A"
+      # - "Example B"
+      # Do not download chapters from <Example C>
+      exclude:
+        - "Example C"
diff --git a/internal/sources/cubari/config.go b/internal/sources/cubari/config.go
@@ -3,7 +3,14 @@ package cubari
 var config Config
 
 type Config struct {
+	// Scraper
 	FilenameTemplate string `yaml:"filenameTemplate"`
+
+	// Groups
+	Groups struct {
+		Include []string `yaml:"include"`
+		Exclude []string `yaml:"exclude"`
+	} `yaml:"groups"`
 }
 
 func SetConfig(cfg Config) {
@@ -12,4 +19,6 @@ func SetConfig(cfg Config) {
 
 func (c *Config) Default() {
 	c.FilenameTemplate = "" // No override of downloader.output.filenameTemplate
+	c.Groups.Include = []string{}
+	c.Groups.Exclude = []string{}
 }
diff --git a/internal/sources/mangadex/config.go b/internal/sources/mangadex/config.go
@@ -9,6 +9,12 @@ type Config struct {
 	LanguageFilter   []string `yaml:"languageFilter"`
 	DataSaver        bool     `yaml:"dataSaver"`
 
+	// Groups
+	Groups struct {
+		Include []string `yaml:"include"`
+		Exclude []string `yaml:"exclude"`
+	} `yaml:"groups"`
+
 	// Connection
 	SyncDeletions bool `yaml:"syncDeletions"`
 }
@@ -23,5 +29,8 @@ func (c *Config) Default() {
 	c.LanguageFilter = []string{"en"}
 	c.DataSaver = false
 
+	c.Groups.Include = []string{}
+	c.Groups.Exclude = []string{}
+
 	c.SyncDeletions = false
 }