diff --git a/src/cocoatweet/api/api.cc b/src/cocoatweet/api/api.cc index 6d5abd7..5249a36 100644 --- a/src/cocoatweet/api/api.cc +++ b/src/cocoatweet/api/api.cc @@ -6,6 +6,7 @@ API::API(CocoaTweet::OAuth::Key _key) { status_ = Statuses::Status(oauth_); favorite_ = Favorites::Favorite(oauth_); media_ = Medias::Media(oauth_); + directMessage_ = DirectMessages::DirectMessage(oauth_); } Statuses::Status API::status() const { @@ -19,4 +20,8 @@ Favorites::Favorite API::favorite() const { Medias::Media API::media() const { return media_; } + +DirectMessages::DirectMessage API::directMessage() const{ + return directMessage_; +} } // namespace CocoaTweet::API diff --git a/src/cocoatweet/api/api.h b/src/cocoatweet/api/api.h index 21ba046..0366517 100644 --- a/src/cocoatweet/api/api.h +++ b/src/cocoatweet/api/api.h @@ -4,6 +4,7 @@ #include #include #include +#include #include namespace CocoaTweet::API { @@ -24,10 +25,13 @@ public: Medias::Media media() const; + DirectMessages::DirectMessage directMessage() const; + private: Statuses::Status status_; Favorites::Favorite favorite_; Medias::Media media_; + DirectMessages::DirectMessage directMessage_; std::shared_ptr oauth_; }; } // namespace CocoaTweet::API