Query.cxx

   1 /*
   2 * Copyright (C) 2010 Toni Gundogdu.
   3 *
   4 * This program is free software: you can redistribute it and/or modify
   5 * it under the terms of the GNU General Public License as published by
   6 * the Free Software Foundation, either version 3 of the License, or
   7 * (at your option) any later version.
   8 *
   9 * This program 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
  12 * GNU General Public License for more details.
  13 *
  14 * You should have received a copy of the GNU General Public License
  15 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <string>
  19 #include <stdexcept>
  20 #include <cassert>
  21
  22 #include <curl/curl.h>
  23
  24 #include "Quvi.h"
  25
  26 Query::Query ()
  27   :_quvi(NULL), _curl(NULL), quvi_code (QUVI_OK), resp_code (-1)
  28 {
  29   _init ();
  30 }
  31
  32 Query::Query (const Query& q)
  33   :_quvi(NULL), _curl(NULL), quvi_code (QUVI_OK), resp_code (-1)
  34 {
  35   _init ();
  36 }
  37
  38 Query&
  39 Query::operator=(const Query& q)
  40 {
  41   if (this != &q)
  42     {
  43       _close ();
  44       _init ();
  45     }
  46   return *this;
  47 }
  48
  49 Query::~Query ()
  50 {
  51   _close ();
  52 }
  53
  54 void
  55 Query::_init ()
  56 {
  57
  58   quvi_code = quvi_init (&_quvi);
  59
  60   if (quvi_code != QUVI_OK)
  61     {
  62       const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
  63       throw std::runtime_error (e);
  64     }
  65
  66   quvi_getinfo (_quvi, QUVIINFO_CURL, &_curl);
  67   assert (_curl != NULL);
  68 }
  69
  70 void
  71 Query::_close ()
  72 {
  73   if (_quvi) quvi_close (&_quvi);
  74   assert (_quvi == NULL);
  75   _curl = NULL;
  76 }
  77
  78 Video
  79 Query::parse (const std::string& url, const Options& opts)
  80 {
  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   if ( !opts.user_agent.empty() )
  89     curl_easy_setopt (_curl, CURLOPT_USERAGENT, opts.user_agent.c_str());
  90
  91   if ( !opts.http_proxy.empty() )
  92     curl_easy_setopt (_curl, CURLOPT_PROXY, opts.http_proxy.c_str () );
  93
  94   curl_easy_setopt (_curl, CURLOPT_VERBOSE, opts.verbose_libcurl ? 1L:0L);
  95
  96   quvi_video_t qv;
  97
  98   quvi_code = quvi_parse (_quvi, const_cast<char*>(url.c_str ()), &qv);
  99
 100   quvi_getinfo (_quvi, QUVIINFO_HTTPCODE, &resp_code);
 101
 102   if (quvi_code != QUVI_OK)
 103     {
 104
 105       const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
 106
 107       last_error = std::string (e);
 108
 109       return Video();
 110     }
 111
 112   return Video (qv);
 113 }
 114
 115 int
 116 Query::next_website (std::string& domain, std::string& formats)
 117 {
 118   char *dom=NULL, *fmt=NULL;
 119   const QUVIcode rc = quvi_next_supported_website (_quvi, &dom, &fmt);
 120   if (rc == QUVI_OK)
 121     {
 122       domain  = dom;
 123       formats = fmt;
 124       quvi_free (dom);
 125       quvi_free (fmt);
 126     }
 127   return static_cast<int>(rc);
 128 }
 129
 130