// Copyright 2017 The Fuchsia 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 LIB_FIDL_LLCPP_STRING_VIEW_H_
#define LIB_FIDL_LLCPP_STRING_VIEW_H_

#include <cstring>
#include <string>
#if defined(__cplusplus) && __cplusplus >= 201703L
#include <string_view>
#else
#include <lib/fit/string_view.h>
#endif

#include <lib/fidl/walker.h>
#include <zircon/fidl.h>

#include <type_traits>

#include "fidl_allocator.h"
#include "vector_view.h"

namespace fidl {

// A FIDL string that either borrows or owns its contents.
//
// To borrow the contents of an |std::string|,
// use |fidl::unowned_str(my_str)|.
class StringView final : private VectorView<const char> {
 public:
  StringView() : VectorView() {}
  StringView(tracking_ptr<const char[]>&& data, uint64_t size)
      : VectorView(std::move(data), size) {}
  explicit StringView(VectorView<char>&& vv) : VectorView(std::move(vv)) {}
  explicit StringView(VectorView<const char>&& vv) : VectorView(std::move(vv)) {}
  // Allocates a string using the allocator.
#if defined(__cplusplus) && __cplusplus >= 201703L
  StringView(AnyAllocator& allocator, std::string_view from) : VectorView(allocator, from.size()) {
    memcpy(const_cast<char*>(VectorView::mutable_data()), from.data(), from.size());
  }
#else
  StringView(AnyAllocator& allocator, const std::string& from)
      : VectorView(allocator, from.size()) {
    memcpy(const_cast<char*>(VectorView::mutable_data()), from.data(), from.size());
  }
#endif

  // Constructs a fidl::StringView referencing a string literal. For example:
  //
  //     fidl::StringView view("hello");
  //     view.size() == 5;
  //
  template <size_t N>
  constexpr StringView(const char (&literal)[N], uint64_t size = N - 1)
      : VectorView(fidl::unowned_ptr_t<const char>(literal), size) {
    static_assert(N > 0, "String should not be empty");
  }

  // Assigns a new value using the allocator.
#if defined(__cplusplus) && __cplusplus >= 201703L
  void Set(AnyAllocator& allocator, std::string_view from) {
    Allocate(allocator, from.size());
    memcpy(const_cast<char*>(VectorView::mutable_data()), from.data(), from.size());
  }
#else
  void Set(AnyAllocator& allocator, fit::string_view& from) {
    Allocate(allocator, from.size());
    memcpy(const_cast<char*>(VectorView::mutable_data()), from.data(), from.size());
  }
#endif

#if defined(__cplusplus) && __cplusplus >= 201703L
  std::string_view get() const { return std::string_view(data(), size()); }
#endif

  uint64_t size() const { return count(); }
  void set_size(uint64_t size) { set_count(size); }

  const char* data() const { return VectorView::data(); }
  void set_data(tracking_ptr<const char[]> data) { VectorView::set_data(std::move(data)); }

  bool is_null() const { return data() == nullptr; }
  bool empty() const { return size() == 0; }

  const char& at(size_t offset) const { return data()[offset]; }

  const char& operator[](size_t offset) const { return at(offset); }

  const char* begin() const { return data(); }
  const char* cbegin() const { return data(); }

  const char* end() const { return data() + size(); }
  const char* cend() const { return data() + size(); }
};

}  // namespace fidl

#endif  // LIB_FIDL_LLCPP_STRING_VIEW_H_