#pragma once

#include <string>

#include "envoy/extensions/path/match/uri_template/v3/uri_template_match.pb.h"

#include "envoy/extensions/path/match/uri_template/v3/uri_template_match.pb.validate.h"

#include "envoy/extensions/path/rewrite/uri_template/v3/uri_template_rewrite.pb.h"

#include "envoy/extensions/path/rewrite/uri_template/v3/uri_template_rewrite.pb.validate.h"

#include "envoy/router/path_matcher.h"

#include "envoy/router/path_rewriter.h"

#include "source/common/protobuf/message_validator_impl.h"

#include "source/common/protobuf/utility.h"

#include "source/extensions/path/uri_template_lib/uri_template.h"

#include "absl/status/statusor.h"

#include "absl/strings/string_view.h"

namespace Envoy {

namespace Extensions {

namespace UriTemplate {

namespace Rewrite {

const absl::string_view NAME = "envoy.path.rewrite.uri_template.uri_template_rewriter";

/**

 * UriTemplateRewriter allows rewriting paths based on match pattern variables provided

 * in UriTemplateMatcher.

 *

 * Example:

 * UriTemplateMatcher = /foo/bar/{var}

 * UriTemplateRewriter = /foo/{var}

 *    Will replace segment of path with value of {var}

 *    e.g. /foo/bar/cat -> /foo/cat

 */

class UriTemplateRewriter : public Router::PathRewriter {

public:

  explicit UriTemplateRewriter(

      const envoy::extensions::path::rewrite::uri_template::v3::UriTemplateRewriteConfig&

          rewrite_config)

  // Router::PathRewriter

  /**

   * Concatenates literals and extracts variable values to form the final rewritten path.

   * For example:

   * rewrite_pattern: [capture_index=2, literal="cat"]

   * path: "/bar/var"

   * capture_regex: "(1)/(2)"

   * Rewrite would result in rewrite of "/var/cat".

   */

  absl::StatusOr<std::string> rewritePath(absl::string_view pattern,

                                          absl::string_view matched_path) const override;

  absl::Status isCompatiblePathMatcher(Router::PathMatcherSharedPtr path_matcher) const override;

private:

  std::string rewrite_pattern_;

};

} // namespace Rewrite

} // namespace UriTemplate

} // namespace Extensions

} // namespace Envoy