examples/script.c

   1 /* libquvi
   2  * Copyright (C) 2012  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 "config.h"
  21
  22 #include <locale.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <glib.h>
  26 #include <quvi.h>
  27
  28 #include "examples.h"
  29
  30 struct _opts_s
  31 {
  32   gboolean autoproxy;
  33   gchar **property;
  34   gboolean verbose;
  35   gchar *type;
  36 };
  37
  38 static struct _opts_s opts;
  39
  40 static const GOptionEntry entries[] =
  41 {
  42   {
  43     "property", 'p', 0, G_OPTION_ARG_STRING_ARRAY, &opts.property,
  44     "Script property to output", "PROPERTY"
  45   },
  46   {
  47     "type", 't', 0, G_OPTION_ARG_STRING, &opts.type,
  48     "Script type", "TYPE"
  49   },
  50   {
  51     "autoproxy", 'a', 0, G_OPTION_ARG_NONE, &opts.autoproxy,
  52     "Enable the autoproxy feature", NULL
  53   },
  54   {
  55     "verbose", 'v', 0, G_OPTION_ARG_NONE, &opts.verbose,
  56     "Verbose libcurl output", NULL
  57   },
  58   {NULL, 0, 0, 0, NULL, NULL, NULL}
  59 };
  60
  61 struct _property_lookup_s
  62 {
  63   QuviScriptProperty to;
  64   const gchar *from;
  65 };
  66
  67 static const struct _property_lookup_s property_conv[] =
  68 {
  69   {QUVI_SCRIPT_PROPERTY_FILEPATH, "filepath"},
  70   {QUVI_SCRIPT_PROPERTY_FILENAME, "filename"},
  71   {QUVI_SCRIPT_PROPERTY_DOMAINS,  "domains"},
  72   {QUVI_SCRIPT_PROPERTY_SHA1,     "sha1"},
  73   {0, NULL}
  74 };
  75
  76 static void dump_script(QuviScriptType type)
  77 {
  78   QuviScriptProperty qsp;
  79   gchar *s;
  80   gint i,j;
  81
  82   for (i=0; opts.property[i] != NULL; ++i)
  83     {
  84       const gchar *p = opts.property[i];
  85       for (j=0; property_conv[j].from != NULL; ++j)
  86         {
  87           if (g_strcmp0(p, property_conv[j].from) == 0)
  88             {
  89               qsp = property_conv[j].to;
  90               break;
  91             }
  92         }
  93       quvi_script_get(q, type, qsp, &s);
  94       if (i == 0)
  95         g_print("[%s]\n", __func__);
  96       g_print("  %s=%s\n", p, s);
  97     }
  98 }
  99
 100 static gchar **property_sv()
 101 {
 102   gchar **r;
 103   gint i,j;
 104
 105   i=0;
 106   while (property_conv[i].from != NULL) ++i;
 107   r = g_new(gchar*, i+1);
 108
 109   i=j=0;
 110   while (property_conv[j].from != NULL)
 111     r[i++] = g_strdup(property_conv[j++].from);
 112   r[i] = NULL;
 113
 114   return (r);
 115 }
 116
 117 static gboolean chk_property_values()
 118 {
 119   gchar **v, *s;
 120   gboolean r;
 121
 122   v = property_sv();
 123   r = examples_chk_val_sv(opts.property, v, &s);
 124   if (r == FALSE)
 125     {
 126       g_printerr(
 127         "error: invalid value (`%s') for the option `--property'\n", s);
 128     }
 129
 130   g_strfreev(v);
 131   v = NULL;
 132
 133   return (r);
 134 }
 135
 136 struct _type_lookup_s
 137 {
 138   QuviScriptType to;
 139   const gchar *from;
 140 };
 141
 142 static const struct _type_lookup_s type_conv[] =
 143 {
 144   {QUVI_SCRIPT_TYPE_PLAYLIST, "playlist"},
 145   {QUVI_SCRIPT_TYPE_MEDIA,    "media"},
 146   {QUVI_SCRIPT_TYPE_SCAN,     "scan"},
 147   {0, NULL}
 148 };
 149
 150 static gchar **type_sv()
 151 {
 152   gchar **r;
 153   gint i,j;
 154
 155   i=0;
 156   while (type_conv[i].from != NULL) ++i;
 157   r = g_new(gchar*, i+1);
 158
 159   i=j=0;
 160   while (type_conv[j].from != NULL)
 161     r[i++] = g_strdup(type_conv[j++].from);
 162   r[i] = NULL;
 163
 164   return (r);
 165 }
 166
 167 static gboolean chk_type_values()
 168 {
 169   gchar **v, *s;
 170   gboolean r;
 171
 172   v = type_sv();
 173   r = examples_chk_val_s(opts.type, v, &s);
 174   if (r == FALSE)
 175     {
 176       g_printerr(
 177         "error: invalid value (`%s') for the option `--type'\n", s);
 178     }
 179
 180   g_strfreev(v);
 181   v = NULL;
 182
 183   return (r);
 184 }
 185
 186 static QuviScriptType type_n()
 187 {
 188   gint i;
 189   for (i=0; type_conv[i].from != NULL; ++i)
 190     {
 191       if (g_strcmp0(opts.type, type_conv[i].from) == 0)
 192         return (type_conv[i].to);
 193     }
 194   return (QUVI_SCRIPT_TYPE_MEDIA);
 195 }
 196
 197 static gint opts_new(gint argc, gchar **argv)
 198 {
 199   GOptionContext *c;
 200   GOptionGroup *g;
 201   GError *e;
 202   gint r;
 203
 204   c = g_option_context_new(NULL);
 205   r = EXIT_SUCCESS;
 206   g = NULL;
 207   e = NULL;
 208
 209   g_option_context_set_help_enabled(c, TRUE);
 210   g_option_context_add_main_entries(c, entries, NULL);
 211
 212   if (g_option_context_parse(c, &argc, &argv, &e) == FALSE)
 213     {
 214       g_printerr("error: %s\n", e->message);
 215       g_error_free(e);
 216       r = EXIT_FAILURE;
 217       e = NULL;
 218     }
 219
 220   g_option_context_free(c);
 221   c = NULL;
 222
 223   /* Set the defaults. */
 224
 225   if (opts.property == NULL)
 226     {
 227       gchar *v[] = {"filename", "domains", NULL};
 228       opts.property = g_strdupv(v);
 229     }
 230
 231   if (opts.type == NULL)
 232     opts.type = g_strdup("media");
 233
 234   /* Check input. */
 235
 236   if (chk_property_values() == FALSE)
 237     return (EXIT_FAILURE);
 238
 239   if (chk_type_values() == FALSE)
 240     return (EXIT_FAILURE);
 241
 242   return (r);
 243 }
 244
 245 static void opts_free()
 246 {
 247   g_strfreev(opts.property);
 248   opts.property = NULL;
 249
 250   g_free(opts.type);
 251   opts.type = NULL;
 252 }
 253
 254 typedef quvi_callback_status qcs;
 255
 256 gint main(gint argc, gchar **argv)
 257 {
 258   QuviScriptType type;
 259   gint r;
 260
 261   setlocale(LC_ALL, "");
 262
 263   memset(&opts, 0, sizeof(struct _opts_s));
 264   g_assert(q == NULL);
 265
 266   r = opts_new(argc, argv);
 267   if (r != EXIT_SUCCESS)
 268     return (r);
 269
 270   q = quvi_new();
 271   examples_exit_if_error();
 272
 273   if (opts.autoproxy == TRUE)
 274     examples_enable_autoproxy();
 275
 276   if (opts.verbose == TRUE)
 277     examples_enable_verbose();
 278
 279   type = type_n();
 280
 281   {
 282     gchar *p = g_strjoinv(",", opts.property);;
 283     g_printerr("[%s] type=%s (0x%x), property=%s\n",
 284                __func__, opts.type, type, p);
 285     g_free(p);
 286     p = NULL;
 287   }
 288
 289   while (quvi_script_next(q, type) == QUVI_TRUE)
 290     dump_script(type);
 291
 292   opts_free();
 293   examples_cleanup();
 294
 295   g_assert(q == NULL);
 296   return (r);
 297 }
 298
 299 /* vim: set ts=2 sw=2 tw=72 expandtab: */