// Copyright 2016 The PDFium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com

#include "core/fpdfapi/parser/cpdf_simple_parser.h"

#include <stdint.h>

#include <optional>

#include "core/fpdfapi/parser/fpdf_parser_utility.h"
#include "core/fxcrt/check_op.h"

CPDF_SimpleParser::CPDF_SimpleParser(pdfium::span<const uint8_t> input)
    : data_(input) {}

CPDF_SimpleParser::~CPDF_SimpleParser() = default;

ByteStringView CPDF_SimpleParser::GetWord() {
  std::optional<uint8_t> start_char = SkipSpacesAndComments();
  if (!start_char.has_value()) {
    return ByteStringView();
  }

  CHECK_GT(cur_position_, 0);
  uint32_t start_position = cur_position_ - 1;
  CHECK_LT(start_position, data_.size());

  if (!PDFCharIsDelimiter(start_char.value())) {
    return HandleNonDelimiter();
  }

  switch (start_char.value()) {
    case '/':
      return HandleName();
    case '<':
      return HandleBeginAngleBracket();
    case '>':
      return HandleEndAngleBracket();
    case '(':
      return HandleParentheses();
    default:
      return GetDataToCurrentPosition(start_position);
  }
}

ByteStringView CPDF_SimpleParser::GetDataToCurrentPosition(
    uint32_t start_position) const {
  return ByteStringView(
      data_.subspan(start_position, cur_position_ - start_position));
}

std::optional<uint8_t> CPDF_SimpleParser::SkipSpacesAndComments() {
  while (true) {
    if (cur_position_ >= data_.size()) {
      return std::nullopt;
    }

    // Skip whitespaces.
    uint8_t cur_char = data_[cur_position_++];
    while (PDFCharIsWhitespace(cur_char)) {
      if (cur_position_ >= data_.size()) {
        return std::nullopt;
      }
      cur_char = data_[cur_position_++];
    }

    if (cur_char != '%') {
      return cur_char;
    }

    // Skip comments.
    while (true) {
      if (cur_position_ >= data_.size()) {
        return std::nullopt;
      }

      cur_char = data_[cur_position_++];
      if (PDFCharIsLineEnding(cur_char)) {
        break;
      }
    }
  }
}

ByteStringView CPDF_SimpleParser::HandleName() {
  uint32_t start_position = cur_position_ - 1;
  while (cur_position_ < data_.size()) {
    uint8_t cur_char = data_[cur_position_];
    // Stop parsing after encountering a whitespace or delimiter.
    if (PDFCharIsWhitespace(cur_char) || PDFCharIsDelimiter(cur_char)) {
      return GetDataToCurrentPosition(start_position);
    }
    ++cur_position_;
  }
  return ByteStringView();
}

ByteStringView CPDF_SimpleParser::HandleBeginAngleBracket() {
  uint32_t start_position = cur_position_ - 1;
  if (cur_position_ >= data_.size()) {
    return GetDataToCurrentPosition(start_position);
  }

  uint8_t cur_char = data_[cur_position_++];
  // Stop parsing if encountering "<<".
  if (cur_char == '<') {
    return GetDataToCurrentPosition(start_position);
  }

  // Continue parsing until end of `data_` or closing bracket.
  while (cur_position_ < data_.size() && cur_char != '>') {
    cur_char = data_[cur_position_++];
  }
  return GetDataToCurrentPosition(start_position);
}

ByteStringView CPDF_SimpleParser::HandleEndAngleBracket() {
  uint32_t start_position = cur_position_ - 1;
  if (cur_position_ < data_.size() && data_[cur_position_] == '>') {
    ++cur_position_;
  }
  return GetDataToCurrentPosition(start_position);
}

ByteStringView CPDF_SimpleParser::HandleParentheses() {
  uint32_t start_position = cur_position_ - 1;
  int level = 1;
  while (cur_position_ < data_.size() && level > 0) {
    uint8_t cur_char = data_[cur_position_++];
    if (cur_char == '(') {
      ++level;
    } else if (cur_char == ')') {
      --level;
    }
  }
  return GetDataToCurrentPosition(start_position);
}

ByteStringView CPDF_SimpleParser::HandleNonDelimiter() {
  uint32_t start_position = cur_position_ - 1;
  while (cur_position_ < data_.size()) {
    uint8_t cur_char = data_[cur_position_];
    if (PDFCharIsDelimiter(cur_char) || PDFCharIsWhitespace(cur_char)) {
      break;
    }
    ++cur_position_;
  }
  return GetDataToCurrentPosition(start_position);
}