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

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * art.c : Art metadata handling
  3.  *****************************************************************************
  4.  * Copyright (C) 1998-2008 the VideoLAN team
  5.  * $Id: 8b65121b48f777ff5c081875bd6bc5531931342b $
  6.  *
  7.  * Authors: Antoine Cellerier <dionoea@videolan.org>
  8.  *          Clément Stenac <zorglub@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. #ifdef HAVE_CONFIG_H
  25. # include "config.h"
  26. #endif
  27. #include <assert.h>
  28. #include <vlc_common.h>
  29. #include <vlc_playlist.h>
  30. #include <vlc_charset.h>
  31. #include <vlc_strings.h>
  32. #include <vlc_stream.h>
  33. #include <vlc_url.h>
  34. #include <limits.h>                                             /* PATH_MAX */
  35. #ifdef HAVE_SYS_STAT_H
  36. #   include <sys/stat.h>
  37. #endif
  38. #include "../libvlc.h"
  39. #include "playlist_internal.h"
  40. static void ArtCacheCreateDir( const char *psz_dir )
  41. {
  42.     char newdir[strlen( psz_dir ) + 1];
  43.     strcpy( newdir, psz_dir );
  44.     char * psz_newdir = newdir;
  45.     char * psz = psz_newdir;
  46.     while( *psz )
  47.     {
  48.         while( *psz && *psz != DIR_SEP_CHAR) psz++;
  49.         if( !*psz ) break;
  50.         *psz = 0;
  51.         if( !EMPTY_STR( psz_newdir ) )
  52.             utf8_mkdir( psz_newdir, 0700 );
  53.         *psz = DIR_SEP_CHAR;
  54.         psz++;
  55.     }
  56.     utf8_mkdir( psz_dir, 0700 );
  57. }
  58. static void ArtCacheGetDirPath( char *psz_dir,
  59.                                 const char *psz_title,
  60.                                 const char *psz_artist, const char *psz_album )
  61. {
  62.     char *psz_cachedir = config_GetCacheDir();
  63.     if( !EMPTY_STR(psz_artist) && !EMPTY_STR(psz_album) )
  64.     {
  65.         char *psz_album_sanitized = filename_sanitize( psz_album );
  66.         char *psz_artist_sanitized = filename_sanitize( psz_artist );
  67.         snprintf( psz_dir, PATH_MAX, "%s" DIR_SEP
  68.                   "art" DIR_SEP "artistalbum" DIR_SEP "%s" DIR_SEP "%s",
  69.                   psz_cachedir, psz_artist_sanitized, psz_album_sanitized );
  70.         free( psz_album_sanitized );
  71.         free( psz_artist_sanitized );
  72.     }
  73.     else
  74.     {
  75.         char * psz_title_sanitized = filename_sanitize( psz_title );
  76.         snprintf( psz_dir, PATH_MAX, "%s" DIR_SEP
  77.                   "art" DIR_SEP "title" DIR_SEP "%s",
  78.                   psz_cachedir, psz_title_sanitized );
  79.         free( psz_title_sanitized );
  80.     }
  81.     free( psz_cachedir );
  82. }
  83. static char *ArtCachePath( input_item_t *p_item )
  84. {
  85.     char psz_path[PATH_MAX+1]; /* FIXME */
  86.     vlc_mutex_lock( &p_item->lock );
  87.     if( !p_item->p_meta )
  88.         p_item->p_meta = vlc_meta_New();
  89.     if( !p_item->p_meta )
  90.     {
  91.         vlc_mutex_unlock( &p_item->lock );
  92.         return NULL;
  93.     }
  94.     const char *psz_artist = vlc_meta_Get( p_item->p_meta, vlc_meta_Artist );
  95.     const char *psz_album = vlc_meta_Get( p_item->p_meta, vlc_meta_Album );
  96.     const char *psz_title = vlc_meta_Get( p_item->p_meta, vlc_meta_Title );
  97.     if( !psz_title )
  98.         psz_title = p_item->psz_name;
  99.     if( (!psz_artist || !psz_album ) && !psz_title )
  100.     {
  101.         vlc_mutex_unlock( &p_item->lock );
  102.         return NULL;
  103.     }
  104.     ArtCacheGetDirPath( psz_path, psz_title, psz_artist, psz_album );
  105.     vlc_mutex_unlock( &p_item->lock );
  106.     return strdup( psz_path );
  107. }
  108. static char *ArtCacheName( input_item_t *p_item, const char *psz_type )
  109. {
  110.     char *psz_path = ArtCachePath( p_item );
  111.     if( !psz_path )
  112.         return NULL;
  113.     ArtCacheCreateDir( psz_path );
  114.     char *psz_ext = filename_sanitize( psz_type ? psz_type : "" );
  115.     char *psz_filename;
  116.     if( asprintf( &psz_filename, "%s" DIR_SEP "art%s", psz_path, psz_ext ) < 0 )
  117.         psz_filename = NULL;
  118.     free( psz_ext );
  119.     free( psz_path );
  120.     return psz_filename;
  121. }
  122. /* */
  123. int playlist_FindArtInCache( input_item_t *p_item )
  124. {
  125.     char *psz_path = ArtCachePath( p_item );
  126.     if( !psz_path )
  127.         return VLC_EGENERIC;
  128.     /* Check if file exists */
  129.     DIR *p_dir = utf8_opendir( psz_path );
  130.     if( !p_dir )
  131.     {
  132.         free( psz_path );
  133.         return VLC_EGENERIC;
  134.     }
  135.     bool b_found = false;
  136.     char *psz_filename;
  137.     while( !b_found && (psz_filename = utf8_readdir( p_dir )) )
  138.     {
  139.         if( !strncmp( psz_filename, "art", 3 ) )
  140.         {
  141.             char *psz_file;
  142.             if( asprintf( &psz_file, "%s" DIR_SEP "%s",
  143.                           psz_path, psz_filename ) < 0 )
  144.                 psz_file = NULL;
  145.             if( psz_file )
  146.             {
  147.                 char *psz_uri = make_URI( psz_file );
  148.                 if( psz_uri )
  149.                 {
  150.                     input_item_SetArtURL( p_item, psz_uri );
  151.                     free( psz_uri );
  152.                 }
  153.                 free( psz_file );
  154.             }
  155.             b_found = true;
  156.         }
  157.         free( psz_filename );
  158.     }
  159.     /* */
  160.     closedir( p_dir );
  161.     free( psz_path );
  162.     return b_found ? VLC_SUCCESS : VLC_EGENERIC;
  163. }
  164. /* */
  165. int playlist_SaveArt( playlist_t *p_playlist, input_item_t *p_item,
  166.                       const uint8_t *p_buffer, int i_buffer, const char *psz_type )
  167. {
  168.     char *psz_filename = ArtCacheName( p_item, psz_type );
  169.     if( !psz_filename )
  170.         return VLC_EGENERIC;
  171.     char *psz_uri = make_URI( psz_filename );
  172.     if( !psz_uri )
  173.     {
  174.         free( psz_filename );
  175.         return VLC_EGENERIC;
  176.     }
  177.     /* Check if we already dumped it */
  178.     struct stat s;
  179.     if( !utf8_stat( psz_filename, &s ) )
  180.     {
  181.         input_item_SetArtURL( p_item, psz_uri );
  182.         free( psz_filename );
  183.         free( psz_uri );
  184.         return VLC_SUCCESS;
  185.     }
  186.     /* Dump it otherwise */
  187.     FILE *f = utf8_fopen( psz_filename, "wb" );
  188.     if( f )
  189.     {
  190.         if( fwrite( p_buffer, i_buffer, 1, f ) != 1 )
  191.         {
  192.             msg_Err( p_playlist, "%s: %m", psz_filename );
  193.         }
  194.         else
  195.         {
  196.             msg_Dbg( p_playlist, "album art saved to %s", psz_filename );
  197.             input_item_SetArtURL( p_item, psz_uri );
  198.         }
  199.         fclose( f );
  200.     }
  201.     free( psz_filename );
  202.     free( psz_uri );
  203.     return VLC_SUCCESS;
  204. }