Converting a string for using it in xml without using it is STUPID
[vlc.git] / modules / misc / playlist / xspf.c
1 /******************************************************************************
2  * xspf.c : XSPF playlist export functions
3  ******************************************************************************
4  * Copyright (C) 2006 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Daniel Stränger <vlc at schmaller dot de>
8  *          Yoann Peronneau <yoann@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 /**
26  * \file modules/misc/playlist/xspf.c
27  * \brief XSPF playlist export functions
28  */
29 #include <vlc/vlc.h>
30 #include <vlc_interface.h>
31 #include <vlc_playlist.h>
32 #include <vlc_input.h>
33 #include <vlc_meta.h>
34 #include <vlc_strings.h>
35 #include <vlc_charset.h>
36 #include "xspf.h"
37
38 /**
39  * \brief Prints the XSPF header to file, writes each item by xspf_export_item()
40  * and closes the open xml elements
41  * \param p_this the VLC playlist object
42  * \return VLC_SUCCESS if some memory is available, otherwise VLC_ENONMEM
43  */
44 int E_(xspf_export_playlist)( vlc_object_t *p_this )
45 {
46     const playlist_t *p_playlist = (playlist_t *)p_this;
47     const playlist_export_t *p_export =
48         (playlist_export_t *)p_playlist->p_private;
49     int               i, i_count;
50     char             *psz_temp;
51     playlist_item_t  *p_node = p_export->p_root;
52
53     /* write XSPF XML header */
54     fprintf( p_export->p_file, "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n" );
55     fprintf( p_export->p_file,
56              "<playlist version=\"1\" xmlns=\"http://xspf.org/ns/0/\">\n" );
57
58     if( !p_node ) return VLC_SUCCESS;
59
60     /* save name of the playlist node */
61     psz_temp = convert_xml_special_chars( p_node->p_input->psz_name );
62     if( *psz_temp )
63     {
64         fprintf(  p_export->p_file, "\t<title>%s</title>\n", psz_temp );
65     }
66     free( psz_temp );
67
68     /* save location of the playlist node */
69     psz_temp = assertUTF8URI( p_export->psz_filename );
70     if( psz_temp && *psz_temp )
71     {
72         fprintf( p_export->p_file, "\t<location>%s</location>\n", psz_temp );
73         free( psz_temp );
74     }
75
76     /* export all items in a flat format */
77     fprintf( p_export->p_file, "\t<trackList>\n" );
78     i_count = 0;
79     for( i = 0; i < p_node->i_children; i++ )
80     {
81         xspf_export_item( p_node->pp_children[i], p_export->p_file,
82                           &i_count );
83     }
84     fprintf( p_export->p_file, "\t</trackList>\n" );
85
86     /* export the tree structure in <extension> */
87     fprintf( p_export->p_file, "\t<extension>\n" );
88     i_count = 0;
89     for( i = 0; i < p_node->i_children; i++ )
90     {
91         xspf_extension_item( p_node->pp_children[i], p_export->p_file,
92                              &i_count );
93     }
94     fprintf( p_export->p_file, "\t</extension>\n" );
95
96     /* close the header elements */
97     fprintf( p_export->p_file, "</playlist>\n" );
98
99     return VLC_SUCCESS;
100 }
101
102 /**
103  * \brief exports one item to file or traverse if item is a node
104  * \param p_item playlist item to export
105  * \param p_file file to write xml-converted item to
106  * \param p_i_count counter for track identifiers
107  */
108 static void xspf_export_item( playlist_item_t *p_item, FILE *p_file,
109                               int *p_i_count )
110 {
111     char *psz;
112     char *psz_temp;
113     mtime_t i_duration;
114
115     if( !p_item ) return;
116
117     /* if we get a node here, we must traverse it */
118     if( p_item->i_children > 0 )
119     {
120         int i;
121         for( i = 0; i < p_item->i_children; i++ )
122         {
123             xspf_export_item( p_item->pp_children[i], p_file, p_i_count );
124         }
125         return;
126     }
127
128     /* don't write empty nodes */
129     if( p_item->i_children == 0 )
130     {
131         return;
132     }
133
134     /* leaves can be written directly */
135     fprintf( p_file, "\t\t<track>\n" );
136
137     /* print identifier and increase the counter */
138     fprintf( p_file, "\t\t\t<identifier>%i</identifier>\n", *p_i_count );
139     ( *p_i_count )++;
140
141     /* -> the location */
142
143     char *psz_uri = input_item_GetURI( p_item->p_input );
144
145     if( psz_uri && *psz_uri )
146     {
147         psz = assertUTF8URI( psz_uri );
148         fprintf( p_file, "\t\t\t<location>%s</location>\n", psz );
149         free( psz );
150     }
151
152     /* -> the name/title (only if different from uri)*/
153     char *psz_name = input_item_GetTitle( p_item->p_input );
154     if( psz_name && psz_uri && strcmp( psz_uri, psz_name ) )
155     {
156         psz_temp = convert_xml_special_chars( psz_name );
157         if( *psz_temp )
158             fprintf( p_file, "\t\t\t<title>%s</title>\n", psz_temp );
159         free( psz_temp );
160     }
161     free( psz_name );
162     free( psz_uri );
163
164     if( p_item->p_input->p_meta == NULL )
165     {
166         goto xspfexportitem_end;
167     }
168
169     /* -> the artist/creator */
170     psz = input_item_GetArtist( p_item->p_input );
171     if( psz == NULL ) psz = strdup( "" );
172     psz_temp = convert_xml_special_chars( psz );
173     free( psz );
174     if( *psz_temp )
175     {
176         fprintf( p_file, "\t\t\t<creator>%s</creator>\n", psz_temp );
177     }
178     free( psz_temp );
179
180     /* -> the album */
181     psz = input_item_GetAlbum( p_item->p_input );
182     if( psz == NULL ) psz = strdup( "" );
183     psz_temp = convert_xml_special_chars( psz );
184     free( psz );
185     if( *psz_temp )
186     {
187         fprintf( p_file, "\t\t\t<album>%s</album>\n", psz_temp );
188     }
189     free( psz_temp );
190
191     /* -> the track number */
192     psz = input_item_GetTrackNum( p_item->p_input );
193     if( psz == NULL ) psz = strdup( "" );
194     if( psz )
195     {
196         if( *psz )
197         {
198             fprintf( p_file, "\t\t\t<trackNum>%i</trackNum>\n", atoi( psz ) );
199         }
200     }
201     free( psz );
202
203     /* -> the description */
204     psz = input_item_GetDescription( p_item->p_input );
205     if( psz == NULL ) psz = strdup( "" );
206     psz_temp = convert_xml_special_chars( psz );
207     free( psz );
208     if( *psz_temp )
209     {
210         fprintf( p_file, "\t\t\t<annotation>%s</annotation>\n", psz_temp );
211     }
212     free( psz_temp );
213
214     psz = input_item_GetArtURL( p_item->p_input );
215     if( psz == NULL ) psz = strdup( "" );
216     psz_temp = convert_xml_special_chars( psz );
217     free( psz );
218     if( !EMPTY_STR( psz_temp ) )
219     {
220         fprintf( p_file, "\t\t\t<image>%s</image>\n", psz_temp );
221     }
222     free( psz_temp );
223
224 xspfexportitem_end:
225     /* -> the duration */
226     i_duration = input_item_GetDuration( p_item->p_input );
227     if( i_duration > 0 )
228     {
229         fprintf( p_file, "\t\t\t<duration>%ld</duration>\n",
230                  (long)(i_duration / 1000) );
231     }
232
233     fprintf( p_file, "\t\t</track>\n" );
234
235     return;
236 }
237
238 /**
239  * \brief exports one item in extension to file and traverse if item is a node
240  * \param p_item playlist item to export
241  * \param p_file file to write xml-converted item to
242  * \param p_i_count counter for track identifiers
243  */
244 static void xspf_extension_item( playlist_item_t *p_item, FILE *p_file,
245                                  int *p_i_count )
246 {
247     if( !p_item ) return;
248
249     /* if we get a node here, we must traverse it */
250     if( p_item->i_children >= 0 )
251     {
252         int i;
253         char *psz_temp;
254         psz_temp = convert_xml_special_chars( p_item->p_input->psz_name );
255         fprintf( p_file, "\t\t<node title=\"%s\">\n",
256                  *psz_temp ? psz_temp : "" );
257         free( psz_temp );
258
259         for( i = 0; i < p_item->i_children; i++ )
260         {
261             xspf_extension_item( p_item->pp_children[i], p_file, p_i_count );
262         }
263
264         fprintf( p_file, "\t\t</node>\n" );
265         return;
266     }
267
268
269     /* print leaf and increase the counter */
270     fprintf( p_file, "\t\t\t<item href=\"%i\" />\n", *p_i_count );
271     ( *p_i_count )++;
272
273     return;
274 }
275
276 /**
277  * \param psz_name the location of the media ressource (e.g. local file,
278  *        device, network stream, etc.)
279  * \return a new char buffer which asserts that the location is valid UTF-8
280  *         and a valid URI
281  * \note the returned buffer must be freed, when it isn't used anymore
282  */
283 static char *assertUTF8URI( char *psz_name )
284 {
285     char *psz_ret = NULL;              /**< the new result buffer to return */
286     char *psz_s = NULL, *psz_d = NULL; /**< src & dest pointers for URI conversion */
287     vlc_bool_t b_name_is_uri = VLC_FALSE;
288
289     if( !psz_name || !*psz_name )
290         return NULL;
291
292     /* check that string is valid UTF-8 */
293     /* XXX: Why do we even need to do that ? (all strings in core are UTF-8 encoded */
294     if( !( psz_s = EnsureUTF8( psz_name ) ) )
295         return NULL;
296
297     /* max. 3x for URI conversion (percent escaping) and
298        8 bytes for "file://" and NULL-termination */
299     psz_ret = (char *)malloc( sizeof(char)*strlen(psz_name)*6*3+8 );
300     if( !psz_ret )
301         return NULL;
302
303     /** \todo check for a valid scheme part preceding the colon */
304     if( strchr( psz_s, ':' ) )
305     {
306         psz_d = psz_ret;
307         b_name_is_uri = VLC_TRUE;
308     }
309     /* assume "file" scheme if no scheme-part is included */
310     else
311     {
312         strcpy( psz_ret, "file://" );
313         psz_d = psz_ret + 7;
314     }
315
316     while( *psz_s )
317     {
318         /* percent-encode all non-ASCII and the XML special characters and the percent sign itself */
319         if( *psz_s & B10000000 ||
320             *psz_s == '<' ||
321             *psz_s == '>' ||
322             *psz_s == '&' ||
323             *psz_s == ' ' ||
324             ( *psz_s == '%' && !b_name_is_uri ) )
325         {
326             *psz_d++ = '%';
327             *psz_d++ = hexchars[(*psz_s >> 4) & B00001111];
328             *psz_d++ = hexchars[*psz_s & B00001111];
329         }
330         else
331         {
332             *psz_d++ = *psz_s;
333         }
334
335         psz_s++;
336     }
337     *psz_d = '\0';
338
339     return (char *)realloc( psz_ret, sizeof(char)*strlen( psz_ret ) + 1 );
340 }