added a helper to prepare curl request based on http_request_t

* added curl_slist to http_request_t structure
* added http_request_t::setup_curl(CURL*)

Signed-off-by: Michael Krelin <hacker@klever.net>

2 files changed, 33 insertions, 0 deletions

diff --git a/include/opkele/oauth/consumer.h b/include/opkele/oauth/consumer.h
index 1e2784c..9196297 100644
--- a/include/opkele/oauth/consumer.h
+++ b/include/opkele/oauth/consumer.h
@@ -1,124 +1,129 @@
 #ifndef __OPKELE_OAUTH_CONSUMER_H
 #define __OPKELE_OAUTH_CONSUMER_H
 
 #include <string>
 #include <opkele/types.h>
 #include <opkele/oauth.h>
+#include <opkele/curl.h>
 
 namespace opkele {
     namespace oauth {
         using std::string;
 
         enum oauth_method_t {
             oauth_auth_header, oauth_post_body, oauth_url_query
         };
 
         struct service_endpoint_t {
             string url;
             string signature_method;
             oauth_method_t oauth_method;
 
             service_endpoint_t(const string& u,const string& sm,oauth_method_t om)
                 : url(u), signature_method(sm), oauth_method(om) { }
         };
 
         class basic_provider_endpoints {
             public:
 
                 virtual ~basic_provider_endpoints() { }
                 
                 virtual const service_endpoint_t& get_request_token_endpoint() const = 0;
                 virtual const service_endpoint_t& get_authorize_user_endpoint() const = 0;
                 virtual const service_endpoint_t& get_access_token_endpoint() const = 0;
 
                 virtual service_endpoint_t& get_url_endpoint(service_endpoint_t& sep,
                         const string& url) const = 0;
         };
 
         struct http_request_t {
             string authorize_header;
             string method;
             string url;
             string body;
 
+            util::curl_slist_t _curl_headers_list;
+
             http_request_t(const string& m,const string& u)
                 : method(m), url(u) { }
+
+            void setup_curl(CURL *curl);
         };
 
         class basic_consumer {
             public:
                 token_t consumer_token;
 
                 basic_consumer(const token_t& ct)
                     : consumer_token(ct) { }
                 virtual ~basic_consumer() { }
 
                 virtual const basic_provider_endpoints& get_endpoints() const = 0;
                 virtual const string allocate_nonce(time_t ts) = 0;
 
                 token_t get_request_token();
                 const string get_authorize_url(const token_t& rt,const string& callback="");
                 token_t get_access_token(const token_t& rt);
 
                 void prepare_request(
                         http_request_t& req,
                         const basic_fields& qf,const basic_fields& pf,
                         oauth_method_t om,const string& sm,
                         const token_t *t=0,const string& realm="");
                 void prepare_request(
                         http_request_t& req,
                         const basic_fields& qf,const basic_fields& pf,
                         const service_endpoint_t& sep,
                         const token_t *t=0,const string& realm="");
 
                 const string signature(
                         const string& method,
                         const string& url,
                         const basic_fields& fields,
                         const token_t* rt=0);
 
                 token_t acquire_token(
                         const service_endpoint_t& sep,
                         const token_t* rt=0);
         };
 
         class simple_provider_endpoints : public basic_provider_endpoints {
             public:
                 service_endpoint_t sep_request_token;
                 service_endpoint_t sep_authorize_user;
                 service_endpoint_t sep_access_token;
                 service_endpoint_t sep_generic;
 
                 simple_provider_endpoints(
                         const string& rt,const string& au,const string& at,
                         const string& sm,
                         oauth_method_t ams=oauth_post_body,
                         oauth_method_t amr=oauth_auth_header )
                     : sep_request_token(rt,sm,ams),
                     sep_authorize_user(au,sm,oauth_url_query),
                     sep_access_token(at,sm,ams),
                     sep_generic("",sm,amr) { }
 
                 const service_endpoint_t& get_request_token_endpoint() const;
                 const service_endpoint_t& get_authorize_user_endpoint() const;
                 const service_endpoint_t& get_access_token_endpoint() const;
                 service_endpoint_t& get_url_endpoint(service_endpoint_t& sep,
                         const string& url) const;
         };
 
         class simple_consumer : public basic_consumer {
             public:
                 simple_provider_endpoints peps;
 
                 simple_consumer(const simple_provider_endpoints& eps,
                         const token_t& ct)
                     : basic_consumer(ct), peps(eps) { }
 
                 const basic_provider_endpoints& get_endpoints() const;
                 const string allocate_nonce(time_t ts);
         };
 
     }
 }
 
 #endif /* __OPKELE_OAUTH_CONSUMER_H */
