//

// Copyright (c) 2019 Vinnie Falco (vinnie.falco@gmail.com)

//

// Distributed under the Boost Software License, Version 1.0. (See accompanying

// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

//

// Official repository: https://github.com/boostorg/json

//

#ifndef BOOST_JSON_IMPL_STRING_IPP

#define BOOST_JSON_IMPL_STRING_IPP

#include <boost/json/detail/except.hpp>

#include <algorithm>

#include <new>

#include <ostream>

#include <stdexcept>

#include <string>

#include <utility>

namespace boost {

namespace json {

//----------------------------------------------------------

//

// Construction

//

//----------------------------------------------------------

string::

string(

    std::size_t count,

    char ch,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(count, ch);

}

string::

string(

    char const* s,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(s);

}

string::

string(

    char const* s,

    std::size_t count,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(s, count);

}

string::

string(string const& other)

    : sp_(other.sp_)

{

    assign(other);

}

string::

string(

    string const& other,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(other);

}

string::

string(

    string&& other,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(std::move(other));

}

string::

string(

    string_view s,

    storage_ptr sp)

    : sp_(std::move(sp))

{

    assign(s);

}

//----------------------------------------------------------

//

// Assignment

//

//----------------------------------------------------------

string&

string::

operator=(string const& other)

{

    return assign(other);

}

string&

string::

operator=(string&& other)

{

    return assign(std::move(other));

}

string&

string::

operator=(char const* s)

{

    return assign(s);

}

string&

string::

operator=(string_view s)

{

    return assign(s);

}

string&

string::

assign(

    size_type count,

    char ch)

{

    std::char_traits<char>::assign(

        impl_.assign(count, sp_),

        count,

        ch);

    return *this;

}

string&

string::

assign(

    string const& other)

{

    if(this == &other)

        return *this;

    return assign(

        other.data(),

        other.size());

}

string&

string::

assign(string&& other)

{

    if( &other == this )

        return *this;

    if(*sp_ == *other.sp_)

    {

        impl_.destroy(sp_);

        impl_ = other.impl_;

        ::new(&other.impl_) detail::string_impl();

        return *this;

    }

    // copy

    return assign(other);

}

string&

string::

assign(

    char const* s,

    size_type count)

{

    std::char_traits<char>::copy(

        impl_.assign(count, sp_),

        s, count);

    return *this;

}

string&

string::

assign(

    char const* s)

{

    return assign(s, std::char_traits<

        char>::length(s));

}

//----------------------------------------------------------

//

// Capacity

//

//----------------------------------------------------------

void

string::

shrink_to_fit()

{

    impl_.shrink_to_fit(sp_);

}

//----------------------------------------------------------

//

// Access

//

//----------------------------------------------------------

system::result<char&>

string::try_at(std::size_t pos) noexcept

{

    if( pos < size() )

        return impl_.data()[pos];

    system::error_code ec;

    BOOST_JSON_FAIL(ec, error::out_of_range);

    return ec;

}

system::result<char const&>

string::try_at(std::size_t pos) const noexcept

{

    if( pos < size() )

        return impl_.data()[pos];

    system::error_code ec;

    BOOST_JSON_FAIL(ec, error::out_of_range);

    return ec;

}

char const&

string::at(std::size_t pos, source_location const& loc) const

{

    return try_at(pos).value(loc);

}

//----------------------------------------------------------

//

// Operations

//

//----------------------------------------------------------

void

string::

clear() noexcept

{

    impl_.term(0);

}

//----------------------------------------------------------

void

string::

push_back(char ch)

{

    *impl_.append(1, sp_) = ch;

}

void

string::

pop_back()

{

    back() = 0;

    impl_.size(impl_.size() - 1);

}

//----------------------------------------------------------

string&

string::

append(size_type count, char ch)

{

    std::char_traits<char>::assign(

        impl_.append(count, sp_),

        count, ch);

    return *this;

}

string&

string::

append(string_view sv)

{

    std::char_traits<char>::copy(

        impl_.append(sv.size(), sp_),

        sv.data(), sv.size());

    return *this;

}

//----------------------------------------------------------

string&

string::

insert(

    size_type pos,

    string_view sv)

{

    impl_.insert(pos, sv.data(), sv.size(), sp_);

    return *this;

}

string&

string::

insert(

    std::size_t pos,

    std::size_t count,

    char ch)

{

    std::char_traits<char>::assign(

        impl_.insert_unchecked(pos, count, sp_),

        count, ch);

    return *this;

}

//----------------------------------------------------------

string&

string::

replace(

    std::size_t pos,

    std::size_t count,

    string_view sv)

{

    impl_.replace(pos, count, sv.data(), sv.size(), sp_);

    return *this;

}

string&

string::

replace(

    std::size_t pos,

    std::size_t count,

    std::size_t count2,

    char ch)

{

    std::char_traits<char>::assign(

        impl_.replace_unchecked(pos, count, count2, sp_),

        count2, ch);

    return *this;

}

//----------------------------------------------------------

string&

string::

erase(

    size_type pos,

    size_type count)

{

    if(pos > impl_.size())

    {

        BOOST_STATIC_CONSTEXPR source_location loc = BOOST_CURRENT_LOCATION;

        detail::throw_system_error( error::out_of_range, &loc );

    }

    if( count > impl_.size() - pos)

        count = impl_.size() - pos;

    std::char_traits<char>::move(

        impl_.data() + pos,

        impl_.data() + pos + count,

        impl_.size() - pos - count + 1);

    impl_.term(impl_.size() - count);

    return *this;

}

auto

string::

erase(const_iterator pos) ->

    iterator

{

    return erase(pos, pos+1);

}

auto

string::

erase(

    const_iterator first,

    const_iterator last) ->

        iterator

{

    auto const pos = first - begin();

    auto const count = last - first;

    erase(pos, count);

    return data() + pos;

}

//----------------------------------------------------------

void

string::

resize(size_type count, char ch)

{

    if(count <= impl_.size())

    {

        impl_.term(count);

        return;

    }

    reserve(count);

    std::char_traits<char>::assign(

        impl_.end(),

        count - impl_.size(),

        ch);

    grow(count - size());

}

//----------------------------------------------------------

void

string::

swap(string& other)

{

    if(*sp_ == *other.sp_)

    {

        std::swap(impl_, other.impl_);

        return;

    }

    string temp1(

        std::move(*this), other.sp_);

    string temp2(

        std::move(other), sp_);

    this->~string();

    ::new(this) string(pilfer(temp2));

    other.~string();

    ::new(&other) string(pilfer(temp1));

}

//----------------------------------------------------------

void

string::

reserve_impl(size_type new_cap)

{

    BOOST_ASSERT(

        new_cap >= impl_.capacity());

    if(new_cap > impl_.capacity())

    {

        // grow

        new_cap = detail::string_impl::growth(

            new_cap, impl_.capacity());

        detail::string_impl tmp(new_cap, sp_);

        std::char_traits<char>::copy(tmp.data(),

            impl_.data(), impl_.size() + 1);

        tmp.size(impl_.size());

        impl_.destroy(sp_);

        impl_ = tmp;

        return;

    }

}

} // namespace json

} // namespace boost

//----------------------------------------------------------

std::size_t

std::hash< ::boost::json::string >::operator()(

    ::boost::json::string const& js ) const noexcept

{

    return ::boost::hash< ::boost::json::string >()( js );

}

#endif