// 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 THIRD_PARTY_BLINK_RENDERER_CORE_CSS_PARSER_MEDIA_QUERY_PARSER_H_
#define THIRD_PARTY_BLINK_RENDERER_CORE_CSS_PARSER_MEDIA_QUERY_PARSER_H_

#include "base/macros.h"
#include "third_party/blink/renderer/core/core_export.h"
#include "third_party/blink/renderer/core/css/media_list.h"
#include "third_party/blink/renderer/core/css/media_query.h"
#include "third_party/blink/renderer/core/css/media_query_exp.h"
#include "third_party/blink/renderer/core/css/parser/css_parser_mode.h"
#include "third_party/blink/renderer/core/css/parser/css_parser_token.h"
#include "third_party/blink/renderer/core/css/parser/css_parser_token_range.h"
#include "third_party/blink/renderer/core/css/parser/media_query_block_watcher.h"
#include "third_party/blink/renderer/platform/wtf/text/wtf_string.h"

namespace blink {

class MediaQuerySet;

class MediaQueryData {
  STACK_ALLOCATED();

 private:
  MediaQuery::RestrictorType restrictor_;
  String media_type_;
  ExpressionHeapVector expressions_;
  String media_feature_;
  bool media_type_set_;

 public:
  MediaQueryData();
  void Clear();
  void AddExpression(CSSParserTokenRange&);
  bool LastExpressionValid();
  void RemoveLastExpression();
  void SetMediaType(const String&);
  std::unique_ptr<MediaQuery> TakeMediaQuery();

  inline bool CurrentMediaQueryChanged() const {
    return (restrictor_ != MediaQuery::kNone || media_type_set_ ||
            expressions_.size() > 0);
  }
  inline MediaQuery::RestrictorType Restrictor() { return restrictor_; }

  inline void SetRestrictor(MediaQuery::RestrictorType restrictor) {
    restrictor_ = restrictor;
  }

  inline void SetMediaFeature(const String& str) { media_feature_ = str; }
  DISALLOW_COPY_AND_ASSIGN(MediaQueryData);
};

class CORE_EXPORT MediaQueryParser {
  STACK_ALLOCATED();

 public:
  static scoped_refptr<MediaQuerySet> ParseMediaQuerySet(const String&);
  static scoped_refptr<MediaQuerySet> ParseMediaQuerySet(CSSParserTokenRange);
  static scoped_refptr<MediaQuerySet> ParseMediaCondition(CSSParserTokenRange);
  static scoped_refptr<MediaQuerySet> ParseMediaQuerySetInMode(
      CSSParserTokenRange,
      CSSParserMode);

 private:
  enum ParserType {
    kMediaQuerySetParser,
    kMediaConditionParser,
  };

  MediaQueryParser(ParserType, CSSParserMode);
  virtual ~MediaQueryParser();

  scoped_refptr<MediaQuerySet> ParseImpl(CSSParserTokenRange);

  void ProcessToken(const CSSParserToken&, CSSParserTokenRange&);

  void ReadRestrictor(CSSParserTokenType,
                      const CSSParserToken&,
                      CSSParserTokenRange&);
  void ReadMediaNot(CSSParserTokenType,
                    const CSSParserToken&,
                    CSSParserTokenRange&);
  void ReadMediaType(CSSParserTokenType,
                     const CSSParserToken&,
                     CSSParserTokenRange&);
  void ReadAnd(CSSParserTokenType, const CSSParserToken&, CSSParserTokenRange&);
  void ReadFeatureStart(CSSParserTokenType,
                        const CSSParserToken&,
                        CSSParserTokenRange&);
  void ReadFeature(CSSParserTokenType,
                   const CSSParserToken&,
                   CSSParserTokenRange&);
  void ReadFeatureColon(CSSParserTokenType,
                        const CSSParserToken&,
                        CSSParserTokenRange&);
  void ReadFeatureValue(CSSParserTokenType,
                        const CSSParserToken&,
                        CSSParserTokenRange&);
  void ReadFeatureEnd(CSSParserTokenType,
                      const CSSParserToken&,
                      CSSParserTokenRange&);
  void SkipUntilComma(CSSParserTokenType,
                      const CSSParserToken&,
                      CSSParserTokenRange&);
  void SkipUntilBlockEnd(CSSParserTokenType,
                         const CSSParserToken&,
                         CSSParserTokenRange&);
  void Done(CSSParserTokenType, const CSSParserToken&, CSSParserTokenRange&);

  using State = void (MediaQueryParser::*)(CSSParserTokenType,
                                           const CSSParserToken&,
                                           CSSParserTokenRange&);

  void SetStateAndRestrict(State, MediaQuery::RestrictorType);
  void HandleBlocks(const CSSParserToken&);

  bool IsMediaFeatureAllowedInMode(const String& media_feature) const;

  State state_;
  ParserType parser_type_;
  MediaQueryData media_query_data_;
  scoped_refptr<MediaQuerySet> query_set_;
  MediaQueryBlockWatcher block_watcher_;
  CSSParserMode mode_;

  const static State kReadRestrictor;
  const static State kReadMediaNot;
  const static State kReadMediaType;
  const static State kReadAnd;
  const static State kReadFeatureStart;
  const static State kReadFeature;
  const static State kReadFeatureColon;
  const static State kReadFeatureValue;
  const static State kReadFeatureEnd;
  const static State kSkipUntilComma;
  const static State kSkipUntilBlockEnd;
  const static State kDone;
  DISALLOW_COPY_AND_ASSIGN(MediaQueryParser);
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_CORE_CSS_PARSER_MEDIA_QUERY_PARSER_H_