DCP: fixed include-guards
[vlc.git] / modules / meta_engine / folder.c
1 /*****************************************************************************
2  * folder.c
3  *****************************************************************************
4  * Copyright (C) 2006 VLC authors and VideoLAN
5  * $Id$
6  *
7  * Authors: Antoine Cellerier <dionoea -at- videolan -dot- org>
8  *
9  * This program is free software; you can redistribute it and/or modify it
10  * under the terms of the GNU Lesser General Public License as published by
11  * the Free Software Foundation; either version 2.1 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17  * GNU Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <sys/stat.h>
33
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
36 #include <vlc_meta_fetcher.h>
37 #include <vlc_fs.h>
38 #include <vlc_url.h>
39 #include <vlc_input_item.h>
40
41 static const char* cover_files[] = {
42     "Folder.jpg",           /* Windows */
43     "AlbumArtSmall.jpg",    /* Windows */
44     "AlbumArt.jpg",         /* Windows */
45     "Album.jpg",
46     ".folder.png",          /* KDE?    */
47     "cover.jpg",            /* rockbox */
48     "thumb.jpg",
49 };
50
51 static const int i_covers = (sizeof(cover_files)/sizeof(cover_files[0]));
52
53 /*****************************************************************************
54  * Local prototypes
55  *****************************************************************************/
56 static int FindMeta( vlc_object_t * );
57
58 /*****************************************************************************
59  * Module descriptor
60  *****************************************************************************/
61
62 vlc_module_begin ()
63     set_shortname( N_( "Folder" ) )
64     set_description( N_("Folder meta data") )
65     add_loadfile( "album-art-filename", NULL,
66         N_("Album art filename"), N_("Filename to look for album art in current directory"), false );
67     set_capability( "art finder", 90 )
68     set_callbacks( FindMeta, NULL )
69 vlc_module_end ()
70
71 /*****************************************************************************
72  *****************************************************************************/
73 static int FindMeta( vlc_object_t *p_this )
74 {
75     meta_fetcher_t *p_finder = (meta_fetcher_t *)p_this;
76     input_item_t *p_item = p_finder->p_item;
77     bool b_have_art = false;
78     struct stat statinfo;
79     char *psz_path = NULL;
80
81     if( !p_item )
82         return VLC_EGENERIC;
83
84     char *psz_uri = input_item_GetURI( p_item );
85     if( !psz_uri )
86         return VLC_EGENERIC;
87
88     if ( *psz_uri && psz_uri[strlen( psz_uri ) - 1] != DIR_SEP_CHAR )
89     {
90         if ( asprintf( &psz_path, "%s"DIR_SEP, psz_uri ) == -1 )
91         {
92             free( psz_uri );
93             return VLC_EGENERIC;
94         }
95         char *psz_basedir = vlc_uri2path( psz_path );
96         FREENULL( psz_path );
97         if( psz_basedir == NULL )
98         {
99             free( psz_uri );
100             return VLC_EGENERIC;
101         }
102         if( vlc_stat( psz_basedir, &statinfo ) == 0 && S_ISDIR(statinfo.st_mode) )
103             psz_path = psz_basedir;
104         else
105             free( psz_basedir );
106     }
107
108     if ( psz_path == NULL )
109     {
110         char *psz_basedir = vlc_uri2path( psz_uri );
111         if( psz_basedir == NULL )
112         {
113             free( psz_uri );
114             return VLC_EGENERIC;
115         }
116
117         char *psz_buf = strrchr( psz_basedir, DIR_SEP_CHAR );
118         if( psz_buf )
119             *++psz_buf = '\0';
120         else
121             *psz_basedir = '\0'; /* relative path */
122         psz_path = psz_basedir;
123     }
124
125     free( psz_uri );
126
127     for( int i = -1; !b_have_art && i < i_covers; i++ )
128     {
129         const char *filename;
130         char *filebuf, *filepath;
131
132         if( i == -1 ) /* higher priority : configured filename */
133         {
134             filebuf = var_InheritString( p_this, "album-art-filename" );
135             if( filebuf == NULL )
136                 continue;
137             filename = filebuf;
138         }
139         else
140         {
141             filename = cover_files[i];
142             filebuf = NULL;
143         }
144
145         if( asprintf( &filepath, "%s%s", psz_path, filename ) == -1 )
146             filepath = NULL;
147         free( filebuf );
148         if( unlikely(filepath == NULL) )
149             continue;
150
151         if( vlc_stat( filepath, &statinfo ) == 0 && S_ISREG(statinfo.st_mode) )
152         {
153             char *psz_uri = vlc_path2uri( filepath, "file" );
154             if( psz_uri )
155             {
156                 input_item_SetArtURL( p_item, psz_uri );
157                 free( psz_uri );
158                 b_have_art = true;
159             }
160         }
161         free( filepath );
162     }
163     free( psz_path );
164
165     return b_have_art ? VLC_SUCCESS : VLC_EGENERIC;
166 }