src/net/fetch.c

   1 /* libquvi
   2  * Copyright (C) 2012  Toni Gundogdu <legatvs@gmail.com>
   3  *
   4  * This file is part of libquvi <http://quvi.sourceforge.net/>.
   5  *
   6  * This library is free software: you can redistribute it and/or
   7  * modify it under the terms of the GNU Affero General Public
   8  * License as published by the Free Software Foundation, either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General
  17  * Public License along with this library.  If not, see
  18  * <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include "config.h"
  22
  23 #include <lauxlib.h>
  24 #include <string.h>
  25 #include <glib.h>
  26
  27 #include "quvi.h"
  28 /* -- */
  29 #include "_quvi_s.h"
  30 #include "_quvi_net_s.h"
  31 #include "_quvi_macro.h"
  32 /* -- */
  33 #include "lua/quvi/opts.h"
  34 #include "net/def.h"
  35 #include "net/handle.h"
  36
  37 extern gchar* to_utf8(const gchar*, const gchar*);
  38 extern QuviError c_fetch(_quvi_net_t);
  39
  40 void n_fetch(_quvi_t q, _quvi_net_t *n, const gchar *url, GSList *opts)
  41 {
  42   *n = n_new(q, url);
  43
  44   if (q->cb.status != NULL)
  45     {
  46       const glong p = q_makelong(QUVI_CALLBACK_STATUS_FETCH, 0);
  47
  48       if (q->cb.status(p, (gpointer) url) != QUVI_OK)
  49         {
  50           q->status.rc = QUVI_ERROR_CALLBACK_ABORTED;
  51           return;
  52         }
  53     }
  54
  55   if (q->cb.fetch != NULL)
  56     q->status.rc = q->cb.fetch(*n);
  57   else /* Fetch using cURL (default). */
  58     q->status.rc = c_fetch(*n);
  59
  60   if (quvi_ok(q) == QUVI_TRUE && (*n)->status.resp_code == 200)
  61     {
  62       /* To UTF8. */
  63
  64       if (opts != NULL)
  65         {
  66           GSList *from;
  67
  68           if (l_quvi_object_opts_is_set(opts,
  69                                         QUVI_OBJECT_OPTION_FETCH_FROM_CHARSET,
  70                                         &from) == TRUE)
  71             {
  72               l_quvi_object_opt_t o;
  73               gchar *c;
  74
  75               o = (l_quvi_object_opt_t) from->data;
  76               c = to_utf8((*n)->fetch.content->str, o->value.str);
  77
  78               if (c != NULL)
  79                 {
  80                   g_string_assign((*n)->fetch.content, c);
  81                   g_free(c);
  82                 }
  83             }
  84         }
  85
  86       /* Update status. */
  87       if (q->cb.status != NULL)
  88         {
  89           const glong p = q_makelong(QUVI_CALLBACK_STATUS_FETCH,
  90                                      QUVI_CALLBACK_STATUS_DONE);
  91
  92           if (q->cb.status(p, 0) != QUVI_OK)
  93             q->status.rc = QUVI_ERROR_CALLBACK_ABORTED;
  94         }
  95     }
  96   else
  97     {
  98       /* Save returned error message. */
  99       if ((*n)->status.errmsg->len >0)
 100         g_string_assign(q->status.errmsg, (*n)->status.errmsg->str);
 101       else
 102         {
 103           g_string_assign(q->status.errmsg,
 104                           "unknown error: fetch: callback returned "
 105                           "empty errmsg");
 106         }
 107     }
 108
 109   q->status.resp_code = (*n)->status.resp_code;
 110 }
 111
 112 /* vim: set ts=2 sw=2 tw=72 expandtab: */