contrib/llvm/tools/lldb/source/API/SBStructuredData.cpp

   1 //===-- SBStructuredData.cpp ------------------------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "lldb/API/SBStructuredData.h"
  11
  12 #include "lldb/API/SBStream.h"
  13 #include "lldb/Core/Error.h"
  14 #include "lldb/Core/Event.h"
  15 #include "lldb/Core/Stream.h"
  16 #include "lldb/Core/StructuredData.h"
  17 #include "lldb/Target/StructuredDataPlugin.h"
  18
  19 using namespace lldb;
  20 using namespace lldb_private;
  21
  22 #pragma mark--
  23 #pragma mark StructuredDataImpl
  24
  25 class StructuredDataImpl {
  26 public:
  27   StructuredDataImpl() : m_plugin_wp(), m_data_sp() {}
  28
  29   StructuredDataImpl(const StructuredDataImpl &rhs) = default;
  30
  31   StructuredDataImpl(const EventSP &event_sp)
  32       : m_plugin_wp(
  33             EventDataStructuredData::GetPluginFromEvent(event_sp.get())),
  34         m_data_sp(EventDataStructuredData::GetObjectFromEvent(event_sp.get())) {
  35   }
  36
  37   ~StructuredDataImpl() = default;
  38
  39   StructuredDataImpl &operator=(const StructuredDataImpl &rhs) = default;
  40
  41   bool IsValid() const { return m_data_sp.get() != nullptr; }
  42
  43   void Clear() {
  44     m_plugin_wp.reset();
  45     m_data_sp.reset();
  46   }
  47
  48   SBError GetAsJSON(lldb_private::Stream &stream) const {
  49     SBError sb_error;
  50
  51     if (!m_data_sp) {
  52       sb_error.SetErrorString("No structured data.");
  53       return sb_error;
  54     }
  55
  56     m_data_sp->Dump(stream);
  57     return sb_error;
  58   }
  59
  60   Error GetDescription(lldb_private::Stream &stream) const {
  61     Error error;
  62
  63     if (!m_data_sp) {
  64       error.SetErrorString("Cannot pretty print structured data: "
  65                            "no data to print.");
  66       return error;
  67     }
  68
  69     // Grab the plugin.
  70     auto plugin_sp = StructuredDataPluginSP(m_plugin_wp);
  71     if (!plugin_sp) {
  72       error.SetErrorString("Cannot pretty print structured data: "
  73                            "plugin doesn't exist.");
  74       return error;
  75     }
  76
  77     // Get the data's description.
  78     return plugin_sp->GetDescription(m_data_sp, stream);
  79   }
  80
  81 private:
  82   StructuredDataPluginWP m_plugin_wp;
  83   StructuredData::ObjectSP m_data_sp;
  84 };
  85
  86 #pragma mark--
  87 #pragma mark SBStructuredData
  88
  89 SBStructuredData::SBStructuredData() : m_impl_up(new StructuredDataImpl()) {}
  90
  91 SBStructuredData::SBStructuredData(const lldb::SBStructuredData &rhs)
  92     : m_impl_up(new StructuredDataImpl(*rhs.m_impl_up.get())) {}
  93
  94 SBStructuredData::SBStructuredData(const lldb::EventSP &event_sp)
  95     : m_impl_up(new StructuredDataImpl(event_sp)) {}
  96
  97 SBStructuredData::~SBStructuredData() {}
  98
  99 SBStructuredData &SBStructuredData::
 100 operator=(const lldb::SBStructuredData &rhs) {
 101   *m_impl_up = *rhs.m_impl_up;
 102   return *this;
 103 }
 104
 105 bool SBStructuredData::IsValid() const { return m_impl_up->IsValid(); }
 106
 107 void SBStructuredData::Clear() { m_impl_up->Clear(); }
 108
 109 SBError SBStructuredData::GetAsJSON(lldb::SBStream &stream) const {
 110   return m_impl_up->GetAsJSON(stream.ref());
 111 }
 112
 113 lldb::SBError SBStructuredData::GetDescription(lldb::SBStream &stream) const {
 114   Error error = m_impl_up->GetDescription(stream.ref());
 115   SBError sb_error;
 116   sb_error.SetError(error);
 117   return sb_error;
 118 }