/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <memory>
#include <optional>
#include <string>
#include <string_view>
#include <vector>

namespace simpleperf {

class RegExMatch {
 public:
  virtual ~RegExMatch();
  virtual bool IsValid() const = 0;
  virtual std::string GetField(size_t index) const = 0;
  virtual void MoveToNextMatch() = 0;
};

// A wrapper of the regex engine (currently libpcre2).
class RegEx {
 public:
  static std::unique_ptr<RegEx> Create(std::string_view pattern);
  virtual ~RegEx() {}
  const std::string& GetPattern() const { return pattern_; }

  // Thread-safe for concurrent calls on the same object.
  virtual bool Match(std::string_view s) const = 0;
  virtual bool Search(std::string_view s) const = 0;

  // Not thread-safe for concurrent calls on the same object. But it is about 2x faster than
  // Match() and Search() because of reusing match data.
  virtual bool ThreadUnsafeMatch(std::string_view s) const = 0;
  virtual bool ThreadUnsafeSearch(std::string_view s) const = 0;

  // Always return a not-null RegExMatch. If no match, RegExMatch->IsValid() is false.
  virtual std::unique_ptr<RegExMatch> SearchAll(std::string_view s) const = 0;
  virtual std::optional<std::string> Replace(const std::string& s,
                                             const std::string& format) const = 0;

 protected:
  RegEx(std::string_view pattern) : pattern_(pattern) {}

  std::string pattern_;
};

bool SearchInRegs(std::string_view s, const std::vector<std::unique_ptr<RegEx>>& regs);

}  // namespace simpleperf
