//===-- SBStructuredData.cpp ------------------------------------*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#include "lldb/API/SBStructuredData.h"

#include "lldb/API/SBStream.h"
#include "lldb/Core/Error.h"
#include "lldb/Core/Event.h"
#include "lldb/Core/Stream.h"
#include "lldb/Core/StructuredData.h"
#include "lldb/Target/StructuredDataPlugin.h"

using namespace lldb;
using namespace lldb_private;

#pragma mark --
#pragma mark Impl

class SBStructuredData::Impl
{
public:

    Impl() :
        m_plugin_wp(),
        m_data_sp()
    {
    }

    Impl(const Impl &rhs) = default;

    Impl(const EventSP &event_sp) :
       m_plugin_wp(EventDataStructuredData::GetPluginFromEvent(event_sp.get())),
        m_data_sp(EventDataStructuredData::GetObjectFromEvent(event_sp.get()))
    {
    }

    ~Impl() = default;

    Impl&
    operator =(const Impl &rhs) = default;

    bool
    IsValid() const
    {
        return m_data_sp.get() != nullptr;
    }

    void
    Clear()
    {
        m_plugin_wp.reset();
        m_data_sp.reset();
    }

    SBError
    GetAsJSON(lldb::SBStream &stream) const
    {
        SBError sb_error;

        if (!m_data_sp)
        {
            sb_error.SetErrorString("No structured data.");
            return sb_error;
        }

        m_data_sp->Dump(stream.ref());
        return sb_error;
    }

    lldb::SBError
    GetDescription(lldb::SBStream &stream) const
    {
        SBError sb_error;

        if (!m_data_sp)
        {
            sb_error.SetErrorString("Cannot pretty print structured data: "
                                    "no data to print.");
            return sb_error;
        }

        // Grab the plugin.
        auto plugin_sp = StructuredDataPluginSP(m_plugin_wp);
        if (!plugin_sp)
        {
            sb_error.SetErrorString("Cannot pretty print structured data: "
                                    "plugin doesn't exist.");
            return sb_error;
        }

        // Get the data's description.
        auto error = plugin_sp->GetDescription(m_data_sp, stream.ref());
        if (!error.Success())
            sb_error.SetError(error);

        return sb_error;
    }

private:

    StructuredDataPluginWP    m_plugin_wp;
    StructuredData::ObjectSP  m_data_sp;

};

#pragma mark --
#pragma mark SBStructuredData


SBStructuredData::SBStructuredData() :
    m_impl_up(new Impl())
{
}

SBStructuredData::SBStructuredData(const lldb::SBStructuredData &rhs) :
    m_impl_up(new Impl(*rhs.m_impl_up.get()))
{
}

SBStructuredData::SBStructuredData(const lldb::EventSP &event_sp) :
    m_impl_up(new Impl(event_sp))
{
}

SBStructuredData::~SBStructuredData()
{
}

SBStructuredData &
SBStructuredData::operator =(const lldb::SBStructuredData &rhs)
{
    *m_impl_up = *rhs.m_impl_up;
    return *this;
}

bool
SBStructuredData::IsValid() const
{
    return m_impl_up->IsValid();
}

void
SBStructuredData::Clear()
{
    m_impl_up->Clear();
}

SBError
SBStructuredData::GetAsJSON(lldb::SBStream &stream) const
{
    return m_impl_up->GetAsJSON(stream);
}

lldb::SBError
SBStructuredData::GetDescription(lldb::SBStream &stream) const
{
    return m_impl_up->GetDescription(stream);
}