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 Impl
24 
25 class SBStructuredData::Impl {
26 public:
27   Impl() : m_plugin_wp(), m_data_sp() {}
28 
29   Impl(const Impl &rhs) = default;
30 
31   Impl(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   ~Impl() = default;
38 
39   Impl &operator=(const Impl &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::SBStream &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.ref());
57     return sb_error;
58   }
59 
60   lldb::SBError GetDescription(lldb::SBStream &stream) const {
61     SBError sb_error;
62 
63     if (!m_data_sp) {
64       sb_error.SetErrorString("Cannot pretty print structured data: "
65                               "no data to print.");
66       return sb_error;
67     }
68 
69     // Grab the plugin.
70     auto plugin_sp = StructuredDataPluginSP(m_plugin_wp);
71     if (!plugin_sp) {
72       sb_error.SetErrorString("Cannot pretty print structured data: "
73                               "plugin doesn't exist.");
74       return sb_error;
75     }
76 
77     // Get the data's description.
78     auto error = plugin_sp->GetDescription(m_data_sp, stream.ref());
79     if (!error.Success())
80       sb_error.SetError(error);
81 
82     return sb_error;
83   }
84 
85 private:
86   StructuredDataPluginWP m_plugin_wp;
87   StructuredData::ObjectSP m_data_sp;
88 };
89 
90 #pragma mark--
91 #pragma mark SBStructuredData
92 
93 SBStructuredData::SBStructuredData() : m_impl_up(new Impl()) {}
94 
95 SBStructuredData::SBStructuredData(const lldb::SBStructuredData &rhs)
96     : m_impl_up(new Impl(*rhs.m_impl_up.get())) {}
97 
98 SBStructuredData::SBStructuredData(const lldb::EventSP &event_sp)
99     : m_impl_up(new Impl(event_sp)) {}
100 
101 SBStructuredData::~SBStructuredData() {}
102 
103 SBStructuredData &SBStructuredData::
104 operator=(const lldb::SBStructuredData &rhs) {
105   *m_impl_up = *rhs.m_impl_up;
106   return *this;
107 }
108 
109 bool SBStructuredData::IsValid() const { return m_impl_up->IsValid(); }
110 
111 void SBStructuredData::Clear() { m_impl_up->Clear(); }
112 
113 SBError SBStructuredData::GetAsJSON(lldb::SBStream &stream) const {
114   return m_impl_up->GetAsJSON(stream);
115 }
116 
117 lldb::SBError SBStructuredData::GetDescription(lldb::SBStream &stream) const {
118   return m_impl_up->GetDescription(stream);
119 }
120