[LIB-9] Set of entities for PGN parsing and formatting
[chesshog.git] / chesshog-format / src / main / java / org / hedgecode / chess / pgn / token / Tokenizer.java
1 /*
2  * Copyright (c) 2018-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.pgn.token;
18
19 import java.util.regex.Pattern;
20
21 import org.hedgecode.chess.pgn.entity.Game;
22 import org.hedgecode.chess.position.ParseException;
23
24 /**
25  * Tokenizer
26  *
27  * @author Dmitry Samoshin aka gotty
28  */
29 public class Tokenizer {
30
31     enum Type {
32
33         TAG   ( new TagToken()   ),
34         EMPTY ( new EmptyToken() ),
35         MOVES ( new MovesToken() );
36
37         private Token<Game> token;
38
39         public Token<Game> token() {
40             return token;
41         }
42
43         Type(Token<Game> token) {
44             this.token = token;
45         }
46     }
47
48     private static final Pattern TAG_PATTERN = Pattern.compile("^\\[[^]]+\\]$", Pattern.MULTILINE);
49     private static final Pattern EMPTY_PATTERN = Pattern.compile("^\\s*$", Pattern.MULTILINE);
50     private static final Pattern MOVES_PATTERN = Pattern.compile("^\\s*([^\\s])", Pattern.MULTILINE);
51
52     private final String pgn;
53
54     private Token<Game> token;
55     private String tokenPgn;
56
57     public Tokenizer(String pgn) {
58         this.pgn = pgn;
59         this.tokenPgn = pgn;
60     }
61
62     public boolean hasToken() {
63         assignToken();
64         return token != null;
65     }
66
67     public void token(Game game) throws ParseException {
68         if (token == null) {
69             throw new ParseException("parse.pgn.null.token");
70         }
71         int index = token.token(game, tokenPgn);
72         tokenPgn = tokenPgn.substring(index);
73         token = null;
74     }
75
76     public String sourcePgn() {
77         return pgn;
78     }
79
80     private void assignToken() {
81         token = null;
82         if (!tokenPgn.isEmpty()) {
83             if (TAG_PATTERN.matcher(tokenPgn).find()) {
84                 token = Type.TAG.token();
85             } else if (EMPTY_PATTERN.matcher(tokenPgn).find()) {
86                 token = Type.EMPTY.token();
87             } else if (MOVES_PATTERN.matcher(tokenPgn).find()) {
88                 token = Type.MOVES.token();
89             }
90         }
91     }
92
93 }