Add "split" xpath in post-processing , newlines in replace support

pkg/scraper/xpath.go

@@ -75,6 +75,9 @@ func (c xpathRegexConfig) apply(value string) string {
 		}
 
 		ret := re.ReplaceAllString(value, with)
+		if with == "\n" {
+			ret = replaceLines(ret)
+		}
 
 		logger.Debugf(`Replace: '%s' with '%s'`, regex, with)
 		logger.Debugf("Before: %s", value)
@@ -94,6 +97,8 @@ func (c xpathRegexConfigs) apply(value string) string {
 	// remove whitespace again
 	value = commonPostProcess(value)
 
+	// restore replaced lines
+	value = restoreLines(value)
 	return value
 }
 
@@ -129,6 +134,15 @@ func (c xpathScraperAttrConfig) getParseDate() string {
 	return c.getString(parseDateKey)
 }
 
+func (c xpathScraperAttrConfig) getSplit() string {
+	const splitKey = "split"
+	return c.getString(splitKey)
+}
+
+func (c xpathScraperAttrConfig) hasSplit() bool {
+	return c.getSplit() != ""
+}
+
 func (c xpathScraperAttrConfig) getReplace() xpathRegexConfigs {
 	const replaceKey = "replace"
 	val, _ := c[replaceKey]
@@ -198,6 +212,23 @@ func (c xpathScraperAttrConfig) parseDate(value string) string {
 	return parsedValue.Format(internalDateFormat)
 }
 
+func (c xpathScraperAttrConfig) splitString(value string) []string {
+	separator := c.getSplit()
+	var res []string
+
+	if separator == "" {
+		return []string{value}
+	}
+
+	for _, str := range strings.Split(value, separator) {
+		if str != "" {
+			res = append(res, str)
+		}
+	}
+
+	return res
+}
+
 func (c xpathScraperAttrConfig) replaceRegex(value string) string {
 	replace := c.getReplace()
 	return replace.apply(value)
@@ -258,6 +289,24 @@ func commonPostProcess(value string) string {
 	return value
 }
 
+// func replaceLines replaces all newlines ("\n") with carriage returns ("\r")
+func replaceLines(value string) string {
+	re := regexp.MustCompile("\r")         // carriage returns shouldn't exist in the string
+	value = re.ReplaceAllString(value, "") // remove them
+	re = regexp.MustCompile("\n")          // replace newlines with CR's so that they don't get removed by commonPostprocess
+	value = re.ReplaceAllString(value, "\r")
+
+	return value
+}
+
+// func restoreLines replaces all carriage returns ("\r") with newlines ("\n")
+func restoreLines(value string) string {
+	re := regexp.MustCompile("\r")
+	value = re.ReplaceAllString(value, "\n")
+
+	return value
+}
+
 func runXPathQuery(doc *html.Node, xpath string, common commonXPathConfig) []*html.Node {
 	// apply common
 	if common != nil {
@@ -299,15 +348,26 @@ func (s xpathScraperConfig) process(doc *html.Node, common commonXPathConfig) xP
 				if attrConfig.hasConcat() {
 					result := attrConfig.concatenateResults(found)
 					result = attrConfig.postProcess(result)
-					const i = 0
-					ret = ret.setKey(i, k, result)
+					if attrConfig.hasSplit() {
+						for j, txt := range attrConfig.splitString(result) {
+							ret = ret.setKey(j, k, txt)
+						}
+					} else {
+						const i = 0
+						ret = ret.setKey(i, k, result)
+					}
 				} else {
 					for i, elem := range found {
 						text := NodeText(elem)
 						text = commonPostProcess(text)
 						text = attrConfig.postProcess(text)
-
-						ret = ret.setKey(i, k, text)
+						if attrConfig.hasSplit() {
+							for j, txt := range attrConfig.splitString(text) {
+								ret = ret.setKey(i+j, k, txt)
+							}
+						} else {
+							ret = ret.setKey(i, k, text)
+						}
 					}
 				}
 			}

pkg/scraper/xpath_test.go

@@ -663,6 +663,14 @@ func makeSceneXPathConfig() xpathScraper {
 	studioConfig["URL"] = `$studioElem/@href`
 	config["Studio"] = studioConfig
 
+	const sep = " "
+	moviesNameConfig := make(map[interface{}]interface{})
+	moviesNameConfig["selector"] = `//i[@class="isMe tooltipTrig"]/@data-title`
+	moviesNameConfig["split"] = sep
+	moviesConfig := make(map[interface{}]interface{})
+	moviesConfig["Name"] = moviesNameConfig
+	config["Movies"] = moviesConfig
+
 	scraper := xpathScraper{
 		Scene:  config,
 		Common: common,
@@ -692,6 +700,27 @@ func verifyTags(t *testing.T, expectedTagNames []string, actualTags []*models.Sc
 	}
 }
 
+func verifyMovies(t *testing.T, expectedMovieNames []string, actualMovies []*models.ScrapedSceneMovie) {
+	t.Helper()
+
+	i := 0
+	for i < len(expectedMovieNames) || i < len(actualMovies) {
+		expectedMovie := ""
+		actualMovie := ""
+		if i < len(expectedMovieNames) {
+			expectedMovie = expectedMovieNames[i]
+		}
+		if i < len(actualMovies) {
+			actualMovie = actualMovies[i].Name
+		}
+
+		if expectedMovie != actualMovie {
+			t.Errorf("Expected movie %s, got %s", expectedMovie, actualMovie)
+		}
+		i++
+	}
+}
+
 func verifyPerformers(t *testing.T, expectedNames []string, expectedURLs []string, actualPerformers []*models.ScrapedScenePerformer) {
 	t.Helper()
 
@@ -761,6 +790,15 @@ func TestApplySceneXPathConfig(t *testing.T) {
 	}
 	verifyTags(t, expectedTags, scene.Tags)
 
+	// verify movies
+	expectedMovies := []string{
+		"Video",
+		"of",
+		"verified",
+		"member",
+	}
+	verifyMovies(t, expectedMovies, scene.Movies)
+
 	expectedPerformerNames := []string{
 		"Alex D",
 		"Mia Malkova",

ui/v2.5/src/index.scss

@@ -483,3 +483,7 @@ div.dropdown-menu {
     text-transform: uppercase;
   }
 }
+
+.pre {
+  white-space: pre-line;
+}