/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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

#ifndef GEODE_UTIL_STRING_H_
#define GEODE_UTIL_STRING_H_

#include <codecvt>
#include <locale>
#include <string>

#include "type_traits.hpp"

namespace apache {
namespace geode {
namespace client {

/**
 * Native codecvt_mode endianess
 */
constexpr std::codecvt_mode codecvt_mode_native_endian =
    endian::native == endian::little ? std::little_endian
                                     : static_cast<std::codecvt_mode>(0);

inline std::u16string to_utf16(const std::string& utf8) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto int16String =
      std::wstring_convert<std::codecvt_utf8_utf16<int16_t>, int16_t>{}
          .from_bytes(utf8);
  return std::u16string(reinterpret_cast<const char16_t*>(int16String.data()),
                        int16String.size());
#else
  return std::wstring_convert<std::codecvt_utf8_utf16<char16_t>, char16_t>{}
      .from_bytes(utf8);
#endif
}

inline std::u16string to_utf16(const std::u32string& ucs4) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto data = reinterpret_cast<const int32_t*>(ucs4.data());
  auto bytes =
      std::wstring_convert<
          std::codecvt_utf16<int32_t, 0x10ffff, codecvt_mode_native_endian>,
          int32_t>{}
          .to_bytes(data, data + ucs4.size());
#else
  auto bytes =
      std::wstring_convert<
          std::codecvt_utf16<char32_t, 0x10ffff, codecvt_mode_native_endian>,
          char32_t>{}
          .to_bytes(ucs4);
#endif

  return std::u16string(reinterpret_cast<const char16_t*>(bytes.c_str()),
                        bytes.length() / sizeof(char16_t));
}

inline std::u16string to_utf16(const char32_t* ucs4, size_t len) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto data = reinterpret_cast<const int32_t*>(ucs4);
  auto bytes =
      std::wstring_convert<
          std::codecvt_utf16<int32_t, 0x10ffff, codecvt_mode_native_endian>,
          int32_t>{}
          .to_bytes(data, data + len);
#else
  auto bytes =
      std::wstring_convert<
          std::codecvt_utf16<char32_t, 0x10ffff, codecvt_mode_native_endian>,
          char32_t>{}
          .to_bytes(ucs4, ucs4 + len);
#endif

  return std::u16string(reinterpret_cast<const char16_t*>(bytes.c_str()),
                        bytes.length() / sizeof(char16_t));
}

inline std::u32string to_ucs4(const std::u16string& utf16) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto data = reinterpret_cast<const char*>(utf16.data());
  auto tmp =
      std::wstring_convert<
          std::codecvt_utf16<int32_t, 0x10ffff, codecvt_mode_native_endian>,
          int32_t>{}
          .from_bytes(data, data + (utf16.length() * sizeof(char16_t)));
  return std::u32string(reinterpret_cast<const char32_t*>(tmp.data()),
                        tmp.length());
#else
  auto data = reinterpret_cast<const char*>(utf16.data());
  return std::wstring_convert<
             std::codecvt_utf16<char32_t, 0x10ffff, codecvt_mode_native_endian>,
             char32_t>{}
      .from_bytes(data, data + (utf16.length() * sizeof(char16_t)));
#endif
}

inline std::string to_utf8(const std::u16string& utf16) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto data = reinterpret_cast<const int16_t*>(utf16.data());
  return std::wstring_convert<std::codecvt_utf8_utf16<int16_t>, int16_t>{}
      .to_bytes(data, data + utf16.size());
#else
  return std::wstring_convert<std::codecvt_utf8_utf16<char16_t>, char16_t>{}
      .to_bytes(utf16);
#endif
}

inline std::string to_utf8(const std::u32string& ucs4) {
#if defined(_MSC_VER) && _MSC_VER >= 1900
  /*
   * Workaround for missing std:codecvt identifier.
   * https://connect.microsoft.com/VisualStudio/feedback/details/1403302
   */
  auto data = reinterpret_cast<const int32_t*>(ucs4.data());
  return std::wstring_convert<std::codecvt_utf8<int32_t>, int32_t>{}.to_bytes(
      data, data + ucs4.size());
#else
  return std::wstring_convert<std::codecvt_utf8<char32_t>, char32_t>{}.to_bytes(
      ucs4);
#endif
}

}  // namespace client
}  // namespace geode
}  // namespace apache

#endif  // GEODE_UTIL_STRING_H_