Add support for body(...) directives and wrapper.

mikhail_beris · mikhail_beris · commit 0c1b5085028b · 2007-05-20T06:49:47.000Z
diff --git a/boost/network/.message.hpp.swp b/boost/network/.message.hpp.swp
diff --git a/boost/network/message.hpp b/boost/network/message.hpp
@@ -46,12 +46,17 @@ namespace boost { namespace network {
             return _headers;
         };
 
+        std::string & body() const {
+            return _body;
+        };
+
         private:
         
         friend struct detail::directive_base<tag> ;
         friend struct detail::wrapper_base<tag> ;
 
         mutable headers_container_type _headers;
+        mutable std::string _body;
     };
 
     typedef basic_message<> message; // default message type
diff --git a/boost/network/message/directives.hpp b/boost/network/message/directives.hpp
@@ -11,6 +11,7 @@
  */
 
 #include <boost/network/message/directives/header.hpp>
+#include <boost/network/message/directives/body.hpp>
 
 namespace boost { namespace network {
 
diff --git a/boost/network/message/directives/body.hpp b/boost/network/message/directives/body.hpp
@@ -0,0 +1,52 @@
+
+//          Copyright Dean Michael Berris 2007.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+
+#ifndef __NETWORK_MESSAGE_DIRECTIVES_BODY_HPP__
+#define __NETWORK_MESSAGE_DIRECTIVES_BODY_HPP__
+
+/** body.hpp
+ *
+ * Defines the types involved and the semantics of adding
+ * body contents into message objects.
+ *
+ * WARNING: DO NOT INCLUDE THIS HEADER DIRECTLY. THIS REQUIRES
+ * TYPES TO BE DEFINED FROM EARLIER FILES THAT INCLUDE THIS
+ * HEADER.
+ */
+namespace boost { namespace network {
+
+    namespace impl {
+        template <class Tag>
+            struct body_directive : public detail::directive_base<Tag> {
+                typedef Tag tag;
+
+                explicit body_directive(
+                        std::string const & body
+                        ) :
+                    _body(body)
+                { };
+
+                void operator() (basic_message<tag> & msg) const {
+                    msg.body() = _body;
+                };
+
+                private:
+
+                std::string _body;
+            };
+    }; // namespace impl
+
+    inline impl::body_directive<tags::default_>
+        body(std::string const & body_) {
+            return impl::body_directive<tags::default_>(body_);
+        };
+
+}; // namespace network
+
+}; // namespace boost
+
+#endif // __NETWORK_MESSAGE_DIRECTIVES_BODY_HPP__
+
diff --git a/boost/network/message/wrappers.hpp b/boost/network/message/wrappers.hpp
@@ -12,5 +12,6 @@
  * Pulls in all the wrapper header files.
  */
 #include <boost/network/message/wrappers/headers.hpp>
+#include <boost/network/message/wrappers/body.hpp>
 
 #endif // __NETWORK_MESSAGE_WRAPPERS_HPP__
diff --git a/boost/network/message/wrappers/body.hpp b/boost/network/message/wrappers/body.hpp
@@ -0,0 +1,39 @@
+
+//          Copyright Dean Michael Berris 2007.
+// Distributed under the Boost Software License, Version 1.0.
+//    (See accompanying file LICENSE_1_0.txt or copy at
+//          http://www.boost.org/LICENSE_1_0.txt)
+
+#ifndef __NETWORK_MESSAGE_WRAPPERS_BODY_HPP__
+#define __NETWORK_MESSAGE_WRAPPERS_BODY_HPP__
+
+namespace boost { namespace network {
+
+    namespace impl {
+        template <class Tag>
+            struct body_wrapper : public detail::wrapper_base<Tag> {
+                typedef Tag tag;
+                typedef basic_message<tag> message_type;
+                
+                explicit body_wrapper(basic_message<tag> & message_)
+                    : detail::wrapper_base<tag>(message_)
+                { };
+
+                operator std::string () const {
+                    return std::string(_message.body());
+                };
+            };
+    }; // namespace impl
+
+    template <class Tag>
+        inline std::string
+        body(basic_message<Tag> & message_) {
+            return impl::body_wrapper<Tag>(message_);
+        };
+
+}; // namespace network
+
+}; // namespace boost
+
+#endif // __NETWORK_MESSAGE_WRAPPERS_BODY_HPP__
+
diff --git a/libs/network/test/message_test.cpp b/libs/network/test/message_test.cpp
@@ -30,3 +30,11 @@ BOOST_AUTO_TEST_CASE(header_directives_test) {
     headers_range<message>::type range = headers(msg)["SOME_HEADER"];
     BOOST_CHECK( range.first != range.second );
 }
+
+BOOST_AUTO_TEST_CASE(body_directive_test) {
+    using namespace boost::network;
+    message msg;
+    msg << body("The quick brown fox jumps over the lazy dog.") ;
+
+    BOOST_CHECK_EQUAL( body(msg), "The quick brown fox jumps over the lazy dog.");
+}
