Git

1 /*\r
2  * FAPG 0.1 released under GPL\r
3  * http://royale.zerezo.com/fapg/\r
4  */\r
5 \r
6 #include <stdio.h>\r
7 #include <stdlib.h>\r
8 #include <getopt.h>\r
9 #include <dirent.h>\r
10 #include <sys/stat.h>\r
11 #include <sys/types.h>\r
12 #include <string.h>\r
13 #include <limits.h>\r
14 #include <unistd.h>\r
15 #include <ctype.h>\r
16 \r
17 #define MAX 10240\r
18 \r
19 int debug=0;\r
20 int format=0; /* 0 = m3u ; 1 = pls ; 2 = html */\r
21 unsigned char *prefix="";\r
22 int recursive=0;\r
23 int separator='/';\r
24 int skip=0;\r
25 int windows=0;\r
26 unsigned char buffer[MAX];\r
27 \r
28 int counter=0;\r
29 \r
30 unsigned char artist[1024];\r
31 unsigned char title[1024];\r
32 int duration;\r
33 \r
34 unsigned char unix2dos[256]={0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,70,35,36,37,38,39,40,41,82,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,84,59,36,61,65,71,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,36,93,94,95,96,97,98,99,100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,115,116,117,118,119,120,121,122,123,36,125,126,127,199,252,233,226,228,224,229,231,234,235,232,239,238,236,196,197,201,230,198,244,246,242,251,249,255,214,220,248,163,216,215,131,225,237,243,250,241,209,170,186,191,174,172,189,188,161,171,187,166,166,166,166,166,193,194,192,169,166,166,43,43,162,165,43,43,45,45,43,45,43,227,195,43,43,45,45,166,45,43,164,240,208,202,203,200,105,205,206,207,43,43,166,220,166,204,175,211,223,212,210,245,213,181,254,222,218,219,217,253,221,175,180,173,177,61,190,182,167,247,184,176,168,183,185,179,178,166,160};\r
35 \r
36 void usage()\r
37 {\r
38   fprintf(stderr,"Usage >> fapg [-b|--backslash] [-d|--debug] [-f|--format=m3u|pls|html] [-o|--output=/path/to/file.m3u] [-p|--prefix=/the/prefix] [-r|--recursive] [-w|--windows] /path/to/mp3/dir1 [/path/to/mp3/dir2 ...]\n");\r
39   exit(1);\r
40 }\r
41 \r
42 void parse_options(int argc,char **argv)\r
43 {\r
44   static char const short_options[]="bdf:o:p:rw";\r
45   static struct option long_options[]=\r
46   {\r
47     {"backslash",no_argument,NULL,'b'},\r
48     {"debug",no_argument,NULL,'d'},\r
49     {"format",required_argument,NULL,'f'},\r
50     {"output",required_argument,NULL,'o'},\r
51     {"prefix",required_argument,NULL,'p'},\r
52     {"recursive",no_argument,NULL,'r'},\r
53     {"windows",no_argument,NULL,'w'}\r
54   };\r
55   int c;\r
56   int option_index=0;\r
57   while ((c=getopt_long(argc,argv,short_options,long_options,&option_index))!=-1)\r
58   {\r
59     switch(c)\r
60     {\r
61       case 'b':\r
62         separator='\\';\r
63         break;\r
64       case 'd':\r
65         debug=1;\r
66         break;\r
67       case 'f':\r
68         if (strcmp(optarg,"m3u")==0)  format=0; else\r
69         if (strcmp(optarg,"pls")==0)  format=1; else\r
70         if (strcmp(optarg,"html")==0) format=2; else\r
71         usage();\r
72         break;\r
73       case 'o':\r
74         close(1);\r
75         if (fopen(optarg,"w")==NULL) { fprintf(stderr,"Error >> unable to open output file : %s\n",optarg); exit(2); }\r
76         break;\r
77       case 'p':\r
78         prefix=malloc(strlen(optarg)+1);\r
79         strcpy(prefix,optarg);\r
80         break;\r
81       case 'r':\r
82         recursive=1;\r
83         break;\r
84       case 'w':\r
85         windows=1;\r
86         break;\r
87       default:\r
88         usage();\r
89     }\r
90   }\r
91 }\r
92 \r
93 void parse_mp3(unsigned char *file)\r
94 {\r
95   int bitrates[2][3][15]=\r
96     {{{0, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 352, 384, 416, 448},\r
97       {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384},\r
98       {0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320}},\r
99      {{0, 32, 48, 56, 64, 80, 96, 112, 128, 144, 160, 176, 192, 224, 256},\r
100       {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160},\r
101       {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}}};\r
102   FILE *fic;\r
103   unsigned char *c;\r
104   int lus;\r
105 \r
106   if (debug) fprintf(stderr,"Debug >> parsing mp3 : %s\n",file);\r
107 \r
108   /* read header */\r
109   if ((fic=fopen(file,"r"))==NULL) { fprintf(stderr,"Warning >> can't open file : %s\n",file); return; }\r
110   lus=fread(buffer,1,MAX,fic);\r
111   c=buffer;  \r
112 \r
113   /* try ID3v2 */\r
114   if (buffer[0]=='I' && buffer[1]=='D' && buffer[2]=='3')\r
115   {\r
116     int size;\r
117     int version;\r
118     version=*(buffer+3);\r
119     if (version<2 || version>4)\r
120       fprintf(stderr,"Warning >> ID3 v2.%d not implemented ! trying anyway : %s\n",version,file);\r
121     if (*(buffer+5)!=0)\r
122       fprintf(stderr,"Warning >> specials headers not implemented (%d) ! trying anyway : %s\n",*(buffer+5),file);\r
123     c=buffer+6;\r
124     size=(*c<<24)+(*(c+1)<<16)+(*(c+2)<<8)+(*(c+3));\r
125     if (size>lus) size=lus;\r
126     c+=4;\r
127     if (version==2) while (c<buffer+size-10)\r
128     {\r
129       int size=(*(c+3)<<16)+(*(c+4)<<8)+(*(c+5));\r
130       if (*c==0) break;\r
131       if (strncmp(c,"TT2",3)==0)\r
132       {\r
133         strncpy(title,c+7,size-1);\r
134         title[size-1]='\0';\r
135       }\r
136       if (strncmp(c,"TP1",3)==0)\r
137       {\r
138         strncpy(artist,c+7,size-1);\r
139         artist[size-1]='\0';\r
140       }\r
141       c+=size+6;\r
142     }\r
143     if (version==3 || version==4) while (c<buffer+size-10)\r
144     {\r
145       int size=(*(c+4)<<24)+(*(c+5)<<16)+(*(c+6)<<8)+(*(c+7));\r
146       if (*c==0) break;\r
147       if (strncmp(c,"TIT2",4)==0)\r
148       {\r
149         strncpy(title,c+11,size-1);\r
150         title[size-1]='\0';\r
151       }\r
152       if (strncmp(c,"TPE1",4)==0)\r
153       {\r
154         strncpy(artist,c+11,size-1);\r
155         artist[size-1]='\0';\r
156       }\r
157       c+=size+10;\r
158     }\r
159   }\r
160   \r
161   while (c<buffer+lus-10)\r
162   {\r
163     if (*c==0xFF && (*(c+1)&0xF0)==0xF0)\r
164     {\r
165       int version;\r
166       int lay;\r
167       int bitrate_index;\r
168       int bitrate;\r
169       version=2-(*(c+1)>>3&1);\r
170       lay=4-(*(c+1)>>1&3);\r
171       bitrate_index=*(c+2)>>4&0xF;\r
172       if (version>=1 && version<=2 && lay-1>=0 && lay-1<=2 && bitrate_index>=0 && bitrate_index<=14)\r
173         bitrate=bitrates[version-1][lay-1][bitrate_index];\r
174       else\r
175         bitrate=0;\r
176       if (bitrate!=0)\r
177       {\r
178         fseek(fic,0,SEEK_END);\r
179         duration=(ftell(fic)+buffer-c)/125/bitrate;\r
180       }\r
181       else\r
182         duration=0;\r
183       break;\r
184     }\r
185     c++;\r
186   }\r
187 \r
188   /* try ID3v1 */\r
189   if (strlen(artist)==0 && strlen(title)==0)\r
190   {\r
191     fseek(fic,-128,SEEK_END);\r
192     lus=fread(buffer,1,128,fic);\r
193     if (lus==128 && buffer[0]=='T' && buffer[1]=='A' && buffer[2]=='G')\r
194     {\r
195       strncpy(title,buffer+3,30);\r
196       title[30]='\0';\r
197       c=title+29;\r
198       while (c>title && *c==' ') *(c--)='\0';\r
199       strncpy(artist,buffer+33,30);\r
200       artist[30]='\0';\r
201       c=artist+29;\r
202       while (c>artist && *c==' ') *(c--)='\0';\r
203     }\r
204   }\r
205 \r
206   fclose(fic);\r
207 }  \r
208 \r
209 void parse_ogg(unsigned char *file)\r
210 {\r
211   FILE *fic;\r
212   unsigned char *c;\r
213   int lus;\r
214   int sample_rate;\r
215   int samples;\r
216 \r
217   if (debug) fprintf(stderr,"Debug >> parsing ogg : %s\n",file);\r
218 \r
219   /* read header */\r
220   if ((fic=fopen(file,"r"))==NULL) { fprintf(stderr,"Warning >> can't open file : %s\n",file); return; }\r
221   lus=fread(buffer,1,MAX,fic);\r
222 \r
223   /* try Ogg */\r
224   if (buffer[0]!='O' && buffer[1]!='g' && buffer[2]!='g')\r
225   {\r
226     fprintf(stderr,"Warning >> not a Ogg header : %s\n",file);\r
227     return;\r
228   }\r
229   \r
230   c=buffer+0x28;\r
231   sample_rate=(*c)+(*(c+1)<<8)+(*(c+2)<<16)+(*(c+3)<<24);\r
232 \r
233   while (c<buffer+lus-10)\r
234   {\r
235     int size;\r
236     if (strncmp(c,"TITLE=",6)==0)\r
237     {\r
238       size=*(c-4)+(*(c-3)<<8)+(*(c-2)<<16)+(*(c-1)<<24);\r
239       strncpy(title,c+6,size-6);\r
240       title[size-6]='\0';\r
241       c+=size;\r
242     }\r
243     if (strncmp(c,"ARTIST=",7)==0)\r
244     {\r
245       size=*(c-4)+(*(c-3)<<8)+(*(c-2)<<16)+(*(c-1)<<24);\r
246       strncpy(artist,c+7,size-7);\r
247       artist[size-7]='\0';\r
248       c+=size;\r
249     }\r
250     c++;\r
251   }\r
252   \r
253   fseek(fic,-MAX,SEEK_END);\r
254   lus=fread(buffer,1,MAX,fic);\r
255   c=buffer+lus-1;\r
256   while (strncmp(c,"OggS",4)!=0 && c>buffer) c--;\r
257   if (c!=buffer)\r
258   {\r
259     c+=6;\r
260     samples=(*c)+(*(c+1)<<8)+(*(c+2)<<16)+(*(c+3)<<24);\r
261     duration=samples/sample_rate;\r
262   }\r
263   \r
264   fclose(fic);\r
265 }\r
266 \r
267 void parse_directory(unsigned char *path)\r
268 {\r
269   int i,n;\r
270   struct dirent **namelist;\r
271   unsigned char newpath[PATH_MAX];\r
272   unsigned char *c;\r
273   struct stat infos;\r
274 \r
275   void print_faketitle()\r
276   {\r
277     c=newpath+strlen(newpath);\r
278     while (c>newpath && *c!='/') c--;\r
279     while (c<newpath+strlen(newpath)-5)\r
280     {\r
281       c++;\r
282       if (*c=='_') putchar(' '); else if (windows) putchar(unix2dos[*c]); else putchar(*c);\r
283     }\r
284   }\r
285   \r
286   void print_path()\r
287   {\r
288     printf(prefix);\r
289     for (c=newpath+skip;*c!='\0';c++) if (*c=='/') putchar(separator); else if (windows) putchar(unix2dos[*c]); else putchar(*c);\r
290   }\r
291 \r
292   if (debug) fprintf(stderr,"Debug >> parsing directory : %s\n",path);\r
293   if ((n=scandir(path,&namelist,0,alphasort))<0) { fprintf(stderr,"Warning >> can't open directory : %s\n",path); return; }\r
294   for (i=0;i<n;i++)\r
295   {\r
296     sprintf(newpath,"%s/%s",path,namelist[i]->d_name);\r
297     if (stat(newpath,&infos)!=0) { fprintf(stderr,"Warning >> can't stat file : %s\n",newpath); continue; }\r
298     if (recursive && S_ISDIR(infos.st_mode) && strcmp(namelist[i]->d_name,".")!=0 && strcmp(namelist[i]->d_name,"..")!=0) parse_directory(newpath);\r
299     if (S_ISREG(infos.st_mode))\r
300     {\r
301       unsigned char ext[5];\r
302       int j;\r
303       for (j=0;j<5;j++) ext[j]=tolower(namelist[i]->d_name[strlen(namelist[i]->d_name)-4+j]);\r
304       artist[0]='\0';\r
305       title[0]='\0';\r
306       duration=-2;\r
307       if (strcmp(".mp3",ext)==0) { duration=-1; parse_mp3(newpath); }\r
308       if (strcmp(".ogg",ext)==0) { duration=-1; parse_ogg(newpath); }\r
309       if (strcmp(".wav",ext)==0) { duration=-1; /* parse_wav(newpath); */ }\r
310       \r
311       if (duration!=-2) /* is it an audio file ? */\r
312       {\r
313         counter++;\r
314         switch (format)\r
315         {\r
316           case 0:\r
317             if (duration!=-1)\r
318             {\r
319               printf("#EXTINF:%d,",duration);\r
320               if (strlen(artist)==0 && strlen(title)==0) print_faketitle();\r
321               else printf("%s - %s",artist,title);\r
322               putchar('\n');\r
323             }\r
324             print_path();\r
325             putchar('\n');\r
326             break;\r
327           case 1:\r
328             printf("File%d=",counter);\r
329             print_path();\r
330             putchar('\n');\r
331             printf("Title%d=",counter);\r
332             if (strlen(artist)==0 && strlen(title)==0) print_faketitle();\r
333             else printf("%s - %s",artist,title);\r
334             putchar('\n');\r
335             if (duration!=-1) printf("Length%d=%d\n",counter,duration);\r
336             break;\r
337           case 2:\r
338             printf("<tr><td>%d</td><td>%s</td><td>%s</td><td>",counter,artist,title);\r
339             if (duration==-1) printf("?"); else printf("%d:%s%d",duration/60,duration%60<10?"0":"",duration%60);\r
340             printf("</td></tr>\n");\r
341             break;\r
342         }\r
343       }\r
344     }\r
345     free(namelist[i]);\r
346   }\r
347   free(namelist);\r
348 }\r
349 \r
350 int main(int argc,char **argv)\r
351 {\r
352   parse_options(argc,argv);\r
353   if (optind==argc) usage();\r
354   switch (format)\r
355   {\r
356     case 0:\r
357       printf("#EXTM3U\n");\r
358       break;\r
359     case 1:\r
360       printf("[playlist]\n");\r
361       break;\r
362     case 2:\r
363       printf("<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n\n<html>\n\n<head>\n<title>Playlist generated by FAPG 0.1</title>\n<meta http-equiv=\"Content-Type\" content=\"text/html; charset=iso-8859-1\" />\n<style type=\"text/css\">\n<!--\n\nbody,td,tr {\n  font-family: Verdana, Arial, Helvetica, sans-serif;\n  font-size: 12px;\n  color: #000000;\n}\n\nbody {\n  background: #ffffff;\n}\n\nth {\n  text-align: center;\n  background: #ffcccc;\n  padding-left: 15px;\n  padding-right: 15px;\n  border: 1px #dd8888 solid;\n}\n\ntd {\n  text-align: center;\n  background: #eeeeee;\n  padding-left: 15px;\n  padding-right: 15px;\n  border: 1px #cccccc solid;\n}\n\nh1 {\n  font-size: 25px;\n}\n\np {\n  font-size: 10px;\n}\n\na {\n  color: #993333;\n  text-decoration: none;\n}\n\na:hover {\n  text-decoration: underline;\n}\n\n-->\n</style>\n</head>\n\n<body>\n\n<h1>Playlist</h1>\n\n<table>\n<tr><th>Entry</th><th>Artist</th><th>Title</th><th>Length</th></tr>\n");\r
364       break;\r
365   }\r
366   for (;optind<argc;optind++)\r
367   {\r
368     if (strlen(prefix)!=0) skip=strlen(argv[optind]);\r
369     parse_directory(argv[optind]);\r
370   }\r
371   switch(format)\r
372   {\r
373     case 1:\r
374       printf("NumberOfEntries=%d\nVersion=2\n",counter);\r
375       break;\r
376     case 2:\r
377       printf("</table>\n\n<p>Playlist generated by <a href=\"http://royale.zerezo.com/fapg/\">FAPG 0.1</a></p>\n\n</body>\n\n</html>");\r
378       break;\r
379   }\r
380   exit(0);\r
381 }\r