Query.cxx

   1 /* WWW::Quvi
   2  * Copyright (C) 2010,2011  Toni Gundogdu <legatvs@gmail.com>
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  17  * 02110-1301  USA
  18  */
  19
  20 #include <string>
  21 #include <stdexcept>
  22 #include <cassert>
  23
  24 #include <curl/curl.h>
  25
  26 #include "Quvi.h"
  27
  28 Query::Query ()
  29   :_quvi(NULL), _curl(NULL), quvi_code (QUVI_OK), resp_code (-1)
  30 {
  31   _init ();
  32 }
  33
  34 Query::Query (const Query& q)
  35   :_quvi(NULL), _curl(NULL), quvi_code (QUVI_OK), resp_code (-1)
  36 {
  37   _init ();
  38 }
  39
  40 Query&
  41 Query::operator=(const Query& q)
  42 {
  43   if (this != &q)
  44     {
  45       _close ();
  46       _init ();
  47     }
  48   return *this;
  49 }
  50
  51 Query::~Query ()
  52 {
  53   _close ();
  54 }
  55
  56 void
  57 Query::_init ()
  58 {
  59   quvi_code = quvi_init (&_quvi);
  60
  61   if (quvi_code != QUVI_OK)
  62     {
  63       const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
  64       throw std::runtime_error (e);
  65     }
  66
  67   quvi_getinfo (_quvi, QUVIINFO_CURL, &_curl);
  68   assert (_curl != NULL);
  69 }
  70
  71 void
  72 Query::_close ()
  73 {
  74   if (_quvi) quvi_close (&_quvi);
  75   assert (_quvi == NULL);
  76   _curl = NULL;
  77 }
  78
  79 Video
  80 Query::parse (const std::string& url, const Options& opts)
  81 {
  82   if ( !opts.format.empty () )
  83     quvi_setopt (_quvi, QUVIOPT_FORMAT, opts.format.c_str () );
  84
  85   quvi_setopt (_quvi, QUVIOPT_NOVERIFY,    !opts.verify ? 1L:0L);
  86   quvi_setopt (_quvi, QUVIOPT_NOSHORTENED, !opts.shortened ? 1L:0L);
  87
  88   quvi_setopt (_quvi, QUVIOPT_CATEGORY, opts.category);
  89
  90   if ( !opts.user_agent.empty() )
  91     curl_easy_setopt (_curl, CURLOPT_USERAGENT, opts.user_agent.c_str());
  92
  93   if ( !opts.http_proxy.empty() )
  94     curl_easy_setopt (_curl, CURLOPT_PROXY, opts.http_proxy.c_str () );
  95
  96   curl_easy_setopt (_curl, CURLOPT_VERBOSE, opts.verbose_libcurl ? 1L:0L);
  97
  98   quvi_video_t qv;
  99
 100   quvi_code = quvi_parse (_quvi, const_cast<char*>(url.c_str ()), &qv);
 101
 102   quvi_getinfo (_quvi, QUVIINFO_HTTPCODE, &resp_code);
 103
 104   if (quvi_code != QUVI_OK)
 105     {
 106       const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
 107
 108       last_error = std::string (e);
 109
 110       return Video();
 111     }
 112
 113   return Video (qv);
 114 }
 115
 116 int
 117 Query::next_website (std::string& domain, std::string& formats)
 118 {
 119   char *dom=NULL, *fmt=NULL;
 120   const QUVIcode rc = quvi_next_supported_website (_quvi, &dom, &fmt);
 121   if (rc == QUVI_OK)
 122     {
 123       domain  = dom;
 124       formats = fmt;
 125       quvi_free (dom);
 126       quvi_free (fmt);
 127     }
 128   return static_cast<int>(rc);
 129 }
 130
 131 // vim: set ts=2 sw=2 tw=72 expandtab: