001 /* 002 * Sonar, open source software quality management tool. 003 * Copyright (C) 2009 SonarSource SA 004 * mailto:contact AT sonarsource DOT com 005 * 006 * Sonar is free software; you can redistribute it and/or 007 * modify it under the terms of the GNU Lesser General Public 008 * License as published by the Free Software Foundation; either 009 * version 3 of the License, or (at your option) any later version. 010 * 011 * Sonar is distributed in the hope that it will be useful, 012 * but WITHOUT ANY WARRANTY; without even the implied warranty of 013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 014 * Lesser General Public License for more details. 015 * 016 * You should have received a copy of the GNU Lesser General Public 017 * License along with Sonar; if not, write to the Free Software 018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02 019 */ 020 package org.sonar.squid.text.colorizer; 021 022 import java.util.Collections; 023 import java.util.HashSet; 024 import java.util.Set; 025 026 import org.sonar.squid.text.CodeReader; 027 import org.sonar.squid.text.EndTokenMatcher; 028 029 public class KeywordsTokenizer extends Tokenizer { 030 031 private Set<String> keywords = new HashSet<String>(); 032 private EndTokenMatcher endTokenMatcher = new EndTokenMatcher() { 033 034 public boolean match(int endFlag) { 035 return !Character.isJavaIdentifierPart(endFlag); 036 } 037 }; 038 039 public KeywordsTokenizer(String tagBefore, String tagAfter, Set<String> keywords) { 040 super(tagBefore, tagAfter); 041 this.keywords = keywords; 042 } 043 044 public KeywordsTokenizer(String tagBefore, String tagAfter, String... keywords) { 045 super(tagBefore, tagAfter); 046 Collections.addAll(this.keywords, keywords); 047 } 048 049 public KeywordsTokenizer(String keyword) { 050 this("", "", keyword); 051 } 052 053 @Override 054 public boolean hasNextToken(CodeReader code) { 055 return Character.isJavaIdentifierStart(code.peek()); 056 } 057 058 @Override 059 public String nextToken(CodeReader code) { 060 String keyword = code.readTo(endTokenMatcher); 061 return keywords.contains(keyword) ? (tagBefore + keyword + tagAfter) : keyword; 062 } 063 064 }