meta.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:6k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * meta.c: Get meta/artwork using lua scripts
  3.  *****************************************************************************
  4.  * Copyright (C) 2007-2008 the VideoLAN team
  5.  * $Id: 733f2b73b771df3d66d545b7766361713189cafc $
  6.  *
  7.  * Authors: Antoine Cellerier <dionoea at videolan tod org>
  8.  *          Pierre d'Herbemont <pdherbemont # videolan.org>
  9.  *
  10.  * This program is free software; you can redistribute it and/or modify
  11.  * it under the terms of the GNU General Public License as published by
  12.  * the Free Software Foundation; either version 2 of the License, or
  13.  * (at your option) any later version.
  14.  *
  15.  * This program is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18.  * GNU General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU General Public License
  21.  * along with this program; if not, write to the Free Software
  22.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23.  *****************************************************************************/
  24. /*****************************************************************************
  25.  * Preamble
  26.  *****************************************************************************/
  27. #ifndef  _GNU_SOURCE
  28. #   define  _GNU_SOURCE
  29. #endif
  30. #ifdef HAVE_CONFIG_H
  31. # include "config.h"
  32. #endif
  33. #include <vlc_common.h>
  34. #include <vlc_input.h>
  35. #include <vlc_playlist.h>
  36. #include <vlc_meta.h>
  37. #include <vlc_url.h>
  38. #include <vlc_strings.h>
  39. #include <vlc_stream.h>
  40. #include <vlc_charset.h>
  41. #include "vlc.h"
  42. #include "libs.h"
  43. /*****************************************************************************
  44.  * Local prototypes
  45.  *****************************************************************************/
  46. static int fetch_art( vlc_object_t *p_this, const char * psz_filename,
  47.                       lua_State * L, void * user_data );
  48. static lua_State *vlclua_meta_init( vlc_object_t *p_this,
  49.                                     input_item_t * p_item );
  50. /*****************************************************************************
  51.  * Init lua
  52.  *****************************************************************************/
  53. static const luaL_Reg p_reg[] = { { NULL, NULL } };
  54. static lua_State * vlclua_meta_init( vlc_object_t *p_this, input_item_t * p_item )
  55. {
  56.     lua_State * L = luaL_newstate();
  57.     if( !L )
  58.     {
  59.         msg_Err( p_this, "Could not create new Lua State" );
  60.         return NULL;
  61.     }
  62.     char *psz_meta;
  63.     /* Load Lua libraries */
  64.     luaL_openlibs( L ); /* XXX: Don't open all the libs? */
  65.     luaL_register( L, "vlc", p_reg );
  66.     luaopen_msg( L );
  67.     luaopen_stream( L );
  68.     luaopen_strings( L );
  69.     luaopen_variables( L );
  70.     luaopen_object( L );
  71.     luaopen_misc( L );
  72.     lua_pushlightuserdata( L, p_this );
  73.     lua_setfield( L, -2, "private" );
  74.     psz_meta = input_item_GetName( p_item );
  75.     lua_pushstring( L, psz_meta );
  76.     lua_setfield( L, -2, "name" );
  77.     free( psz_meta );
  78.     psz_meta = input_item_GetArtist( p_item );
  79.     lua_pushstring( L, psz_meta );
  80.     lua_setfield( L, -2, "artist" );
  81.     free( psz_meta );
  82.     psz_meta = input_item_GetTitle( p_item ) ;
  83.     lua_pushstring( L, psz_meta );
  84.     lua_setfield( L, -2, "title" );
  85.     free( psz_meta );
  86.     psz_meta = input_item_GetAlbum( p_item );
  87.     lua_pushstring( L, psz_meta );
  88.     lua_setfield( L, -2, "album" );
  89.     free( psz_meta );
  90.     psz_meta = input_item_GetArtURL( p_item );
  91.     lua_pushstring( L, psz_meta );
  92.     lua_setfield( L, -2, "arturl" );
  93.     free( psz_meta );
  94.     /* XXX: all should be passed ( could use macro ) */
  95.     return L;
  96. }
  97. /*****************************************************************************
  98.  * Called through lua_scripts_batch_execute to call 'fetch_art' on the script
  99.  * pointed by psz_filename.
  100.  *****************************************************************************/
  101. static int fetch_art( vlc_object_t *p_this, const char * psz_filename,
  102.                       lua_State * L, void * user_data )
  103. {
  104.     int i_ret = VLC_EGENERIC;
  105.     input_item_t * p_input = user_data;
  106.     int s;
  107.     /* Ugly hack to delete previous versions of the fetchart()
  108.     * functions. */
  109.     lua_pushnil( L );
  110.     lua_setglobal( L, "fetch_art" );
  111.     /* Load and run the script(s) */
  112.     if( luaL_dofile( L, psz_filename ) )
  113.     {
  114.         msg_Warn( p_this, "Error loading script %s: %s", psz_filename,
  115.                   lua_tostring( L, lua_gettop( L ) ) );
  116.         lua_pop( L, 1 );
  117.         return VLC_EGENERIC;
  118.     }
  119.     lua_getglobal( L, "fetch_art" );
  120.     if( !lua_isfunction( L, lua_gettop( L ) ) )
  121.     {
  122.         msg_Warn( p_this, "Error while runing script %s, "
  123.                   "function fetch_art() not found", psz_filename );
  124.         lua_pop( L, 1 );
  125.         return VLC_EGENERIC;
  126.     }
  127.     if( lua_pcall( L, 0, 1, 0 ) )
  128.     {
  129.         msg_Warn( p_this, "Error while runing script %s, "
  130.                   "function fetch_art(): %s", psz_filename,
  131.                   lua_tostring( L, lua_gettop( L ) ) );
  132.         lua_pop( L, 1 );
  133.         return VLC_EGENERIC;
  134.     }
  135.     if((s = lua_gettop( L )))
  136.     {
  137.         const char * psz_value;
  138.         if( lua_isstring( L, s ) )
  139.         {
  140.             psz_value = lua_tostring( L, s );
  141.             if( psz_value && *psz_value != 0 )
  142.             {
  143.                 lua_Dbg( p_this, "setting arturl: %s", psz_value );
  144.                 input_item_SetArtURL ( p_input, psz_value );
  145.                 i_ret = VLC_SUCCESS;
  146.             }
  147.         }
  148.         else if( !lua_isnil( L, s ) )
  149.         {
  150.             msg_Err( p_this, "Lua art fetcher script %s: "
  151.                  "didn't return a string", psz_filename );
  152.         }
  153.     }
  154.     else
  155.     {
  156.         msg_Err( p_this, "Script went completely foobar" );
  157.     }
  158.     return i_ret;
  159. }
  160. /*****************************************************************************
  161.  * Module entry point for art.
  162.  *****************************************************************************/
  163. int FindArt( vlc_object_t *p_this )
  164. {
  165.     playlist_t *p_playlist = pl_Hold( p_this );
  166.     if( !p_playlist )
  167.         return VLC_EGENERIC;
  168.     input_item_t *p_item = (input_item_t *)p_this->p_private;
  169.     lua_State *L = vlclua_meta_init( p_this, p_item );
  170.     int i_ret = vlclua_scripts_batch_execute( p_this, "meta", &fetch_art, L, p_item );
  171.     lua_close( L );
  172.     pl_Release( p_this );
  173.     return i_ret;
  174. }