// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CSSTokenizer_h
#define CSSTokenizer_h

#include "core/CoreExport.h"
#include "core/css/parser/CSSParserToken.h"
#include "core/css/parser/CSSTokenizerInputStream.h"
#include "core/html/parser/InputStreamPreprocessor.h"
#include "wtf/Allocator.h"
#include "wtf/text/StringView.h"
#include "wtf/text/WTFString.h"

#include <climits>

namespace blink {

class CSSTokenizerInputStream;
class CSSParserObserverWrapper;
class CSSParserTokenRange;

class CORE_EXPORT CSSTokenizer {
  WTF_MAKE_NONCOPYABLE(CSSTokenizer);
  DISALLOW_NEW();

 public:
  CSSTokenizer(const String&);
  CSSTokenizer(const String&, CSSParserObserverWrapper&);  // For the inspector

  CSSParserTokenRange tokenRange();
  unsigned tokenCount();

  Vector<String> takeEscapedStrings() { return std::move(m_stringPool); }

 private:
  CSSParserToken nextToken();

  UChar consume();
  void reconsume(UChar);

  CSSParserToken consumeNumericToken();
  CSSParserToken consumeIdentLikeToken();
  CSSParserToken consumeNumber();
  CSSParserToken consumeStringTokenUntil(UChar);
  CSSParserToken consumeUnicodeRange();
  CSSParserToken consumeUrlToken();

  void consumeBadUrlRemnants();
  void consumeSingleWhitespaceIfNext();
  void consumeUntilCommentEndFound();

  bool consumeIfNext(UChar);
  StringView consumeName();
  UChar32 consumeEscape();

  bool nextTwoCharsAreValidEscape();
  bool nextCharsAreNumber(UChar);
  bool nextCharsAreNumber();
  bool nextCharsAreIdentifier(UChar);
  bool nextCharsAreIdentifier();

  CSSParserToken blockStart(CSSParserTokenType);
  CSSParserToken blockStart(CSSParserTokenType blockType,
                            CSSParserTokenType,
                            StringView);
  CSSParserToken blockEnd(CSSParserTokenType, CSSParserTokenType startType);

  CSSParserToken whiteSpace(UChar);
  CSSParserToken leftParenthesis(UChar);
  CSSParserToken rightParenthesis(UChar);
  CSSParserToken leftBracket(UChar);
  CSSParserToken rightBracket(UChar);
  CSSParserToken leftBrace(UChar);
  CSSParserToken rightBrace(UChar);
  CSSParserToken plusOrFullStop(UChar);
  CSSParserToken comma(UChar);
  CSSParserToken hyphenMinus(UChar);
  CSSParserToken asterisk(UChar);
  CSSParserToken lessThan(UChar);
  CSSParserToken solidus(UChar);
  CSSParserToken colon(UChar);
  CSSParserToken semiColon(UChar);
  CSSParserToken hash(UChar);
  CSSParserToken circumflexAccent(UChar);
  CSSParserToken dollarSign(UChar);
  CSSParserToken verticalLine(UChar);
  CSSParserToken tilde(UChar);
  CSSParserToken commercialAt(UChar);
  CSSParserToken reverseSolidus(UChar);
  CSSParserToken asciiDigit(UChar);
  CSSParserToken letterU(UChar);
  CSSParserToken nameStart(UChar);
  CSSParserToken stringStart(UChar);
  CSSParserToken endOfFile(UChar);

  StringView registerString(const String&);

  using CodePoint = CSSParserToken (CSSTokenizer::*)(UChar);
  static const CodePoint codePoints[];

  CSSTokenizerInputStream m_input;
  Vector<CSSParserTokenType, 8> m_blockStack;

  Vector<CSSParserToken, 32> m_tokens;
  // We only allocate strings when escapes are used.
  Vector<String> m_stringPool;
};

}  // namespace blink

#endif  // CSSTokenizer_h
