Add option to strip Royal Road site title suffix from filename (#33)

* Add option to strip Royal Road site title suffix from filename * Adjust option and invert logic so it's a keep instead of skip option * Remove the --keep-site-title option again and just always remove the title
Aivean · Jan 19, 2024 · 7017d77 · 7017d77
1 parent d05a0b0
commit 7017d77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/scala/com/aivean/royalroad/Args.scala b/src/main/scala/com/aivean/royalroad/Args.scala
@@ -56,4 +56,4 @@ class Args(args: Seq[String]) extends ScallopConf(args) {
 
 object Args {
   val urlValidationRegex = "(https?://(www\\.)?royalroadl?.com/fiction/\\d+)(/.*)?"
-}
+}
diff --git a/src/main/scala/com/aivean/royalroad/Main.scala b/src/main/scala/com/aivean/royalroad/Main.scala
@@ -74,7 +74,9 @@ object Main extends App {
     chapQ.put(None)
   }
 
-  val filename = title.replaceAll("[^\\w\\d]+", "_") + {
+  val filesafeTitle = title.stripSuffix(" | Royal Road").replaceAll("[^\\w\\d]+", "_")
+
+  val filename = filesafeTitle + {
     // when chapter range is specified, add it to the filename
     if (chapUrls.size != chapUrlsConstrained.size) {
       val firstChapter = chapUrls.indexOf(chapUrlsConstrained.head)
-Original file line number
+Diff line change
@@ Expand Up / @@ -56,4 +56,4 @@ class Args(args: Seq[String]) extends ScallopConf(args) { @@
     object Args {
       val urlValidationRegex = "(https?://(www\\.)?royalroadl?.com/fiction/\\d+)(/.*)?"
-    }
+    }