src/main/java/org/hedgecode/chess/scanner/portal/ChessGamesScanner.java

   1 /*
   2  * Copyright (c) 2019-2020. Developed by Hedgecode.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *   http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.hedgecode.chess.scanner.portal;
  18
  19 import java.util.ArrayList;
  20 import java.util.List;
  21 import java.util.Map;
  22
  23 import org.hedgecode.chess.scanner.Scanner;
  24 import org.hedgecode.chess.scanner.ScannerException;
  25 import org.hedgecode.chess.scanner.entity.PGNGame;
  26 import org.hedgecode.chess.scanner.entity.PGNTournament;
  27
  28 /**
  29  * ChessGamesScanner
  30  *
  31  * @author Dmitry Samoshin aka gotty
  32  */
  33 public class ChessGamesScanner extends AbstractSettingsScanner implements Scanner {
  34
  35     private static final String SETTINGS_FILENAME = "chessgames.settings";
  36
  37     @Override
  38     protected String getResourceName() {
  39         return SETTINGS_FILENAME;
  40     }
  41
  42     @Override
  43     public PGNTournament scanTournament(String tournamentId) throws ScannerException {
  44         PGNTournament tournament = new PGNTournament(tournamentId);
  45         assignTournamentGames(tournament);
  46         return tournament;
  47     }
  48
  49     @Override
  50     public PGNTournament findTournament(String tournamentName) throws ScannerException {
  51         PGNTournament tournament = null;
  52         Map<String, String> result = matchMap(
  53                 assignUrl(
  54                         tournamentName, true
  55                 ),
  56                 getSettings().getTournamentQueryUrlRegex(),
  57                 true
  58         );
  59         for (Map.Entry<String, String> entry : result.entrySet()) {
  60             if (entry.getValue().contains(tournamentName)) { // todo: contains?
  61                 tournament = new PGNTournament(
  62                         entry.getKey(),
  63                         entry.getValue()
  64                 );
  65             }
  66         }
  67         if (tournament != null) {
  68             assignTournamentGames(tournament);
  69         }
  70         return tournament;
  71     }
  72
  73     @Override
  74     public PGNGame scanGame(String gameId) throws ScannerException {
  75         String pgn = request(
  76                 assignUrl(gameId)
  77         );
  78         if (!isPgnFormat(pgn)) {
  79             throw new ScannerException(
  80                     String.format("Failed to get PGN for requesting game ID: %s", gameId)
  81             );
  82         }
  83         return new PGNGame(
  84                 gameId, pgn
  85         );
  86     }
  87
  88     @Override
  89     public PGNGame scanGame(String gameId, String tournamentId) throws ScannerException {
  90         return scanGame(gameId);
  91     }
  92
  93     @Override
  94     public PGNGame scanUrl(String gameUrl) throws ScannerException {
  95         String pgn = regex(
  96                 request(
  97                         gameUrl
  98                 ),
  99                 getSettings().getGameUrlRegex()
 100         );
 101         if (pgn == null || !isPgnFormat(pgn)) {
 102             throw new ScannerException(
 103                     String.format("Failed to get PGN for requesting URL: %s", gameUrl)
 104             );
 105         }
 106         return new PGNGame(
 107                 null,
 108                 pgn
 109         );
 110     }
 111
 112     private void assignTournamentGames(PGNTournament tournament) throws ScannerException {
 113         tournament.clearGames();
 114         List<String> gamesId = new ArrayList<>();
 115
 116         List<String> pageGamesId;
 117         int pageId = 0;
 118         do {
 119             pageGamesId = match(
 120                     assignUrl(
 121                             tournament.id(),
 122                             Integer.toString(++pageId)
 123                     ),
 124                     getSettings().getTournamentGamesUrlRegex(),
 125                     false
 126             );
 127             gamesId.addAll(
 128                     pageGamesId
 129             );
 130         } while (!pageGamesId.isEmpty());
 131
 132         for (String gameId : gamesId) {
 133             tournament.addGame(
 134                     scanGame(gameId)
 135             );
 136         }
 137     }
 138
 139 }