src/lua.c

   1 /*
   2  * Copyright (C) 2008 - 2016 The Geeqie Team
   3  *
   4  * Author: Klaus Ethgen
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program 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 General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include "config.h"
  22
  23 #ifdef HAVE_LUA
  24
  25 #define _XOPEN_SOURCE
  26
  27 #include <lua.h>
  28 #include <lauxlib.h>
  29 #include <lualib.h>
  30
  31 #include <stdio.h>
  32 #include <glib.h>
  33 #include <string.h>
  34 #include <time.h>
  35
  36 #include "main.h"
  37 #include "glua.h"
  38 #include "ui_fileops.h"
  39 #include "exif.h"
  40
  41 /**
  42  * @file
  43  * User API consists of the following namespaces:
  44  *
  45  * @link image_methods Image:@endlink basic image information
  46  *
  47  * <b>Collection</b>: not implemented
  48  *
  49  * @link exif_methods <exif-structure>:get_datum() @endlink get single exif parameter
  50  *
  51  */
  52
  53 static lua_State *L; /** The LUA object needed for all operations (NOTE: That is
  54                        * a upper-case variable to match the documentation!) */
  55
  56 /* Taking that definition from lua 5.1 source */
  57 #if defined(LUA_VERSION_NUM) && LUA_VERSION_NUM >= 502
  58 int luaL_typerror(lua_State *L, int narg, const char *tname)
  59 {
  60         const char *msg = lua_pushfstring(L, "%s expected, got %s", tname, luaL_typename(L, narg));
  61         return luaL_argerror(L, narg, msg);
  62 }
  63
  64 # define LUA_register_meta(L, meta) luaL_setfuncs(L, meta, 0);
  65 # define LUA_register_global(L, string, func) \
  66         lua_newtable(L); \
  67         luaL_setfuncs(L, func, 0); \
  68         lua_pushvalue(L, -1); \
  69         lua_setglobal(L, string)
  70 #else
  71 # define LUA_register_meta(L, meta) luaL_register(L, NULL, meta)
  72 # define LUA_register_global(L, string, func) luaL_register(L, string, func)
  73 #endif
  74
  75 static FileData *lua_check_image(lua_State *L, int index)
  76 {
  77         FileData **fd;
  78         luaL_checktype(L, index, LUA_TUSERDATA);
  79         fd = (FileData **)luaL_checkudata(L, index, "Image");
  80         if (fd == NULL) luaL_typerror(L, index, "Image");
  81         return *fd;
  82 }
  83
  84 /**
  85  * @brief Get exif structure of selected image
  86  * @param L
  87  * @returns An @ref ExifData data structure containing the entire exif data
  88  *
  89  * To be used in conjunction with @link lua_exif_get_datum <exif-structure>:get_datum() @endlink
  90  */
  91 static int lua_image_get_exif(lua_State *L)
  92 {
  93         FileData *fd;
  94         ExifData *exif;
  95         ExifData **exif_data;
  96
  97         fd = lua_check_image(L, 1);
  98         exif = exif_read_fd(fd);
  99
 100         exif_data = (ExifData **)lua_newuserdata(L, sizeof(ExifData *));
 101         luaL_getmetatable(L, "Exif");
 102         lua_setmetatable(L, -2);
 103
 104         *exif_data = exif;
 105
 106         return 1;
 107 }
 108
 109 /**
 110  * @brief Get full path of selected image
 111  * @param L
 112  * @returns char The full path of the file, including filename and extension
 113  *
 114  *
 115  */
 116 static int lua_image_get_path(lua_State *L)
 117 {
 118         FileData *fd;
 119
 120         fd = lua_check_image(L, 1);
 121         lua_pushstring(L, fd->path);
 122         return 1;
 123 }
 124
 125 /**
 126  * @brief Get full filename of selected image
 127  * @param L
 128  * @returns char The full filename including extension
 129  *
 130  *
 131  */
 132 static int lua_image_get_name(lua_State *L)
 133 {
 134         FileData *fd;
 135
 136         fd = lua_check_image(L, 1);
 137         lua_pushstring(L, fd->name);
 138         return 1;
 139 }
 140
 141 /**
 142  * @brief Get file extension of selected image
 143  * @param L
 144  * @returns char The file extension including preceding dot
 145  *
 146  *
 147  */
 148 static int lua_image_get_extension(lua_State *L)
 149 {
 150         FileData *fd;
 151
 152         fd = lua_check_image(L, 1);
 153         lua_pushstring(L, fd->extension);
 154         return 1;
 155 }
 156
 157 /**
 158  * @brief Get file date of selected image
 159  * @param L
 160  * @returns time_t The file date in Unix timestamp format.
 161  *
 162  * time_t - signed integer which represents the number of seconds since
 163  * the start of the Unix epoch: midnight UTC of January 1, 1970
 164  */
 165 static int lua_image_get_date(lua_State *L)
 166 {
 167         FileData *fd;
 168
 169         fd = lua_check_image(L, 1);
 170         lua_pushnumber(L, fd->date);
 171         return 1;
 172 }
 173
 174 /**
 175  * @brief Get file size of selected image
 176  * @param L
 177  * @returns integer The file size in bytes
 178  *
 179  *
 180  */
 181 static int lua_image_get_size(lua_State *L)
 182 {
 183         FileData *fd;
 184
 185         fd = lua_check_image(L, 1);
 186         lua_pushnumber(L, fd->size);
 187         return 1;
 188 }
 189
 190 /**
 191  * @brief Get marks of selected image
 192  * @param L
 193  * @returns unsigned integer Bit map of marks set
 194  *
 195  * Bit 0 == Mark 1 etc.
 196  *
 197  *
 198  */
 199 static int lua_image_get_marks(lua_State *L)
 200 {
 201         FileData *fd;
 202
 203         fd = lua_check_image(L, 1);
 204         lua_pushnumber(L, fd->marks);
 205         return 1;
 206 }
 207
 208 static ExifData *lua_check_exif(lua_State *L, int index)
 209 {
 210         ExifData **exif;
 211         luaL_checktype(L, index, LUA_TUSERDATA);
 212         exif = (ExifData **)luaL_checkudata(L, index, "Exif");
 213         if (exif == NULL) luaL_typerror(L, index, "Exif");
 214         return *exif;
 215 }
 216
 217 /**
 218  * @brief Interface for EXIF data
 219  * @param L
 220  * @returns <i>return</i> A single exif tag extracted from a structure output by the @link lua_image_get_exif Image:get_exif() @endlink command
 221  *
 222  * e.g. \n
 223  * exif_structure = Image:get_exif(); \n
 224  * DateTimeDigitized = exif_structure:get_datum("Exif.Photo.DateTimeDigitized");
 225  *
 226  * Where <i>return</i> is: \n
 227  * Exif.Photo.DateTimeOriginal = signed integer time_t \n
 228  * Exif.Photo.DateTimeDigitized = signed integer time_t \n
 229  * otherwise char
 230  *
 231  */
 232 static int lua_exif_get_datum(lua_State *L)
 233 {
 234         const gchar *key;
 235         gchar *value = NULL;
 236         ExifData *exif;
 237         struct tm tm;
 238         time_t datetime;
 239
 240         exif = lua_check_exif(L, 1);
 241         key = luaL_checkstring(L, 2);
 242         if (key == (gchar*)NULL || key[0] == '\0')
 243                 {
 244                 lua_pushnil(L);
 245                 return 1;
 246                 }
 247         if (!exif)
 248                 {
 249                 lua_pushnil(L);
 250                 return 1;
 251                 }
 252         value = exif_get_data_as_text(exif, key);
 253         if (strcmp(key, "Exif.Photo.DateTimeOriginal") == 0)
 254                 {
 255                 memset(&tm, 0, sizeof(tm));
 256                 if (value && strptime(value, "%Y:%m:%d %H:%M:%S", &tm))
 257                         {
 258                         datetime = mktime(&tm);
 259                         lua_pushnumber(L, datetime);
 260                         return 1;
 261                         }
 262                 else
 263                         {
 264                         lua_pushnil(L);
 265                         return 1;
 266                         }
 267                 }
 268         else if (strcmp(key, "Exif.Photo.DateTimeDigitized") == 0)
 269                 {
 270                 memset(&tm, 0, sizeof(tm));
 271                 if (value && strptime(value, "%Y:%m:%d %H:%M:%S", &tm))
 272                         {
 273                         datetime = mktime(&tm);
 274                         lua_pushnumber(L, datetime);
 275                         return 1;
 276                         }
 277                 else
 278                         {
 279                         lua_pushnil(L);
 280                         return 1;
 281                         }
 282                 }
 283         lua_pushstring(L, value);
 284         return 1;
 285 }
 286
 287 /**
 288  * @brief  <b>Image:</b> metatable and methods \n
 289  * Call by e.g. \n
 290  * path_name = @link lua_image_get_path Image:getpath() @endlink \n
 291  * where the keyword <b>Image</b> represents the currently selected image
 292  */
 293 static const luaL_Reg image_methods[] = {
 294                 {"get_path", lua_image_get_path},
 295                 {"get_name", lua_image_get_name},
 296                 {"get_extension", lua_image_get_extension},
 297                 {"get_date", lua_image_get_date},
 298                 {"get_size", lua_image_get_size},
 299                 {"get_exif", lua_image_get_exif},
 300                 {"get_marks", lua_image_get_marks},
 301                 {NULL, NULL}
 302 };
 303
 304 /**
 305  * @brief  <b>exif:</b> table and methods \n
 306  * Call by e.g. \n
 307  * @link lua_exif_get_datum <exif-structure>:get_datum() @endlink \n
 308  * where <exif-structure> is the output of @link lua_image_get_exif Image:get_exif() @endlink
 309  *
 310  * exif_structure = Image:get_exif(); \n
 311  * DateTimeDigitized = exif_structure:get_datum("Exif.Photo.DateTimeDigitized");
 312  */
 313 static const luaL_Reg exif_methods[] = {
 314                 {"get_datum", lua_exif_get_datum},
 315                 {NULL, NULL}
 316 };
 317
 318 /**
 319  * @brief Initialize the lua interpreter.
 320  */
 321 void lua_init(void)
 322 {
 323         L = luaL_newstate();
 324         luaL_openlibs(L); /* Open all libraries for lua programs */
 325
 326         /* Now create custom methodes to do something */
 327         static const luaL_Reg meta_methods[] = {
 328                         {NULL, NULL}
 329         };
 330
 331         LUA_register_global(L, "Image", image_methods);
 332         luaL_newmetatable(L, "Image");
 333         LUA_register_meta(L, meta_methods);
 334         lua_pushliteral(L, "__index");
 335         lua_pushvalue(L, -3);
 336         lua_settable(L, -3);
 337         lua_pushliteral(L, "__metatable");
 338         lua_pushvalue(L, -3);
 339         lua_settable(L, -3);
 340         lua_pop(L, 1);
 341         lua_pop(L, 1);
 342
 343         LUA_register_global(L, "Exif", exif_methods);
 344         luaL_newmetatable(L, "Exif");
 345         LUA_register_meta(L, meta_methods);
 346         lua_pushliteral(L, "__index");
 347         lua_pushvalue(L, -3);
 348         lua_settable(L, -3);
 349         lua_pushliteral(L, "__metatable");
 350         lua_pushvalue(L, -3);
 351         lua_settable(L, -3);
 352         lua_pop(L, 1);
 353         lua_pop(L, 1);
 354 }
 355
 356 /**
 357  * @brief Call a lua function to get a single value.
 358  */
 359 gchar *lua_callvalue(FileData *fd, const gchar *file, const gchar *function)
 360 {
 361         gint result;
 362         gchar *data = NULL;
 363         gchar *path;
 364         FileData **image_data;
 365         gchar *tmp;
 366         GError *error = NULL;
 367         gboolean ok;
 368
 369         ok = access(g_build_filename(get_rc_dir(), "lua", file, NULL), R_OK);
 370         if (ok == 0)
 371                 {
 372                 path = g_build_filename(get_rc_dir(), "lua", file, NULL);
 373                 }
 374         else
 375                 {
 376                 /** @FIXME what is the correct way to find the scripts folder? */
 377                 ok = access(g_build_filename("/usr/local/lib", GQ_APPNAME_LC, file, NULL), R_OK);
 378                 if (ok == 0)
 379                         {
 380                         path = g_build_filename("/usr/local/lib", GQ_APPNAME_LC, file, NULL);
 381                         }
 382                 else
 383                         {
 384                         return g_strdup("");
 385                         }
 386                 }
 387
 388         /* Collection Table (Dummy at the moment) */
 389         lua_newtable(L);
 390         lua_setglobal(L, "Collection");
 391
 392         /* Current Image */
 393         image_data = (FileData **)lua_newuserdata(L, sizeof(FileData *));
 394         luaL_getmetatable(L, "Image");
 395         lua_setmetatable(L, -2);
 396         lua_setglobal(L, "Image");
 397
 398         *image_data = fd;
 399         if (file[0] == '\0')
 400                 {
 401                 result = luaL_dostring(L, function);
 402                 }
 403         else
 404                 {
 405                 result = luaL_dofile(L, path);
 406                 g_free(path);
 407                 }
 408
 409         if (result)
 410                 {
 411                 data = g_strdup_printf("Error running lua script: %s", lua_tostring(L, -1));
 412                 return data;
 413                 }
 414         data = g_strdup(lua_tostring(L, -1));
 415         tmp = g_locale_to_utf8(data, strlen(data), NULL, NULL, &error);
 416         if (error)
 417                 {
 418                 log_printf("Error converting lua output from locale to UTF-8: %s\n", error->message);
 419                 g_error_free(error);
 420                 }
 421         else
 422                 {
 423                 g_free(data);
 424                 data = g_strdup(tmp);
 425                 } // if (error) { ... } else
 426         return data;
 427 }
 428
 429 #endif
 430 /* vim: set shiftwidth=8 softtabstop=0 cindent cinoptions={1s: */