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     { _init (); }
  29
  30 Query::Query (const Query& q)
  31     :_quvi(NULL), _curl(NULL), quvi_code (QUVI_OK), resp_code (-1)
  32     { _init (); }
  33
  34 Query&
  35 Query::operator=(const Query& q) {
  36     if (this != &q) {
  37         _close ();
  38         _init ();
  39     }
  40     return *this;
  41 }
  42
  43 Query::~Query () { _close (); }
  44
  45 void
  46 Query::_init () {
  47
  48     quvi_code = quvi_init (&_quvi);
  49
  50     if (quvi_code != QUVI_OK) {
  51         const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
  52         throw std::runtime_error (e);
  53     }
  54
  55     quvi_getinfo (_quvi, QUVIINFO_CURL, &_curl);
  56     assert (_curl != NULL);
  57 }
  58
  59 void
  60 Query::_close () {
  61     if (_quvi) quvi_close (&_quvi);
  62     assert (_quvi == NULL);
  63     _curl = NULL;
  64 }
  65
  66 Video
  67 Query::parse (const std::string& url, const Options& opts) {
  68
  69     if ( !opts.format.empty () )
  70         quvi_setopt (_quvi, QUVIOPT_FORMAT, opts.format.c_str () );
  71
  72     quvi_setopt (_quvi, QUVIOPT_NOVERIFY,   !opts.verify ? 1L:0L);
  73
  74     if ( !opts.user_agent.empty() )
  75         curl_easy_setopt (_curl, CURLOPT_USERAGENT, opts.user_agent.c_str());
  76
  77     if ( !opts.http_proxy.empty() )
  78         curl_easy_setopt (_curl, CURLOPT_PROXY, opts.http_proxy.c_str () );
  79
  80     curl_easy_setopt (_curl, CURLOPT_VERBOSE, opts.verbose_libcurl ? 1L:0L);
  81
  82     quvi_video_t qv;
  83
  84     quvi_code = quvi_parse (_quvi, const_cast<char*>(url.c_str ()), &qv);
  85
  86     quvi_getinfo (_quvi, QUVIINFO_HTTPCODE, &resp_code);
  87
  88     if (quvi_code != QUVI_OK) {
  89
  90         const char *e = quvi_strerror (_quvi, static_cast<QUVIcode>(quvi_code));
  91
  92         last_error = std::string (e);
  93
  94         return Video();
  95     }
  96
  97     return Video (qv);
  98 }
  99
 100 int
 101 Query::next_website (std::string& domain, std::string& formats) {
 102     char *dom=NULL, *fmt=NULL;
 103     const QUVIcode rc = quvi_next_supported_website (_quvi, &dom, &fmt);
 104     if (rc == QUVI_OK) {
 105         domain  = dom;
 106         formats = fmt;
 107         quvi_free (dom);
 108         quvi_free (fmt);
 109     }
 110     return static_cast<int>(rc);
 111 }
 112
 113