diff --git a/lib/oauth-consumer.cc b/lib/oauth-consumer.cc
index d717ed3..0c4c9e3 100644
--- a/lib/oauth-consumer.cc
+++ b/lib/oauth-consumer.cc
@@ -129,112 +129,140 @@ namespace opkele {
             if( (r=curl.easy_setopt(CURLOPT_URL,hr.url.c_str())) )
                 throw exception_curl(OPKELE_CP_ "failed to set curly urlie",r);
             if( (r=curl.easy_perform()) )
                 throw exception_curl(OPKELE_CP_ "failed to perform curly request",r);
             token_t rv;
             string::size_type p=0;
             while(p!=string::npos) {
                 string::size_type np = curl.response.find('&',p);
                 string part;
                 if(np==string::npos) {
                     part.assign(curl.response.c_str()+p); p = string::npos;
                 }else{
                     part.assign(curl.response,p,np-p); p = np+1;
                 }
                 string::size_type eq = part.find('=');
                 if(eq==string::npos) continue;
                 string n(part,0,eq);
                 if(n=="oauth_token") {
                     if(!rv.key.empty()) /* TODO: specialize */
                         throw opkele::exception(OPKELE_CP_ "found oauth_token twice");
                     rv.key = util::url_decode(part.substr(eq+1));
                 }else if(n=="oauth_token_secret") {
                     if(!rv.secret.empty()) /* TODO: specialize */
                         throw opkele::exception(OPKELE_CP_ "found oauth_secret twice");
                     rv.secret = util::url_decode(part.substr(eq+1));
                 }
             }
             return rv;
         }
 
         void basic_consumer::prepare_request(
                 http_request_t& req,
                 const basic_fields& qf,const basic_fields& pf,
                 oauth_method_t om,const string& sm,
                 const token_t *t,const string& realm) {
             fields_t op;
             op.set_field("oauth_consumer_key",consumer_token.key);
             if(t) op.set_field("oauth_token",t->key);
             op.set_field("oauth_signature_method",sm);
             time_t now;
             op.set_field("oauth_timestamp",
                     util::long_to_string(time(&now)));
             op.set_field("oauth_nonce",allocate_nonce(now));
             op.set_field("oauth_version","1.0");
             /* TODO: normalize and strip down url */
             {
                 fields_t af; /* TODO: optimize, I don't want it to be copied */
                 qf.copy_to(af); pf.append_to(af); op.append_to(af);
                 op.set_field("oauth_signature", signature(
                             req.method,req.url,af,t) );
             }
             req.authorize_header.clear();
             if(om==oauth_auth_header) {
                 req.authorize_header = "OAuth ";
                 req.authorize_header += "realm=\"";
                 req.authorize_header += util::url_encode(realm);
                 req.authorize_header += '\"';
                 for(basic_fields::fields_iterator
                         i=op.fields_begin(),ie=op.fields_end();
                         i!=ie;++i) {
                     req.authorize_header += ", ";
                     req.authorize_header += *i;
                     req.authorize_header += "=\"";
                     req.authorize_header += util::url_encode(op.get_field(*i));
                     req.authorize_header += "\"";
                 }
                 req.url = qf.append_query(req.url);
                 req.body = pf.query_string();
             }else if(om==oauth_post_body) {
                 assert(req.method=="POST");
                 /* TODO: optimize, don't copy it over and over */
                 fields_t p;
                 pf.append_to(p); op.append_to(p);
                 req.url = qf.append_query(req.url);
                 req.body = p.query_string();
             }else if(om==oauth_url_query) {
                 fields_t q;
                 qf.append_to(q); op.append_to(q);
                 req.url = q.append_query(req.url);
                 req.body = pf.query_string();
             }else
                 throw opkele::exception(OPKELE_CP_ /* TODO: specialize */
                         "Unknown oauth method");
         }
 
         void basic_consumer::prepare_request(
                 http_request_t& req,
                 const basic_fields& qf,const basic_fields& pf,
                 const service_endpoint_t& sep,
                 const token_t *t,const string& realm) {
             prepare_request(
                     req, qf, pf,
                     sep.oauth_method,sep.signature_method,
                     t,realm);
         }
 
+        void http_request_t::setup_curl(CURL *curl) {
+            CURLcode r;
+            r = curl_easy_setopt(curl,CURLOPT_URL,url.c_str());
+            if(r)
+                throw exception_curl(OPKELE_CP_ "failed to set curly urlie",r);
+            if(method=="POST") {
+                (r = curl_easy_setopt(curl,CURLOPT_POST,1))
+                    || (r = curl_easy_setopt(curl,CURLOPT_POSTFIELDS,body.c_str()))
+                    || (r = curl_easy_setopt(curl,CURLOPT_POSTFIELDSIZE,body.size()));
+            }else if(method=="GET") {
+                r = curl_easy_setopt(curl,CURLOPT_HTTPGET,1);
+            }else if(method=="HEAD") {
+                r = curl_easy_setopt(curl,CURLOPT_NOBODY,1);
+            }else /* TODO: specialize exception */
+                throw exception(OPKELE_CP_ "don't know how to handle http method");
+            if(r)
+                throw exception_curl(OPKELE_CP_ "failed to set curly options",r);
+            if(!authorize_header.empty()) {
+                r = curl_easy_setopt(curl,CURLOPT_HTTPHEADER,(curl_slist*)(
+                            _curl_headers_list = curl_slist_append(
+                                0,string("Authorization: "+authorize_header).c_str()
+                                )
+                            ) );
+                if(r)
+                    throw exception_curl(OPKELE_CP_ "failed to setup curlie header");
+            }
+        }
+
 
         const basic_provider_endpoints& simple_consumer::get_endpoints() const {
             return peps; }
 
         const string simple_consumer::allocate_nonce(time_t ts) {
 #           ifndef HAVE_LIBUUID
                 throw opkele::not_implemented(OPKELE_CP_
                         "not implemented consumer's allocate_nonce()");
 #           else /* HAVE_LIBUUID */
                 uuid_t uuid; uuid_generate(uuid);
                 return util::encode_base64(uuid,sizeof(uuid));
 #           endif /* HAVE_LIBUUID */
         }
 
     }
 }