9867def383
* Start adding video metadata support using libdlna. We might need to carry libdlna because it needs some changes, but that project appears to be heading in another direction.
516 lines
15 KiB
C
516 lines
15 KiB
C
/* MiniDLNA media server
|
|
* Copyright (C) 2008 Justin Maggard
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
#include <locale.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include <sqlite3.h>
|
|
|
|
#include "upnpglobalvars.h"
|
|
#include "metadata.h"
|
|
#include "sql.h"
|
|
#include "scanner.h"
|
|
|
|
int
|
|
is_video(const char * file)
|
|
{
|
|
return (ends_with(file, ".mpg") || ends_with(file, ".mpeg") ||
|
|
ends_with(file, ".ts") || ends_with(file, ".avi"));
|
|
}
|
|
|
|
int
|
|
is_audio(const char * file)
|
|
{
|
|
return (ends_with(file, ".mp3") ||
|
|
ends_with(file, ".m4a") || ends_with(file, ".aac"));
|
|
}
|
|
|
|
int
|
|
is_image(const char * file)
|
|
{
|
|
return (ends_with(file, ".jpg") || ends_with(file, ".jpeg"));
|
|
}
|
|
|
|
long long int
|
|
insert_container(const char * tmpTable, const char * item, const char * rootParent, const char *subParent, const char *class, const char *artist)
|
|
{
|
|
char **result;
|
|
char *sql;
|
|
int cols, rows, ret;
|
|
char *zErrMsg = NULL;
|
|
int parentID = 0, objectID = 0;
|
|
sqlite_int64 detailID;
|
|
|
|
sql = sqlite3_mprintf("SELECT * from %s where ITEM = '%q' and SUBITEM = '%q'", tmpTable, item, subParent);
|
|
ret = sql_get_table(db, sql, &result, &rows, &cols, &zErrMsg);
|
|
sqlite3_free(sql);
|
|
if( cols )
|
|
{
|
|
sscanf(result[4], "%X", &parentID);
|
|
sqlite3_free_table(result);
|
|
sql = sqlite3_mprintf("SELECT OBJECT_ID, max(ID) from OBJECTS where PARENT_ID = '%s$%X'", rootParent, parentID);
|
|
ret = sql_get_table(db, sql, &result, 0, &cols, &zErrMsg);
|
|
sqlite3_free(sql);
|
|
if( result[2] && (sscanf(rindex(result[2], '$')+1, "%X", &objectID) == 1) )
|
|
{
|
|
objectID++;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
sqlite3_free_table(result);
|
|
sql = sqlite3_mprintf("SELECT OBJECT_ID, max(ID) from OBJECTS where PARENT_ID = '%s'", rootParent);
|
|
sql_get_table(db, sql, &result, &rows, &cols, &zErrMsg);
|
|
sqlite3_free(sql);
|
|
if( result[2] && (sscanf(rindex(result[2], '$')+1, "%X", &parentID) == 1) )
|
|
{
|
|
parentID++;
|
|
}
|
|
else
|
|
{
|
|
parentID = 0;
|
|
}
|
|
detailID = GetFolderMetadata(item, artist);
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, DETAIL_ID, CLASS, NAME) "
|
|
"VALUES"
|
|
" ('%s$%X', '%s', %lld, 'container.%s', '%q')",
|
|
rootParent, parentID, rootParent, detailID, class, item);
|
|
ret = sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
sql = sqlite3_mprintf("INSERT into %s values ('%q', '%X', '%q')", tmpTable, item, parentID, subParent);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
sqlite3_free_table(result);
|
|
|
|
return (long long)parentID<<32|objectID;
|
|
}
|
|
|
|
void
|
|
insert_containers(const char * name, const char *path, const char * refID, const char * class, long unsigned int detailID)
|
|
{
|
|
char sql_buf[128];
|
|
char *sql;
|
|
char **result;
|
|
int ret;
|
|
int cols, row;
|
|
char *zErrMsg = NULL;
|
|
long long int container;
|
|
int parentID;
|
|
int objectID = -1;
|
|
|
|
sprintf(sql_buf, "SELECT * from DETAILS where ID = %lu", detailID);
|
|
ret = sql_get_table(db, sql_buf, &result, &row, &cols, &zErrMsg);
|
|
|
|
if( strstr(class, "imageItem") )
|
|
{
|
|
char *date = result[12+cols], *cam = result[16+cols];
|
|
char date_taken[11]; date_taken[10] = '\0';
|
|
static int last_all_objectID = 0;
|
|
if( date )
|
|
strncpy(date_taken, date, 10);
|
|
if( date )
|
|
{
|
|
container = insert_container("DATES", date_taken, "3$12", NULL, "album.photoAlbum", NULL);
|
|
parentID = container>>32;
|
|
objectID = container;
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('3$12$%X$%X', '3$12$%X', '%s', '%s', %lu, %Q, %Q)",
|
|
parentID, objectID, parentID, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
if( cam && date )
|
|
{
|
|
container = insert_container("CAMS", cam, "3$13", NULL, "storageFolder", NULL);
|
|
parentID = container>>32;
|
|
//objectID = container;
|
|
char parent[64];
|
|
sprintf(parent, "3$13$%X", parentID);
|
|
long long int subcontainer = insert_container("CAMDATE", date_taken, parent, cam, "storageFolder", NULL);
|
|
int subParentID = subcontainer>>32;
|
|
int subObjectID = subcontainer;
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('3$13$%X$%X$%X', '3$13$%X$%X', '%s', '%s', %lu, %Q, %Q)",
|
|
parentID, subParentID, subObjectID, parentID, subParentID, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
/* All Images */
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('3$11$%X', '3$11', '%s', '%s', %lu, %Q, %Q)",
|
|
last_all_objectID++, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
else if( strstr(class, "audioItem") )
|
|
{
|
|
char *artist = result[6+cols], *album = result[7+cols], *genre = result[8+cols];
|
|
static char last_artist[1024];
|
|
static int last_artist_parentID, last_artist_objectID;
|
|
static char last_album[1024];
|
|
static int last_album_parentID, last_album_objectID;
|
|
static char last_genre[1024];
|
|
static int last_genre_parentID, last_genre_objectID;
|
|
static int last_all_objectID = 0;
|
|
|
|
if( artist )
|
|
{
|
|
if( strcmp(artist, last_artist) == 0 )
|
|
{
|
|
objectID = ++last_artist_objectID;
|
|
parentID = last_artist_parentID;
|
|
}
|
|
else
|
|
{
|
|
strcpy(last_artist, artist);
|
|
container = insert_container("ARTISTS", artist, "1$6", NULL, "person.musicArtist", NULL);
|
|
parentID = container>>32;
|
|
objectID = container;
|
|
last_artist_objectID = objectID;
|
|
last_artist_parentID = parentID;
|
|
}
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('1$6$%X$%X', '1$6$%X', '%s', '%s', %lu, %Q, %Q)",
|
|
parentID, objectID, parentID, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
if( album )
|
|
{
|
|
if( strcmp(album, last_album) == 0 )
|
|
{
|
|
objectID = ++last_album_objectID;
|
|
parentID = last_album_parentID;
|
|
}
|
|
else
|
|
{
|
|
strcpy(last_album, album);
|
|
container = insert_container("ALBUMS", album, "1$7", NULL, "album.musicAlbum", artist);
|
|
parentID = container>>32;
|
|
objectID = container;
|
|
last_album_objectID = objectID;
|
|
last_album_parentID = parentID;
|
|
}
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('1$7$%X$%X', '1$7$%X', '%s', '%s', %lu, %Q, %Q)",
|
|
parentID, objectID, parentID, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
if( genre )
|
|
{
|
|
if( strcmp(genre, last_genre) == 0 )
|
|
{
|
|
objectID = ++last_genre_objectID;
|
|
parentID = last_genre_parentID;
|
|
}
|
|
else
|
|
{
|
|
strcpy(last_genre, genre);
|
|
container = insert_container("GENRES", genre, "1$5", NULL, "genre.musicGenre", NULL);
|
|
parentID = container>>32;
|
|
objectID = container;
|
|
last_genre_objectID = objectID;
|
|
last_genre_parentID = parentID;
|
|
}
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('1$5$%X$%X', '1$5$%X', '%s', '%s', %lu, %Q, %Q)",
|
|
parentID, objectID, parentID, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
/* All Music */
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('1$4$%X', '1$4', '%s', '%s', %lu, %Q, %Q)",
|
|
last_all_objectID++, refID, class, detailID, path, name);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
}
|
|
sqlite3_free_table(result);
|
|
}
|
|
|
|
int
|
|
insert_directory(const char * name, const char * path, const char * parentID, int objectID)
|
|
{
|
|
char * sql;
|
|
int ret, i;
|
|
sqlite_int64 detailID;
|
|
char * refID = NULL;
|
|
char class[] = "container.storageFolder";
|
|
const char * const base[] = { BROWSEDIR_ID, MUSIC_DIR_ID, VIDEO_DIR_ID, IMAGE_DIR_ID, 0 };
|
|
|
|
detailID = GetFolderMetadata(name, NULL);
|
|
for( i=0; base[i]; i++ )
|
|
{
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, DETAIL_ID, CLASS, PATH, NAME) "
|
|
"VALUES"
|
|
" ('%s%s$%X', '%s%s', %Q, '%lld', '%s', '%q', '%q')",
|
|
base[i], parentID, objectID, base[i], parentID, refID, detailID, class, path, name);
|
|
//DEBUG printf("SQL: %s\n", sql);
|
|
ret = sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
if( !i )
|
|
asprintf(&refID, "%s%s$%X", base[0], parentID, objectID);
|
|
}
|
|
if( refID )
|
|
free(refID);
|
|
|
|
return -1;
|
|
}
|
|
|
|
int
|
|
insert_file(char * name, const char * path, const char * parentID, int object)
|
|
{
|
|
char *sql;
|
|
char class[32];
|
|
char objectID[64];
|
|
unsigned long int detailID = 0;
|
|
char base[8];
|
|
|
|
static long unsigned int fileno = 0;
|
|
printf("Scanned %lu files...\r", fileno++); fflush(stdout);
|
|
|
|
sprintf(objectID, "%s%s$%X", BROWSEDIR_ID, parentID, object);
|
|
|
|
if( is_image(name) )
|
|
{
|
|
strcpy(base, IMAGE_DIR_ID);
|
|
strcpy(class, "item.imageItem");
|
|
detailID = GetImageMetadata(path, name);
|
|
}
|
|
else if( is_audio(name) )
|
|
{
|
|
strcpy(base, MUSIC_DIR_ID);
|
|
strcpy(class, "item.audioItem.musicTrack");
|
|
detailID = GetAudioMetadata(path, name);
|
|
}
|
|
else if( is_video(name) )
|
|
{
|
|
strcpy(base, VIDEO_DIR_ID);
|
|
strcpy(class, "item.videoItem");
|
|
detailID = GetVideoMetadata(path, name);
|
|
}
|
|
//DEBUG printf("Got DetailID %lu!\n", detailID);
|
|
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('%s', '%s%s', '%s', %lu, '%q', '%q')",
|
|
objectID, BROWSEDIR_ID, parentID, class, detailID, path, name);
|
|
//DEBUG printf("SQL: %s\n", sql);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
#if 0
|
|
sql = sqlite3_mprintf( "INSERT into OBJECTS"
|
|
" (OBJECT_ID, PARENT_ID, REF_ID, CLASS, DETAIL_ID, PATH, NAME) "
|
|
"VALUES"
|
|
" ('%s%s$%X', '%s%s', '%s', '%s', %lu, '%q', '%q')",
|
|
base, parentID, object, base, parentID, objectID, class, detailID, path, name);
|
|
//DEBUG printf("SQL: %s\n", sql);
|
|
sql_exec(db, sql);
|
|
sqlite3_free(sql);
|
|
#else
|
|
insert_containers(name, path, objectID, class, detailID);
|
|
#endif
|
|
return -1;
|
|
}
|
|
|
|
int
|
|
create_database(void)
|
|
{
|
|
int ret, i;
|
|
char sql_buf[512];
|
|
const char * containers[] = { "0","-1", "root",
|
|
"1", "0", "Music",
|
|
"1$4", "1", "All Music",
|
|
"1$5", "1", "Genre",
|
|
"1$6", "1", "Artist",
|
|
"1$7", "1", "Album",
|
|
"1$20", "1", "Folders",
|
|
"2", "0", "Video",
|
|
"2$8", "2", "All Video",
|
|
"2$21", "2", "Folders",
|
|
"3", "0", "Pictures",
|
|
"3$11", "3", "All Pictures",
|
|
"3$12", "3", "Date Taken",
|
|
"3$13", "3", "Camera",
|
|
"3$22", "3", "Folders",
|
|
"64", "0", "Browse Folders",
|
|
0 };
|
|
|
|
sql_exec(db, "pragma temp_store = MEMORY");
|
|
sql_exec(db, "pragma synchronous = OFF;");
|
|
sql_exec(db, "pragma cache_size = 8192;");
|
|
|
|
ret = sql_exec(db, "CREATE TABLE OBJECTS ( "
|
|
"ID INTEGER PRIMARY KEY AUTOINCREMENT, "
|
|
"OBJECT_ID TEXT NOT NULL, "
|
|
"PARENT_ID TEXT NOT NULL, "
|
|
"REF_ID TEXT DEFAULT NULL, "
|
|
"CLASS TEXT NOT NULL, "
|
|
"DETAIL_ID INTEGER DEFAULT NULL, "
|
|
"PATH TEXT DEFAULT NULL, "
|
|
"NAME TEXT DEFAULT NULL"
|
|
");");
|
|
if( ret != SQLITE_OK )
|
|
goto sql_failed;
|
|
ret = sql_exec(db, "CREATE TABLE DETAILS ( "
|
|
"ID INTEGER PRIMARY KEY AUTOINCREMENT, "
|
|
"SIZE INTEGER, "
|
|
"TITLE TEXT, "
|
|
"DURATION TEXT, "
|
|
"BITRATE INTEGER, "
|
|
"SAMPLERATE INTEGER, "
|
|
"ARTIST TEXT, "
|
|
"ALBUM TEXT, "
|
|
"GENRE TEXT, "
|
|
"COMMENT TEXT, "
|
|
"CHANNELS INTEGER, "
|
|
"TRACK INTEGER, "
|
|
"DATE DATE, "
|
|
"RESOLUTION TEXT, "
|
|
"THUMBNAIL BOOL DEFAULT 0, "
|
|
"CREATOR TEXT, "
|
|
"DLNA_PN TEXT, "
|
|
"MIME TEXT"
|
|
");");
|
|
if( ret != SQLITE_OK )
|
|
goto sql_failed;
|
|
for( i=0; containers[i]; i=i+3 )
|
|
{
|
|
sprintf(sql_buf, "INSERT into OBJECTS (OBJECT_ID, PARENT_ID, DETAIL_ID, CLASS, NAME)"
|
|
" values "
|
|
"('%s', '%s', %lld, 'container.storageFolder', '%s')",
|
|
containers[i], containers[i+1], GetFolderMetadata(containers[i+2], NULL), containers[i+2]);
|
|
ret = sql_exec(db, sql_buf);
|
|
if( ret != SQLITE_OK )
|
|
goto sql_failed;
|
|
}
|
|
sql_exec(db, "create TEMP TABLE ARTISTS (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
sql_exec(db, "create TEMP TABLE ALBUMS (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
sql_exec(db, "create TEMP TABLE GENRES (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
sql_exec(db, "create TEMP TABLE DATES (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
sql_exec(db, "create TEMP TABLE CAMS (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
sql_exec(db, "create TEMP TABLE CAMDATE (ITEM TEXT, OBJECT_ID TEXT, SUBITEM TEXT DEFAULT NULL);");
|
|
|
|
sql_exec(db, "create INDEX IDX_OBJECTS_OBJECT_ID ON OBJECTS(OBJECT_ID);");
|
|
sql_exec(db, "create INDEX IDX_OBJECTS_PARENT_ID ON OBJECTS(PARENT_ID);");
|
|
sql_exec(db, "create INDEX IDX_OBJECTS_DETAIL_ID ON OBJECTS(DETAIL_ID);");
|
|
sql_exec(db, "create INDEX IDX_OBJECTS_CLASS ON OBJECTS(CLASS);");
|
|
sql_exec(db, "create INDEX IDX_OBJECTS_PATH ON OBJECTS(PATH);");
|
|
sql_exec(db, "create INDEX IDX_DETAILS_ID ON DETAILS(ID);");
|
|
|
|
|
|
sql_failed:
|
|
if( ret != SQLITE_OK )
|
|
fprintf(stderr, "Error creating SQLite3 database!\n");
|
|
return (ret != SQLITE_OK);
|
|
}
|
|
|
|
int
|
|
filter_media(const struct dirent *d)
|
|
{
|
|
struct stat entry;
|
|
return ( (*d->d_name != '.') &&
|
|
(stat(d->d_name, &entry) == 0) &&
|
|
(S_ISDIR(entry.st_mode) ||
|
|
(S_ISREG(entry.st_mode) &&
|
|
(is_image(d->d_name) ||
|
|
is_audio(d->d_name) ||
|
|
is_video(d->d_name)
|
|
)
|
|
)) );
|
|
}
|
|
|
|
void
|
|
ScanDirectory(const char * dir, const char * parent)
|
|
{
|
|
struct dirent **namelist;
|
|
struct stat entry;
|
|
int n, i;
|
|
char parent_id[PATH_MAX];
|
|
char full_path[PATH_MAX];
|
|
char * name;
|
|
|
|
if( !parent )
|
|
{
|
|
if( create_database() != 0 )
|
|
{
|
|
fprintf(stderr, "Error creating database!\n");
|
|
return;
|
|
}
|
|
}
|
|
|
|
setlocale(LC_COLLATE, "");
|
|
if( chdir(dir) != 0 )
|
|
return;
|
|
printf("\nScanning %s\n", dir);
|
|
n = scandir(".", &namelist, filter_media, alphasort);
|
|
if (n < 0) {
|
|
fprintf(stderr, "Error scanning %s [scandir]\n", dir);
|
|
return;
|
|
}
|
|
for (i=0; i < n; i++) {
|
|
if( stat(namelist[i]->d_name, &entry) == 0 )
|
|
{
|
|
name = NULL;
|
|
sprintf(full_path, "%s/%s", dir, namelist[i]->d_name);
|
|
if( index(namelist[i]->d_name, '&') )
|
|
{
|
|
name = modifyString(strdup(namelist[i]->d_name), "&", "&amp;", 0);
|
|
}
|
|
if( S_ISDIR(entry.st_mode) )
|
|
{
|
|
insert_directory(name?name:namelist[i]->d_name, full_path, (parent ? parent:""), i);
|
|
sprintf(parent_id, "%s$%X", (parent ? parent:""), i);
|
|
ScanDirectory(full_path, parent_id);
|
|
}
|
|
else
|
|
{
|
|
insert_file(name?name:namelist[i]->d_name, full_path, (parent ? parent:""), i);
|
|
}
|
|
}
|
|
if( name )
|
|
free(name);
|
|
free(namelist[i]);
|
|
}
|
|
free(namelist);
|
|
chdir("..");
|
|
}
|