//===- FormatAdapters.h - Formatters for common LLVM types -----*- C++ -*-===//

//

// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.

// See https://llvm.org/LICENSE.txt for license information.

// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

//

//===----------------------------------------------------------------------===//

#ifndef LLVM_SUPPORT_FORMATADAPTERS_H

#define LLVM_SUPPORT_FORMATADAPTERS_H

#include "llvm/ADT/StringRef.h"

#include "llvm/Support/Error.h"

#include "llvm/Support/FormatCommon.h"

#include "llvm/Support/FormatVariadicDetails.h"

#include "llvm/Support/raw_ostream.h"

namespace llvm {

template <typename T> class FormatAdapter : public detail::format_adapter {

protected:

  explicit FormatAdapter(T &&Item) : Item(std::forward<T>(Item)) {}

  T Item;

};

namespace detail {

template <typename T> class AlignAdapter final : public FormatAdapter<T> {

  AlignStyle Where;

  size_t Amount;

  char Fill;

public:

  AlignAdapter(T &&Item, AlignStyle Where, size_t Amount, char Fill)

      : FormatAdapter<T>(std::forward<T>(Item)), Where(Where), Amount(Amount),

        Fill(Fill) {}

  void format(llvm::raw_ostream &Stream, StringRef Style) {

    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));

    FmtAlign(Adapter, Where, Amount, Fill).format(Stream, Style);

  }

};

template <typename T> class PadAdapter final : public FormatAdapter<T> {

  size_t Left;

  size_t Right;

public:

  PadAdapter(T &&Item, size_t Left, size_t Right)

      : FormatAdapter<T>(std::forward<T>(Item)), Left(Left), Right(Right) {}

  void format(llvm::raw_ostream &Stream, StringRef Style) {

    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));

    Stream.indent(Left);

    Adapter.format(Stream, Style);

    Stream.indent(Right);

  }

};

template <typename T> class RepeatAdapter final : public FormatAdapter<T> {

  size_t Count;

public:

  RepeatAdapter(T &&Item, size_t Count)

      : FormatAdapter<T>(std::forward<T>(Item)), Count(Count) {}

  void format(llvm::raw_ostream &Stream, StringRef Style) {

    auto Adapter = detail::build_format_adapter(std::forward<T>(this->Item));

    for (size_t I = 0; I < Count; ++I) {

      Adapter.format(Stream, Style);

    }

  }

};

class ErrorAdapter : public FormatAdapter<Error> {

public:

  ErrorAdapter(Error &&Item) : FormatAdapter(std::move(Item)) {}

  ErrorAdapter(ErrorAdapter &&) = default;

  ~ErrorAdapter() { consumeError(std::move(Item)); }

  void format(llvm::raw_ostream &Stream, StringRef Style) { Stream << Item; }

};

}

template <typename T>

detail::AlignAdapter<T> fmt_align(T &&Item, AlignStyle Where, size_t Amount,

                                  char Fill = ' ') {

  return detail::AlignAdapter<T>(std::forward<T>(Item), Where, Amount, Fill);

}

template <typename T>

detail::PadAdapter<T> fmt_pad(T &&Item, size_t Left, size_t Right) {

  return detail::PadAdapter<T>(std::forward<T>(Item), Left, Right);

}

template <typename T>

detail::RepeatAdapter<T> fmt_repeat(T &&Item, size_t Count) {

  return detail::RepeatAdapter<T>(std::forward<T>(Item), Count);

}

// llvm::Error values must be consumed before being destroyed.

// Wrapping an error in fmt_consume explicitly indicates that the formatv_object

// should take ownership and consume it.

inline detail::ErrorAdapter fmt_consume(Error &&Item) {

  return detail::ErrorAdapter(std::move(Item));

}

}

#endif