[LIB-13] Method scanUrl implementation for several scanners

[chesshog-scanner.git] / src / main / java / org / hedgecode / chess / scanner / portal / AbstractSettingsScanner.java

diff --git a/src/main/java/org/hedgecode/chess/scanner/portal/AbstractSettingsScanner.java b/src/main/java/org/hedgecode/chess/scanner/portal/AbstractSettingsScanner.java
index 58ac5b0..445b603 100644 (file)
--- a/src/main/java/org/hedgecode/chess/scanner/portal/AbstractSettingsScanner.java
+++ b/src/main/java/org/hedgecode/chess/scanner/portal/AbstractSettingsScanner.java
@@ -16,6 +16,9 @@
 
 package org.hedgecode.chess.scanner.portal;
 
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
 import org.hedgecode.chess.scanner.ChessHogScannerException;
 import org.hedgecode.chess.scanner.Initiable;
 import org.hedgecode.chess.scanner.Settings;
@@ -24,6 +27,7 @@ import org.hedgecode.chess.scanner.regex.RegexBuilder;
 import org.hedgecode.chess.scanner.regex.RegexParams;
 import org.hedgecode.chess.scanner.spi.ServiceRegistry;
 
+import static org.hedgecode.chess.scanner.ChessHogScannerConstants.*;
 import static org.hedgecode.chess.scanner.regex.RegexBuilder.Type;
 
 /**
@@ -53,7 +57,7 @@ public abstract class AbstractSettingsScanner extends AbstractRequestScanner imp
 
     protected String assignUrl(String tournamentId, String pageId) {
         String tournamentUrl = assignUrlWithParams(
-                settings.getTournamentUrl(), settings.getGameQueryParams()
+                settings.getTournamentUrl(), settings.getTournamentQueryParams()
         );
         return RegexBuilder.build(
                 Type.TOURNAMENT,
@@ -76,14 +80,44 @@ public abstract class AbstractSettingsScanner extends AbstractRequestScanner imp
         );
     }
 
+    protected String assignUrl(String gameId, String tournamentId, boolean withTournament) {
+        String gameUrl = settings.getGamePgnUrl() != null
+                ? settings.getGamePgnUrl()
+                : settings.getGameUrl();
+        return RegexBuilder.build(
+                Type.GAME,
+                assignUrlWithParams(gameUrl, settings.getGameQueryParams()),
+                new RegexParams(
+                        gameId,
+                        tournamentId,
+                        withTournament
+                )
+        );
+    }
+
     protected String assignUrl(String query, boolean isUrlEncode) {
         return RegexBuilder.build(
                 Type.QUERY,
                 settings.getTournamentQuery(),
-                new RegexParams(query, isUrlEncode)
+                new RegexParams(
+                        query,
+                        isUrlEncode
+                )
         );
     }
 
+    protected String regex(String source, String regex) {
+        Matcher matcher = Pattern.compile(regex, Pattern.MULTILINE).matcher(source);
+        if (matcher.find()) {
+            return matcher.groupCount() > 0 ? matcher.group(1) : matcher.group();
+        }
+        return null;
+    }
+
+    protected boolean isPgnFormat(String source) {
+        return regex(source, PGN_DETECT_REGEX) != null;
+    }
+
     private String assignUrlWithParams(String url, String params) {
         return params != null
                 ? url.concat(params)