//------------------------------------------------------------------------------
/*
    This file is part of Beast: https://github.com/vinniefalco/Beast
    Copyright 2013, Vinnie Falco <vinnie.falco@gmail.com>

    Permission to use, copy, modify, and/or distribute this software for any
    purpose  with  or without fee is hereby granted, provided that the above
    copyright notice and this permission notice appear in all copies.

    THE  SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
    WITH  REGARD  TO  THIS  SOFTWARE  INCLUDING  ALL  IMPLIED  WARRANTIES  OF
    MERCHANTABILITY  AND  FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
    ANY  SPECIAL ,  DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
    WHATSOEVER  RESULTING  FROM  LOSS  OF USE, DATA OR PROFITS, WHETHER IN AN
    ACTION  OF  CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
    OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
//==============================================================================

#include <beast/utility/PropertyStream.h>
#include <beast/unit_test/suite.h>

#include <limits>
#include <iostream>

namespace beast {

//------------------------------------------------------------------------------
//
// Item
//
//------------------------------------------------------------------------------

PropertyStream::Item::Item (Source* source)
    : m_source (source)
{
}

PropertyStream::Source& PropertyStream::Item::source() const
{
    return *m_source;
}

PropertyStream::Source* PropertyStream::Item::operator-> () const
{
    return &source();
}

PropertyStream::Source& PropertyStream::Item::operator* () const
{
    return source();
}

//------------------------------------------------------------------------------
//
// Proxy
//
//------------------------------------------------------------------------------

PropertyStream::Proxy::Proxy (
    Map const& map, std::string const& key)
    : m_map (&map)
    , m_key (key)
{
}

PropertyStream::Proxy::Proxy (Proxy const& other)
    : m_map (other.m_map)
    , m_key (other.m_key)
{
}

PropertyStream::Proxy::~Proxy ()
{
    std::string const s (m_ostream.str());
    if (! s.empty())
        m_map->add (m_key, s);
}

std::ostream& PropertyStream::Proxy::operator<< (
    std::ostream& manip (std::ostream&)) const
{
    return m_ostream << manip;
}

//------------------------------------------------------------------------------
//
// Map
//
//------------------------------------------------------------------------------

PropertyStream::Map::Map (PropertyStream& stream)
    : m_stream (stream)
{
}

PropertyStream::Map::Map (Set& parent)
    : m_stream (parent.stream())
{
    m_stream.map_begin ();
}

PropertyStream::Map::Map (std::string const& key, Map& map)
    : m_stream (map.stream())
{
    m_stream.map_begin (key);
}

PropertyStream::Map::Map (std::string const& key, PropertyStream& stream)
    : m_stream (stream)
{
    m_stream.map_begin (key);
}

PropertyStream::Map::~Map ()
{
    m_stream.map_end ();
}

PropertyStream& PropertyStream::Map::stream()
{
    return m_stream;
}

PropertyStream const& PropertyStream::Map::stream() const
{
    return m_stream;
}

PropertyStream::Proxy PropertyStream::Map::operator[] (std::string const& key)
{
    return Proxy (*this, key);
}

//------------------------------------------------------------------------------
//
// Set
//
//------------------------------------------------------------------------------

PropertyStream::Set::Set (std::string const& key, Map& map)
    : m_stream (map.stream())
{
    m_stream.array_begin (key);
}

PropertyStream::Set::Set (std::string const& key, PropertyStream& stream)
    : m_stream (stream)
{
    m_stream.array_begin (key);
}

PropertyStream::Set::~Set ()
{
    m_stream.array_end ();
}

PropertyStream& PropertyStream::Set::stream()
{
    return m_stream;
}

PropertyStream const& PropertyStream::Set::stream() const
{
    return m_stream;
}

//------------------------------------------------------------------------------
//
// Source
//
//------------------------------------------------------------------------------

PropertyStream::Source::Source (std::string const& name)
    : m_name (name)
    , item_ (this)
    , parent_ (nullptr)
{
}

PropertyStream::Source::~Source ()
{
    std::lock_guard<std::recursive_mutex> _(lock_);
    if (parent_ != nullptr)
        parent_->remove (*this);
    removeAll ();
}

std::string const& PropertyStream::Source::name () const
{
    return m_name;
}

void PropertyStream::Source::add (Source& source)
{
    std::lock(lock_, source.lock_);
    std::lock_guard<std::recursive_mutex> lk1(lock_, std::adopt_lock);
    std::lock_guard<std::recursive_mutex> lk2(source.lock_, std::adopt_lock);

    bassert (source.parent_ == nullptr);
    children_.push_back (source.item_);
    source.parent_ = this;
}

void PropertyStream::Source::remove (Source& child)
{
    std::lock(lock_, child.lock_);
    std::lock_guard<std::recursive_mutex> lk1(lock_, std::adopt_lock);
    std::lock_guard<std::recursive_mutex> lk2(child.lock_, std::adopt_lock);

    bassert (child.parent_ == this);
    children_.erase (
        children_.iterator_to (
            child.item_));
    child.parent_ = nullptr;
}

void PropertyStream::Source::removeAll ()
{
    std::lock_guard<std::recursive_mutex> _(lock_);
    for (auto iter = children_.begin(); iter != children_.end(); )
    {
        std::lock_guard<std::recursive_mutex> _cl((*iter)->lock_);
        remove (*(*iter));
    }
}

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

void PropertyStream::Source::write_one (PropertyStream& stream)
{
    Map map (m_name, stream);
    onWrite (map);
}

void PropertyStream::Source::write (PropertyStream& stream)
{
    Map map (m_name, stream);
    onWrite (map);

    std::lock_guard<std::recursive_mutex> _(lock_);

    for (auto& child : children_)
        child.source().write (stream);
}

void PropertyStream::Source::write (PropertyStream& stream, std::string const& path)
{
    std::pair <Source*, bool> result (find (path));

    if (result.first == nullptr)
        return;

    if (result.second)
        result.first->write (stream);
    else
        result.first->write_one (stream);
}

std::pair <PropertyStream::Source*, bool> PropertyStream::Source::find (std::string path)
{
    bool const deep (peel_trailing_slashstar (&path));
    bool const rooted (peel_leading_slash (&path));
    Source* source (this);
    if (! path.empty())
    {
        if (! rooted)
        {
            std::string const name (peel_name (&path));
            source = find_one_deep (name);
            if (source == nullptr)
                return std::make_pair (nullptr, deep);
        }
        source = source->find_path (path);
    }
    return std::make_pair (source, deep);
}

bool PropertyStream::Source::peel_leading_slash (std::string* path)
{
    if (! path->empty() && path->front() == '/')
    {
        *path = std::string (path->begin() + 1, path->end());
        return true;
    }
    return false;
}

bool PropertyStream::Source::peel_trailing_slashstar (std::string* path)
{
    bool found(false);
    if (path->empty())
        return false;
    if (path->back() == '*')
    {
        found = true;
        path->pop_back();
    }
    if(! path->empty() && path->back() == '/')
        path->pop_back();
    return found;
}

std::string PropertyStream::Source::peel_name (std::string* path)
{
    if (path->empty())
        return "";

    std::string::const_iterator first = (*path).begin();
    std::string::const_iterator last = (*path).end();
    std::string::const_iterator pos (std::find (first, last, '/'));
    std::string s (first, pos);

    if (pos != last)
        *path = std::string (pos+1, last);
    else
        *path = std::string ();

    return s;
}

// Recursive search through the whole tree until name is found
PropertyStream::Source* PropertyStream::Source::find_one_deep (std::string const& name)
{
    Source* found = find_one (name);
    if (found != nullptr)
        return found;

    std::lock_guard<std::recursive_mutex> _(lock_);
    for (auto& s : children_)
    {
        found = s.source().find_one_deep (name);
        if (found != nullptr)
            return found;
    }
    return nullptr;
}

PropertyStream::Source* PropertyStream::Source::find_path (std::string path)
{
    if (path.empty())
        return this;
    Source* source (this);
    do
    {
        std::string const name (peel_name (&path));
        if(name.empty ())
            break;
        source = source->find_one(name);
    }
    while (source != nullptr);
    return source;
}

// This function only looks at immediate children
// If no immediate children match, then return nullptr
PropertyStream::Source* PropertyStream::Source::find_one (std::string const& name)
{
    std::lock_guard<std::recursive_mutex> _(lock_);
    for (auto& s : children_)
    {
        if (s.source().m_name == name)
            return &s.source();
    }
    return nullptr;
}

void PropertyStream::Source::onWrite (Map&)
{
}

//------------------------------------------------------------------------------
//
// PropertyStream
//
//------------------------------------------------------------------------------

PropertyStream::PropertyStream ()
{
}

PropertyStream::~PropertyStream ()
{
}

void PropertyStream::add (std::string const& key, bool value)
{
    if (value)
        add (key, "true");
    else
        add (key, "false");
}

void PropertyStream::add (std::string const& key, char value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, signed char value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, unsigned char value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, wchar_t value)
{
    lexical_add (key, value);
}

#if 0
void PropertyStream::add (std::string const& key, char16_t value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, char32_t value)
{
    lexical_add (key, value);
}
#endif

void PropertyStream::add (std::string const& key, short value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, unsigned short value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, int value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, unsigned int value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, long value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, unsigned long value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, long long value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, unsigned long long value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, float value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, double value)
{
    lexical_add (key, value);
}

void PropertyStream::add (std::string const& key, long double value)
{
    lexical_add (key, value);
}

void PropertyStream::add (bool value)
{
    if (value)
        add ("true");
    else
        add ("false");
}

void PropertyStream::add (char value)
{
    lexical_add (value);
}

void PropertyStream::add (signed char value)
{
    lexical_add (value);
}

void PropertyStream::add (unsigned char value)
{
    lexical_add (value);
}

void PropertyStream::add (wchar_t value)
{
    lexical_add (value);
}

#if 0
void PropertyStream::add (char16_t value)
{
    lexical_add (value);
}

void PropertyStream::add (char32_t value)
{
    lexical_add (value);
}
#endif

void PropertyStream::add (short value)
{
    lexical_add (value);
}

void PropertyStream::add (unsigned short value)
{
    lexical_add (value);
}

void PropertyStream::add (int value)
{
    lexical_add (value);
}

void PropertyStream::add (unsigned int value)
{
    lexical_add (value);
}

void PropertyStream::add (long value)
{
    lexical_add (value);
}

void PropertyStream::add (unsigned long value)
{
    lexical_add (value);
}

void PropertyStream::add (long long value)
{
    lexical_add (value);
}

void PropertyStream::add (unsigned long long value)
{
    lexical_add (value);
}

void PropertyStream::add (float value)
{
    lexical_add (value);
}

void PropertyStream::add (double value)
{
    lexical_add (value);
}

void PropertyStream::add (long double value)
{
    lexical_add (value);
}

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

class PropertyStream_test : public unit_test::suite
{
public:
    using Source = PropertyStream::Source;

    void test_peel_name (std::string s, std::string const& expected,
        std::string const& expected_remainder)
    {
        try
        {
            std::string const peeled_name = Source::peel_name (&s);
            expect (peeled_name == expected);
            expect (s == expected_remainder);
        }
        catch (...)
        {
            fail ("unhandled exception");;
       }
    }

    void test_peel_leading_slash (std::string s, std::string const& expected,
        bool should_be_found)
    {
        try
        {
            bool const found (Source::peel_leading_slash (&s));
            expect (found == should_be_found);
            expect (s == expected);
        }
        catch(...)
        {
            fail ("unhandled exception");;
        }
    }

    void test_peel_trailing_slashstar (std::string s,
        std::string const& expected_remainder, bool should_be_found)
    {
        try
        {
            bool const found (Source::peel_trailing_slashstar (&s));
            expect (found == should_be_found);
            expect (s == expected_remainder);
        }
        catch (...)
        {
            fail ("unhandled exception");;
        }
    }

    void test_find_one (Source& root, Source* expected, std::string const& name)
    {
        try
        {
            Source* source (root.find_one (name));
            expect (source == expected);
        }
        catch (...)
        {
            fail ("unhandled exception");;
        }
    }

    void test_find_path (Source& root, std::string const& path,
        Source* expected)
    {
        try
        {
            Source* source (root.find_path (path));
            expect (source == expected);
        }
        catch (...)
        {
            fail ("unhandled exception");;
        }
    }

    void test_find_one_deep (Source& root, std::string const& name,
        Source* expected)
    {
        try
        {
            Source* source (root.find_one_deep (name));
            expect (source == expected);
        }
        catch(...)
        {
            fail ("unhandled exception");;
        }
    }

    void test_find (Source& root, std::string path, Source* expected,
        bool expected_star)
    {
        try
        {
            auto const result (root.find (path));
            expect (result.first == expected);
            expect (result.second == expected_star);
        }
        catch (...)
        {
            fail ("unhandled exception");;
        }
    }

    void run()
    {
        Source a ("a");
        Source b ("b");
        Source c ("c");
        Source d ("d");
        Source e ("e");
        Source f ("f");
        Source g ("g");

        //
        // a { b { d { f }, e }, c { g } }
        //

        a.add ( b );
        a.add ( c );
        c.add ( g );
        b.add ( d );
        b.add ( e );
        d.add ( f );

        testcase ("peel_name");
        test_peel_name ("a", "a", "");
        test_peel_name ("foo/bar", "foo", "bar");
        test_peel_name ("foo/goo/bar", "foo", "goo/bar");
        test_peel_name ("", "", "");

        testcase ("peel_leading_slash");
        test_peel_leading_slash ("foo/", "foo/", false);
        test_peel_leading_slash ("foo", "foo", false);
        test_peel_leading_slash ("/foo/", "foo/", true);
        test_peel_leading_slash ("/foo", "foo", true);

        testcase ("peel_trailing_slashstar");
        test_peel_trailing_slashstar ("/foo/goo/*", "/foo/goo", true);
        test_peel_trailing_slashstar ("foo/goo/*", "foo/goo", true);
        test_peel_trailing_slashstar ("/foo/goo/", "/foo/goo", false);
        test_peel_trailing_slashstar ("foo/goo", "foo/goo", false);
        test_peel_trailing_slashstar ("", "", false);
        test_peel_trailing_slashstar ("/", "", false);
        test_peel_trailing_slashstar ("/*", "", true);
        test_peel_trailing_slashstar ("//", "/", false);
        test_peel_trailing_slashstar ("**", "*", true);
        test_peel_trailing_slashstar ("*/", "*", false);

        testcase ("find_one");
        test_find_one (a, &b, "b");
        test_find_one (a, nullptr, "d");
        test_find_one (b, &e, "e");
        test_find_one (d, &f, "f");

        testcase ("find_path");
        test_find_path (a, "a", nullptr);
        test_find_path (a, "e", nullptr);
        test_find_path (a, "a/b", nullptr);
        test_find_path (a, "a/b/e", nullptr);
        test_find_path (a, "b/e/g", nullptr);
        test_find_path (a, "b/e/f", nullptr);
        test_find_path (a, "b", &b);
        test_find_path (a, "b/e", &e);
        test_find_path (a, "b/d/f", &f);

        testcase ("find_one_deep");
        test_find_one_deep (a, "z", nullptr);
        test_find_one_deep (a, "g", &g);
        test_find_one_deep (a, "b", &b);
        test_find_one_deep (a, "d", &d);
        test_find_one_deep (a, "f", &f);

        testcase ("find");
        test_find (a, "",     &a, false);
        test_find (a, "*",    &a, true);
        test_find (a, "/b",   &b, false);
        test_find (a, "b",    &b, false);
        test_find (a, "d",    &d, false);
        test_find (a, "/b*",  &b, true);
        test_find (a, "b*",   &b, true);
        test_find (a, "d*",   &d, true);
        test_find (a, "/b/*", &b, true);
        test_find (a, "b/*",  &b, true);
        test_find (a, "d/*",  &d, true);
        test_find (a, "a",    nullptr, false);
        test_find (a, "/d",   nullptr, false);
        test_find (a, "/d*",  nullptr, true);
        test_find (a, "/d/*", nullptr, true);
    }
};

BEAST_DEFINE_TESTSUITE(PropertyStream,utility,beast);

}