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 <stdexcept>
  21 #include <cassert>
  22 #include <string>
  23
  24 #include <curl/curl.h>
  25
  26 #include "Quvi.h"
  27
  28 #define _init_m \
  29  : _quvi(NULL), \
  30    _curl(NULL), \
  31    quvi_code(QUVI_OK), \
  32    resp_code(-1), \
  33    ok(1)
  34
  35 Query::Query()
  36 _init_m
  37 {
  38   _init();
  39 }
  40
  41 Query::Query(const Query& q)
  42 _init_m
  43 {
  44   _init();
  45 }
  46
  47 #undef _init_m
  48
  49 Query& Query::operator=(const Query& q)
  50 {
  51   if (this != &q)
  52     {
  53       _close ();
  54       _init ();
  55     }
  56   return *this;
  57 }
  58
  59 Query::~Query()
  60 {
  61   _close();
  62 }
  63
  64 void Query::_init()
  65 {
  66   quvi_code = quvi_init(&_quvi);
  67
  68   if (quvi_code != QUVI_OK)
  69     {
  70       const char *e = quvi_strerror(_quvi, static_cast<QUVIcode>(quvi_code));
  71       throw std::runtime_error(e);
  72     }
  73
  74   quvi_getinfo(_quvi, QUVIINFO_CURL, &_curl);
  75   assert(_curl != NULL);
  76
  77   set_opts(_opts);
  78 }
  79
  80 void Query::_close()
  81 {
  82   if (_quvi)
  83     quvi_close(&_quvi);
  84   assert(_quvi == NULL);
  85   _curl = NULL;
  86 }
  87
  88 void Query::set_opts(const Options& opts)
  89 {
  90   if ( !opts.format.empty() )
  91     quvi_setopt(_quvi, QUVIOPT_FORMAT, opts.format.c_str() );
  92
  93   quvi_setopt(_quvi, QUVIOPT_NOVERIFY,  !opts.verify ? 1L:0L);
  94   quvi_setopt(_quvi, QUVIOPT_NORESOLVE, !opts.resolve ? 1L:0L);
  95
  96   quvi_setopt(_quvi, QUVIOPT_CATEGORY, opts.category);
  97
  98   if ( !opts.user_agent.empty() )
  99     curl_easy_setopt(_curl, CURLOPT_USERAGENT, opts.user_agent.c_str());
 100
 101   if ( !opts.http_proxy.empty() )
 102     curl_easy_setopt(_curl, CURLOPT_PROXY, opts.http_proxy.c_str());
 103
 104   curl_easy_setopt(_curl, CURLOPT_VERBOSE, opts.verbose_libcurl ? 1L:0L);
 105 }
 106
 107 Media Query::parse(const std::string& url)
 108 {
 109   quvi_media_t m;
 110   quvi_code = quvi_parse(_quvi, const_cast<char*>(url.c_str()), &m);
 111
 112   quvi_getinfo(_quvi, QUVIINFO_HTTPCODE, &resp_code);
 113
 114   Media res;
 115
 116   if (quvi_code != QUVI_OK)
 117     _format_error();
 118   else
 119     {
 120       res = Media(m);
 121       ok = 1;
 122     }
 123
 124   return res;
 125 }
 126
 127 void Query::next_website(std::string& domain,
 128                          std::string& formats)
 129 {
 130   char *d=NULL, *f=NULL;
 131   quvi_code = quvi_next_supported_website(_quvi, &d, &f);
 132
 133   if (quvi_code == QUVI_OK)
 134     {
 135       domain = d;
 136       quvi_free(d);
 137       formats = f;
 138       quvi_free(f);
 139       ok = 1;
 140     }
 141   else
 142     ok = 0;
 143 }
 144
 145 void Query::_format_error()
 146 {
 147   const char *e = quvi_strerror(_quvi, static_cast<QUVIcode>(quvi_code));
 148   errmsg = std::string(e);
 149   ok = 0;
 150 }
 151
 152 int Query::supported(const std::string& url)
 153 {
 154   quvi_code = quvi_supported(_quvi, const_cast<char*>(url.c_str()));
 155
 156   if (quvi_code != QUVI_OK)
 157     _format_error();
 158
 159   return static_cast<int>(quvi_code);
 160 }
 161
 162 std::string Query::formats(const std::string& url)
 163 {
 164   char *s = NULL;
 165
 166   quvi_code =
 167     quvi_query_formats(_quvi, const_cast<char*>(url.c_str()), &s);
 168
 169   quvi_getinfo(_quvi, QUVIINFO_HTTPCODE, &resp_code);
 170
 171   std::string res;
 172
 173   if (quvi_code != QUVI_OK)
 174     _format_error();
 175   else
 176     {
 177       res = s;
 178       quvi_free(s);
 179       ok = 1;
 180     }
 181
 182   return res;
 183 }
 184
 185 // vim: set ts=2 sw=2 tw=72 expandtab: