2 * Copyright (C) 2008 by INdT
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public License
6 * as published by the Free Software Foundation; either version 2
7 * of the License, or (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * @author Andre Moreira Magalhaes <andre.magalhaes@openbossa.org>
31 #include <lightmediascanner_plugin.h>
32 #include <lightmediascanner_db.h>
38 STREAM_TYPE_UNKNOWN = 0,
44 struct lms_string_size title;
45 struct lms_string_size artist;
46 struct lms_string_size album;
47 struct lms_string_size genre;
52 struct lms_plugin plugin;
53 lms_db_audio_t *audio_db;
54 lms_db_video_t *video_db;
57 static const char _name[] = "mp4";
58 static const struct lms_string_size _exts[] = {
59 LMS_STATIC_STRING_SIZE(".mp4"),
60 LMS_STATIC_STRING_SIZE(".m4a"),
61 LMS_STATIC_STRING_SIZE(".mov"),
62 LMS_STATIC_STRING_SIZE(".qt"),
63 LMS_STATIC_STRING_SIZE(".3gp")
67 _match(struct plugin *p, const char *path, int len, int base)
71 i = lms_which_extension(path, len, _exts, LMS_ARRAY_SIZE(_exts));
75 return (void*)(i + 1);
79 _parse(struct plugin *plugin, struct lms_context *ctxt, const struct lms_file_info *finfo, void *match)
81 struct mp4_info info = {{0}, {0}, {0}, {0}};
82 struct lms_audio_info audio_info = {0, {0}, {0}, {0}, {0}, 0, 0, 0};
83 struct lms_video_info video_info = {0, {0}, {0}};
84 int r, stream_type = STREAM_TYPE_AUDIO;
88 mp4_fh = MP4Read(finfo->path, 0);
89 if (mp4_fh == MP4_INVALID_FILE_HANDLE) {
90 fprintf(stderr, "ERROR: cannot read mp4 file %s\n", finfo->path);
94 /* check if the file contains a video track */
95 num_tracks = MP4GetNumberOfTracks(mp4_fh, MP4_VIDEO_TRACK_TYPE, 0);
97 stream_type = STREAM_TYPE_VIDEO;
99 MP4GetMetadataName(mp4_fh, &info.title.str);
101 info.title.len = strlen(info.title.str);
102 MP4GetMetadataArtist(mp4_fh, &info.artist.str);
104 info.artist.len = strlen(info.artist.str);
106 if (stream_type == STREAM_TYPE_AUDIO) {
107 u_int16_t total_tracks;
109 MP4GetMetadataAlbum(mp4_fh, &info.album.str);
111 info.album.len = strlen(info.album.str);
112 MP4GetMetadataGenre(mp4_fh, &info.genre.str);
114 info.genre.len = strlen(info.genre.str);
116 MP4GetMetadataTrack(mp4_fh, &info.trackno, &total_tracks);
119 lms_string_size_strip_and_free(&info.title);
120 lms_string_size_strip_and_free(&info.artist);
121 lms_string_size_strip_and_free(&info.album);
122 lms_string_size_strip_and_free(&info.genre);
124 if (!info.title.str) {
126 ext_idx = ((int)match) - 1;
127 info.title.len = finfo->path_len - finfo->base - _exts[ext_idx].len;
128 info.title.str = malloc((info.title.len + 1) * sizeof(char));
129 memcpy(info.title.str, finfo->path + finfo->base, info.title.len);
130 info.title.str[info.title.len] = '\0';
132 lms_charset_conv(ctxt->cs_conv, &info.title.str, &info.title.len);
135 lms_charset_conv(ctxt->cs_conv, &info.artist.str, &info.artist.len);
137 lms_charset_conv(ctxt->cs_conv, &info.album.str, &info.album.len);
139 lms_charset_conv(ctxt->cs_conv, &info.genre.str, &info.genre.len);
142 fprintf(stderr, "file %s info\n", finfo->path);
143 fprintf(stderr, "\ttitle='%s'\n", info.title.str);
144 fprintf(stderr, "\tartist='%s'\n", info.artist.str);
145 fprintf(stderr, "\talbum='%s'\n", info.album.str);
146 fprintf(stderr, "\tgenre='%s'\n", info.genre.str);
149 if (stream_type == STREAM_TYPE_AUDIO) {
150 audio_info.id = finfo->id;
151 audio_info.title = info.title;
152 audio_info.artist = info.artist;
153 audio_info.album = info.album;
154 audio_info.genre = info.genre;
155 audio_info.trackno = info.trackno;
156 r = lms_db_audio_add(plugin->audio_db, &audio_info);
159 video_info.id = finfo->id;
160 video_info.title = info.title;
161 video_info.artist = info.artist;
162 r = lms_db_video_add(plugin->video_db, &video_info);
168 free(info.title.str);
170 free(info.artist.str);
172 free(info.album.str);
174 free(info.genre.str);
180 _setup(struct plugin *plugin, struct lms_context *ctxt)
182 plugin->audio_db = lms_db_audio_new(ctxt->db);
183 if (!plugin->audio_db)
185 plugin->video_db = lms_db_video_new(ctxt->db);
186 if (!plugin->video_db)
193 _start(struct plugin *plugin, struct lms_context *ctxt)
196 r = lms_db_audio_start(plugin->audio_db);
197 r |= lms_db_video_start(plugin->video_db);
202 _finish(struct plugin *plugin, struct lms_context *ctxt)
204 if (plugin->audio_db)
205 lms_db_audio_free(plugin->audio_db);
206 if (plugin->video_db)
207 lms_db_video_free(plugin->video_db);
213 _close(struct plugin *plugin)
219 API struct lms_plugin *
220 lms_plugin_open(void)
222 struct plugin *plugin;
224 plugin = (struct plugin *)malloc(sizeof(*plugin));
225 plugin->plugin.name = _name;
226 plugin->plugin.match = (lms_plugin_match_fn_t)_match;
227 plugin->plugin.parse = (lms_plugin_parse_fn_t)_parse;
228 plugin->plugin.close = (lms_plugin_close_fn_t)_close;
229 plugin->plugin.setup = (lms_plugin_setup_fn_t)_setup;
230 plugin->plugin.start = (lms_plugin_start_fn_t)_start;
231 plugin->plugin.finish = (lms_plugin_finish_fn_t)_finish;
233 return (struct lms_plugin *)plugin;