scraper: Fix issue getting DarkroomVR Scene Id (xbapps#1694)

vt-idiot · May 3, 2024 · 5a00289 · 5a00289
1 parent e376fa4
commit 5a00289
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pkg/scrape/darkroomvr.go b/pkg/scrape/darkroomvr.go
@@ -81,11 +81,12 @@ func DarkRoomVR(wg *sync.WaitGroup, updateSite bool, knownScenes []string, out c
 		})
 
 		// Scene ID
-		e.ForEach(`a[href*="signup.php?vid"]`, func(id int, e *colly.HTMLElement) {
+		e.ForEach(`a[href*="signup.php"]`, func(id int, e *colly.HTMLElement) {
 			url := e.Attr("href")
-			sc.SiteID = url[strings.LastIndex(url, "=")+1:]
+			if strings.Contains(url, "vid=") {
+				sc.SiteID = url[strings.LastIndex(url, "=")+1:]
+			}
 		})
-
 		// Filenames (only a guess for now, according to the sample files)
 		suffixes := []string{"4k", "5k", "5k10", "6k", "7k", "960p", "1440p", "psvr_1440p"}
 		base := e.Request.URL.Path