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/Event.h"
14 #include "lldb/Core/StructuredData.h"
15 #include "lldb/Target/StructuredDataPlugin.h"
16 #include "lldb/Utility/Error.h"
17 #include "lldb/Utility/Stream.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 }
119