TDE
/
amarok
mirror of https://mirror.git.trinitydesktop.org/gitea/TDE/amarok
0
0
Fork 0
Code Issues Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'd815bc1f86'
${ noResults }
amarok/amarok/src/collectiondb.cpp

8075 lines
290 KiB
Raw Blame History

// (c) 2004 Mark Kretschmann <markey@web.de>
// (c) 2004 Christian Muehlhaeuser <chris@chris.de>
// (c) 2004 Sami Nieminen <sami.nieminen@iki.fi>
// (c) 2005 Ian Monroe <ian@monroe.nu>
// (c) 2005 Jeff Mitchell <kde-dev@emailgoeshere.com>
// (c) 2005 Isaiah Damron <xepo@trifault.net>
// (c) 2005-2006 Alexandre Pereira de Oliveira <aleprj@gmail.com>
// (c) 2006 Jonas Hurrelmann <j@outpo.st>
// (c) 2006 Shane King <kde@dontletsstart.com>
// (c) 2006 Peter C. Ndikuwera <pndiku@gmail.com>
// (c) 2006 Stanislav Nikolov <valsinats@gmail.com>
// See COPYING file for licensing information.
#define DEBUG_PREFIX "CollectionDB"
#include "app.h"
#include "amarok.h"
#include "amarokconfig.h"
#include "config.h"
#include "debug.h"
#include "collectionbrowser.h" //updateTags()
#include "collectiondb.h"
#include "coverfetcher.h"
#include "enginecontroller.h"
#include "expression.h"
#include "mediabrowser.h"
#include "metabundle.h" //updateTags()
#include "mountpointmanager.h" //buildQuery()
#include "organizecollectiondialog.h"
#include "playlist.h"
#include "playlistloader.h"
#include "playlistbrowser.h"
#include "podcastbundle.h" //addPodcast
#include "qstringx.h"
#include "scancontroller.h"
#include "scriptmanager.h"
#include "scrobbler.h"
#include "statusbar.h"
#include "threadmanager.h"
#include <tqbuffer.h>
#include <tqcheckbox.h>
#include <tqdeepcopy.h>
#include <tqfile.h>
#include <tqmap.h>
#include <tqmutex.h>
#include <tqregexp.h> //setHTMLLyrics()
#include <tqtimer.h>
#include <tqpainter.h> //createDragPixmap()
#include <tqpalette.h>
#include <pthread.h> //debugging, can be removed later
#include <kcharsets.h> //setHTMLLyrics()
#include <kcombobox.h>
#include <kconfig.h>
#include <kdialogbase.h> //checkDatabase()
#include <kglobal.h>
#include <kinputdialog.h> //setupCoverFetcher()
#include <klineedit.h> //setupCoverFetcher()
#include <klocale.h>
#include <kmdcodec.h>
#include <kmessagebox.h>
#include <ksimpleconfig.h>
#include <kstandarddirs.h>
#include <kio/job.h>
#include <kio/netaccess.h>
#include <cmath> //DbConnection::sqlite_power()
#include <ctime> //query()
#include <cstdlib> //exit()
#include <unistd.h> //usleep()
#include <taglib/audioproperties.h>
#include "sqlite/sqlite3.h"
#ifdef USE_MYSQL
#include <mysql/mysql.h>
#include <mysql/mysql_version.h>
#endif
#ifdef USE_POSTGRESQL
#include <libpq-fe.h>
#endif
#undef HAVE_INOTIFY // NOTE Disabled for now, due to stability issues
#ifdef HAVE_INOTIFY
#include <linux/inotify.h>
#include "inotify/inotify-syscalls.h"
#endif
using Amarok::QStringx;
#define DEBUG 0
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS INotify
//////////////////////////////////////////////////////////////////////////////////////////
INotify* INotify::s_instance = 0;
INotify::INotify( CollectionDB *parent, int fd )
: DependentJob( parent, "INotify" )
, m_parent( parent )
, m_fd( fd )
{
s_instance = this;
}
INotify::~INotify()
{}
bool
INotify::watchDir( const TQString directory )
{
#ifdef HAVE_INOTIFY
int wd = inotify_add_watch( m_fd, directory.local8Bit(), IN_CLOSE_WRITE | IN_DELETE | IN_MOVE |
IN_MODIFY | IN_ATTRIB );
if ( wd < 0 )
debug() << "Could not add INotify watch for: " << directory << endl;
return ( wd >= 0 );
#else
Q_UNUSED(directory);
#endif
return false;
}
bool
INotify::doJob()
{
#ifdef HAVE_INOTIFY
DEBUG_BLOCK
IdList list = MountPointManager::instance()->getMountedDeviceIds();
TQString deviceIds;
foreachType( IdList, list )
{
if ( !deviceIds.isEmpty() ) deviceIds += ',';
deviceIds += TQString::number(*it);
}
const TQStringList values = m_parent->query( TQString( "SELECT dir, deviceid FROM directories WHERE deviceid IN (%1);" )
.tqarg( deviceIds ) );
foreach( values )
{
TQString rpath = *it;
int deviceid = (*(++it)).toInt();
TQString abspath = MountPointManager::instance()->getAbsolutePath( deviceid, rpath );
watchDir( abspath );
}
/* size of the event structure, not counting name */
const int EVENT_SIZE = ( sizeof( struct inotify_event ) );
/* reasonable guess as to size of 1024 events */
const int BUF_LEN = 1024 * ( EVENT_SIZE + 16 );
while ( 1 )
{
char buf[BUF_LEN];
int len, i = 0;
len = read( m_fd, buf, BUF_LEN );
if ( len < 0 )
{
debug() << "Read from INotify failed" << endl;
return false;
}
else
{
if ( !len )
{
/* BUF_LEN too small? */
}
else
{
while ( i < len )
{
struct inotify_event *event;
event = (struct inotify_event *) &buf[i];
i += EVENT_SIZE + event->len;
}
TQTimer::singleShot( 0, m_parent, TQT_SLOT( scanMonitor() ) );
}
}
}
#endif
// this shouldn't happen
return false;
}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS CollectionDB
//////////////////////////////////////////////////////////////////////////////////////////
TQMutex* CollectionDB::connectionMutex = new TQMutex();
TQMutex* CollectionDB::itemCoverMapMutex = new TQMutex();
//we don't have to worry about this map leaking memory since ThreadManager limits the total
//number of TQThreads ever created
TQMap<TQThread *, DbConnection *> *CollectionDB::threadConnections = new TQMap<TQThread *, DbConnection *>();
TQMap<TQListViewItem*, CoverFetcher*> *CollectionDB::itemCoverMap = new TQMap<TQListViewItem*, CoverFetcher*>();
CollectionDB* CollectionDB::instance()
{
static CollectionDB db;
return &db;
}
CollectionDB::CollectionDB()
: EngineObserver( EngineController::instance() )
, m_autoScoring( true )
, m_noCover( locate( "data", "amarok/images/nocover.png" ) )
, m_shadowImage( locate( "data", "amarok/images/shadow_albumcover.png" ) )
, m_scanInProgress( false )
, m_rescanRequired( false )
, m_aftEnabledPersistentTables()
, m_moveFileJobCancelled( false )
{
DEBUG_BLOCK
#ifdef USE_MYSQL
if ( AmarokConfig::databaseEngine().toInt() == DbConnection::mysql )
m_dbConnType = DbConnection::mysql;
else
#endif
#ifdef USE_POSTGRESQL
if ( AmarokConfig::databaseEngine().toInt() == DbConnection::postgresql )
m_dbConnType = DbConnection::postgresql;
else
#endif
m_dbConnType = DbConnection::sqlite;
//perform all necessary operations to allow MountPointManager to access the devices table here
//there is a recursive dependency between CollectionDB and MountPointManager and this is the workaround
//checkDatabase has to be able to access MountPointManager
//<OPEN DATABASE>
initialize();
//</OPEN DATABASE>
// Remove cached "nocover" images, so that a new version actually gets shown
// The asterisk is for also deleting the shadow-caches.
const TQStringList entryList = cacheCoverDir().entryList( "*nocover.png*", TQDir::Files );
foreach( entryList )
cacheCoverDir().remove( *it );
connect( this, TQT_SIGNAL(fileMoved(const TQString&, const TQString&, const TQString&)),
this, TQT_SLOT(aftMigratePermanentTablesUrl(const TQString&, const TQString&, const TQString&)) );
connect( this, TQT_SIGNAL(uniqueIdChanged(const TQString&, const TQString&, const TQString&)),
this, TQT_SLOT(aftMigratePermanentTablesUniqueId(const TQString&, const TQString&, const TQString&)) );
connect( tqApp, TQT_SIGNAL( aboutToQuit() ), this, TQT_SLOT( disableAutoScoring() ) );
connect( this, TQT_SIGNAL( coverRemoved( const TQString&, const TQString& ) ),
TQT_SIGNAL( coverChanged( const TQString&, const TQString& ) ) );
connect( Scrobbler::instance(), TQT_SIGNAL( similarArtistsFetched( const TQString&, const TQStringList& ) ),
this, TQT_SLOT( similarArtistsFetched( const TQString&, const TQStringList& ) ) );
// If we're supposed to monitor dirs for changes, make sure we run it once
// on startup, since inotify can't inform us about old events
// TQTimer::singleShot( 0, this, TQT_SLOT( scanMonitor() ) )
initDirOperations();
m_aftEnabledPersistentTables << "lyrics" << "statistics" << "tags_labels";
}
CollectionDB::~CollectionDB()
{
DEBUG_BLOCK
#ifdef HAVE_INOTIFY
if ( INotify::instance()->fd() >= 0 )
close( INotify::instance()->fd() );
#endif
destroy();
}
inline TQString
CollectionDB::exactCondition( const TQString &right )
{
if ( DbConnection::mysql == instance()->getDbConnectionType() )
return TQString( "= BINARY '" + instance()->escapeString( right ) + '\'' );
else
return TQString( "= '" + instance()->escapeString( right ) + '\'' );
}
TQString
CollectionDB::likeCondition( const TQString &right, bool anyBegin, bool anyEnd )
{
TQString escaped = right;
escaped.replace( '/', "//" ).replace( '%', "/%" ).replace( '_', "/_" );
escaped = instance()->escapeString( escaped );
TQString ret;
if ( DbConnection::postgresql == instance()->getDbConnectionType() )
ret = " ILIKE "; //case-insensitive according to locale
else
ret = " LIKE ";
ret += '\'';
if ( anyBegin )
ret += '%';
ret += escaped;
if ( anyEnd )
ret += '%';
ret += '\'';
//Use / as the escape character
ret += " ESCAPE '/' ";
return ret;
}
//////////////////////////////////////////////////////////////////////////////////////////
// PUBLIC
//////////////////////////////////////////////////////////////////////////////////////////
void
CollectionDB::initDirOperations()
{
//this code was originally part of the ctor. It has to call MountPointManager to
//generate absolute paths from deviceids and relative paths. MountPointManager's ctor
//absolutely has to access the database, which resulted in a recursive ctor call. To
//solve this problem, the directory access code was moved into its own method, which can
//only be called when the CollectionDB object already exists.
//FIXME max: make sure we check additional directories if we connect a new device
#ifdef HAVE_INOTIFY
// Try to initialize inotify, if not available use the old timer approach.
int inotify_fd = inotify_init();
if ( inotify_fd < 0 )
#endif
{
// debug() << "INotify not available, using TQTimer!" << endl;
startTimer( MONITOR_INTERVAL * 1000 );
}
#ifdef HAVE_INOTIFY
else
{
debug() << "INotify enabled!" << endl;
ThreadManager::instance()->onlyOneJob( new INotify( this, inotify_fd ) );
}
#endif
}
/**
* Executes a SQL query on the already opened database
* @param statement SQL program to execute. Only one SQL statement is allowed.
* @return The queried data, or TQStringList() on error.
*/
TQStringList
CollectionDB::query( const TQString& statement, bool suppressDebug )
{
m_mutex.lock();
clock_t start;
if ( DEBUG )
{
debug() << "Query-start: " << statement << endl;
start = clock();
}
if ( statement.stripWhiteSpace().isEmpty() )
{
m_mutex.unlock();
return TQStringList();
}
DbConnection *dbConn;
dbConn = getMyConnection();
TQStringList values = dbConn->query( statement, suppressDebug );
if ( DEBUG )
{
clock_t finish = clock();
const double duration = (double) (finish - start) / CLOCKS_PER_SEC;
debug() << "SQL-query (" << duration << "s): " << statement << endl;
}
m_mutex.unlock();
return values;
}
/**
* Executes a SQL insert on the already opened database
* @param statement SQL statement to execute. Only one SQL statement is allowed.
* @return The rowid of the inserted item.
*/
int
CollectionDB::insert( const TQString& statement, const TQString& table )
{
m_mutex.lock();
clock_t start;
if ( DEBUG )
{
debug() << "insert-start: " << statement << endl;
start = clock();
}
DbConnection *dbConn;
dbConn = getMyConnection();
int id = dbConn->insert( statement, table );
if ( DEBUG )
{
clock_t finish = clock();
const double duration = (double) (finish - start) / CLOCKS_PER_SEC;
debug() << "SQL-insert (" << duration << "s): " << statement << endl;
}
m_mutex.unlock();
return id;
}
TQString
CollectionDB::deviceidSelection( const bool showAll )
{
if ( !showAll )
{
IdList list = MountPointManager::instance()->getMountedDeviceIds();
TQString deviceIds = "";
foreachType( IdList, list )
{
if ( it != list.begin() ) deviceIds += ',';
deviceIds += TQString::number(*it);
}
return " AND tags.deviceid IN (" + deviceIds + ')';
}
else return "";
}
TQStringList
CollectionDB::URLsFromQuery( const TQStringList &result ) const
{
TQStringList values;
foreach( result )
{
const int id = (*it).toInt();
values << MountPointManager::instance()->getAbsolutePath( id, *(++it) );
}
return values;
}
KURL::List
CollectionDB::URLsFromSqlDrag( const TQStringList &values ) const
{
KURL::List urls;
for( TQStringList::const_iterator it = values.begin();
it != values.end();
it++ )
{
const TQString &rel = *it;
it++;
int id = (*it).toInt();
urls += KURL::fromPathOrURL( MountPointManager::instance()->getAbsolutePath( id, rel ) );
for( int i = 0;
i < QueryBuilder::dragFieldCount-1 && it != values.end();
i++ )
it++;
}
return urls;
}
bool
CollectionDB::isEmpty( )
{
TQStringList values;
values = query( "SELECT COUNT( url ) FROM tags LIMIT 1 OFFSET 0;" );
return values.isEmpty() ? true : values.first() == "0";
}
bool
CollectionDB::isValid( )
{
TQStringList values1;
TQStringList values2;
TQStringList values3;
TQStringList values4;
TQStringList values5;
values1 = query( "SELECT COUNT( url ) FROM tags LIMIT 1 OFFSET 0;" );
values2 = query( "SELECT COUNT( url ) FROM statistics LIMIT 1 OFFSET 0;" );
values3 = query( "SELECT COUNT( url ) FROM podcastchannels LIMIT 1 OFFSET 0;" );
values4 = query( "SELECT COUNT( url ) FROM podcastepisodes LIMIT 1 OFFSET 0;" );
values5 = query( "SELECT COUNT( id ) FROM devices LIMIT 1 OFFSET 0;" );
//It's valid as long as we've got _some_ tables that have something in.
return !( values1.isEmpty() && values2.isEmpty() && values3.isEmpty() && values4.isEmpty() && values5.isEmpty() );
}
TQString
CollectionDB::adminValue( TQString noption ) {
TQStringList values;
values = query (
TQString( "SELECT value FROM admin WHERE noption = '%1';").tqarg(noption)
);
return values.isEmpty() ? "" : values.first();
}
void
CollectionDB::setAdminValue( TQString noption, TQString value ) {
TQStringList values = query( TQString( "SELECT value FROM admin WHERE noption = '%1';").tqarg( noption ));
if(values.count() > 0)
{
query( TQString( "UPDATE admin SET value = '%1' WHERE noption = '%2';" ).tqarg( value, noption ) );
}
else
{
insert( TQString( "INSERT INTO admin (value, noption) values ( '%1', '%2' );" ).tqarg( value, noption ),
NULL );
}
}
void
CollectionDB::createTables( const bool temporary )
{
DEBUG_BLOCK
//create tag table
query( TQString( "CREATE %1 TABLE tags%2 ("
"url " + exactTextColumnType() + ","
"dir " + exactTextColumnType() + ","
"createdate INTEGER,"
"modifydate INTEGER,"
"album INTEGER,"
"artist INTEGER,"
"composer INTEGER,"
"genre INTEGER,"
"title " + textColumnType() + ","
"year INTEGER,"
"comment " + longTextColumnType() + ","
"track NUMERIC(4),"
"discnumber INTEGER,"
"bitrate INTEGER,"
"length INTEGER,"
"samplerate INTEGER,"
"filesize INTEGER,"
"filetype INTEGER,"
"sampler BOOL,"
"bpm FLOAT,"
"deviceid INTEGER);" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" ) );
TQString albumAutoIncrement = "";
TQString artistAutoIncrement = "";
TQString composerAutoIncrement = "";
TQString genreAutoIncrement = "";
TQString yearAutoIncrement = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
if(!temporary)
{
query( TQString( "CREATE SEQUENCE album_seq;" ) );
query( TQString( "CREATE SEQUENCE artist_seq;" ) );
query( TQString( "CREATE SEQUENCE composer_seq;" ) );
query( TQString( "CREATE SEQUENCE genre_seq;" ) );
query( TQString( "CREATE SEQUENCE year_seq;" ) );
}
albumAutoIncrement = TQString("DEFAULT nextval('album_seq')");
artistAutoIncrement = TQString("DEFAULT nextval('artist_seq')");
composerAutoIncrement = TQString("DEFAULT nextval('composer_seq')");
genreAutoIncrement = TQString("DEFAULT nextval('genre_seq')");
yearAutoIncrement = TQString("DEFAULT nextval('year_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
albumAutoIncrement = "AUTO_INCREMENT";
artistAutoIncrement = "AUTO_INCREMENT";
composerAutoIncrement = "AUTO_INCREMENT";
genreAutoIncrement = "AUTO_INCREMENT";
yearAutoIncrement = "AUTO_INCREMENT";
}
//create album table
query( TQString( "CREATE %1 TABLE album%2 ("
"id INTEGER PRIMARY KEY %3,"
"name " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( albumAutoIncrement ) );
//create artist table
query( TQString( "CREATE %1 TABLE artist%2 ("
"id INTEGER PRIMARY KEY %3,"
"name " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( artistAutoIncrement ) );
//create composer table
query( TQString( "CREATE %1 TABLE composer%2 ("
"id INTEGER PRIMARY KEY %3,"
"name " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( composerAutoIncrement ) );
//create genre table
query( TQString( "CREATE %1 TABLE genre%2 ("
"id INTEGER PRIMARY KEY %3,"
"name " + textColumnType() +");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( genreAutoIncrement ) );
//create year table
query( TQString( "CREATE %1 TABLE year%2 ("
"id INTEGER PRIMARY KEY %3,"
"name " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( yearAutoIncrement ) );
//create images table
query( TQString( "CREATE %1 TABLE images%2 ("
"path " + exactTextColumnType() + ","
"deviceid INTEGER,"
"artist " + textColumnType() + ","
"album " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" ) );
//create embed table
query( TQString( "CREATE %1 TABLE embed%2 ("
"url " + exactTextColumnType() + ","
"deviceid INTEGER,"
"hash " + exactTextColumnType() + ","
"description " + textColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" ) );
// create directory statistics table
query( TQString( "CREATE %1 TABLE directories%2 ("
"dir " + exactTextColumnType() + ","
"deviceid INTEGER,"
"changedate INTEGER);" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" ) );
//create uniqueid table
query( TQString( "CREATE %1 TABLE uniqueid%2 ("
"url " + exactTextColumnType() + ","
"deviceid INTEGER,"
"uniqueid " + exactTextColumnType(32) + " UNITQUE,"
"dir " + exactTextColumnType() + ");" )
.tqarg( temporary ? "TEMPORARY" : "" )
.tqarg( temporary ? "_temp" : "" ) );
//create indexes
query( TQString( "CREATE INDEX album_idx%1 ON album%2( name );" )
.tqarg( temporary ? "_temp" : "" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "CREATE INDEX artist_idx%1 ON artist%2( name );" )
.tqarg( temporary ? "_temp" : "" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "CREATE INDEX composer_idx%1 ON composer%2( name );" )
.tqarg( temporary ? "_temp" : "" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "CREATE INDEX genre_idx%1 ON genre%2( name );" )
.tqarg( temporary ? "_temp" : "" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "CREATE INDEX year_idx%1 ON year%2( name );" )
.tqarg( temporary ? "_temp" : "" ).tqarg( temporary ? "_temp" : "" ) );
if ( !temporary )
{
//create admin table -- holds the db version, put here other stuff if necessary
query( TQString( "CREATE TABLE admin ("
"noption " + textColumnType() + ", "
"value " + textColumnType() + ");" ) );
// create related artists cache
query( TQString( "CREATE TABLE related_artists ("
"artist " + textColumnType() + ","
"suggestion " + textColumnType() + ","
"changedate INTEGER );" ) );
createIndices();
}
else
{
query( "CREATE UNITQUE INDEX url_tagtemp ON tags_temp( url, deviceid );" );
query( "CREATE UNITQUE INDEX embed_urltemp ON embed_temp( url, deviceid );" );
query( "CREATE UNITQUE INDEX dir_temp_dir ON directories_temp( dir, deviceid );" );
query( "CREATE INDEX album_tagtemp ON tags_temp( album );" );
query( "CREATE INDEX artist_tagtemp ON tags_temp( artist );" );
query( "CREATE INDEX sampler_tagtemp ON tags_temp( sampler );" );
query( "CREATE INDEX uniqueidtemp_uniqueid ON uniqueid_temp( uniqueid );");
query( "CREATE INDEX uniqueidtemp_url ON uniqueid_temp( url, deviceid );");
}
}
void
CollectionDB::createIndices()
{
//This creates the indices for tables created in createTables. It should not refer to
//tables which are not created in that function.
debug() << "Creating indices, ignore errors about already existing indices" << endl;
query( "CREATE UNITQUE INDEX url_tag ON tags( url, deviceid );" );
query( "CREATE INDEX album_tag ON tags( album );" );
query( "CREATE INDEX artist_tag ON tags( artist );" );
query( "CREATE INDEX composer_tag ON tags( composer );" );
query( "CREATE INDEX genre_tag ON tags( genre );" );
query( "CREATE INDEX year_tag ON tags( year );" );
query( "CREATE INDEX sampler_tag ON tags( sampler );" );
query( "CREATE INDEX images_album ON images( album );" );
query( "CREATE INDEX images_artist ON images( artist );" );
query( "CREATE INDEX images_url ON images( path, deviceid );" );
query( "CREATE UNITQUE INDEX embed_url ON embed( url, deviceid );" );
query( "CREATE INDEX embed_hash ON embed( hash );" );
query( "CREATE UNITQUE INDEX directories_dir ON directories( dir, deviceid );" );
query( "CREATE INDEX uniqueid_uniqueid ON uniqueid( uniqueid );");
query( "CREATE INDEX uniqueid_url ON uniqueid( url, deviceid );");
query( "CREATE INDEX album_idx ON album( name );" );
query( "CREATE INDEX artist_idx ON artist( name );" );
query( "CREATE INDEX composer_idx ON composer( name );" );
query( "CREATE INDEX genre_idx ON genre( name );" );
query( "CREATE INDEX year_idx ON year( name );" );
query( "CREATE INDEX tags_artist_index ON tags( artist );" );
query( "CREATE INDEX tags_album_index ON tags( album );" );
query( "CREATE INDEX tags_deviceid_index ON tags( deviceid ); ");
query( "CREATE INDEX tags_url_index ON tags( url ); ");
query( "CREATE INDEX embed_deviceid_index ON embed( deviceid ); ");
query( "CREATE INDEX embed_url_index ON embed( url ); ");
query( "CREATE INDEX related_artists_artist ON related_artists( artist );" );
debug() << "Finished creating indices, stop ignoring errors" << endl;
}
void
CollectionDB::createPermanentIndices()
{
//this method creates all indices which are not referred to in createTables
//this method is called on each startup of amarok
//until we figure out a way to handle this better it produces SQL errors if the indices
//already exist, but these can be ignored
debug() << "Creating permanent indices, ignore errors about already existing indices" << endl;
query( "CREATE UNITQUE INDEX lyrics_url ON lyrics( url, deviceid );" );
query( "CREATE INDEX lyrics_uniqueid ON lyrics( uniqueid );" );
query( "CREATE INDEX playlist_playlists ON playlists( playlist );" );
query( "CREATE INDEX url_playlists ON playlists( url );" );
query( "CREATE UNITQUE INDEX labels_name ON labels( name, type );" );
query( "CREATE INDEX tags_labels_uniqueid ON tags_labels( uniqueid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
query( "CREATE INDEX tags_labels_url ON tags_labels( url, deviceid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
query( "CREATE INDEX tags_labels_labelid ON tags_labels( labelid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
query( "CREATE UNITQUE INDEX url_stats ON statistics( deviceid, url );" );
query( "CREATE INDEX percentage_stats ON statistics( percentage );" );
query( "CREATE INDEX rating_stats ON statistics( rating );" );
query( "CREATE INDEX playcounter_stats ON statistics( playcounter );" );
query( "CREATE INDEX uniqueid_stats ON statistics( uniqueid );" );
query( "CREATE INDEX url_podchannel ON podcastchannels( url );" );
query( "CREATE INDEX url_podepisode ON podcastepisodes( url );" );
query( "CREATE INDEX localurl_podepisode ON podcastepisodes( localurl );" );
query( "CREATE INDEX url_podfolder ON podcastfolders( id );" );
debug() << "Finished creating permanent indices, stop ignoring errors" << endl;
}
void
CollectionDB::dropTables( const bool temporary )
{
query( TQString( "DROP TABLE tags%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE album%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE artist%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE composer%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE genre%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE year%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE images%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE embed%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE directories%1;" ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "DROP TABLE uniqueid%1;" ).tqarg( temporary ? "_temp" : "" ) );
if ( !temporary )
{
query( TQString( "DROP TABLE related_artists;" ) );
debug() << "Dropping media table" << endl;
}
if ( getDbConnectionType() == DbConnection::postgresql )
{
if (temporary == false) {
query( TQString( "DROP SEQUENCE album_seq;" ) );
query( TQString( "DROP SEQUENCE artist_seq;" ) );
query( TQString( "DROP SEQUENCE composer_seq;" ) );
query( TQString( "DROP SEQUENCE genre_seq;" ) );
query( TQString( "DROP SEQUENCE year_seq;" ) );
}
}
}
void
CollectionDB::clearTables( const bool temporary )
{
TQString clearCommand = "DELETE FROM";
if ( getDbConnectionType() == DbConnection::mysql || getDbConnectionType() == DbConnection::postgresql)
{
// TRUNCATE TABLE is faster than DELETE FROM TABLE, so use it when supported.
clearCommand = "TRUNCATE TABLE";
}
query( TQString( "%1 tags%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 album%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 artist%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 composer%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 genre%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 year%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 images%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 embed%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 directories%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
query( TQString( "%1 uniqueid%2;" ).tqarg( clearCommand ).tqarg( temporary ? "_temp" : "" ) );
if ( !temporary )
{
query( TQString( "%1 related_artists;" ).tqarg( clearCommand ) );
//debug() << "Clearing media table" << endl;
//query( TQString( "%1 media;" ).tqarg( clearCommand ) );
}
}
void
CollectionDB::copyTempTables( )
{
DEBUG_BLOCK
insert( "INSERT INTO tags SELECT * FROM tags_temp;", NULL );
//mysql 5 supports subqueries with IN, mysql 4 doesn't. this way will work for all SQL servers
TQStringList albumIdList = query( "SELECT album.id FROM album;" );
//in an empty database, albumIdList is empty. This would result in a SQL query like NOT IN ( ) without
//the -1 below which is invalid SQL. The auto generated values start at 1 so this is fine
TQString albumIds = "-1";
foreach( albumIdList )
{
albumIds += ',';
albumIds += *it;
}
insert( TQString ( "INSERT INTO album SELECT * FROM album_temp WHERE album_temp.id NOT IN ( %1 );" ).tqarg( albumIds ), NULL );
TQStringList artistIdList = query( "SELECT artist.id FROM artist;" );
TQString artistIds = "-1";
foreach( artistIdList )
{
artistIds += ',';
artistIds += *it;
}
insert( TQString ( "INSERT INTO artist SELECT * FROM artist_temp WHERE artist_temp.id NOT IN ( %1 );" ).tqarg( artistIds ), NULL );
TQStringList composerIdList = query( "SELECT composer.id FROM composer;" );
TQString composerIds = "-1";
foreach( composerIdList )
{
composerIds += ',';
composerIds += *it;
}
insert( TQString ( "INSERT INTO composer SELECT * FROM composer_temp WHERE composer_temp.id NOT IN ( %1 );" ).tqarg( composerIds ), NULL );
TQStringList genreIdList = query( "SELECT genre.id FROM genre;" );
TQString genreIds = "-1";
foreach( genreIdList )
{
genreIds += ',';
genreIds += *it;
}
insert( TQString ( "INSERT INTO genre SELECT * FROM genre_temp WHERE genre_temp.id NOT IN ( %1 );" ).tqarg( genreIds ), NULL );
TQStringList yearIdList = query( "SELECT year.id FROM year;" );
TQString yearIds = "-1";
foreach( yearIdList )
{
yearIds += ',';
yearIds += *it;
}
insert( TQString ( "INSERT INTO year SELECT * FROM year_temp WHERE year_temp.id NOT IN ( %1 );" ).tqarg( yearIds ), NULL );
insert( "INSERT INTO images SELECT * FROM images_temp;", NULL );
insert( "INSERT INTO embed SELECT * FROM embed_temp;", NULL );
insert( "INSERT INTO directories SELECT * FROM directories_temp;", NULL );
insert( "INSERT INTO uniqueid SELECT * FROM uniqueid_temp;", NULL );
}
void
CollectionDB::prepareTempTables()
{
DEBUG_BLOCK
insert( "INSERT INTO album_temp SELECT * from album;", 0 );
insert( "INSERT INTO artist_temp SELECT * from artist;", 0 );
insert( "INSERT INTO composer_temp SELECT * from composer;", 0 );
insert( "INSERT INTO genre_temp SELECT * from genre;", 0 );
insert( "INSERT INTO year_temp SELECT * from year;", 0 );
}
void
CollectionDB::createDevicesTable()
{
debug() << "Creating DEVICES table" << endl;
TQString deviceAutoIncrement = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
query( TQString( "CREATE SEQUENCE devices_seq;" ) );
deviceAutoIncrement = TQString("DEFAULT nextval('devices_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
deviceAutoIncrement = "AUTO_INCREMENT";
}
query( TQString( "CREATE TABLE devices ("
"id INTEGER PRIMARY KEY %1,"
"type " + textColumnType() + ","
"label " + textColumnType() + ","
"lastmountpoint " + textColumnType() + ","
"uuid " + textColumnType() + ","
"servername " + textColumnType() + ","
"sharename " + textColumnType() + ");" )
.tqarg( deviceAutoIncrement ) );
query( "CREATE INDEX devices_type ON devices( type );" );
query( "CREATE INDEX devices_uuid ON devices( uuid );" );
query( "CREATE INDEX devices_rshare ON devices( servername, sharename );" );
}
void
CollectionDB::createStatsTable()
{
// create music statistics database
query( TQString( "CREATE TABLE statistics ("
"url " + exactTextColumnType() + ","
"deviceid INTEGER,"
"createdate INTEGER,"
"accessdate INTEGER,"
"percentage FLOAT,"
"rating INTEGER DEFAULT 0,"
"playcounter INTEGER,"
"uniqueid " + exactTextColumnType(32) + " UNITQUE,"
"deleted BOOL DEFAULT " + boolF() + ","
"PRIMARY KEY(url, deviceid) );" ) );
}
//Old version, used in upgrade code. This should never be changed.
void
CollectionDB::createStatsTableV8()
{
// create music statistics database - old form, for upgrade code.
query( TQString( "CREATE TABLE statistics ("
"url " + textColumnType() + " UNITQUE,"
"createdate INTEGER,"
"accessdate INTEGER,"
"percentage FLOAT,"
"rating INTEGER DEFAULT 0,"
"playcounter INTEGER,"
"uniqueid " + textColumnType(8) + " UNITQUE,"
"deleted BOOL DEFAULT " + boolF() + ");" ) );
query( "CREATE INDEX url_stats ON statistics( url );" );
query( "CREATE INDEX percentage_stats ON statistics( percentage );" );
query( "CREATE INDEX rating_stats ON statistics( rating );" );
query( "CREATE INDEX playcounter_stats ON statistics( playcounter );" );
query( "CREATE INDEX uniqueid_stats ON statistics( uniqueid );" );
}
//Old version, used in upgrade code
void
CollectionDB::createStatsTableV10( bool temp )
{
// create music statistics database
query( TQString( "CREATE %1 TABLE statistics%2 ("
"url " + exactTextColumnType() + ","
"deviceid INTEGER,"
"createdate INTEGER,"
"accessdate INTEGER,"
"percentage FLOAT,"
"rating INTEGER DEFAULT 0,"
"playcounter INTEGER,"
"uniqueid " + exactTextColumnType(32) + " UNITQUE,"
"deleted BOOL DEFAULT " + boolF() + ","
"PRIMARY KEY(url, deviceid) );"
).tqarg( temp ? "TEMPORARY" : "" )
.tqarg( temp ? "_fix_ten" : "" ) );
if ( !temp )
{
query( "CREATE UNITQUE INDEX url_stats ON statistics( deviceid, url );" );
query( "CREATE INDEX percentage_stats ON statistics( percentage );" );
query( "CREATE INDEX rating_stats ON statistics( rating );" );
query( "CREATE INDEX playcounter_stats ON statistics( playcounter );" );
query( "CREATE INDEX uniqueid_stats ON statistics( uniqueid );" );
}
}
void
CollectionDB::dropStatsTable()
{
query( "DROP TABLE statistics;" );
}
void
CollectionDB::dropStatsTableV1()
{
query( "DROP TABLE statistics;" );
}
void
CollectionDB::createPersistentTables()
{
// create amazon table
query( "CREATE TABLE amazon ( "
"asin " + textColumnType(20) + ", "
"locale " + textColumnType(2) + ", "
"filename " + exactTextColumnType(33) + ", "
"refetchdate INTEGER );" );
// create lyrics table
query( TQString( "CREATE TABLE lyrics ("
"url " + exactTextColumnType() + ", "
"deviceid INTEGER,"
"lyrics " + longTextColumnType() + ", "
"uniqueid " + exactTextColumnType(32) + ");" ) );
query( TQString( "CREATE TABLE playlists ("
"playlist " + textColumnType() + ", "
"url " + exactTextColumnType() + ", "
"tracknum INTEGER );" ) );
TQString labelsAutoIncrement = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
query( TQString( "CREATE SEQUENCE labels_seq;" ) );
labelsAutoIncrement = TQString("DEFAULT nextval('labels_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
labelsAutoIncrement = "AUTO_INCREMENT";
}
//create labels tables
query( TQString( "CREATE TABLE labels ("
"id INTEGER PRIMARY KEY " + labelsAutoIncrement + ", "
"name " + textColumnType() + ", "
"type INTEGER);" ) );
query( TQString( "CREATE TABLE tags_labels ("
"deviceid INTEGER,"
"url " + exactTextColumnType() + ", "
"uniqueid " + exactTextColumnType(32) + ", " //m:n relationship, DO NOT MAKE UNITQUE!
"labelid INTEGER REFERENCES labels( id ) ON DELETE CASCADE );" ) );
}
void
CollectionDB::createPersistentTablesV12()
{
// create amazon table
query( "CREATE TABLE amazon ( "
"asin " + textColumnType(20) + ", "
"locale " + textColumnType(2) + ", "
"filename " + textColumnType(33) + ", "
"refetchdate INTEGER );" );
// create lyrics table
query( TQString( "CREATE TABLE lyrics ("
"url " + textColumnType() + ", "
"lyrics " + longTextColumnType() + ");" ) );
// create labels table
query( TQString( "CREATE TABLE label ("
"url " + textColumnType() + ","
"label " + textColumnType() + ");" ) );
query( TQString( "CREATE TABLE playlists ("
"playlist " + textColumnType() + ", "
"url " + textColumnType() + ", "
"tracknum INTEGER );" ) );
query( "CREATE INDEX url_label ON label( url );" );
query( "CREATE INDEX label_label ON label( label );" );
query( "CREATE INDEX playlist_playlists ON playlists( playlist );" );
query( "CREATE INDEX url_playlists ON playlists( url );" );
}
void
CollectionDB::createPersistentTablesV14( bool temp )
{
const TQString a( temp ? "TEMPORARY" : "" );
const TQString b( temp ? "_fix" : "" );
// create amazon table
query( TQString( "CREATE %1 TABLE amazon%2 ( "
"asin " + textColumnType(20) + ", "
"locale " + textColumnType(2) + ", "
"filename " + exactTextColumnType(33) + ", "
"refetchdate INTEGER );" ).tqarg( a,b ) );
// create lyrics table
query( TQString( "CREATE %1 TABLE lyrics%2 ("
"url " + exactTextColumnType() + ", "
"deviceid INTEGER,"
"lyrics " + longTextColumnType() + ");" ).tqarg( a,b ) );
query( TQString( "CREATE %1 TABLE playlists%2 ("
"playlist " + textColumnType() + ", "
"url " + exactTextColumnType() + ", "
"tracknum INTEGER );" ).tqarg( a,b ) );
if ( !temp )
{
query( "CREATE UNITQUE INDEX lyrics_url ON lyrics( url, deviceid );" );
query( "CREATE INDEX playlist_playlists ON playlists( playlist );" );
query( "CREATE INDEX url_playlists ON playlists( url );" );
}
}
void
CollectionDB::createPodcastTables()
{
TQString podcastAutoIncrement = "";
TQString podcastFolderAutoInc = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
query( TQString( "CREATE SEQUENCE podcastepisode_seq;" ) );
query( TQString( "CREATE SEQUENCE podcastfolder_seq;" ) );
podcastAutoIncrement = TQString("DEFAULT nextval('podcastepisode_seq')");
podcastFolderAutoInc = TQString("DEFAULT nextval('podcastfolder_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
podcastAutoIncrement = "AUTO_INCREMENT";
podcastFolderAutoInc = "AUTO_INCREMENT";
}
// create podcast channels table
query( TQString( "CREATE TABLE podcastchannels ("
"url " + exactTextColumnType() + " UNITQUE,"
"title " + textColumnType() + ","
"weblink " + exactTextColumnType() + ","
"image " + exactTextColumnType() + ","
"comment " + longTextColumnType() + ","
"copyright " + textColumnType() + ","
"parent INTEGER,"
"directory " + textColumnType() + ","
"autoscan BOOL, fetchtype INTEGER, "
"autotransfer BOOL, haspurge BOOL, purgecount INTEGER );" ) );
// create podcast episodes table
query( TQString( "CREATE TABLE podcastepisodes ("
"id INTEGER PRIMARY KEY %1, "
"url " + exactTextColumnType() + " UNITQUE,"
"localurl " + exactTextColumnType() + ","
"parent " + exactTextColumnType() + ","
"guid " + exactTextColumnType() + ","
"title " + textColumnType() + ","
"subtitle " + textColumnType() + ","
"composer " + textColumnType() + ","
"comment " + longTextColumnType() + ","
"filetype " + textColumnType() + ","
"createdate " + textColumnType() + ","
"length INTEGER,"
"size INTEGER,"
"isNew BOOL );" )
.tqarg( podcastAutoIncrement ) );
// create podcast folders table
query( TQString( "CREATE TABLE podcastfolders ("
"id INTEGER PRIMARY KEY %1, "
"name " + textColumnType() + ","
"parent INTEGER, isOpen BOOL );" )
.tqarg( podcastFolderAutoInc ) );
query( "CREATE INDEX url_podchannel ON podcastchannels( url );" );
query( "CREATE INDEX url_podepisode ON podcastepisodes( url );" );
query( "CREATE INDEX localurl_podepisode ON podcastepisodes( localurl );" );
query( "CREATE INDEX url_podfolder ON podcastfolders( id );" );
}
void
CollectionDB::createPodcastTablesV2( bool temp )
{
const TQString a( temp ? "TEMPORARY" : "" );
const TQString b( temp ? "_fix" : "" );
TQString podcastAutoIncrement = "";
TQString podcastFolderAutoInc = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
query( TQString( "CREATE SEQUENCE podcastepisode_seq;" ) );
query( TQString( "CREATE SEQUENCE podcastfolder_seq;" ) );
podcastAutoIncrement = TQString("DEFAULT nextval('podcastepisode_seq')");
podcastFolderAutoInc = TQString("DEFAULT nextval('podcastfolder_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
podcastAutoIncrement = "AUTO_INCREMENT";
podcastFolderAutoInc = "AUTO_INCREMENT";
}
// create podcast channels table
query( TQString( "CREATE %1 TABLE podcastchannels%2 ("
"url " + exactTextColumnType() + " UNITQUE,"
"title " + textColumnType() + ","
"weblink " + exactTextColumnType() + ","
"image " + exactTextColumnType() + ","
"comment " + longTextColumnType() + ","
"copyright " + textColumnType() + ","
"parent INTEGER,"
"directory " + textColumnType() + ","
"autoscan BOOL, fetchtype INTEGER, "
"autotransfer BOOL, haspurge BOOL, purgecount INTEGER );" ).tqarg( a,b ) );
// create podcast episodes table
query( TQString( "CREATE %2 TABLE podcastepisodes%3 ("
"id INTEGER PRIMARY KEY %1, "
"url " + exactTextColumnType() + " UNITQUE,"
"localurl " + exactTextColumnType() + ","
"parent " + exactTextColumnType() + ","
"guid " + exactTextColumnType() + ","
"title " + textColumnType() + ","
"subtitle " + textColumnType() + ","
"composer " + textColumnType() + ","
"comment " + longTextColumnType() + ","
"filetype " + textColumnType() + ","
"createdate " + textColumnType() + ","
"length INTEGER,"
"size INTEGER,"
"isNew BOOL );" )
.tqarg( podcastAutoIncrement, a, b ) );
// create podcast folders table
query( TQString( "CREATE %2 TABLE podcastfolders%3 ("
"id INTEGER PRIMARY KEY %1, "
"name " + textColumnType() + ","
"parent INTEGER, isOpen BOOL );" )
.tqarg( podcastFolderAutoInc, a, b ) );
if ( !temp )
{
query( "CREATE INDEX url_podchannel ON podcastchannels( url );" );
query( "CREATE INDEX url_podepisode ON podcastepisodes( url );" );
query( "CREATE INDEX localurl_podepisode ON podcastepisodes( localurl );" );
query( "CREATE INDEX url_podfolder ON podcastfolders( id );" );
}
}
void
CollectionDB::dropPersistentTables()
{
query( "DROP TABLE amazon;" );
query( "DROP TABLE lyrics;" );
query( "DROP TABLE playlists;" );
query( "DROP TABLE tags_labels;" );
query( "DROP TABLE labels;" );
}
void
CollectionDB::dropPersistentTablesV14()
{
query( "DROP TABLE amazon;" );
query( "DROP TABLE lyrics;" );
query( "DROP TABLE label;" );
query( "DROP TABLE playlists;" );
}
void
CollectionDB::dropPodcastTables()
{
query( "DROP TABLE podcastchannels;" );
query( "DROP TABLE podcastepisodes;" );
query( "DROP TABLE podcastfolders;" );
}
void
CollectionDB::dropPodcastTablesV2()
{
query( "DROP TABLE podcastchannels;" );
query( "DROP TABLE podcastepisodes;" );
query( "DROP TABLE podcastfolders;" );
}
void
CollectionDB::dropDevicesTable()
{
query( "DROP TABLE devices;" );
}
uint
CollectionDB::artistID( TQString value, bool autocreate, const bool temporary, bool exact /* = true */ )
{
// lookup cache
if ( m_validArtistCache && m_cacheArtist[(int)temporary] == value )
return m_cacheArtistID[(int)temporary];
uint id;
if ( exact )
id = IDFromExactValue( "artist", value, autocreate, temporary ).toUInt();
else
id = IDFromValue( "artist", value, autocreate, temporary );
// cache values
m_cacheArtist[(int)temporary] = value;
m_cacheArtistID[(int)temporary] = id;
m_validArtistCache = 1;
return id;
}
TQString
CollectionDB::artistValue( uint id )
{
// lookup cache
if ( m_cacheArtistID[0] == id )
return m_cacheArtist[0];
TQString value = valueFromID( "artist", id );
// cache values
m_cacheArtist[0] = value;
m_cacheArtistID[0] = id;
return value;
}
uint
CollectionDB::composerID( TQString value, bool autocreate, const bool temporary, bool exact /* = true */ )
{
// lookup cache
if ( m_validComposerCache && m_cacheComposer[(int)temporary] == value )
return m_cacheComposerID[(int)temporary];
uint id;
if ( exact )
id = IDFromExactValue( "composer", value, autocreate, temporary ).toUInt();
else
id = IDFromValue( "composer", value, autocreate, temporary );
// cache values
m_cacheComposer[(int)temporary] = value;
m_cacheComposerID[(int)temporary] = id;
m_validComposerCache = 1;
return id;
}
TQString
CollectionDB::composerValue( uint id )
{
// lookup cache
if ( m_cacheComposerID[0] == id )
return m_cacheComposer[0];
TQString value = valueFromID( "composer", id );
// cache values
m_cacheComposer[0] = value;
m_cacheComposerID[0] = id;
return value;
}
uint
CollectionDB::albumID( TQString value, bool autocreate, const bool temporary, bool exact /* = true */ )
{
// lookup cache
if ( m_validAlbumCache && m_cacheAlbum[(int)temporary] == value )
return m_cacheAlbumID[(int)temporary];
uint id;
if ( exact )
id = IDFromExactValue( "album", value, autocreate, temporary ).toUInt();
else
id = IDFromValue( "album", value, autocreate, temporary );
// cache values
m_cacheAlbum[(int)temporary] = value;
m_cacheAlbumID[(int)temporary] = id;
m_validAlbumCache = 1;
return id;
}
TQString
CollectionDB::albumValue( uint id )
{
// lookup cache
if ( m_cacheAlbumID[0] == id )
return m_cacheAlbum[0];
TQString value = valueFromID( "album", id );
// cache values
m_cacheAlbum[0] = value;
m_cacheAlbumID[0] = id;
return value;
}
uint
CollectionDB::genreID( TQString value, bool autocreate, const bool temporary, bool exact /* = true */ )
{
return exact ?
IDFromExactValue( "genre", value, autocreate, temporary ).toUInt() :
IDFromValue( "genre", value, autocreate, temporary );
}
TQString
CollectionDB::genreValue( uint id )
{
return valueFromID( "genre", id );
}
uint
CollectionDB::yearID( TQString value, bool autocreate, const bool temporary, bool exact /* = true */ )
{
return exact ?
IDFromExactValue( "year", value, autocreate, temporary ).toUInt() :
IDFromValue( "year", value, autocreate, temporary );
}
TQString
CollectionDB::yearValue( uint id )
{
return valueFromID( "year", id );
}
uint
CollectionDB::IDFromValue( TQString name, TQString value, bool autocreate, const bool temporary )
{
if ( temporary )
name.append( "_temp" );
// what the hell is the reason for this?
// else
// conn = NULL;
TQStringList values =
query( TQString(
"SELECT id, name FROM %1 WHERE name %2;" )
.tqarg( name )
.tqarg( CollectionDB::likeCondition( value ) ) );
//check if item exists. if not, should we autocreate it?
uint id;
if ( values.isEmpty() && autocreate )
{
id = insert( TQString( "INSERT INTO %1 ( name ) VALUES ( '%2' );" )
.tqarg( name )
.tqarg( CollectionDB::instance()->escapeString( value ) ), name );
return id;
}
return values.isEmpty() ? 0 : values.first().toUInt();
}
TQString
CollectionDB::valueFromID( TQString table, uint id )
{
TQStringList values =
query( TQString(
"SELECT name FROM %1 WHERE id=%2;" )
.tqarg( table )
.tqarg( id ) );
return values.isEmpty() ? 0 : values.first();
}
TQString
CollectionDB::albumSongCount( const TQString &artist_id, const TQString &album_id )
{
TQStringList values =
query( TQString(
"SELECT COUNT( url ) FROM tags WHERE album = %1 AND artist = %2;" )
.tqarg( album_id )
.tqarg( artist_id ) );
return values.first();
}
bool
CollectionDB::albumIsCompilation( const TQString &album_id )
{
TQStringList values =
query( TQString(
"SELECT sampler FROM tags WHERE sampler=%1 AND album=%2" )
.tqarg( CollectionDB::instance()->boolT() )
.tqarg( album_id ) );
return (values.count() != 0);
}
TQStringList
CollectionDB::albumTracks( const TQString &artist_id, const TQString &album_id )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valURL );
qb.addMatch( QueryBuilder::tabAlbum, QueryBuilder::valID, album_id );
const bool isCompilation = albumIsCompilation( album_id );
if( !isCompilation )
qb.addMatch( QueryBuilder::tabArtist, QueryBuilder::valID, artist_id );
qb.sortBy( QueryBuilder::tabSong, QueryBuilder::valDiscNumber );
qb.sortBy( QueryBuilder::tabSong, QueryBuilder::valTrack );
TQStringList ret = qb.run();
uint returnValues = qb.countReturnValues();
if ( returnValues > 1 )
{
TQStringList ret2;
for ( TQStringList::size_type i = 0; i < ret.size(); i += returnValues )
ret2 << ret[ i ];
return ret2;
}
else
return ret;
}
TQStringList
CollectionDB::albumDiscTracks( const TQString &artist_id, const TQString &album_id, const TQString &discNumber)
{
TQStringList rs;
rs = query( TQString( "SELECT tags.deviceid, tags.url FROM tags, year WHERE tags.album = %1 AND "
"tags.artist = %2 AND year.id = tags.year AND tags.discnumber = %3 "
+ deviceidSelection() + " ORDER BY tags.track;" )
.tqarg( album_id )
.tqarg( artist_id )
.tqarg( discNumber ) );
TQStringList result;
foreach( rs )
{
const int id = (*it).toInt();
result << MountPointManager::instance()->getAbsolutePath( id, *(++it) );
}
return result;
}
TQStringList
CollectionDB::artistTracks( const TQString &artist_id )
{
TQStringList rs = query( TQString( "SELECT tags.deviceid, tags.url FROM tags, album "
"WHERE tags.artist = '%1' AND album.id = tags.album " + deviceidSelection() +
"ORDER BY album.name, tags.discnumber, tags.track;" )
.tqarg( artist_id ) );
TQStringList result = TQStringList();
foreach( rs )
{
const int id = (*it).toInt();
result << MountPointManager::instance()->getAbsolutePath( id, *(++it) );
}
return result;
}
void
CollectionDB::addImageToAlbum( const TQString& image, TQValueList< TQPair<TQString, TQString> > info, const bool temporary )
{
int deviceid = MountPointManager::instance()->getIdForUrl( image );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, image );
for ( TQValueList< TQPair<TQString, TQString> >::ConstIterator it = info.begin(); it != info.end(); ++it )
{
if ( (*it).first.isEmpty() || (*it).second.isEmpty() )
continue;
TQString sql = TQString( "INSERT INTO images%1 ( path, deviceid, artist, album ) VALUES ( '%3', %2" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) );
sql += TQString( ", '%1'" ).tqarg( escapeString( (*it).first ) );
sql += TQString( ", '%1' );" ).tqarg( escapeString( (*it).second ) );
// debug() << "Added image for album: " << (*it).first << " - " << (*it).second << ": " << image << endl;
insert( sql, NULL );
}
}
void
CollectionDB::addEmbeddedImage( const TQString& path, const TQString& hash, const TQString& description )
{
// debug() << "Added embedded image hash " << hash << " for file " << path << endl;
//TODO: figure out what this embedded table does and then add the necessary code
//what are embedded images anyway?
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath(deviceid, path );
insert( TQString( "INSERT INTO embed_temp ( url, deviceid, hash, description ) VALUES ( '%2', %1, '%3', '%4' );" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ), escapeString( hash ), escapeString( description ) ), NULL );
}
void
CollectionDB::removeOrphanedEmbeddedImages()
{
//TODO refactor
// do it the hard way, since a delete subquery wont work on MySQL
TQStringList orphaned = query( "SELECT embed.deviceid, embed.url FROM embed LEFT JOIN tags ON embed.url = tags.url AND embed.deviceid = tags.deviceid WHERE tags.url IS NULL;" );
foreach( orphaned ) {
TQString deviceid = *it;
TQString rpath = *(++it);
query( TQString( "DELETE FROM embed WHERE embed.deviceid = %1 AND embed.url = '%2';" )
.tqarg( deviceid, escapeString( rpath ) ) );
}
}
TQPixmap
CollectionDB::createDragPixmapFromSQL( const TQString &sql, TQString textOverRide )
{
// it is too slow to check if the url is actually in the colleciton.
//TODO mountpointmanager: figure out what has to be done here
TQStringList values = instance()->query( sql );
KURL::List list;
foreach( values )
{
KURL u = KURL::fromPathOrURL( *it );
if( u.isValid() )
list += u;
}
return createDragPixmap( list, textOverRide );
}
TQPixmap
CollectionDB::createDragPixmap( const KURL::List &urls, TQString textOverRide )
{
// settings
const int maxCovers = 4; // maximum number of cover images to show
const int coverSpacing = 20; // spacing between stacked covers
const int fontSpacing = 5; // spacing between covers and info text
const int coverW = AmarokConfig::coverPreviewSize() > 100 ? 100 : AmarokConfig::coverPreviewSize();
const int coverH = coverW;
const int margin = 2; //px margin
int covers = 0;
int songs = 0;
int pixmapW = 0;
int pixmapH = 0;
int remoteUrls = 0;
int playlists = 0;
TQMap<TQString, int> albumMap;
TQPixmap coverPm[maxCovers];
TQString song, album;
// iterate urls, get covers and count artist/albums
bool correctAlbumCount = true;
KURL::List::ConstIterator it = urls.begin();
for ( ; it != urls.end(); ++it )
{
if( PlaylistFile::isPlaylistFile( *it )
|| (*it).protocol() == "playlist" || (*it).protocol() == "smartplaylist"
|| (*it).protocol() == "dynamic" )
{
playlists++;
}
else if( (*it).isLocalFile() )
{
songs++;
if( covers >= maxCovers )
{
correctAlbumCount = false;
continue;
}
MetaBundle mb( *it );
song = mb.title();
album = mb.album();
TQString artist = mb.artist();
if( mb.compilation() == MetaBundle::CompilationYes )
artist = TQString( "Various_AMAROK_Artists" ); // magic key for the albumMap!
if( !albumMap.contains( artist + album ) )
{
albumMap[ artist + album ] = 1;
TQString coverName = CollectionDB::instance()->albumImage( mb.artist(), album, false, coverW );
if ( !coverName.endsWith( "@nocover.png" ) )
coverPm[covers++].load( coverName );
}
}
else
{
MetaBundle mb( *it );
if( !albumMap.contains( mb.artist() + mb.album() ) )
{
albumMap[ mb.artist() + mb.album() ] = 1;
TQString coverName = CollectionDB::instance()->podcastImage( mb, false, coverW );
if ( covers < maxCovers && !coverName.endsWith( "@nocover.png" ) )
coverPm[covers++].load( coverName );
}
remoteUrls++;
}
}
// make better text...
int albums = albumMap.count();
TQString text;
if( !textOverRide.isEmpty() )
{
text = textOverRide;
}
else if( ( songs && remoteUrls ) ||
( songs && playlists ) ||
( playlists && remoteUrls ) )
{
text = i18n( "One item", "%n items", songs + remoteUrls + playlists );
}
else if( songs > 0 )
{
if( correctAlbumCount ) {
text = i18n( "X songs from X albums", "%2 from %1" );
text = text.tqarg( albums == 1 && !album.isEmpty() ? album : i18n( "one album", "%n albums",albums ) );
}
else
text = "%1";
text = text.tqarg( songs == 1 && !song.isEmpty() ? song : i18n( "One song", "%n songs", songs ) );
}
else if( playlists > 0 )
text = i18n( "One playlist", "%n playlists", playlists );
else if ( remoteUrls > 0 )
text = i18n( "One remote file", "%n remote files", remoteUrls );
else
text = i18n( "Unknown item" );
TQFont font;
TQFontMetrics fm( font );
int fontH = fm.height() + margin;
int minWidth = fm.width( text ) + margin*2; //margin either side
if ( covers > 0 )
{
// insert "..." cover as first image if appropriate
if ( covers < albums )
{
if ( covers < maxCovers ) covers++;
for ( int i = maxCovers-1; i > 0; i-- )
coverPm[i] = coverPm[i-1];
TQImage im( locate( "data","amarok/images/more_albums.png" ) );
coverPm[0].convertFromImage( im.smoothScale( coverW, coverH, TQ_ScaleMin ) );
}
pixmapH = coverPm[0].height();
pixmapW = coverPm[0].width();
// caluclate pixmap height
int dW, dH;
for ( int i = 1; i < covers; i++ )
{
dW = coverPm[i].width() - coverPm[i-1].width() + coverSpacing;
dH = coverPm[i].height() - coverPm[i-1].height() + coverSpacing;
if ( dW > 0 ) pixmapW += dW;
if ( dH > 0 ) pixmapH += dH;
}
pixmapH += fontSpacing + fontH;
if ( pixmapW < minWidth )
pixmapW = minWidth;
}
else
{
pixmapW = minWidth;
pixmapH = fontH;
}
TQPixmap pmdrag( pixmapW, pixmapH );
TQPixmap pmtext( pixmapW, fontH );
TQPalette palette = TQToolTip::palette();
TQPainter p;
p.begin( &pmtext );
p.fillRect( 0, 0, pixmapW, fontH, TQBrush( TQt::black ) ); // border
p.fillRect( 1, 1, pixmapW-margin, fontH-margin, palette.brush( TQPalette::Normal, TQColorGroup::Background ) );
p.setBrush( palette.color( TQPalette::Normal, TQColorGroup::Text ) );
p.setFont( font );
p.drawText( margin, fm.ascent() + 1, text );
p.end();
TQBitmap pmtextMask(pixmapW, fontH);
pmtextMask.fill( TQt::color1 );
// when we have found no covers, just display the text message
if( !covers )
{
pmtext.setMask(pmtextMask);
return pmtext;
}
// compose image
p.begin( &pmdrag );
p.setBackgroundMode( Qt::TransparentMode );
for ( int i = 0; i < covers; i++ )
bitBlt( &pmdrag, i * coverSpacing, i * coverSpacing, &coverPm[i], 0, TQt::CopyROP );
bitBlt( &pmdrag, 0, pixmapH - fontH, &pmtext, 0, TQt::CopyROP );
p.end();
TQBitmap pmdragMask( pmdrag.size(), true );
for ( int i = 0; i < covers; i++ )
{
TQBitmap coverMask( coverPm[i].width(), coverPm[i].height() );
coverMask.fill( TQt::color1 );
bitBlt( &pmdragMask, i * coverSpacing, i * coverSpacing, &coverMask, 0, TQt::CopyROP );
}
bitBlt( &pmdragMask, 0, pixmapH - fontH, &pmtextMask, 0, TQt::CopyROP );
pmdrag.setMask( pmdragMask );
return pmdrag;
}
TQImage
CollectionDB::fetchImage( const KURL& url, TQString &/*tmpFile*/ )
{
if ( url.protocol() != "file" )
{
TQString tmpFile;
KIO::NetAccess::download( url, tmpFile, 0 ); //TODO set 0 to the window, though it probably doesn't really matter
return TQImage( tmpFile );
}
else
{
return TQImage( url.path() );
}
}
bool
CollectionDB::setAlbumImage( const TQString& artist, const TQString& album, const KURL& url )
{
TQString tmpFile;
bool success = setAlbumImage( artist, album, fetchImage(url, tmpFile) );
KIO::NetAccess::removeTempFile( tmpFile ); //only removes file if it was created with NetAccess
return success;
}
bool
CollectionDB::setAlbumImage( const TQString& artist, const TQString& album, TQImage img, const TQString& amazonUrl, const TQString& asin )
{
//show a wait cursor for the duration
Amarok::OverrideCursor keep;
const bool isCompilation = albumIsCompilation( TQString::number( albumID( album, false, false, true ) ) );
const TQString artist_ = isCompilation ? "" : artist;
// remove existing album covers
removeAlbumImage( artist_, album );
TQCString key = md5sum( artist_, album );
newAmazonReloadDate(asin, AmarokConfig::amazonLocale(), key);
// Save Amazon product page URL as embedded string, for later retreival
if ( !amazonUrl.isEmpty() )
img.setText( "amazon-url", 0, amazonUrl );
const bool b = img.save( largeCoverDir().filePath( key ), "PNG");
emit coverChanged( artist_, album );
return b;
}
TQString
CollectionDB::podcastImage( const MetaBundle &bundle, const bool withShadow, uint width )
{
PodcastEpisodeBundle peb;
PodcastChannelBundle pcb;
KURL url = bundle.url().url();
if( getPodcastEpisodeBundle( url, &peb ) )
{
url = peb.parent().url();
}
if( getPodcastChannelBundle( url, &pcb ) )
{
if( pcb.imageURL().isValid() )
return podcastImage( pcb.imageURL().url(), withShadow, width );
}
return notAvailCover( withShadow, width );
}
TQString
CollectionDB::podcastImage( const TQString &remoteURL, const bool withShadow, uint width )
{
// we aren't going to need a 1x1 size image. this is just a quick hack to be able to show full size images.
// width of 0 == full size
if( width == 1 )
width = AmarokConfig::coverPreviewSize();
TQString s = findAmazonImage( "Podcast", remoteURL, width );
if( s.isEmpty() )
{
s = notAvailCover( withShadow, width );
const KURL url = KURL::fromPathOrURL( remoteURL );
if( url.isValid() ) //KIO crashes with invalid URLs
{
KIO::Job *job = KIO::storedGet( url, false, false );
m_podcastImageJobs[job] = remoteURL;
connect( job, TQT_SIGNAL( result( KIO::Job* ) ), TQT_SLOT( podcastImageResult( KIO::Job* ) ) );
}
}
if ( withShadow )
s = makeShadowedImage( s );
return s;
}
void
CollectionDB::podcastImageResult( KIO::Job *gjob )
{
TQString url = m_podcastImageJobs[gjob];
m_podcastImageJobs.remove( gjob );
KIO::StoredTransferJob *job = dynamic_cast<KIO::StoredTransferJob *>( gjob );
if( !job )
{
debug() << "connected to wrong job type" << endl;
return;
}
if( job->error() )
{
debug() << "job finished with error" << endl;
return;
}
if( job->isErrorPage() )
{
debug() << "error page" << endl;
return;
}
TQImage image( job->data() );
if( !image.isNull() )
{
if( url.isEmpty() )
url = job->url().url();
TQCString key = md5sum( "Podcast", url );
if( image.save( largeCoverDir().filePath( key ), "PNG") )
emit imageFetched( url );
}
}
TQString
CollectionDB::albumImage( const TQString &artist, const TQString &album, bool withShadow, uint width, bool* embedded )
{
TQString s;
// we aren't going to need a 1x1 size image. this is just a quick hack to be able to show full size images.
// width of 0 == full size
if( width == 1 )
width = AmarokConfig::coverPreviewSize();
if( embedded )
*embedded = false;
s = findAmazonImage( artist, album, width );
if( s.isEmpty() )
s = findAmazonImage( "", album, width ); // handle compilations
if( s.isEmpty() )
s = findDirectoryImage( artist, album, width );
if( s.isEmpty() )
{
s = findEmbeddedImage( artist, album, width );
if( embedded && !s.isEmpty() )
*embedded = true;
}
if( s.isEmpty() )
s = notAvailCover( withShadow, width );
if ( withShadow )
s = makeShadowedImage( s );
return s;
}
TQString
CollectionDB::albumImage( const uint artist_id, const uint album_id, bool withShadow, uint width, bool* embedded )
{
return albumImage( artistValue( artist_id ), albumValue( album_id ), withShadow, width, embedded );
}
TQString
CollectionDB::albumImage( const MetaBundle &trackInformation, bool withShadow, uint width, bool* embedded )
{
TQString s;
if( width == 1 )
width = AmarokConfig::coverPreviewSize();
TQString album = trackInformation.album();
TQString artist = trackInformation.artist();
// this art is per track, so should check for it first
s = findMetaBundleImage( trackInformation, width );
if( embedded )
*embedded = !s.isEmpty();
if( s.isEmpty() )
s = findAmazonImage( artist, album, width );
if( s.isEmpty() )
s = findAmazonImage( "", album, width ); // handle compilations
if( s.isEmpty() )
s = findDirectoryImage( artist, album, width );
if( s.isEmpty() )
s = notAvailCover( withShadow, width );
if ( withShadow )
s = makeShadowedImage( s );
return s;
}
TQString
CollectionDB::makeShadowedImage( const TQString& albumImage, bool cache )
{
tqApp->lock();
const TQImage original( albumImage, "PNG" );
tqApp->unlock();
if( original.hasAlphaBuffer() )
return albumImage;
const TQFileInfo fileInfo( albumImage );
const uint shadowSize = static_cast<uint>( original.width() / 100.0 * 6.0 );
const TQString cacheFile = fileInfo.fileName() + "@shadow";
if ( !cache && cacheCoverDir().exists( cacheFile ) )
return cacheCoverDir().filePath( cacheFile );
TQImage shadow;
const TQString folder = Amarok::saveLocation( "covershadow-cache/" );
const TQString file = TQString( "shadow_albumcover%1x%2.png" ).tqarg( original.width() + shadowSize ).tqarg( original.height() + shadowSize );
if ( TQFile::exists( folder + file ) ) {
tqApp->lock();
shadow.load( folder + file, "PNG" );
tqApp->unlock();
}
else {
shadow = TQDeepCopy<TQImage>(instance()->m_shadowImage);
shadow = shadow.smoothScale( original.width() + shadowSize, original.height() + shadowSize );
shadow.save( folder + file, "PNG" );
}
TQImage target(shadow);
bitBlt( &target, 0, 0, &original );
if ( cache ) {
target.save( cacheCoverDir().filePath( cacheFile ), "PNG" );
return cacheCoverDir().filePath( cacheFile );
}
target.save( albumImage, "PNG" );
return albumImage;
}
// Amazon Image
TQString
CollectionDB::findAmazonImage( const TQString &artist, const TQString &album, uint width )
{
TQCString widthKey = makeWidthKey( width );
if ( artist.isEmpty() && album.isEmpty() )
return TQString();
TQCString key = md5sum( artist, album );
// check cache for existing cover
if ( cacheCoverDir().exists( widthKey + key ) )
return cacheCoverDir().filePath( widthKey + key );
// we need to create a scaled version of this cover
TQDir imageDir = largeCoverDir();
if ( imageDir.exists( key ) )
{
if ( width > 1 )
{
TQImage img( imageDir.filePath( key ) );
img.smoothScale( width, width, TQ_ScaleMin ).save( cacheCoverDir().filePath( widthKey + key ), "PNG" );
return cacheCoverDir().filePath( widthKey + key );
}
else
return imageDir.filePath( key );
}
return TQString();
}
TQString
CollectionDB::findDirectoryImage( const TQString& artist, const TQString& album, uint width )
{
if ( width == 1 )
width = AmarokConfig::coverPreviewSize();
TQCString widthKey = makeWidthKey( width );
if ( album.isEmpty() )
return TQString();
IdList list = MountPointManager::instance()->getMountedDeviceIds();
TQString deviceIds;
foreachType( IdList, list )
{
if ( !deviceIds.isEmpty() ) deviceIds += ',';
deviceIds += TQString::number(*it);
}
TQStringList rs;
if ( artist == i18n( "Various Artists" ) || artist.isEmpty() )
{
rs = query( TQString(
"SELECT distinct images.deviceid,images.path FROM images, artist, tags "
"WHERE images.artist = artist.name "
"AND artist.id = tags.artist "
"AND tags.sampler = %1 "
"AND images.album %2 "
"AND images.deviceid IN (%3) " )
.tqarg( boolT() )
.tqarg( CollectionDB::likeCondition( album ) )
.tqarg( deviceIds ) );
}
else
{
rs = query( TQString(
"SELECT distinct images.deviceid,images.path FROM images WHERE artist %1 AND album %2 AND deviceid IN (%3) ORDER BY path;" )
.tqarg( CollectionDB::likeCondition( artist ) )
.tqarg( CollectionDB::likeCondition( album ) )
.tqarg( deviceIds ) );
}
TQStringList values = URLsFromQuery( rs );
if ( !values.isEmpty() )
{
TQString image( values.first() );
uint matches = 0;
uint maxmatches = 0;
TQRegExp iTunesArt( "^AlbumArt_.*Large" );
for ( uint i = 0; i < values.count(); i++ )
{
matches = values[i].contains( "front", false ) + values[i].contains( "cover", false ) + values[i].contains( "folder", false ) + values[i].contains( iTunesArt );
if ( matches > maxmatches )
{
image = values[i];
maxmatches = matches;
}
}
TQCString key = md5sum( artist, album, image );
if ( width > 1 )
{
TQString path = cacheCoverDir().filePath( widthKey + key );
if ( !TQFile::exists( path ) )
{
TQImage img( image );
img.smoothScale( width, width, TQ_ScaleMin ).save( path, "PNG" );
}
return path;
}
else //large image
return image;
}
return TQString();
}
TQString
CollectionDB::findEmbeddedImage( const TQString& artist, const TQString& album, uint width )
{
// In the case of multiple embedded images, we arbitrarily choose one from the newest file
// could potentially select multiple images within a file based on description, although a
// lot of tagging software doesn't fill in that field, so we just get whatever the DB
// happens to return for us
TQStringList rs;
if ( artist == i18n("Various Artists") || artist.isEmpty() ) {
// VAs need special handling to not match on artist name but instead check for sampler flag
rs = query( TQString(
"SELECT embed.hash, embed.deviceid, embed.url FROM "
"tags INNER JOIN embed ON tags.url = embed.url "
"INNER JOIN album ON tags.album = album.id "
"WHERE "
"album.name = '%1' "
"AND tags.sampler = %2 "
"ORDER BY modifydate DESC LIMIT 1;" )
.tqarg( escapeString( album ) )
.tqarg( boolT() ) );
} else {
rs = query( TQString(
"SELECT embed.hash, embed.deviceid, embed.url FROM "
"tags INNER JOIN embed ON tags.url = embed.url "
"INNER JOIN artist ON tags.artist = artist.id "
"INNER JOIN album ON tags.album = album.id "
"WHERE "
"artist.name = '%1' "
"AND album.name = '%2' "
"ORDER BY modifydate DESC LIMIT 1;" )
.tqarg( escapeString( artist ) )
.tqarg( escapeString( album ) ) );
}
TQStringList values = TQStringList();
if ( rs.count() == 3 ) {
values += rs.first();
values += MountPointManager::instance()->getAbsolutePath( rs[1].toInt(), rs[2] );
}
if ( values.count() == 2 ) {
TQCString hash = values.first().utf8();
TQString result = loadHashFile( hash, width );
if ( result.isEmpty() ) {
// need to get original from file first
MetaBundle mb( KURL::fromPathOrURL( values.last() ) );
if ( extractEmbeddedImage( mb, hash ) ) {
// try again, as should be possible now
result = loadHashFile( hash, width );
}
}
return result;
}
return TQString();
}
TQString
CollectionDB::findMetaBundleImage( const MetaBundle& trackInformation, uint width )
{
int deviceid = MountPointManager::instance()->getIdForUrl( trackInformation.url() );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, trackInformation.url().path() );
TQStringList values =
query( TQString(
"SELECT embed.hash FROM tags LEFT JOIN embed ON tags.url = embed.url "
" AND tags.deviceid = embed.deviceid WHERE tags.url = '%2' AND tags.deviceid = %1 ORDER BY hash DESC LIMIT 1;" )
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
if ( values.empty() || !values.first().isEmpty() ) {
TQCString hash;
TQString result;
if( !values.empty() ) { // file in collection, so we know the hash
hash = values.first().utf8();
result = loadHashFile( hash, width );
}
if ( result.isEmpty() ) {
// need to get original from file first
if ( extractEmbeddedImage( trackInformation, hash ) ) {
// try again, as should be possible now
result = loadHashFile( hash, width );
}
}
return result;
}
return TQString();
}
TQCString
CollectionDB::makeWidthKey( uint width )
{
return TQString::number( width ).local8Bit() + '@';
}
bool
CollectionDB::removeAlbumImage( const TQString &artist, const TQString &album )
{
DEBUG_BLOCK
TQCString widthKey = "*@";
TQCString key = md5sum( artist, album );
query( "DELETE FROM amazon WHERE filename='" + key + '\'' );
// remove scaled versions of images (and add the asterisk for the shadow-caches)
TQStringList scaledList = cacheCoverDir().entryList( TQString(widthKey + key + '*') );
if ( scaledList.count() > 0 )
for ( uint i = 0; i < scaledList.count(); i++ )
TQFile::remove( cacheCoverDir().filePath( scaledList[ i ] ) );
bool deleted = false;
// remove large, original image
if ( largeCoverDir().exists( key ) && TQFile::remove( largeCoverDir().filePath( key ) ) )
deleted = true;
TQString hardImage = findDirectoryImage( artist, album );
debug() << "hardImage: " << hardImage << endl;
if( !hardImage.isEmpty() )
{
int id = MountPointManager::instance()->getIdForUrl( hardImage );
TQString rpath = MountPointManager::instance()->getRelativePath( id, hardImage );
query( "DELETE FROM images WHERE path='" + escapeString( hardImage ) + "' AND deviceid = " + TQString::number( id ) + ';' );
deleted = true;
}
if ( deleted )
{
emit coverRemoved( artist, album );
return true;
}
return false;
}
bool
CollectionDB::removeAlbumImage( const uint artist_id, const uint album_id )
{
return removeAlbumImage( artistValue( artist_id ), albumValue( album_id ) );
}
TQString
CollectionDB::notAvailCover( const bool withShadow, int width )
{
if ( width <= 1 )
width = AmarokConfig::coverPreviewSize();
TQString widthKey = TQString::number( width ) + '@';
TQString s;
if( cacheCoverDir().exists( widthKey + "nocover.png" ) )
s = cacheCoverDir().filePath( widthKey + "nocover.png" );
else
{
m_noCover.smoothScale( width, width, TQ_ScaleMin ).save( cacheCoverDir().filePath( widthKey + "nocover.png" ), "PNG" );
s = cacheCoverDir().filePath( widthKey + "nocover.png" );
}
if ( withShadow )
s = makeShadowedImage( s );
return s;
}
TQStringList
CollectionDB::artistList( bool withUnknowns, bool withCompilations )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabArtist, QueryBuilder::valName );
if ( !withUnknowns )
qb.excludeMatch( QueryBuilder::tabArtist, i18n( "Unknown" ) );
if ( !withCompilations )
qb.setOptions( QueryBuilder::optNoCompilations );
qb.groupBy( QueryBuilder::tabArtist, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabArtist, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::composerList( bool withUnknowns, bool withCompilations )
{
DEBUG_BLOCK
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabComposer, QueryBuilder::valName );
if ( !withUnknowns )
qb.excludeMatch( QueryBuilder::tabComposer, i18n( "Unknown" ) );
if ( !withCompilations )
qb.setOptions( QueryBuilder::optNoCompilations );
qb.groupBy( QueryBuilder::tabComposer, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabComposer, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::albumList( bool withUnknowns, bool withCompilations )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabAlbum, QueryBuilder::valName );
if ( !withUnknowns )
qb.excludeMatch( QueryBuilder::tabAlbum, i18n( "Unknown" ) );
if ( !withCompilations )
qb.setOptions( QueryBuilder::optNoCompilations );
qb.groupBy( QueryBuilder::tabAlbum, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabAlbum, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::genreList( bool withUnknowns, bool withCompilations )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabGenre, QueryBuilder::valName );
//Only report genres that currently have at least one song
qb.addFilter( QueryBuilder::tabSong, "" );
if ( !withUnknowns )
qb.excludeMatch( QueryBuilder::tabGenre, i18n( "Unknown" ) );
if ( !withCompilations )
qb.setOptions( QueryBuilder::optNoCompilations );
qb.groupBy( QueryBuilder::tabGenre, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabGenre, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::yearList( bool withUnknowns, bool withCompilations )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabYear, QueryBuilder::valName );
if ( !withUnknowns )
qb.excludeMatch( QueryBuilder::tabYear, i18n( "Unknown" ) );
if ( !withCompilations )
qb.setOptions( QueryBuilder::optNoCompilations );
qb.groupBy( QueryBuilder::tabYear, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabYear, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::labelList()
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabLabels, QueryBuilder::valName );
qb.groupBy( QueryBuilder::tabLabels, QueryBuilder::valName );
qb.setOptions( QueryBuilder::optShowAll );
qb.sortBy( QueryBuilder::tabLabels, QueryBuilder::valName );
return qb.run();
}
TQStringList
CollectionDB::albumListOfArtist( const TQString &artist, bool withUnknown, bool withCompilations )
{
if (getDbConnectionType() == DbConnection::postgresql)
{
return query( "SELECT DISTINCT album.name, lower( album.name ) AS __discard FROM tags, album, artist WHERE "
"tags.album = album.id AND tags.artist = artist.id "
"AND lower(artist.name) = lower('" + escapeString( artist ) + "') " +
( withUnknown ? TQString() : "AND album.name <> '' " ) +
( withCompilations ? TQString() : "AND tags.sampler = " + boolF() ) + deviceidSelection() +
" ORDER BY lower( album.name );" );
}
// mysql is case insensitive and lower() is very slow
else if (getDbConnectionType() == DbConnection::mysql)
{
return query( "SELECT DISTINCT album.name FROM tags, album, artist WHERE "
"tags.album = album.id AND tags.artist = artist.id "
"AND artist.name = '" + escapeString( artist ) + "' " +
( withUnknown ? TQString() : "AND album.name <> '' " ) +
( withCompilations ? TQString() : "AND tags.sampler = " + boolF() ) + deviceidSelection() +
" ORDER BY album.name;" );
}
else // sqlite
{
return query( "SELECT DISTINCT album.name FROM tags, album, artist WHERE "
"tags.album = album.id AND tags.artist = artist.id "
"AND lower(artist.name) = lower('" + escapeString( artist ) + "') " +
( withUnknown ? TQString() : "AND album.name <> '' " ) +
( withCompilations ? TQString() : "AND tags.sampler = " + boolF() ) + deviceidSelection() +
" ORDER BY lower( album.name );" );
}
}
TQStringList
CollectionDB::artistAlbumList( bool withUnknown, bool withCompilations )
{
if (getDbConnectionType() == DbConnection::postgresql)
{
return query( "SELECT DISTINCT artist.name, album.name, lower( album.name ) AS __discard FROM tags, album, artist WHERE "
"tags.album = album.id AND tags.artist = artist.id " +
( withUnknown ? TQString() : "AND album.name <> '' AND artist.name <> '' " ) +
( withCompilations ? TQString() : "AND tags.sampler = " + boolF() ) + deviceidSelection() +
" ORDER BY lower( album.name );" );
}
else
{
return query( "SELECT DISTINCT artist.name, album.name FROM tags, album, artist WHERE "
"tags.album = album.id AND tags.artist = artist.id " +
( withUnknown ? TQString() : "AND album.name <> '' AND artist.name <> '' " ) +
( withCompilations ? TQString() : "AND tags.sampler = " + boolF() ) + deviceidSelection() +
" ORDER BY lower( album.name );" );
}
}
bool
CollectionDB::addPodcastChannel( const PodcastChannelBundle &pcb, const bool &replace )
{
TQString command;
if( replace ) {
command = "REPLACE INTO podcastchannels "
"( url, title, weblink, image, comment, copyright, parent, directory"
", autoscan, fetchtype, autotransfer, haspurge, purgecount ) "
"VALUES (";
} else {
command = "INSERT INTO podcastchannels "
"( url, title, weblink, image, comment, copyright, parent, directory"
", autoscan, fetchtype, autotransfer, haspurge, purgecount ) "
"VALUES (";
}
TQString title = pcb.title();
KURL link = pcb.link();
KURL image = pcb.imageURL();
TQString description = pcb.description();
TQString copyright = pcb.copyright();
if( title.isEmpty() )
title = pcb.url().prettyURL();
command += '\'' + escapeString( pcb.url().url() ) + "',";
command += ( title.isEmpty() ? "NULL" : '\'' + escapeString( title ) + '\'' ) + ',';
command += ( link.isEmpty() ? "NULL" : '\'' + escapeString( link.url() ) + '\'' ) + ',';
command += ( image.isEmpty() ? "NULL" : '\'' + escapeString( image.url() ) + '\'' ) + ',';
command += ( description.isEmpty() ? "NULL" : '\'' + escapeString( description ) + '\'' ) + ',';
command += ( copyright.isEmpty() ? "NULL" : '\'' + escapeString( copyright ) + '\'' ) + ',';
command += TQString::number( pcb.parentId() ) + ",'";
command += escapeString( pcb.saveLocation() ) + "',";
command += pcb.autoscan() ? boolT() + ',' : boolF() + ',';
command += TQString::number( pcb.fetchType() ) + ',';
command += pcb.autotransfer() ? boolT() + ',' : boolF() + ',';
command += pcb.hasPurge() ? boolT() + ',' : boolF() + ',';
command += TQString::number( pcb.purgeCount() ) + ");";
//FIXME: currently there's no way to check if an INSERT query failed or not - always return true atm.
// Now it might be possible as insert returns the rowid.
insert( command, NULL );
return true;
}
int
CollectionDB::addPodcastEpisode( const PodcastEpisodeBundle &episode, const int idToUpdate )
{
TQString command;
if( idToUpdate ) {
command = "REPLACE INTO podcastepisodes "
"( id, url, localurl, parent, title, subtitle, composer, comment, filetype, createdate, guid, length, size, isNew ) "
"VALUES (";
} else {
command = "INSERT INTO podcastepisodes "
"( url, localurl, parent, title, subtitle, composer, comment, filetype, createdate, guid, length, size, isNew ) "
"VALUES (";
}
TQString localurl = episode.localUrl().url();
TQString title = episode.title();
TQString subtitle = episode.subtitle();
TQString author = episode.author();
TQString description = episode.description();
TQString type = episode.type();
TQString date = episode.date();
TQString guid = episode.guid();
int duration = episode.duration();
uint size = episode.size();
if( title.isEmpty() )
title = episode.url().prettyURL();
if( idToUpdate )
command += TQString::number( idToUpdate ) + ',';
command += '\'' + escapeString( episode.url().url() ) + "',";
command += ( localurl.isEmpty() ? "NULL" : '\'' + escapeString( localurl ) + '\'' ) + ',';
command += '\'' + escapeString( episode.parent().url()) + "',";
command += ( title.isEmpty() ? "NULL" : '\'' + escapeString( title ) + '\'' ) + ',';
command += ( subtitle.isEmpty() ? "NULL" : '\'' + escapeString( subtitle ) + '\'' ) + ',';
command += ( author.isEmpty() ? "NULL" : '\'' + escapeString( author ) + '\'' ) + ',';
command += ( description.isEmpty() ? "NULL" : '\'' + escapeString( description ) + '\'' ) + ',';
command += ( type.isEmpty() ? "NULL" : '\'' + escapeString( type ) + '\'' ) + ',';
command += ( date.isEmpty() ? "NULL" : '\'' + escapeString( date ) + '\'' ) + ',';
command += ( guid.isEmpty() ? "NULL" : '\'' + escapeString( guid ) + '\'' ) + ',';
command += TQString::number( duration ) + ',';
command += TQString::number( size ) + ',';
command += episode.isNew() ? boolT() + " );" : boolF() + " );";
insert( command, NULL );
if( idToUpdate ) return idToUpdate;
//This is a bit of a hack. We have just inserted an item, so it is going to be the one with the
//highest id. Change this if threaded insertions are used in the future.
TQStringList values = query( TQString("SELECT id FROM podcastepisodes WHERE url='%1' ORDER BY id DESC;")
.tqarg( escapeString( episode.url().url() ) ) );
if( values.isEmpty() ) return -1;
return values[0].toInt();
}
TQValueList<PodcastChannelBundle>
CollectionDB::getPodcastChannels()
{
TQString command = "SELECT url, title, weblink, image, comment, copyright, parent, directory "
", autoscan, fetchtype, autotransfer, haspurge, purgecount FROM podcastchannels;";
TQStringList values = query( command );
TQValueList<PodcastChannelBundle> bundles;
foreach( values )
{
PodcastChannelBundle pcb;
pcb.setURL ( KURL::fromPathOrURL(*it) );
pcb.setTitle ( *++it );
pcb.setLink ( KURL::fromPathOrURL(*++it) );
pcb.setImageURL ( KURL::fromPathOrURL(*++it) );
pcb.setDescription ( *++it );
pcb.setCopyright ( *++it );
pcb.setParentId ( (*++it).toInt() );
pcb.setSaveLocation( *++it );
pcb.setAutoScan ( boolFromSql( *++it ) );
pcb.setFetchType ( (*++it).toInt() );
pcb.setAutoTransfer( boolFromSql( *++it ) );
pcb.setPurge ( boolFromSql( *++it ) );
pcb.setPurgeCount ( (*++it).toInt() );
bundles.append( pcb );
}
return bundles;
}
TQValueList<PodcastEpisodeBundle>
CollectionDB::getPodcastEpisodes( const KURL &parent, bool onlyNew, int limit )
{
TQString command = TQString( "SELECT id, url, localurl, parent, guid, title, subtitle, composer, comment, filetype, createdate, length, size, isNew FROM podcastepisodes WHERE ( parent='%1'" ).tqarg( parent.url() );
if( onlyNew )
command += TQString( " AND isNew='%1'" ).tqarg( boolT() );
command += " ) ORDER BY id";
if( limit != -1 )
command += TQString( " DESC LIMIT %1 OFFSET 0" ).tqarg( limit );
command += ';';
TQStringList values = query( command );
TQValueList<PodcastEpisodeBundle> bundles;
foreach( values )
{
PodcastEpisodeBundle peb;
peb.setDBId ( (*it).toInt() );
peb.setURL ( KURL::fromPathOrURL(*++it) );
if( *++it != "NULL" )
peb.setLocalURL ( KURL::fromPathOrURL(*it) );
peb.setParent ( KURL::fromPathOrURL(*++it) );
peb.setGuid ( *++it );
peb.setTitle ( *++it );
if( *++it != NULL )
peb.setSubtitle( *it );
peb.setAuthor ( *++it );
peb.setDescription ( *++it );
peb.setType ( *++it );
peb.setDate ( *++it );
peb.setDuration ( (*++it).toInt() );
if( *++it == NULL )
peb.setSize ( 0 );
else
peb.setSize ( (*it).toInt() );
peb.setNew ( boolFromSql( *++it ) );
bundles.append( peb );
}
return bundles;
}
PodcastEpisodeBundle
CollectionDB::getPodcastEpisodeById( int id )
{
TQString command = TQString( "SELECT url, localurl, parent, guid, title, subtitle, composer, comment, filetype, createdate, length, size, isNew FROM podcastepisodes WHERE id=%1;").tqarg( id );
TQStringList values = query( command );
PodcastEpisodeBundle peb;
foreach( values )
{
peb.setDBId ( id );
peb.setURL ( KURL::fromPathOrURL(*it) );
if( *++it != "NULL" )
peb.setLocalURL( KURL::fromPathOrURL(*it) );
peb.setParent ( KURL::fromPathOrURL(*++it) );
peb.setGuid ( *++it );
peb.setTitle ( *++it );
peb.setSubtitle ( *++it );
peb.setAuthor ( *++it );
peb.setDescription ( *++it );
peb.setType ( *++it );
peb.setDate ( *++it );
peb.setDuration ( (*++it).toInt() );
if( *++it == NULL )
peb.setSize ( 0 );
else
peb.setSize ( (*it).toInt() );
peb.setNew ( boolFromSql( *++it ) );
}
return peb;
}
bool
CollectionDB::getPodcastEpisodeBundle( const KURL &url, PodcastEpisodeBundle *peb )
{
int id = 0;
if( url.isLocalFile() )
{
TQStringList values =
query( TQString( "SELECT id FROM podcastepisodes WHERE localurl = '%1';" )
.tqarg( escapeString( url.url() ) ) );
if( !values.isEmpty() )
id = values[0].toInt();
}
else
{
TQStringList values =
query( TQString( "SELECT id FROM podcastepisodes WHERE url = '%1';" )
.tqarg( escapeString( url.url() ) ) );
if( !values.isEmpty() )
id = values[0].toInt();
}
if( id )
{
*peb = getPodcastEpisodeById( id );
return true;
}
return false;
}
bool
CollectionDB::getPodcastChannelBundle( const KURL &url, PodcastChannelBundle *pcb )
{
TQStringList values = query( TQString(
"SELECT url, title, weblink, image, comment, copyright, parent, directory "
", autoscan, fetchtype, autotransfer, haspurge, purgecount FROM podcastchannels WHERE url = '%1';"
).tqarg( escapeString( url.url() ) ) );
foreach( values )
{
pcb->setURL ( KURL::fromPathOrURL(*it) );
pcb->setTitle ( *++it );
pcb->setLink ( KURL::fromPathOrURL(*++it) );
if( *++it != "NULL" )
pcb->setImageURL( KURL::fromPathOrURL(*it) );
pcb->setDescription ( *++it );
pcb->setCopyright ( *++it );
pcb->setParentId ( (*++it).toInt() );
pcb->setSaveLocation( *++it );
pcb->setAutoScan ( boolFromSql( *++it ) );
pcb->setFetchType ( (*++it).toInt() );
pcb->setAutoTransfer( boolFromSql( *++it ) );
pcb->setPurge ( boolFromSql( *++it ) );
pcb->setPurgeCount ( (*++it).toInt() );
}
return !values.isEmpty();
}
// return newly created folder id
int
CollectionDB::addPodcastFolder( const TQString &name, const int parent_id, const bool isOpen )
{
TQString command = TQString( "INSERT INTO podcastfolders ( name, parent, isOpen ) VALUES ('" );
command += escapeString( name ) + "',";
command += TQString::number( parent_id ) + ",";
command += isOpen ? boolT() + ");" : boolF() + ");";
insert( command, NULL );
command = TQString( "SELECT id FROM podcastfolders WHERE name = '%1' AND parent = '%2';" )
.tqarg( name, TQString::number(parent_id) );
TQStringList values = query( command );
return values[0].toInt();
}
void
CollectionDB::updatePodcastChannel( const PodcastChannelBundle &b )
{
if( getDbConnectionType() == DbConnection::postgresql )
{
query( QStringx( "UPDATE podcastchannels SET title='%1', weblink='%2', comment='%3', "
"copyright='%4', parent=%5, directory='%6', autoscan=%7, fetchtype=%8, "
"autotransfer=%9, haspurge=%10, purgecount=%11 WHERE url='%12';" )
.args ( TQStringList()
<< escapeString( b.title() )
<< escapeString( b.link().url() )
<< escapeString( b.description() )
<< escapeString( b.copyright() )
<< TQString::number( b.parentId() )
<< escapeString( b.saveLocation() )
<< ( b.autoscan() ? boolT() : boolF() )
<< TQString::number( b.fetchType() )
<< (b.hasPurge() ? boolT() : boolF() )
<< (b.autotransfer() ? boolT() : boolF() )
<< TQString::number( b.purgeCount() )
<< escapeString( b.url().url() )
)
);
}
else {
addPodcastChannel( b, true ); //replace the already existing row
}
}
void
CollectionDB::updatePodcastEpisode( const int id, const PodcastEpisodeBundle &b )
{
if( getDbConnectionType() == DbConnection::postgresql )
{
query( QStringx( "UPDATE podcastepisodes SET url='%1', localurl='%2', parent='%3', title='%4', subtitle='%5', composer='%6', comment='%7', "
"filetype='%8', createdate='%9', guid='%10', length=%11, size=%12, isNew=%13 WHERE id=%14;" )
.args( TQStringList()
<< escapeString( b.url().url() )
<< ( b.localUrl().isValid() ? escapeString( b.localUrl().url() ) : "NULL" )
<< escapeString( b.parent().url() )
<< escapeString( b.title() )
<< escapeString( b.subtitle() )
<< escapeString( b.author() )
<< escapeString( b.description() )
<< escapeString( b.type() )
<< escapeString( b.date() )
<< escapeString( b.guid() )
<< TQString::number( b.duration() )
<< escapeString( TQString::number( b.size() ) )
<< ( b.isNew() ? boolT() : boolF() )
<< TQString::number( id )
)
);
}
else {
addPodcastEpisode( b, id );
}
}
void
CollectionDB::updatePodcastFolder( const int folder_id, const TQString &name, const int parent_id, const bool isOpen )
{
if( getDbConnectionType() == DbConnection::postgresql ) {
query( QStringx( "UPDATE podcastfolders SET name='%1', parent=%2, isOpen=%3 WHERE id=%4;" )
.args( TQStringList()
<< escapeString(name)
<< TQString::number(parent_id)
<< ( isOpen ? boolT() : boolF() )
<< TQString::number(folder_id)
)
);
}
else {
query( QStringx( "REPLACE INTO podcastfolders ( id, name, parent, isOpen ) "
"VALUES ( %1, '%2', %3, %4 );" )
.args( TQStringList()
<< TQString::number(folder_id)
<< escapeString(name)
<< TQString::number(parent_id)
<< ( isOpen ? boolT() : boolF() )
)
);
}
}
void
CollectionDB::removePodcastChannel( const KURL &url )
{
//remove channel
query( TQString( "DELETE FROM podcastchannels WHERE url = '%1';" )
.tqarg( escapeString( url.url() ) ) );
//remove all tqchildren
query( TQString( "DELETE FROM podcastepisodes WHERE parent = '%1';" )
.tqarg( escapeString( url.url() ) ) );
}
/// Try not to delete by url, since some podcast feeds have all the same url
void
CollectionDB::removePodcastEpisode( const int id )
{
if( id < 0 ) return;
query( TQString( "DELETE FROM podcastepisodes WHERE id = '%1';" )
.tqarg( TQString::number(id) ) );
}
void
CollectionDB::removePodcastFolder( const int id )
{
if( id < 0 ) return;
query( TQString("DELETE FROM podcastfolders WHERE id=%1;")
.tqarg( TQString::number(id) ) );
}
bool
CollectionDB::addSong( MetaBundle* bundle, const bool incremental )
{
if ( !TQFileInfo( bundle->url().path() ).isReadable() ) return false;
TQString command = "INSERT INTO tags_temp "
"( url, dir, deviceid, createdate, modifydate, album, artist, composer, genre, year, title, "
"comment, track, discnumber, bpm, sampler, length, bitrate, "
"samplerate, filesize, filetype ) "
"VALUES ('";
TQString artist = bundle->artist();
TQString title = bundle->title();
if ( title.isEmpty() )
{
title = bundle->url().fileName();
if ( bundle->url().fileName().find( '-' ) > 0 )
{
if ( artist.isEmpty() )
{
artist = TQString(bundle->url().fileName().section( '-', 0, 0 )).stripWhiteSpace();
bundle->setArtist( artist );
}
title = TQString(bundle->url().fileName().section( '-', 1 )).stripWhiteSpace();
title = title.left( title.findRev( '.' ) ).stripWhiteSpace();
if ( title.isEmpty() ) title = bundle->url().fileName();
}
bundle->setTitle( title );
}
int deviceId = MountPointManager::instance()->getIdForUrl( bundle->url() );
KURL relativePath;
MountPointManager::instance()->getRelativePath( deviceId, bundle->url(), relativePath );
//debug() << "File has deviceId " << deviceId << ", relative path " << relativePath.path() << ", absolute path " << bundle->url().path() << endl;
command += escapeString( relativePath.path() ) + "','";
command += escapeString( relativePath.directory() ) + "',";
command += TQString::number( deviceId ) + ',';
command += TQString::number( TQFileInfo( bundle->url().path() ).created().toTime_t() ) + ',';
command += TQString::number( TQFileInfo( bundle->url().path() ).lastModified().toTime_t() ) + ',';
command += escapeString( TQString::number( albumID( bundle->album(), true, !incremental, true ) ) ) + ',';
command += escapeString( TQString::number( artistID( bundle->artist(), true, !incremental, true ) ) ) + ',';
command += escapeString( TQString::number( composerID( bundle->composer(), true, !incremental, true ) ) ) + ',';
command += escapeString( TQString::number( genreID( bundle->genre(), true, !incremental, true ) ) ) + ",'";
command += escapeString( TQString::number( yearID( TQString::number( bundle->year() ), true, !incremental, true ) ) ) + "','";
command += escapeString( bundle->title() ) + "','";
command += escapeString( bundle->comment() ) + "', ";
command += escapeString( TQString::number( bundle->track() ) ) + " , ";
command += escapeString( TQString::number( bundle->discNumber() ) ) + " , ";
command += escapeString( TQString::number( bundle->bpm() ) ) + " , ";
switch( bundle->compilation() ) {
case MetaBundle::CompilationNo:
command += boolF();
break;
case MetaBundle::CompilationYes:
command += boolT();
break;
case MetaBundle::CompilationUnknown:
default:
command += "NULL";
}
command += ',';
// NOTE any of these may be -1 or -2, this is what we want
// see MetaBundle::Undetermined
command += TQString::number( bundle->length() ) + ',';
command += TQString::number( bundle->bitrate() ) + ',';
command += TQString::number( bundle->sampleRate() ) + ',';
command += TQString::number( bundle->filesize() ) + ',';
command += TQString::number( bundle->fileType() ) + ')';
//FIXME: currently there's no way to check if an INSERT query failed or not - always return true atm.
// Now it might be possible as insert returns the rowid.
insert( command, NULL );
doAFTStuff( bundle, true );
return true;
}
void
CollectionDB::doAFTStuff( MetaBundle* bundle, const bool tempTables )
{
if( bundle->uniqueId().isEmpty() || bundle->url().path().isEmpty() )
return;
MountPointManager *mpm = MountPointManager::instance();
//const to make sure one isn't later modified without the other being changed
const int deviceIdInt = mpm->getIdForUrl( bundle->url().path() );
const TQString currdeviceid = TQString::number( deviceIdInt );
TQString currid = escapeString( bundle->uniqueId() );
TQString currurl = escapeString( mpm->getRelativePath( deviceIdInt, bundle->url().path() ) );
TQString currdir = escapeString( mpm->getRelativePath( deviceIdInt, bundle->url().directory() ) );
//debug() << "Checking currid = " << currid << ", currdir = " << currdir << ", currurl = " << currurl << endl;
//debug() << "tempTables = " << (tempTables?"true":"false") << endl;
TQStringList urls = query( TQString(
"SELECT url, uniqueid "
"FROM uniqueid%1 "
"WHERE deviceid = %2 AND url = '%3';" )
.tqarg( tempTables ? "_temp" : ""
, currdeviceid
, currurl ) );
TQStringList uniqueids = query( TQString(
"SELECT url, uniqueid, deviceid "
"FROM uniqueid%1 "
"WHERE uniqueid = '%2';" )
.tqarg( tempTables ? "_temp" : ""
, currid ) );
TQStringList nonTempIDs = query( TQString(
"SELECT url, uniqueid, deviceid "
"FROM uniqueid "
"WHERE uniqueid = '%1';" )
.tqarg( currid ) );
TQStringList nonTempURLs = query( TQString(
"SELECT url, uniqueid "
"FROM uniqueid "
"WHERE deviceid = %1 AND url = '%2';" )
.tqarg( currdeviceid
, currurl ) );
bool tempTablesAndInPermanent = false;
bool permanentFullMatch = false;
//if we're not using temp tables here, i.e. tempTables is false,
//then the results from both sets of queries above should be equal,
//so behavior should be the same
if( tempTables && ( nonTempURLs.count() > 0 || nonTempIDs.count() > 0 ) )
tempTablesAndInPermanent = true;
if( tempTablesAndInPermanent && nonTempURLs.count() > 0 && nonTempIDs.count() > 0 )
permanentFullMatch = true;
//debug() << "tempTablesAndInPermanent = " << (tempTablesAndInPermanent?"true":"false") << endl;
//debug() << "permanentFullMatch = " << (permanentFullMatch?"true":"false") << endl;
//debug() << "Entering checks" << endl;
//first case: not in permanent table or temporary table
if( !tempTablesAndInPermanent && urls.empty() && uniqueids.empty() )
{
//debug() << "first case" << endl;
TQString insertline = QStringx( "INSERT INTO uniqueid%1 (deviceid, url, uniqueid, dir) "
"VALUES ( %2,'%3', '%4', '%5');" )
.args( TQStringList()
<< ( tempTables ? "_temp" : "" )
<< currdeviceid
<< currurl
<< currid
<< currdir );
insert( insertline, NULL );
//debug() << "aftCheckPermanentTables #1" << endl;
aftCheckPermanentTables( currdeviceid, currid, currurl );
return;
}
//next case: not in permanent table, but a match on one or the other in the temporary table
//OR, we are using permanent tables (and not considering temp ones)
if( !tempTablesAndInPermanent )
{
if( urls.empty() ) //uniqueid already found in temporary table but not url; check the old URL
{
//stat the original URL
TQString absPath = mpm->getAbsolutePath( uniqueids[2].toInt(), uniqueids[0] );
//debug() << "At doAFTStuff, stat-ing file " << absPath << endl;
bool statSuccessful = false;
bool pathsSame = absPath == bundle->url().path();
if( !pathsSame )
statSuccessful = TQFile::exists( absPath );
if( statSuccessful ) //if true, new one is a copy
warning() << "Already-scanned file at " << absPath << " has same UID as new file at " << bundle->url().path() << endl;
else //it's a move, not a copy, or a copy and then both files were moved...can't detect that
{
//debug() << "stat was NOT successful, updating tables with: " << endl;
//debug() << TQString( "UPDATE uniqueid%1 SET url='%2', dir='%3' WHERE uniqueid='%4';" ).tqarg( ( tempTables ? "_temp" : "" ), currurl, currdir, currid ) << endl;
query( QStringx( "UPDATE uniqueid%1 SET deviceid = %2, url='%3', dir='%4' WHERE uniqueid='%5';" )
.args( TQStringList()
<< ( tempTables ? "_temp" : "" )
<< currdeviceid
<< currurl
<< currdir
<< currid ) );
if( !pathsSame )
emit fileMoved( absPath, bundle->url().path(), bundle->uniqueId() );
}
}
//okay then, url already found in temporary table but different uniqueid
//a file exists in the same place as before, but new uniqueid...assume
//that this is desired user behavior
//NOTE: this should never happen during an incremental scan with temporary tables...!
else if( uniqueids.empty() )
{
//debug() << "file exists in same place as before, new uniqueid" << endl;
query( TQString( "UPDATE uniqueid%1 SET uniqueid='%2' WHERE deviceid = %3 AND url='%4';" )
.tqarg( tempTables ? "_temp" : ""
, currid
, currdeviceid
, currurl ) );
emit uniqueIdChanged( bundle->url().path(), urls[1], bundle->uniqueId() );
}
//else uniqueid and url match; nothing happened, so safely exit
return;
}
//okay...being here means, we are using temporary tables, AND it exists in the permanent table
else
{
//first case...full match exists in permanent table, should then be no match in temp table
//(since code below deleted from permanent table after changes)
//in this case, just insert into temp table
if( permanentFullMatch )
{
TQString insertline = TQString( "INSERT INTO uniqueid_temp (deviceid, url, uniqueid, dir) "
"VALUES ( %1, '%2'" )
.tqarg( currdeviceid
, currurl );
insertline += TQString( ", '%1', '%2');" ).tqarg( currid ).tqarg( currdir );
//debug() << "running command: " << insertline << endl;
insert( insertline, NULL );
//debug() << "aftCheckPermanentTables #2" << endl;
aftCheckPermanentTables( currdeviceid, currid, currurl );
return;
}
//second case...full match exists in permanent table, but path is different
if( nonTempURLs.empty() )
{
//stat the original URL
TQString absPath = mpm->getAbsolutePath( nonTempIDs[2].toInt(), nonTempIDs[0] );
//debug() << "At doAFTStuff part 2, stat-ing file " << absPath << endl;
bool statSuccessful = false;
bool pathsSame = absPath == bundle->url().path();
if( !pathsSame )
statSuccessful = TQFile::exists( absPath );
if( statSuccessful ) //if true, new one is a copy
warning() << "Already-scanned file at " << absPath << " has same UID as new file at " << currurl << endl;
else //it's a move, not a copy, or a copy and then both files were moved...can't detect that
{
//debug() << "stat part 2 was NOT successful, updating tables with: " << endl;
query( TQString( "DELETE FROM uniqueid WHERE uniqueid='%1';" )
.tqarg( currid ) );
query( TQString( "INSERT INTO uniqueid_temp (deviceid, url, uniqueid, dir) "
"VALUES ( %1, '%2', '%3', '%4')" )
.tqarg( currdeviceid
, currurl
, currid
, currdir ) );
if( !pathsSame )
emit fileMoved( absPath, bundle->url().path(), bundle->uniqueId() );
}
}
else if( nonTempIDs.empty() )
{
//debug() << "file exists in same place as before, part 2, new uniqueid" << endl;
query( TQString( "DELETE FROM uniqueid WHERE deviceid = %1 AND url='%2';" )
.tqarg( currdeviceid )
.tqarg( currurl ) );
query( TQString( "INSERT INTO uniqueid_temp (deviceid, url, uniqueid, dir) VALUES ( %1, '%2', '%3', '%4')" )
.tqarg( currdeviceid
, currurl
, currid
, currdir ) );
emit uniqueIdChanged( bundle->url().path(), nonTempURLs[1], bundle->uniqueId() );
}
//else do nothing...really this case should never happen
return;
}
}
void
CollectionDB::emitFileDeleted( const TQString &absPath, const TQString &uniqueid )
{
if( uniqueid.isEmpty() )
emit fileDeleted( absPath );
else
emit fileDeleted( absPath, uniqueid );
}
void
CollectionDB::emitFileAdded( const TQString &absPath, const TQString &uniqueid )
{
if( uniqueid.isEmpty() )
emit fileAdded( absPath );
else
emit fileAdded( absPath, uniqueid );
}
TQString
CollectionDB::urlFromUniqueId( const TQString &id )
{
bool scanning = ( ScanController::instance() && ScanController::instance()->tablesCreated() );
TQStringList urls = query( TQString(
"SELECT deviceid, url "
"FROM uniqueid%1 "
"WHERE uniqueid = '%2';" )
.tqarg( scanning ? "_temp" : TQString() )
.tqarg( id ), true );
if( urls.empty() && scanning )
urls = query( TQString(
"SELECT deviceid, url "
"FROM uniqueid "
"WHERE uniqueid = '%1';" )
.tqarg( id ) );
if( urls.empty() )
return TQString();
return MountPointManager::instance()->getAbsolutePath( urls[0].toInt(), urls[1] );
}
TQString
CollectionDB::uniqueIdFromUrl( const KURL &url )
{
MountPointManager *mpm = MountPointManager::instance();
int currdeviceid = mpm->getIdForUrl( url.path() );
TQString currurl = escapeString( mpm->getRelativePath( currdeviceid, url.path() ) );
bool scanning = ( ScanController::instance() && ScanController::instance()->tablesCreated() );
TQStringList uid = query( TQString(
"SELECT uniqueid "
"FROM uniqueid%1 "
"WHERE deviceid = %2 AND url = '%3';" )
.tqarg( scanning ? "_temp" : TQString() )
.tqarg( currdeviceid )
.tqarg( currurl ), true );
if( uid.empty() && scanning )
uid = query( TQString(
"SELECT uniqueid "
"FROM uniqueid "
"WHERE deviceid = %1 AND url = '%2';" )
.tqarg( currdeviceid )
.tqarg( currurl ) );
if( uid.empty() )
return TQString();
return uid[0];
}
TQString
CollectionDB::getURL( const MetaBundle &bundle )
{
uint artID = artistID( bundle.artist(), false );
if( !artID )
return TQString();
uint albID = albumID( bundle.album(), false );
if( !albID )
return TQString();
TQString q = TQString( "SELECT tags.deviceid, tags.url "
"FROM tags "
"WHERE tags.album = '%1' AND tags.artist = '%2' AND tags.track = '%3' AND tags.title = '%4'" +
deviceidSelection() + ';' )
.tqarg( albID )
.tqarg( artID )
.tqarg( bundle.track() )
.tqarg( escapeString( bundle.title() ) );
TQStringList urls = URLsFromQuery( query( q ) );
if( urls.empty() )
return TQString();
if( urls.size() == 1 )
{
return urls.first();
}
TQString url = urls.first();
int maxPlayed = -1;
for( TQStringList::iterator it = urls.begin();
it != urls.end();
it++ )
{
int pc = getPlayCount( *it );
if( pc > maxPlayed )
{
maxPlayed = pc;
url = *it;
}
}
return url;
}
// Helper function to convert the "tags.sampler" column to a MetaBundle::Collection value
//
// We use the first char of boolT / boolF as not all DBs store true/false as
// numerics (and it's only a single-char column)
static int
samplerToCompilation( const TQString &it )
{
if( it == CollectionDB::instance()->boolT().mid( 0, 1 ) )
{
return MetaBundle::CompilationYes;
}
else if( it == CollectionDB::instance()->boolF().mid( 0, 1 ) )
{
return MetaBundle::CompilationNo;
}
return MetaBundle::CompilationUnknown;
}
MetaBundle
CollectionDB::bundleFromQuery( TQStringList::const_iterator *iter )
{
TQStringList::const_iterator &it = *iter;
MetaBundle b;
//QueryBuilder automatically inserts the deviceid as return value if asked for the path
TQString rpath = *it;
int deviceid = (*++it).toInt();
b.setPath ( MountPointManager::instance()->getAbsolutePath( deviceid, rpath ) );
b.setAlbum ( *++it );
b.setArtist ( *++it );
b.setComposer ( *++it );
b.setGenre ( *++it );
b.setTitle ( *++it );
b.setYear ( (*++it).toInt() );
b.setComment ( *++it );
b.setTrack ( (*++it).toInt() );
b.setBitrate ( (*++it).toInt() );
b.setDiscNumber( (*++it).toInt() );
b.setLength ( (*++it).toInt() );
b.setSampleRate( (*++it).toInt() );
b.setFilesize ( (*++it).toInt() );
b.setCompilation( samplerToCompilation( *it ) );
++it;
b.setFileType( (*++it).toInt() );
b.setBpm ( (*++it).toFloat() );
b.setScore ( (*++it).toFloat() );
b.setRating ( (*++it).toInt() );
b.setPlayCount ( (*++it).toInt() );
b.setLastPlay ( (*++it).toInt() );
if( false && b.length() <= 0 ) {
// we try to read the tags, despite the slow-down
debug() << "Audioproperties not known for: " << b.url().fileName() << endl;
b.readTags( TagLib::AudioProperties::Fast);
}
return b;
}
static void
fillInBundle( TQStringList values, MetaBundle &bundle )
{
//TODO use this whenever possible
// crash prevention
while( values.count() < 16 )
values += "IF YOU CAN SEE THIS THERE IS A BUG!";
TQStringList::ConstIterator it = values.begin();
bundle.setAlbum ( *it ); ++it;
bundle.setArtist ( *it ); ++it;
bundle.setComposer ( *it ); ++it;
bundle.setGenre ( *it ); ++it;
bundle.setTitle ( *it ); ++it;
bundle.setYear ( (*it).toInt() ); ++it;
bundle.setComment ( *it ); ++it;
bundle.setDiscNumber( (*it).toInt() ); ++it;
bundle.setTrack ( (*it).toInt() ); ++it;
bundle.setBitrate ( (*it).toInt() ); ++it;
bundle.setLength ( (*it).toInt() ); ++it;
bundle.setSampleRate( (*it).toInt() ); ++it;
bundle.setFilesize ( (*it).toInt() ); ++it;
bundle.setFileType ( (*it).toInt() ); ++it;
bundle.setBpm ( (*it).toFloat() ); ++it;
bundle.setCompilation( samplerToCompilation( *it ) );
++it;
bundle.setUniqueId(*it);
}
bool
CollectionDB::bundleForUrl( MetaBundle* bundle )
{
int deviceid = MountPointManager::instance()->getIdForUrl( bundle->url() );
KURL rpath;
MountPointManager::instance()->getRelativePath( deviceid, bundle->url(), rpath );
TQStringList values = query( TQString(
"SELECT album.name, artist.name, composer.name, genre.name, tags.title, "
"year.name, tags.comment, tags.discnumber, "
"tags.track, tags.bitrate, tags.length, tags.samplerate, "
"tags.filesize, tags.filetype, tags.bpm, tags.sampler, uniqueid.uniqueid "
"FROM tags LEFT OUTER JOIN uniqueid ON tags.url = uniqueid.url AND tags.deviceid = uniqueid.deviceid,"
"album, artist, composer, genre, year "
"WHERE album.id = tags.album AND artist.id = tags.artist AND composer.id = tags.composer AND "
"genre.id = tags.genre AND year.id = tags.year AND tags.url = '%2' AND tags.deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath.path( ) ) ) );
bool valid = false;
if ( !values.empty() )
{
fillInBundle( values, *bundle );
valid = true;
}
else if( MediaBrowser::instance() && MediaBrowser::instance()->getBundle( bundle->url(), bundle ) )
{
valid = true;
}
else
{
// check if it's a podcast
PodcastEpisodeBundle peb;
if( getPodcastEpisodeBundle( bundle->url(), &peb ) )
{
if( bundle->url().protocol() == "file" && TQFile::exists( bundle->url().path() ) )
{
MetaBundle mb( bundle->url(), true /* avoid infinite recursion */ );
*bundle = mb;
}
bundle->copyFrom( peb );
valid = true;
}
}
return valid;
}
TQValueList<MetaBundle>
CollectionDB::bundlesByUrls( const KURL::List& urls )
{
BundleList bundles;
TQStringList paths;
QueryBuilder qb;
for( KURL::List::ConstIterator it = urls.begin(), end = urls.end(), last = urls.fromLast(); it != end; ++it )
{
// non file stuff won't exist in the db, but we still need to
// re-insert it into the list we return, just with no tags assigned
paths += (*it).protocol() == "file" ? (*it).path() : (*it).url();
if( paths.count() == 50 || it == last )
{
qb.clear();
qb.addReturnValue( QueryBuilder::tabAlbum, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabArtist, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabComposer, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabGenre, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTitle );
qb.addReturnValue( QueryBuilder::tabYear, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valComment );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTrack );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valBitrate );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valDiscNumber );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valLength );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valSamplerate );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFilesize );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFileType );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valBPM );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valURL );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valIsCompilation );
qb.addURLFilters( paths );
qb.setOptions( QueryBuilder::optRemoveDuplicates );
const TQStringList values = qb.run();
BundleList buns50;
MetaBundle b;
foreach( values )
{
b.setAlbum ( *it );
b.setArtist ( *++it );
b.setComposer ( *++it );
b.setGenre ( *++it );
b.setTitle ( *++it );
b.setYear ( (*++it).toInt() );
b.setComment ( *++it );
b.setTrack ( (*++it).toInt() );
b.setBitrate ( (*++it).toInt() );
b.setDiscNumber( (*++it).toInt() );
b.setLength ( (*++it).toInt() );
b.setSampleRate( (*++it).toInt() );
b.setFilesize ( (*++it).toInt() );
b.setFileType ( (*++it).toInt() );
b.setBpm ( (*++it).toFloat() );
b.setPath ( *++it );
b.setCompilation( samplerToCompilation( *it ) );
++it;
b.checkExists();
buns50.append( b );
}
// we get no guarantee about the order that the database
// will return our values, and sqlite indeed doesn't return
// them in the desired order :( (MySQL does though)
foreach( paths )
{
for( BundleList::Iterator jt = buns50.begin(), end = buns50.end(); jt != end; ++jt )
{
if ( ( *jt ).url().path() == ( *it ) )
{
bundles += *jt;
buns50.remove( jt );
goto success;
}
}
// if we get here, we didn't find an entry
{
KURL url = KURL::fromPathOrURL( *it );
if( !MediaBrowser::instance()->getBundle( url, &b ) )
{
if( url.isLocalFile() )
{
b = MetaBundle( url );
}
else
{
b = MetaBundle();
b.setUrl( url );
// FIXME: more context for i18n after string freeze
b.setTitle( TQString( "%1 %2 %3%4" )
.tqarg( url.filename(),
i18n( "from" ),
url.hasHost() ? url.host() : TQString(),
url.directory( false ) ) );
}
// check if it's a podcast
PodcastEpisodeBundle peb;
if( getPodcastEpisodeBundle( url, &peb ) )
{
b.copyFrom( peb );
}
else if( b.url().protocol() == "audiocd" || b.url().protocol() == "cdda" )
{
// try to see if the engine has some info about the
// item (the intended behaviour should be that if the
// item is an AudioCD track, the engine can return
// CDDB data for it)
Engine::SimpleMetaBundle smb;
if ( EngineController::engine()->metaDataForUrl( b.url(), smb ) )
{
b.setTitle( smb.title );
b.setArtist( smb.artist );
b.setAlbum( smb.album );
b.setComment( smb.comment );
b.setGenre( smb.genre );
b.setBitrate( smb.bitrate.toInt() );
b.setSampleRate( smb.samplerate.toInt() );
b.setLength( smb.length.toInt() );
b.setYear( smb.year.toInt() );
b.setTrack( smb.tracknr.toInt() );
}
}
}
}
bundles += b;
success: ;
}
paths.clear();
}
}
return bundles;
}
void
CollectionDB::addAudioproperties( const MetaBundle& bundle )
{
int deviceid = MountPointManager::instance()->getIdForUrl( bundle.url() );
KURL rpath;
MountPointManager::instance()->getRelativePath( deviceid, bundle.url(), rpath );
query( TQString( "UPDATE tags SET bitrate='%1', length='%2', samplerate='%3' WHERE url='%5' AND deviceid = %4;" )
.tqarg( bundle.bitrate() )
.tqarg( bundle.length() )
.tqarg( bundle.sampleRate() )
.tqarg( deviceid )
.tqarg( escapeString( rpath.path() ) ) );
}
void
CollectionDB::addSongPercentage( const TQString &url, float percentage,
const TQString &reason, const TQDateTime *playtime )
{
//the URL must always be inserted last! an escaped URL can contain Strings like %1->bug
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
//statistics table might not have those values, but we need them later, so keep them
int statDevId = deviceid;
TQString statRPath = rpath;
TQStringList values =
query( TQString(
"SELECT playcounter, createdate, percentage, rating FROM statistics "
"WHERE url = '%2' AND deviceid = %1;" )
.tqarg( statDevId ).tqarg( escapeString( statRPath ) ) );
//handle corner case: deviceid!=-1 but there is a statistics row for that song with deviceid -1
if ( values.isEmpty() )
{
TQString rpath2 = '.' + url;
values = query( TQString(
"SELECT playcounter, createdate, percentage, rating FROM statistics "
"WHERE url = '%1' AND deviceid = -1;" )
.tqarg( escapeString( rpath2 ) ) );
if ( !values.isEmpty() )
{
statRPath = rpath2;
statDevId = -1;
}
}
uint atime = playtime ? playtime->toTime_t() : TQDateTime::tqcurrentDateTime().toTime_t();
// check boundaries
if ( percentage > 100.f ) percentage = 100.f;
if ( percentage < 1.f ) percentage = 1.f;
if ( !values.isEmpty() )
{
// increment playcounter and update accesstime
query( TQString( "UPDATE statistics SET playcounter=%1, accessdate=%2 WHERE url='%4' AND deviceid= %3;" )
.tqarg( values[0] + " + 1" )
.tqarg( atime )
.tqarg( statDevId )
.tqarg( escapeString( statRPath ) ) );
}
else
{
insert( TQString( "INSERT INTO statistics ( url, deviceid, createdate, accessdate, percentage, playcounter, rating, uniqueid, deleted ) "
"VALUES ( '%6', %5, %1, %2, 0, 1, 0, %3, %4 );" )
.tqarg( atime )
.tqarg( atime )
.tqarg( ( getUniqueId( url ).isNull() ? "NULL" : '\'' + escapeString( getUniqueId( url ) ) + '\'' ) )
.tqarg( boolF() )
.tqarg( statDevId )
.tqarg( escapeString( statRPath ) ), 0 );
}
double prevscore = 50;
int playcount = 0;
if( !values.isEmpty() )
{
playcount = values[ 0 ].toInt();
// This stops setting the Rating (which creates a row) from affecting the
// prevscore of an unplayed track. See bug 127475
if ( playcount )
prevscore = values[ 2 ].toDouble();
}
const TQStringList v = query( TQString( "SELECT length FROM tags WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
const int length = v.isEmpty() ? 0 : v.first().toInt();
ScriptManager::instance()->requestNewScore( url, prevscore, playcount, length, percentage, reason );
}
float
CollectionDB::getSongPercentage( const TQString &url )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabStats, QueryBuilder::valScore );
qb.addMatch( QueryBuilder::tabStats, QueryBuilder::valURL, url );
TQStringList values = qb.run();
if( !values.isEmpty() )
return values.first().toFloat();
return 0;
}
int
CollectionDB::getSongRating( const TQString &url )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabStats, QueryBuilder::valRating );
qb.addMatch( QueryBuilder::tabStats, QueryBuilder::valURL, url );
TQStringList values = qb.run();
if( values.count() )
return kClamp( values.first().toInt(), 0, 10 );
return 0;
}
void
CollectionDB::setSongPercentage( const TQString &url , float percentage)
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQStringList values =
query( TQString(
"SELECT playcounter, createdate, accessdate, rating FROM statistics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
//handle corner case: deviceid!=-1 but there is a statistics row for that song with deviceid -1
if ( values.isEmpty() )
{
TQString rpath2 = '.' + url;
values = query( TQString(
"SELECT playcounter, createdate, accessdate, rating FROM statistics "
"WHERE url = '%1' AND deviceid = -1;" )
.tqarg( escapeString( rpath2 ) ) );
if ( !values.isEmpty() )
{
rpath = rpath2;
deviceid = -1;
}
}
// check boundaries
if ( percentage > 100.f ) percentage = 100.f;
if ( percentage < 0.f ) percentage = 0.f;
if ( !values.isEmpty() )
{
query( TQString( "UPDATE statistics SET percentage=%1 WHERE url='%3' AND deviceid = %2;" )
.tqarg( percentage )
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
}
else
{
insert( TQString( "INSERT INTO statistics ( url, deviceid, createdate, accessdate, percentage, playcounter, rating, uniqueid, deleted ) "
"VALUES ( '%7', %6, %2, %3, %1, 0, 0, %3, %4 );" )
.tqarg( percentage )
.tqarg( TQDateTime::tqcurrentDateTime().toTime_t() )
.tqarg( 0 )
.tqarg( ( getUniqueId( url ).isNull() ? "NULL" : '\'' + escapeString( getUniqueId( url ) ) + '\'' ) )
.tqarg( boolF() )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ),0 );
}
emit scoreChanged( url, percentage );
}
void
CollectionDB::setSongRating( const TQString &url, int rating, bool toggleHalf )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQStringList values =
query( TQString(
"SELECT playcounter, createdate, accessdate, percentage, rating FROM statistics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
//handle corner case: deviceid!=-1 but there is a statistics row for that song with deviceid -1
if( values.isEmpty() )
{
TQString rpath2 = '.' + url;
values = query( TQString(
"SELECT playcounter, createdate, accessdate, percentage, rating FROM statistics "
"WHERE url = '%1' AND deviceid = -1;" )
.tqarg( escapeString( rpath2 ) ) );
if ( !values.isEmpty() )
{
rpath = rpath2;
deviceid = -1;
}
}
bool ok = true;
if( !values.isEmpty() )
{
int prev = values[4].toInt( &ok );
if( ok && toggleHalf && ( prev == rating || ( prev == 1 && rating == 2 ) ) )
{
if( prev == 1 && rating == 2 )
rating = 0;
else if( rating % 2 ) //.5
rating++;
else
rating--;
}
}
// check boundaries
if ( rating > 10 ) rating = 10;
if ( rating < 0 /*|| rating == 1*/ ) rating = 0; //ratings are 1-5
if ( !values.isEmpty() )
{
query( TQString( "UPDATE statistics SET rating=%1 WHERE url='%3' AND deviceid = %2;" )
.tqarg( rating )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
else
{
insert( TQString( "INSERT INTO statistics ( url, deviceid, createdate, accessdate, percentage, rating, playcounter, uniqueid, deleted ) "
"VALUES ( '%7', %6, %2, %3, 0, %1, 0, %4, %5 );" )
.tqarg( rating )
.tqarg( TQDateTime::tqcurrentDateTime().toTime_t() )
.tqarg( 0 )
.tqarg( ( getUniqueId( url ).isNull() ? "NULL" : '\'' + escapeString( getUniqueId( url ) ) + '\'' ) )
.tqarg( boolF() )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ), NULL );
}
emit ratingChanged( url, rating );
}
int
CollectionDB::getPlayCount( const TQString &url )
{
//queryBuilder is good
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabStats, QueryBuilder::valPlayCounter );
qb.addMatch( QueryBuilder::tabStats, QueryBuilder::valURL, url );
TQStringList values = qb.run();
if( values.count() )
return values.first().toInt();
return 0;
}
TQDateTime
CollectionDB::getFirstPlay( const TQString &url )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabStats, QueryBuilder::valCreateDate );
qb.addMatch( QueryBuilder::tabStats, QueryBuilder::valURL, url );
TQStringList values = qb.run();
TQDateTime dt;
if( values.count() )
dt.setTime_t( values.first().toUInt() );
return dt;
}
TQDateTime
CollectionDB::getLastPlay( const TQString &url )
{
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabStats, QueryBuilder::valAccessDate );
qb.addMatch( QueryBuilder::tabStats, QueryBuilder::valURL, url );
TQStringList values = qb.run();
TQDateTime dt;
if( values.count() )
dt.setTime_t( values.first().toUInt() );
else
dt.setTime_t( 0 );
return dt;
}
/*!
* @short: exchange url references in the database for a particular file
* @note: deletes all items for newURL, changes oldURL->newURL, deletes oldURL.
* FIXME: should we check if lyrics etc exist in the newURL and keep them if necessary?
*/
void
CollectionDB::migrateFile( const TQString &oldURL, const TQString &newURL )
{
int oldMediaid = MountPointManager::instance()->getIdForUrl( oldURL );
TQString oldRpath = MountPointManager::instance()->getRelativePath( oldMediaid, oldURL );
int newMediaid = MountPointManager::instance()->getIdForUrl( newURL );
TQString newRpath = MountPointManager::instance()->getRelativePath( newMediaid, newURL );
// Ensure destination is clear.
query( TQString( "DELETE FROM tags WHERE url = '%2' AND deviceid = %1;" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) ) );
query( TQString( "DELETE FROM statistics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) ) );
query( TQString( "DELETE FROM tags_labels WHERE url = '%2' and deviceid = %1;" )
.tqarg( newMediaid).tqarg( escapeString( newRpath ) ) );
if ( !getLyrics( oldURL ).isEmpty() )
query( TQString( "DELETE FROM lyrics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) ) );
// Migrate
//code looks ugly but prevents problems when the URL contains HTTP escaped characters
query( TQString( "UPDATE tags SET url = '%3', deviceid = %1" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) )
+ TQString( " WHERE deviceid=%1 AND url = '%2';" )
.tqarg( oldMediaid ).tqarg( escapeString( oldRpath ) ) );
query( TQString( "UPDATE statistics SET url = '%2', deviceid = %1" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) )
+ TQString( " WHERE deviceid=%1 AND url = '%2';" )
.tqarg( oldMediaid ).tqarg( escapeString( oldRpath ) ) );
query( TQString( "UPDATE lyrics SET url = '%2', deviceid = %1" )
.tqarg( newMediaid ).tqarg( escapeString( newRpath ) )
+ TQString( " WHERE deviceid=%1 AND url = '%2';" )
.tqarg( oldMediaid ).tqarg( escapeString( oldRpath ) ) );
query( TQString( "UPDATE tags_labels SET url = '%2', deviceid = %1 WHERE deviceid = %3 AND url = '%4';" )
.tqarg( TQString::number( newMediaid ), escapeString( newRpath ), TQString::number( oldMediaid ), escapeString( oldRpath ) ) );
query( TQString( "UPDATE uniqueid SET url = '%1', deviceid = %2 WHERE url = '%3' AND deviceid = %4;" )
.tqarg( escapeString( newRpath ), TQString::number( newMediaid ),
escapeString( oldRpath ), TQString::number( oldMediaid ) ) );
query( TQString( "UPDATE playlists SET url = '%1' WHERE url = '%2';" )
.tqarg( escapeString( newURL ),
escapeString( oldURL ) ) );
}
void
CollectionDB::fileOperationResult( KIO::Job *job ) // slot
{
if(job->error())
{
m_fileOperationFailed = true;
debug() << "file operation failed: " << job->errorText() << endl;
}
else
{
m_fileOperationFailed = false;
}
m_waitForFileOperation = false;
}
void CollectionDB::cancelMovingFileJob()
{
m_moveFileJobCancelled = true;
}
bool
CollectionDB::organizeFile( const KURL &src, OrganizeCollectionDialog &dialog, bool copy )
{
if( !MetaBundle::isKioUrl( src ) )
return false;
bool overwrite = dialog.overwriteCheck->isChecked();
bool localFile = src.isLocalFile();
KURL tmpSrc = src;
if( !localFile )
{
TQString tmp;
TQString extension = src.url().section( '.', -1 );
extension = extension.section("?", 0, 0); // remove trailling stuff lead by ?, if any
int count = 0;
do
{
tmp = TQString( dialog.folderCombo->currentText() + "/amarok-tmp-%1." + extension ).tqarg( count );
count++;
} while( TQFile::exists( tmp ) );
tmpSrc = KURL::fromPathOrURL( tmp );
KIO::FileCopyJob *job = 0;
if( copy )
{
job = KIO::file_copy( src, tmpSrc, -1, false, false, false );
}
else
{
job = KIO::file_move( src, tmpSrc, -1, false, false, false );
}
connect( job, TQT_SIGNAL(result( KIO::Job * )), TQT_SLOT(fileOperationResult( KIO::Job * )) );
m_waitForFileOperation = true;
while( m_waitForFileOperation )
{
if( m_moveFileJobCancelled )
{
disconnect( job, TQT_SIGNAL(result( KIO::Job * )), this, TQT_SLOT(fileOperationResult( KIO::Job * )) );
TQString partFile = TQString( "%1.part" ).tqarg( (job->destURL()).path() );
job->kill();
TQFile file( partFile );
if( file.exists() ) file.remove();
m_waitForFileOperation = false;
m_fileOperationFailed = true;
continue;
}
usleep( 10000 );
kapp->tqprocessEvents( 100 );
}
if( m_fileOperationFailed )
{
debug() << "failed to transfer " << src.url() << " to " << tmpSrc << endl;
m_moveFileJobCancelled = false;
return false;
}
}
//Building destination here.
MetaBundle mb( tmpSrc );
TQString dest = dialog.buildDestination( dialog.buildFormatString(), mb );
debug() << "Destination: " << dest << endl;
if( !m_moveFileJobCancelled && tmpSrc.path() != dest ) //suppress error warning that file couldn't be moved
{
if( !CollectionDB::instance()->moveFile( tmpSrc.url(), dest, overwrite, copy && localFile ) )
{
if( !localFile )
TQFile::remove( tmpSrc.path() );
m_moveFileJobCancelled = false;
return false;
}
}
//Use cover image for folder icon
if( !m_moveFileJobCancelled && dialog.coverCheck->isChecked() && !mb.artist().isEmpty() && !mb.album().isEmpty() )
{
KURL dstURL = KURL::fromPathOrURL( dest );
dstURL.cleanPath();
TQString path = dstURL.directory();
TQString cover = CollectionDB::instance()->albumImage( mb.artist(), mb.album(), false, 1 );
if( !TQFile::exists(path + "/.directory") && !cover.endsWith( "nocover.png" ) )
{
//TQPixmap thumb; //Not amazon nice.
//if ( thumb.load( cover ) ){
//thumb.save(path + "/.front.png", "PNG", -1 ); //hide files
KSimpleConfig config(path + "/.directory");
config.setGroup("Desktop Entry");
if( !config.hasKey("Icon") )
{
config.writeEntry( "Icon", cover );
config.sync();
}
//} //Not amazon nice.
}
}
if( localFile && isDirInCollection( src.directory() ) && TQDir().rmdir( src.directory() ) )
{
debug() << "removed: " << src.directory() << endl;
}
m_moveFileJobCancelled = false;
return true;
}
bool
CollectionDB::moveFile( const TQString &src, const TQString &dest, bool overwrite, bool copy )
{
DEBUG_BLOCK
if(src == dest){
debug() << "Source and destination URLs are the same, aborting." << endl;
return false;
}
// Escape URL.
KURL srcURL = KURL::fromPathOrURL( src );
KURL dstURL = KURL::fromPathOrURL( dest );
// Clean it.
srcURL.cleanPath();
dstURL.cleanPath();
// Make sure it is valid.
if(!srcURL.isValid() || !dstURL.isValid())
debug() << "Invalid URL " << endl;
// Get just the directory.
KURL dir = dstURL;
dir.setFileName(TQString());
// Create the directory.
if(!KStandardDirs::exists(dir.path()))
if(!KStandardDirs::makeDir(dir.path())) {
debug() << "Unable to create directory " << dir.path() << endl;
}
m_fileOperationFailed = false;
KIO::FileCopyJob *job = 0;
if( copy )
{
job = KIO::file_copy( srcURL, dstURL, -1, overwrite, false, false );
}
else
{
job = KIO::file_move( srcURL, dstURL, -1, overwrite, false, false );
}
connect( job, TQT_SIGNAL(result( KIO::Job * )), TQT_SLOT(fileOperationResult( KIO::Job * )) );
m_waitForFileOperation = true;
while( m_waitForFileOperation )
{
if( m_moveFileJobCancelled )
{
disconnect( job, TQT_SIGNAL(result( KIO::Job * )), this, TQT_SLOT(fileOperationResult( KIO::Job * )) );
TQString partFile = TQString( "%1.part" ).tqarg( (job->destURL()).path() );
job->kill();
TQFile file( partFile );
if( file.exists() ) file.remove();
m_waitForFileOperation = false;
m_fileOperationFailed = true;
continue;
}
usleep( 10000 );
kapp->tqprocessEvents( 100 );
}
if( !m_fileOperationFailed )
{
if( copy )
{
MetaBundle bundle( dstURL );
if( bundle.isValidMedia() )
{
addSong( &bundle, true );
return true;
}
}
else
{
emit fileMoved( src, dest );
migrateFile( srcURL.path(), dstURL.path() );
if( isFileInCollection( srcURL.path() ) )
{
return true;
}
else
{
MetaBundle bundle( dstURL );
if( bundle.isValidMedia() )
{
addSong( &bundle, true );
return true;
}
}
}
}
return false;
}
void
CollectionDB::updateDirStats( TQString path, const long datetime, const bool temporary )
{
if ( path.endsWith( "/" ) )
path = path.left( path.length() - 1 );
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, path );
if (getDbConnectionType() == DbConnection::postgresql)
{
// REPLACE INTO is not valid SQL for postgres, so we need to check whether we
// should UPDATE() or INSERT()
TQStringList values = query( TQString("SELECT * FROM directories%1 WHERE dir='%3' AND deviceid=%2;")
.tqarg( temporary ? "_temp" : "")
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
if(values.count() > 0 )
{
query( TQString( "UPDATE directories%1 SET changedate=%2 WHERE dir='%4'AND deviceid=%3;")
.tqarg( temporary ? "_temp" : "" )
.tqarg( datetime )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
else
{
query( TQString( "INSERT INTO directories%1 (dir, deviceid,changedate) VALUES ('%4', %3, '%2');")
.tqarg( temporary ? "_temp" : "")
.tqarg( datetime )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
}
else
{
query( TQString( "REPLACE INTO directories%1 ( dir, deviceid, changedate ) VALUES ( '%4', %3, %2 );" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( datetime )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
INotify::instance()->watchDir( path );
}
void
CollectionDB::removeSongsInDir( TQString path, TQMap<TQString,TQString> *tagsRemoved )
{
if ( path.endsWith( "/" ) )
path = path.left( path.length() - 1 );
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, path );
// Pass back the list of tags we actually delete if requested.
if( tagsRemoved )
{
TQStringList result
= query( TQString( "SELECT tags.deviceid, tags.url, uniqueid.uniqueid FROM tags "
"LEFT JOIN uniqueid ON uniqueid.url = tags.url "
"AND uniqueid.deviceid = tags.deviceid "
"WHERE tags.dir = '%2' AND tags.deviceid = %1" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
TQStringList::ConstIterator it = result.begin(), end = result.end();
while( it != end )
{
int deviceid2 = (*(it++)).toInt();
TQString rpath2 = *(it++);
TQString uniqueid = *(it++);
(*tagsRemoved)[uniqueid] = MountPointManager::instance()->getAbsolutePath(
deviceid2, rpath2 );
}
}
query( TQString( "DELETE FROM tags WHERE dir = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
query( TQString( "DELETE FROM uniqueid WHERE dir = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
bool
CollectionDB::isDirInCollection( TQString path )
{
if ( path.endsWith( "/" ) )
path = path.left( path.length() - 1 );
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, path );
TQStringList values =
query( TQString( "SELECT changedate FROM directories WHERE dir = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
return !values.isEmpty();
}
bool
CollectionDB::isFileInCollection( const TQString &url )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQString sql = TQString( "SELECT url FROM tags WHERE url = '%2' AND deviceid = %1" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) );
if ( deviceid == -1 )
{
sql += ';';
}
else
{
TQString rpath2 = '.' + url;
sql += TQString( " OR url = '%1' AND deviceid = -1;" )
.tqarg( escapeString( rpath2 ) );
}
TQStringList values = query( sql );
return !values.isEmpty();
}
void
CollectionDB::removeSongs( const KURL::List& urls )
{
for( KURL::List::ConstIterator it = urls.begin(), end = urls.end(); it != end; ++it )
{
int deviceid = MountPointManager::instance()->getIdForUrl( *it );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, (*it).path() );
query( TQString( "DELETE FROM tags WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
query( TQString( "DELETE FROM uniqueid WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
query( TQString( "UPDATE statistics SET deleted = %1 WHERE url = '%3' AND deviceid = %2;" )
.tqarg( boolT() )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
}
TQStringList
CollectionDB::similarArtists( const TQString &artist, uint count )
{
TQStringList values;
values = query( TQString( "SELECT suggestion FROM related_artists WHERE artist = '%1' ORDER BY %2 LIMIT %3 OFFSET 0;" )
.tqarg( escapeString( artist ), randomFunc(), TQString::number( count ) ) );
if ( values.isEmpty() )
Scrobbler::instance()->similarArtists( artist );
return values;
}
void
CollectionDB::sanitizeCompilations()
{
query( TQString( "UPDATE tags_temp SET sampler = %1 WHERE sampler IS NULL;").tqarg( boolF() ) );
}
void
CollectionDB::checkCompilations( const TQString &path, const bool temporary )
{
TQStringList albums;
TQStringList artists;
TQStringList dirs;
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, path );
albums = query( TQString( "SELECT DISTINCT album.name FROM tags_temp, album%1 AS album WHERE tags_temp.dir = '%3' AND tags_temp.deviceid = %2 AND album.id = tags_temp.album AND tags_temp.sampler IS NULL;" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
for ( uint i = 0; i < albums.count(); i++ )
{
if ( albums[ i ].isEmpty() ) continue;
const uint album_id = albumID( albums[ i ], false, temporary, true );
artists = query( TQString( "SELECT DISTINCT artist.name FROM tags_temp, artist%1 AS artist WHERE tags_temp.album = '%2' AND tags_temp.artist = artist.id;" )
.tqarg( temporary ? "_temp" : "" )
.tqarg( album_id ) );
dirs = query( TQString( "SELECT DISTINCT dir FROM tags_temp WHERE album = '%1';" )
.tqarg( album_id ) );
if ( artists.count() > dirs.count() )
{
debug() << "Detected compilation: " << albums[ i ] << " - " << artists.count() << ':' << dirs.count() << endl;
}
query( TQString( "UPDATE tags_temp SET sampler = %1 WHERE album = '%2' AND sampler IS NULL;" )
.tqarg(artists.count() > dirs.count() ? boolT() : boolF()).tqarg( album_id ) );
}
}
void
CollectionDB::setCompilation( const KURL::List &urls, bool enabled, bool updateView )
{
for ( KURL::List::const_iterator it = urls.begin(); it != urls.end(); ++it )
{
TQString url( ( *it ).path() );
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
query( TQString( "UPDATE tags SET sampler = %1 WHERE tags.url = '%2' AND tags.deviceid = %3;" )
.tqarg( ( enabled ? boolT() : boolF() ), escapeString( rpath ), TQString::number( deviceid ) ) );
}
// Update the Collection-Browser view,
// using TQTimer to make sure we don't manipulate the GUI from a thread
if ( updateView )
TQTimer::singleShot( 0, CollectionView::instance(), TQT_SLOT( renderView() ) );
}
void
CollectionDB::removeDirFromCollection( TQString path )
{
//if ( path.endsWith( "/" ) )
// path = path.left( path.length() - 1 );
int deviceid = MountPointManager::instance()->getIdForUrl( path );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, path );
query( TQString( "DELETE FROM directories WHERE dir = '%2' AND deviceid = %1;" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
}
TQString
CollectionDB::IDFromExactValue( TQString table, TQString value, bool autocreate, bool temporary /* = false */ )
{
if ( temporary )
{
table.append( "_temp" );
}
TQString querystr( TQString( "SELECT id FROM %1 WHERE name " ).tqarg( table ) );
querystr += exactCondition( value ) + ';';
TQStringList result = query( querystr );
if ( result.isEmpty() )
{
if ( autocreate )
return TQString::number( insert( TQString( "INSERT INTO %1 ( name ) VALUES ( '%2' );" )
.tqarg( table, escapeString( value ) ),
table ) );
else
return 0;
}
else
{
if ( result.count() > 1 )
debug() << "More than one entry in the " << table << " database for '" << value << '\'' << endl;
return result.first();
}
}
void
CollectionDB::deleteRedundantName( const TQString &table, const TQString &id )
{
TQString querystr( TQString( "SELECT %1 FROM tags WHERE tags.%1 = %2 LIMIT 1;" ).tqarg( table, id ) );
TQStringList result = query( querystr );
if ( result.isEmpty() )
query( TQString( "DELETE FROM %1 WHERE id = %2;" ).tqarg( table,id ) );
}
void
CollectionDB::deleteAllRedundant( const TQString &table )
{
//This works with MySQL4. I thought it might not do, due to the comment in copyTempTables
query( TQString( "DELETE FROM %1 WHERE id NOT IN ( SELECT %2 FROM tags )" ).tqarg( table, table ) );
}
void
CollectionDB::updateTags( const TQString &url, const MetaBundle &bundle, const bool updateView )
{
DEBUG_BLOCK
QueryBuilder qb;
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTitle );
qb.addReturnValue( QueryBuilder::tabArtist, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabComposer, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabAlbum, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabGenre, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabYear, QueryBuilder::valName );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTrack );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valComment );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valDiscNumber );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFilesize );
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFileType );
// [10] is above. [11] is below.
qb.addReturnValue( QueryBuilder::tabSong, QueryBuilder::valBPM );
qb.addReturnValue( QueryBuilder::tabArtist, QueryBuilder::valID );
qb.addReturnValue( QueryBuilder::tabComposer, QueryBuilder::valID );
qb.addReturnValue( QueryBuilder::tabAlbum, QueryBuilder::valID );
qb.addReturnValue( QueryBuilder::tabGenre, QueryBuilder::valID );
qb.addReturnValue( QueryBuilder::tabYear, QueryBuilder::valID );
qb.addURLFilters ( TQStringList( url ) );
qb.setOptions( QueryBuilder::optRemoveDuplicates );
TQStringList values = qb.run();
if ( values.count() > 17 )
{
error() << "Query returned more than 1 song. Aborting updating metadata" << endl;
return;
}
if ( !values.isEmpty() )
{
bool art=false, comp=false, alb=false, gen=false, year=false;
TQString command = "UPDATE tags SET ";
if ( values[ 0 ] != bundle.title() )
command += "title = '" + escapeString( bundle.title() ) + "', ";
if ( values[ 1 ] != bundle.artist() )
{
art = true;
command += "artist = " + IDFromExactValue( "artist", bundle.artist() ) + ", ";
}
if ( values[ 2 ] != bundle.composer() )
{
comp = true;
command += "composer = " + IDFromExactValue( "composer", bundle.composer() ) + ", ";
}
if ( values[ 3 ] != bundle.album() )
{
alb = true;
command += "album = " + IDFromExactValue( "album", bundle.album() ) + ", ";
}
if ( values[ 4 ] != bundle.genre() )
{
gen = true;
command += "genre = " + IDFromExactValue( "genre", bundle.genre() ) + ", ";
}
if ( values[ 5 ] != TQString::number( bundle.year() ) )
{
year = false;
command += "year = " + IDFromExactValue( "year", TQString::number( bundle.year() ) ) + ", ";
}
if ( values[ 6 ] != TQString::number( bundle.track() ) )
command += "track = " + TQString::number( bundle.track() ) + ", ";
if ( values[ 7 ] != bundle.comment() )
command += "comment = '" + escapeString( bundle.comment() ) + "', ";
if ( values[ 8 ] != TQString::number( bundle.discNumber() ) )
command += "discnumber = '" + TQString::number( bundle.discNumber() ) + "', ";
if ( values[ 9 ] != TQString::number( bundle.filesize() ) )
command += "filesize = '" + TQString::number( bundle.filesize() ) + "', ";
if ( values[ 10 ] != TQString::number( bundle.fileType() ) )
command += "filetype = '" + TQString::number( bundle.fileType() ) + "', ";
if ( values[ 11 ] != TQString::number( bundle.bpm() ) )
command += "bpm = '" + TQString::number( bundle.bpm() ) + "', ";
if ( "UPDATE tags SET " == command )
{
debug() << "No tags selected to be changed" << endl;
}
else
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
//We have to remove the trailing comma from command
query( command.left( command.length() - 2 ) + " WHERE url = '" + escapeString( rpath ) +
"' AND deviceid = " + TQString::number( deviceid ) + ';' );
}
//Check to see if we use the entry anymore. If not, delete it
if ( art )
deleteRedundantName( "artist", values[ 12 ] );
if ( comp )
deleteRedundantName( "composer", values[ 13 ] );
if ( alb )
deleteRedundantName( "album", values[ 14 ] );
if ( gen )
deleteRedundantName( "genre", values[ 15 ] );
if ( year )
deleteRedundantName( "year", values[ 16 ] );
// Update the Collection-Browser view,
// using TQTimer to make sure we don't manipulate the GUI from a thread
if ( updateView )
TQTimer::singleShot( 0, CollectionView::instance(), TQT_SLOT( databaseChanged() ) );
if( art || alb )
emit tagsChanged( values[12], values[14] );
}
if ( EngineController::instance()->bundle().url() == bundle.url() )
{
debug() << "Current song edited, updating widgets: " << bundle.title() << endl;
EngineController::instance()->currentTrackMetaDataChanged( bundle );
}
emit tagsChanged( bundle );
}
void
CollectionDB::updateURL( const TQString &url, const bool updateView )
{
// don't use the KURL ctor as it checks the db first
MetaBundle bundle;
bundle.setPath( url );
bundle.readTags( TagLib::AudioProperties::Fast );
updateTags( url, bundle, updateView);
doAFTStuff( &bundle, false );
}
TQString
CollectionDB::getUniqueId( const TQString &url )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQStringList values = query( TQString( "SELECT uniqueid FROM uniqueid WHERE deviceid = %1 AND url = '%2';" )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ));
if( !values.empty() )
return values[0];
else
return TQString();
}
void
CollectionDB::setLyrics( const TQString &url, const TQString &lyrics, const TQString &uniqueid )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQStringList values = query(TQString("SELECT lyrics FROM lyrics WHERE url = '%2' AND deviceid = %1;")
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
if(values.count() > 0)
{
if ( !lyrics.isEmpty() )
query( TQString( "UPDATE lyrics SET lyrics = '%1' WHERE url = '%3' AND deviceid = %2;" )
.tqarg( escapeString( lyrics), TQString::number(deviceid), escapeString( rpath ) ) );
else
query( TQString( "DELETE FROM lyrics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid).tqarg( escapeString( rpath ) ) );
}
else
{
insert( TQString( "INSERT INTO lyrics (deviceid, url, lyrics, uniqueid) values ( %1, '%2', '%3', '%4' );" )
.tqarg( TQString::number(deviceid), escapeString( rpath ), escapeString( lyrics ), escapeString( uniqueid ) ), NULL);
}
}
TQString
CollectionDB::getLyrics( const TQString &url )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
TQStringList values = query( TQString( "SELECT lyrics FROM lyrics WHERE url = '%2' AND deviceid = %1;" )
.tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
return values[0];
}
void CollectionDB::removeInvalidAmazonInfo( const TQString& md5sum )
{
query( TQString( "DELETE FROM amazon WHERE filename='%1'" ).tqarg( md5sum ) );
}
void CollectionDB::newAmazonReloadDate( const TQString& asin, const TQString& locale, const TQString& md5sum)
{
TQStringList values = query(TQString("SELECT filename FROM amazon WHERE filename = '%1'")
.tqarg(md5sum));
if(values.count() > 0)
{
query( TQString("UPDATE amazon SET asin = '%1', locale = '%2', refetchdate = '%3' WHERE filename = '%4'")
.tqarg(asin)
.tqarg(locale)
.tqarg(TQDateTime::tqcurrentDateTime().addDays(80).toTime_t())
.tqarg(md5sum));
}
else
{
insert( TQString( "INSERT INTO amazon ( asin, locale, filename, refetchdate ) VALUES ( '%1', '%2', '%3', '%4');" )
.tqarg(asin)
.tqarg(locale)
.tqarg(md5sum)
.tqarg(TQDateTime::tqcurrentDateTime().addDays(80).toTime_t()), NULL );
}
}
TQStringList CollectionDB::staleImages()
{
return query(TQString("SELECT asin, locale, filename FROM amazon WHERE refetchdate < %1 ;")
.tqarg(TQDateTime::tqcurrentDateTime().toTime_t() ));
}
void
CollectionDB::applySettings()
{
bool recreateConnections = false;
if ( AmarokConfig::databaseEngine().toInt() != getDbConnectionType() )
{
if ( AmarokConfig::databaseEngine().toInt() == DbConnection::mysql )
m_dbConnType = DbConnection::mysql;
else if ( AmarokConfig::databaseEngine().toInt() == DbConnection::postgresql )
m_dbConnType = DbConnection::postgresql;
else m_dbConnType = DbConnection::sqlite;
recreateConnections = true;
}
else if ( AmarokConfig::databaseEngine().toInt() == DbConnection::mysql )
{
// Using MySQL, so check if MySQL settings were changed
const MySqlConfig *config =
static_cast<const MySqlConfig*> ( m_dbConfig );
if ( AmarokConfig::mySqlHost() != config->host() )
{
recreateConnections = true;
}
else if ( AmarokConfig::mySqlPort() != config->port() )
{
recreateConnections = true;
}
else if ( AmarokConfig::mySqlDbName() != config->database() )
{
recreateConnections = true;
}
else if ( AmarokConfig::mySqlUser() != config->username() )
{
recreateConnections = true;
}
else if ( AmarokConfig::mySqlPassword2() != config->password() )
{
recreateConnections = true;
}
}
else if ( AmarokConfig::databaseEngine().toInt() == DbConnection::postgresql )
{
const PostgresqlConfig *config =
static_cast<const PostgresqlConfig*> ( m_dbConfig );
if ( AmarokConfig::postgresqlHost() != config->host() )
{
recreateConnections = true;
}
else if ( AmarokConfig::postgresqlPort() != config->port() )
{
recreateConnections = true;
}
else if ( AmarokConfig::postgresqlDbName() != config->database() )
{
recreateConnections = true;
}
else if ( AmarokConfig::postgresqlUser() != config->username() )
{
recreateConnections = true;
}
else if ( AmarokConfig::postgresqlPassword() != config->password() )
{
recreateConnections = true;
}
}
if ( recreateConnections )
{
debug()
<< "Database engine settings changed: "
<< "recreating DbConnections" << endl;
// If Database engine was changed, recreate DbConnections.
destroy();
initialize();
CollectionView::instance()->renderView();
PlaylistBrowser::instance()->loadPodcastsFromDatabase();
emit databaseEngineChanged();
}
}
DbConnection * CollectionDB::getMyConnection()
{
//after some thought, to be thread-safe, must lock at the beginning of this function,
//not only if a new connection is made
connectionMutex->lock();
DbConnection *dbConn;
TQThread *currThread = ThreadManager::Thread::getRunning();
if (threadConnections->contains(currThread))
{
TQMap<TQThread *, DbConnection *>::Iterator it = threadConnections->find(currThread);
dbConn = it.data();
connectionMutex->unlock();
return dbConn;
}
#ifdef USE_MYSQL
if ( m_dbConnType == DbConnection::mysql )
{
dbConn = new MySqlConnection( static_cast<MySqlConfig*>( m_dbConfig ) );
}
else
#endif
#ifdef USE_POSTGRESQL
if ( m_dbConnType == DbConnection::postgresql )
{
dbConn = new PostgresqlConnection( static_cast<PostgresqlConfig*>( m_dbConfig ) );
}
else
#endif
{
dbConn = new SqliteConnection( static_cast<SqliteConfig*>( m_dbConfig ) );
}
threadConnections->insert(currThread, dbConn);
connectionMutex->unlock();
return dbConn;
}
void
CollectionDB::releasePreviousConnection(TQThread *currThread)
{
//if something already exists, delete the object, and erase knowledge of it from the TQMap.
connectionMutex->lock();
DbConnection *dbConn;
if (threadConnections->contains(currThread))
{
TQMap<TQThread *, DbConnection *>::Iterator it = threadConnections->find(currThread);
dbConn = it.data();
delete dbConn;
threadConnections->erase(currThread);
}
connectionMutex->unlock();
}
bool
CollectionDB::isConnected()
{
return getMyConnection()->isConnected();
}
//////////////////////////////////////////////////////////////////////////////////////////
// PROTECTED
//////////////////////////////////////////////////////////////////////////////////////////
TQCString
CollectionDB::md5sum( const TQString& artist, const TQString& album, const TQString& file )
{
KMD5 context( artist.lower().local8Bit() + album.lower().local8Bit() + file.local8Bit() );
// debug() << "MD5 SUM for " << artist << ", " << album << ": " << context.hexDigest() << endl;
return context.hexDigest();
}
void CollectionDB::engineTrackEnded( int finalPosition, int trackLength, const TQString &reason )
{
//This is where percentages are calculated
//TODO statistics are not calculated when currentTrack doesn't exist
// Don't update statistics if song has been played for less than 15 seconds
// if ( finalPosition < 15000 ) return;
//below check is necessary because if stop after current track is selected,
//the url's path will be empty, so check the previous URL for the path that
//had just played
const KURL url = EngineController::instance()->bundle().url().path().isEmpty() ?
EngineController::instance()->previousURL() :
EngineController::instance()->bundle().url();
PodcastEpisodeBundle peb;
if( getPodcastEpisodeBundle( url.url(), &peb ) )
{
PodcastEpisode *p = PlaylistBrowser::instance()->findPodcastEpisode( peb.url(), peb.parent() );
if ( p )
p->setListened();
if( !url.isLocalFile() )
return;
}
if ( url.path().isEmpty() || !m_autoScoring ) return;
// sanity check
if ( finalPosition > trackLength || finalPosition <= 0 )
finalPosition = trackLength;
int pct = (int) ( ( (double) finalPosition / (double) trackLength ) * 100 );
// increase song counter & calculate new statistics
addSongPercentage( url.path(), pct, reason );
}
void
CollectionDB::timerEvent( TQTimerEvent* )
{
scanMonitor();
}
//////////////////////////////////////////////////////////////////////////////////////////
// PUBLIC SLOTS
//////////////////////////////////////////////////////////////////////////////////////////
void
CollectionDB::fetchCover( TQWidget* parent, const TQString& artist, const TQString& album, bool noedit, TQListViewItem* item ) //SLOT
{
#ifdef AMAZON_SUPPORT
debug() << "Fetching cover for " << artist << " - " << album << endl;
const bool isCompilation = albumIsCompilation( TQString::number( albumID( album, false, false, true ) ) );
CoverFetcher* fetcher;
if( isCompilation )
// avoid putting various artists in front of album title. this causes problems for locales other than US.
fetcher = new CoverFetcher( parent, "", album );
else
fetcher = new CoverFetcher( parent, artist, album );
if( item )
{
itemCoverMapMutex->lock();
itemCoverMap->insert( item, fetcher );
itemCoverMapMutex->unlock();
}
connect( fetcher, TQT_SIGNAL(result( CoverFetcher* )), TQT_SLOT(coverFetcherResult( CoverFetcher* )) );
fetcher->setUserCanEditQuery( !noedit );
fetcher->startFetch();
#endif
}
void
CollectionDB::scanMonitor() //SLOT
{
if ( AmarokConfig::monitorChanges() )
scanModifiedDirs();
}
void
CollectionDB::startScan() //SLOT
{
TQStringList folders = MountPointManager::instance()->collectionFolders();
if ( folders.isEmpty() )
{
//dropTables( false );
//createTables( false );
clearTables( false );
emit scanDone( true );
}
else if( PlaylistBrowser::instance() )
{
emit scanStarted();
ThreadManager::instance()->queueJob( new ScanController( this, false, folders ) );
}
}
void
CollectionDB::stopScan() //SLOT
{
ThreadManager::instance()->abortAllJobsNamed( "CollectionScanner" );
}
//////////////////////////////////////////////////////////////////////////////////////////
// PRIVATE SLOTS
//////////////////////////////////////////////////////////////////////////////////////////
void
CollectionDB::dirDirty( const TQString& path )
{
debug() << k_funcinfo << "Dirty: " << path << endl;
ThreadManager::instance()->queueJob( new ScanController( this, false, path ) );
}
void
CollectionDB::coverFetcherResult( CoverFetcher *fetcher )
{
if( fetcher->wasError() ) {
error() << fetcher->errors() << endl;
emit coverFetcherError( fetcher->errors().front() );
}
else {
setAlbumImage( fetcher->artist(), fetcher->album(), fetcher->image(), fetcher->amazonURL(), fetcher->asin() );
emit coverFetched( fetcher->artist(), fetcher->album() );
}
//check the validity of the CollectionItem as it may have been deleted e.g. by a
//collection scan while fetching the cover
itemCoverMapMutex->lock();
TQMap<TQListViewItem*, CoverFetcher*>::Iterator it;
for( it = itemCoverMap->begin(); it != itemCoverMap->end(); ++it )
{
if( it.data() == fetcher )
{
if( it.key()->isOpen() )
static_cast<CollectionItem*>(it.key())->setPixmap( 0, TQPixmap() );
itemCoverMap->erase( it );
}
}
itemCoverMapMutex->unlock();
}
/**
* This query is fairly slow with sqlite, and often happens just
* after the OSD is shown. Threading it restores responsivity.
*/
class SimilarArtistsInsertionJob : public ThreadManager::DependentJob
{
virtual bool doJob()
{
CollectionDB::instance()->query( TQString( "DELETE FROM related_artists WHERE artist = '%1';" ).tqarg( escapedArtist ) );
const TQString sql = "INSERT INTO related_artists ( artist, suggestion, changedate ) VALUES ( '%1', '%2', 0 );";
foreach( suggestions )
CollectionDB::instance()->insert( sql
.tqarg( escapedArtist,
CollectionDB::instance()->escapeString( *it ) ), NULL);
return true;
}
virtual void completeJob() { emit CollectionDB::instance()->similarArtistsFetched( artist ); }
const TQString artist;
const TQString escapedArtist;
const TQStringList suggestions;
public:
SimilarArtistsInsertionJob( CollectionDB *parent, const TQString &s, const TQStringList &list )
: ThreadManager::DependentJob( parent, "SimilarArtistsInsertionJob" )
, artist( TQDeepCopy<TQString>(s) )
, escapedArtist( parent->escapeString( TQDeepCopy<TQString>(s) ) )
, suggestions( TQDeepCopy<TQStringList>(list) )
{}
};
void
CollectionDB::similarArtistsFetched( const TQString& artist, const TQStringList& suggestions )
{
debug() << "Received similar artists\n";
ThreadManager::instance()->queueJob( new SimilarArtistsInsertionJob( this, artist, suggestions ) );
}
void
CollectionDB::aftCheckPermanentTables( const TQString &currdeviceid, const TQString &currid, const TQString &currurl )
{
//DEBUG_BLOCK
//debug() << "deviceid = " << currdeviceid << endl << "url = " << currurl << endl << "uid = " << currid << endl;
TQStringList check1, check2;
foreach( m_aftEnabledPersistentTables )
{
//debug() << "Checking " << (*it) << endl;;
check1 = query( TQString(
"SELECT url, deviceid "
"FROM %1 "
"WHERE uniqueid = '%2';" )
.tqarg( escapeString( *it ) )
.tqarg( currid ) );
check2 = query( TQString(
"SELECT url, uniqueid "
"FROM %1 "
"WHERE deviceid = %2 AND url = '%3';" )
.tqarg( escapeString( *it ) )
.tqarg( currdeviceid
, currurl ) );
if( !check1.empty() )
{
//debug() << "uniqueid found, updating url" << endl;
query( TQString( "UPDATE %1 SET deviceid = %2, url = '%4' WHERE uniqueid = '%3';" )
.tqarg( escapeString( *it ) )
.tqarg( currdeviceid
, currid
, currurl ) );
}
else if( !check2.empty() )
{
//debug() << "url found, updating uniqueid" << endl;
query( TQString( "UPDATE %1 SET uniqueid = '%2' WHERE deviceid = %3 AND url = '%4';" )
.tqarg( escapeString( *it ) )
.tqarg( currid
, currdeviceid
, currurl ) );
}
}
}
void
CollectionDB::aftMigratePermanentTablesUrl( const TQString& /*oldUrl*/, const TQString& newUrl, const TQString& uniqueid )
{
//DEBUG_BLOCK
int deviceid = MountPointManager::instance()->getIdForUrl( newUrl );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, newUrl );
//NOTE: if ever do anything with "deleted" in the statistics table, set deleted to false in query
//below; will need special case.
//debug() << "deviceid = " << deviceid << endl << "newurl = " << newUrl << endl << "uid = " << uniqueid << endl;
foreach( m_aftEnabledPersistentTables )
{
query( TQString( "DELETE FROM %1 WHERE deviceid = %2 AND url = '%3';" )
.tqarg( escapeString( *it ) )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) ) );
query( TQString( "UPDATE %1 SET deviceid = %2, url = '%4' WHERE uniqueid = '%3';" )
.tqarg( escapeString( *it ) )
.tqarg( deviceid )
.tqarg( escapeString( uniqueid ) )
.tqarg( escapeString( rpath ) ) );
}
}
void
CollectionDB::aftMigratePermanentTablesUniqueId( const TQString& /*url*/, const TQString& oldid, const TQString& newid )
{
//DEBUG_BLOCK
//debug() << "oldid = " << oldid << endl << "newid = " << newid << endl;
//NOTE: if ever do anything with "deleted" in the statistics table, set deleted to false in query
//below; will need special case.
foreach( m_aftEnabledPersistentTables )
{
query( TQString( "DELETE FROM %1 WHERE uniqueid = '%2';" )
.tqarg( escapeString( *it ) )
.tqarg( escapeString( newid ) ) );
query( TQString( "UPDATE %1 SET uniqueid = '%1' WHERE uniqueid = '%2';" )
.tqarg( escapeString( *it ) )
.tqarg( escapeString( newid ) )
.tqarg( escapeString( oldid ) ) );
}
}
//////////////////////////////////////////////////////////////////////////////////////////
// PRIVATE
//////////////////////////////////////////////////////////////////////////////////////////
void
CollectionDB::initialize()
{
DEBUG_BLOCK
/// Create DBConfig instance:
#ifdef USE_MYSQL
if ( m_dbConnType == DbConnection::mysql )
{
TQString appVersion = Amarok::config( "General Options" )->readEntry( "Version" );
TQString passwd = AmarokConfig::mySqlPassword2(); // stored as string type
if( passwd.isEmpty() )
{
if( appVersion.startsWith( "1.3" ) )
{
/// This is because of the encrypted -> plaintext conversion
passwd = AmarokConfig::mySqlPassword(); // stored as password type
AmarokConfig::setMySqlPassword2( passwd );
}
else if( appVersion.startsWith( "1.4" ) )
{
passwd = Amarok::config( "MySql" )->readEntry( "MySqlPassword" ); //read the field as plaintext
AmarokConfig::setMySqlPassword2( passwd ); // store it in plaintext field
}
}
m_dbConfig = new MySqlConfig(
AmarokConfig::mySqlHost(),
AmarokConfig::mySqlPort(),
AmarokConfig::mySqlDbName(),
AmarokConfig::mySqlUser(),
passwd );
}
else
#endif
#ifdef USE_POSTGRESQL
if ( m_dbConnType == DbConnection::postgresql )
{
TQString appVersion = Amarok::config( "General Options" )->readEntry( "Version" );
TQString passwd = AmarokConfig::postgresqlPassword2();
if( passwd.isEmpty() )
{
if( appVersion.startsWith( "1.3" ) )
{
/// This is because of the encrypted -> plaintext conversion
passwd = AmarokConfig::postgresqlPassword(); // stored as password type
AmarokConfig::setPostgresqlPassword2( passwd );
}
else if( appVersion.startsWith( "1.4" ) &&
( appVersion.contains( "beta", false ) ||
appVersion.contains( "svn", false ) ) )
{
passwd = Amarok::config( "Postgresql" )->readEntry( "PostgresqlPassword" );
AmarokConfig::setPostgresqlPassword2( passwd );
}
}
m_dbConfig = new PostgresqlConfig(
AmarokConfig::postgresqlHost(),
AmarokConfig::postgresqlPort(),
AmarokConfig::postgresqlDbName(),
AmarokConfig::postgresqlUser(),
passwd );
}
else
#endif
{
m_dbConfig = new SqliteConfig(
Amarok::config( "Sqlite" )->readPathEntry( "location",
Amarok::saveLocation() + "collection.db" ) );
}
DbConnection *dbConn = getMyConnection();
if ( !dbConn->isConnected() || !dbConn->isInitialized() )
{
error() << "Failed to connect to or initialise database!" << endl;
Amarok::MessageQueue::instance()->addMessage( dbConn->lastError() );
}
else
{
if ( !isValid() )
{
//No tables seem to exist (as doing a count(url) didn't even return any number, even 0).
warning() << "Tables seem to not exist." << endl;
warning() << "Attempting to create tables (this should be safe; ignore any errors)..." << endl;
createTables(false);
createPersistentTables();
createPodcastTables();
createStatsTable();
warning() << "Tables should now definitely exist. (Stop ignoring errors)" << endl;
//Since we have created the tables, we need to make sure the version numbers are
//set to the correct values. If this is not done now, the database update code may
//run, which could corrupt things.
Amarok::config( "Collection Browser" )->writeEntry( "Database Version", DATABASE_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Stats Version", DATABASE_STATS_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Persistent Tables Version", DATABASE_PERSISTENT_TABLES_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Podcast Tables Version", DATABASE_PODCAST_TABLES_VERSION ); Amarok::config( "Collection Browser" )->writeEntry( "Database AFT Version", DATABASE_AFT_VERSION );
setAdminValue( "Database Version", TQString::number( DATABASE_VERSION ) );
setAdminValue( "Database Stats Version", TQString::number( DATABASE_STATS_VERSION ) );
setAdminValue( "Database Persistent Tables Version", TQString::number( DATABASE_PERSISTENT_TABLES_VERSION ) );
setAdminValue( "Database Podcast Tables Version", TQString::number( DATABASE_PODCAST_TABLES_VERSION ) );
setAdminValue( "Database AFT Version", TQString::number( DATABASE_AFT_VERSION ) );
}
// Due to a bug in our sqllite handling code, we have to recreate the indices.
// We should rmeove this before 1.4.5
if ( m_dbConnType == DbConnection::sqlite ) {
TQStringList indices = query( "SELECT name FROM sqlite_master WHERE type='index' ORDER BY name;" );
if (!indices.contains("url_tag")) {
createIndices();
}
}
//updates for the Devices table go here
//put all other update code into checkDatabase()
//make sure that there is no call to MountPointManager in CollectionDB's ctor
//or in methods called from the ctor.
if ( adminValue( "Database Devices Version" ).isEmpty()
&& Amarok::config( "CollectionBrowser" )->readNumEntry( "Database Devices Version", 0 ) == 0 )
{
createDevicesTable();
}
else if ( adminValue( "Database Devices Version" ).toInt() != DATABASE_DEVICES_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Devices Version", 0 ) != DATABASE_DEVICES_VERSION )
{
int prev = adminValue( "Database Devices Version" ).toInt();
if ( prev > DATABASE_DEVICES_VERSION || prev < 0 )
{
error() << "Database devices version too new for this version of Amarok" << endl;
exit( 1 );
//dropDevicesTable();
}
else
{
debug() << "Updating DEVICES table" << endl;
//add future Devices update code here
}
}
Amarok::config( "Collection Browser" )->writeEntry( "Database Devices Version", DATABASE_DEVICES_VERSION );
setAdminValue( "Database Devices Version", TQString::number( DATABASE_DEVICES_VERSION ) );
//make sure that all indices exist
createIndices();
createPermanentIndices();
}
}
void
CollectionDB::checkDatabase()
{
DEBUG_BLOCK
if ( isValid() )
{
//Inform the user that he should attach as many devices with music as possible
//Hopefully this won't be necessary soon.
//
//Currently broken, so disabled - seems to cause crashes as events are sent to
//the Playlist - maybe it's not fully initialised?
/*
TQString text = i18n( "Amarok has to update your database to be able to use the new Dynamic Collection(insert link) feature. Amarok now has to determine on which physical devices your collection is stored. Please attach all removable devices which contain part of your collection and continue. Cancelling will exit Amarok." );
int result = KMessageBox::warningContinueCancel( 0, text, "Database migration" );
if ( result != KMessageBox::Continue )
{
error() << "Dynamic Collection migration was aborted by user...exiting" << endl;
exit( 1 );
}
*/
bool needsUpdate = ( adminValue( "Database Stats Version" ).toInt() != DATABASE_STATS_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Stats Version", 0 ) != DATABASE_STATS_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Version", 0 ) != DATABASE_VERSION
|| adminValue( "Database Version" ).toInt() != DATABASE_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Persistent Tables Version", 0 ) != DATABASE_PERSISTENT_TABLES_VERSION
|| adminValue( "Database Persistent Tables Version" ).toInt() != DATABASE_PERSISTENT_TABLES_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Podcast Tables Version", 0 ) != DATABASE_PODCAST_TABLES_VERSION
|| adminValue( "Database Podcast Tables Version" ).toInt() != DATABASE_PODCAST_TABLES_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database AFT Version", 0 ) != DATABASE_AFT_VERSION
|| adminValue( "Database AFT Version" ).toInt() != DATABASE_AFT_VERSION );
if ( needsUpdate )
{
KDialogBase *dialog = new KDialogBase( KDialogBase::Swallow,
TQt::WType_TopLevel|TQt::WStyle_Customize|TQt::WStyle_DialogBorder,
0,
"Update database warning dialog",
false,
i18n( "Updating database" ),
0 );
/* TODO: remove the standard window controls from the dialog window, the user should not be able
to close, minimize, maximize the dialog
add additional text, e.g. Amarok is currently updating your database. This may take a while.
Please wait.
Consider using a ProgressBarDialog
*/
TQLabel *label = new TQLabel( i18n( "Updating database" ), dialog );
dialog->setMainWidget( label );
label->show();
TQTimer::singleShot( 0, dialog, TQT_SLOT( show() ) );
//process events in the main event loop so that the dialog actually gets shown
kapp->processEvents();
debug() << "Beginning database update" << endl;
updateStatsTables();
updatePersistentTables();
updatePodcastTables();
//This is really a one-off call that fixes a Collection Browser glitch
updateGroupBy();
//remove database file if version is incompatible
if ( Amarok::config( "Collection Browser" )->readNumEntry( "Database Version", 0 ) != DATABASE_VERSION
|| adminValue( "Database Version" ).toInt() != DATABASE_VERSION )
{
debug() << "Rebuilding database!" << endl;
dropTables(false);
createTables(false);
}
delete dialog;
}
emit databaseUpdateDone();
}
// TODO Should write to config in dtor, but it crashes...
Amarok::config( "Collection Browser" )->writeEntry( "Database Version", DATABASE_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Stats Version", DATABASE_STATS_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Persistent Tables Version", DATABASE_PERSISTENT_TABLES_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database Podcast Tables Version", DATABASE_PODCAST_TABLES_VERSION );
Amarok::config( "Collection Browser" )->writeEntry( "Database AFT Version", DATABASE_AFT_VERSION );
setAdminValue( "Database Version", TQString::number( DATABASE_VERSION ) );
setAdminValue( "Database Stats Version", TQString::number( DATABASE_STATS_VERSION ) );
setAdminValue( "Database Persistent Tables Version", TQString::number( DATABASE_PERSISTENT_TABLES_VERSION ) );
setAdminValue( "Database Podcast Tables Version", TQString::number( DATABASE_PODCAST_TABLES_VERSION ) );
setAdminValue( "Database AFT Version", TQString::number( DATABASE_AFT_VERSION ) );
initDirOperations();
}
void
CollectionDB::updateGroupBy()
{
//This ugly bit of code makes sure the Group BY setting is preserved, after the
//meanings of the values were changed due to the addition of the Composer table.
int version = adminValue( "Database Version" ).toInt();
if (!version) // an even older update
version = Amarok::config( "Collection Browser" )->readNumEntry( "Database Version", 0 );
if ( version && version < 32 )
{
KConfig* config = Amarok::config( "Collection Browser" );
int m_cat1 = config->readNumEntry( "Category1" );
int m_cat2 = config->readNumEntry( "Category2" );
int m_cat3 = config->readNumEntry( "Category3" );
m_cat1 = m_cat1 ? ( m_cat1 > 2 ? m_cat1 << 1 : m_cat1 ) : CollectionBrowserIds::IdArtist;
m_cat2 = m_cat2 ? ( m_cat2 > 2 ? m_cat2 << 1 : m_cat2 ) : CollectionBrowserIds::IdAlbum;
m_cat3 = m_cat3 ? ( m_cat3 > 2 ? m_cat3 << 1 : m_cat3 ) : CollectionBrowserIds::IdNone;
config->writeEntry( "Category1", m_cat1 );
config->writeEntry( "Category2", m_cat2 );
config->writeEntry( "Category3", m_cat3 );
}
}
void
CollectionDB::updateStatsTables()
{
if ( adminValue( "Database Stats Version" ).toInt() != DATABASE_STATS_VERSION
|| Amarok::config( "Collection Browser" )->readNumEntry( "Database Stats Version", 0 ) != DATABASE_STATS_VERSION )
{
debug() << "Different database stats version detected! Stats table will be updated or rebuilt." << endl;
#if 0 // causes mysterious crashes
if( getType() == DbConnection::sqlite && TQFile::exists( Amarok::saveLocation()+"collection.db" ) )
{
debug() << "Creating a backup of the database in "
<< Amarok::saveLocation()+"collection-backup.db" << '.' << endl;
bool copied = KIO::NetAccess::file_copy( Amarok::saveLocation()+"collection.db",
Amarok::saveLocation()+"collection-backup.db",
-1 /*perms*/, true /*overwrite*/, false /*resume*/ );
if( !copied )
{
debug() << "Backup failed! Perhaps the volume is not writable." << endl;
debug() << "Error was: " << KIO::NetAccess::lastErrorString() << endl;
}
}
#endif
int prev = adminValue( "Database Stats Version" ).toInt();
/* If config returns 3 or lower, it came from an Amarok version that was not aware of
admin table, so we can't trust this table at all */
if( !prev || ( Amarok::config( "Collection Browser" )->readNumEntry( "Database Stats Version", 0 )
&& Amarok::config( "Collection Browser" )->readNumEntry( "Database Stats Version", 0 ) <= 3 ) )
prev = Amarok::config( "Collection Browser" )->readNumEntry( "Database Stats Version", 0 );
//pre somewhere in the 1.3-1.4 timeframe, the version wasn't stored in the DB, so try to guess it
const TQString q = "SELECT COUNT( %1 ) FROM statistics;";
if( !prev && query( q.tqarg( "url" ) ).first().toInt()
&& query( q.tqarg( "createdate" ) ).first().toInt()
&& query( q.tqarg( "accessdate" ) ).first().toInt()
&& query( q.tqarg( "percentage" ) ).first().toInt()
&& query( q.tqarg( "playcounter" ) ).first().toInt() )
{
prev = 3;
}
if ( prev < 3 ) //it is from before 1.2, or our poor user is otherwise fucked
{
debug() << "Rebuilding stats-database!" << endl;
dropStatsTableV1();
createStatsTable();
}
else //Incrementally update the stats table to reach the present version
{
if( prev < 4 ) //every version from 1.2 forward had a stats version of 3
{
debug() << "Updating stats-database!" << endl;
query( "ALTER TABLE statistics ADD rating INTEGER DEFAULT 0;" );
query( "CREATE INDEX rating_stats ON statistics( rating );" );
query( "UPDATE statistics SET rating=0 WHERE " + boolT() + ';' );
}
if( prev < 5 )
{
debug() << "Updating stats-database!" << endl;
query( "UPDATE statistics SET rating = rating * 2;" );
}
if( prev < 8 ) //Versions 6, 7 and 8 all were all attempts to add columns for ATF. his code should do it all.
{
query( TQString( "CREATE TABLE statistics_fix ("
"url " + textColumnType() + " UNITQUE,"
"createdate INTEGER,"
"accessdate INTEGER,"
"percentage FLOAT,"
"rating INTEGER DEFAULT 0,"
"playcounter INTEGER);" ) );
insert( "INSERT INTO statistics_fix (url, createdate, accessdate, percentage, playcounter, rating)"
"SELECT url, createdate, accessdate, percentage, playcounter, rating FROM statistics;"
, NULL );
dropStatsTableV1();
createStatsTableV8();
insert( "INSERT INTO statistics (url, createdate, accessdate, percentage, playcounter, rating)"
"SELECT url, createdate, accessdate, percentage, playcounter, rating FROM statistics_fix;"
, NULL );
query( "DROP TABLE statistics_fix" );
}
if( prev < 9 )
{
//Update for Dynamic Collection:
//This is not technically for the stats table, but it is part of the same
//update, so put it here anyway.
MountPointManager::instance()->setCollectionFolders( Amarok::config( "Collection" )->readPathListEntry( "Collection Folders" ) );
query( "ALTER TABLE statistics ADD deviceid INTEGER;" );
//FIXME: (max) i know this is bad but its fast
TQStringList oldURLs = query( "SELECT url FROM statistics;" );
//it might be necessary to use batch updates to improve speed
debug() << "Updating " << oldURLs.count() << " rows in statistics" << endl;
foreach( oldURLs )
{
bool exists = TQFile::exists( *it );
int deviceid = exists ? MountPointManager::instance()->getIdForUrl( *it ) : -2;
TQString rpath = exists ? MountPointManager::instance()->getRelativePath( deviceid, *it ) : *it;
TQString update = TQString( "UPDATE statistics SET deviceid = %1, url = '%2' WHERE " )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) );
update += TQString( "url = '%1';" ).tqarg( escapeString( *it ) );
query ( update );
}
}
if ( prev < 12 )
{
//re-using old method cause just a slight change to one column...
//if people are upgrading from earlier than 11, just get the new column
//earlier :-)
createStatsTableV10( true );
query( "INSERT INTO statistics_fix_ten SELECT url,deviceid,createdate,"
"accessdate,percentage,rating,playcounter,uniqueid,deleted FROM "
"statistics;" );
dropStatsTableV1();
createStatsTableV10( false );
query( "INSERT INTO statistics SELECT * FROM statistics_fix_ten;" );
query( "UPDATE statistics SET uniqueid=NULL;" );
}
else if( prev > DATABASE_STATS_VERSION )
{
error() << "Database statistics version too new for this version of Amarok. Quitting..." << endl;
exit( 1 );
}
}
}
}
void
CollectionDB::updatePersistentTables()
{
TQString PersistentVersion = adminValue( "Database Persistent Tables Version" );
if ( PersistentVersion.isEmpty() )
{
/* persistent tables didn't have a version on 1.3X and older, but let's be nice and try to
copy/keep the good information instead of just deleting the tables */
debug() << "Detected old schema for tables with important data. Amarok will convert the tables, ignore any \"table already exists\" errors." << endl;
createPersistentTables();
/* Copy lyrics */
debug() << "Trying to get lyrics from old db schema." << endl;
TQStringList Lyrics = query( "SELECT url, lyrics FROM tags where tags.lyrics IS NOT NULL;" );
for (uint i=0; i<Lyrics.count(); i+=2 )
setLyrics( Lyrics[i], Lyrics[i+1] );
debug() << "Building podcast tables" << endl;
createPodcastTables();
}
else if ( PersistentVersion == "1" || PersistentVersion == "2" )
{
createPersistentTables(); /* From 1 to 2 nothing changed. There was just a bug on the code, and
on some cases the table wouldn't be created.
From 2 to 3, lyrics were made plain text, instead of HTML */
debug() << "Converting Lyrics to Plain Text." << endl;
TQStringList Lyrics = query( "SELECT url, lyrics FROM lyrics;" );
for (uint i=0; i<Lyrics.count(); i+=2 )
setLyrics( Lyrics[i], Lyrics[i+1] );
debug() << "Building podcast tables" << endl;
createPodcastTables();
}
else if ( PersistentVersion.toInt() < 4 )
{
debug() << "Building podcast tables" << endl;
createPodcastTables();
}
else
{
if ( PersistentVersion.toInt() < 5 )
{
debug() << "Updating podcast tables" << endl;
query( "ALTER TABLE podcastchannels ADD image " + textColumnType() + ';' );
query( "ALTER TABLE podcastepisodes ADD localurl " + textColumnType() + ';' );
query( "ALTER TABLE podcastepisodes ADD subtitle " + textColumnType() + ';' );
query( "ALTER TABLE podcastepisodes ADD size INTEGER;" );
query( "ALTER TABLE podcastepisodes DROP comment;" );
query( "ALTER TABLE podcastepisodes ADD comment " + longTextColumnType() + ';' );
query( "CREATE INDEX localurl_podepisode ON podcastepisodes( localurl );" );
}
if ( PersistentVersion.toInt() < 6 )
{
debug() << "Updating podcast tables" << endl;
query( "ALTER TABLE podcastchannels ADD image " + textColumnType() + ';' );
query( "ALTER TABLE podcastepisodes ADD subtitle " + textColumnType() + ';' );
query( "ALTER TABLE podcastepisodes ADD size INTEGER;" );
query( "ALTER TABLE podcastepisodes DROP comment;" );
query( "ALTER TABLE podcastepisodes ADD comment " + longTextColumnType() + ';' );
}
if ( PersistentVersion.toInt() < 11 )
{
debug() << "This is used to handle problems from uniqueid changeover and should not do anything" << endl;
}
if ( PersistentVersion.toInt() < 12 )
{
debug() << "Adding playlists table..." << endl;
createPersistentTablesV12();
}
if ( PersistentVersion.toInt() < 13 )
{
//Update for Dynamic Collection:
query( "ALTER TABLE lyrics ADD deviceid INTEGER;" );
//FIXME: (max) i know this is bad but its fast
TQStringList oldURLs = query( "SELECT url FROM lyrics;" );
//it might be necessary to use batch updates to improve speed
debug() << "Updating " << oldURLs.count() << " rows in lyrics" << endl;
foreach( oldURLs )
{
int deviceid = MountPointManager::instance()->getIdForUrl( *it );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, *it );
TQString update = TQString( "UPDATE lyrics SET deviceid = %1, url = '%2' WHERE " )
.tqarg( deviceid )
.tqarg( escapeString( rpath ) );
update += TQString( "url = '%1';" ).tqarg( escapeString( *it ) );
query ( update );
}
}
if ( PersistentVersion.toInt() < 15 )
{
createPersistentTablesV14( true );
query( "INSERT INTO amazon_fix SELECT asin,locale,filename,refetchdate FROM amazon;" );
query( "INSERT INTO lyrics_fix SELECT url,deviceid,lyrics FROM lyrics;" );
query( "INSERT INTO playlists_fix SELECT playlist,url,tracknum FROM playlists;" );
dropPersistentTablesV14();
createPersistentTablesV14( false );
query( "INSERT INTO amazon SELECT * FROM amazon_fix;" );
query( "INSERT INTO lyrics SELECT * FROM lyrics_fix;" );
query( "INSERT INTO playlists SELECT * FROM playlists_fix;" );
}
if ( PersistentVersion.toInt() < 17 )
{
//drop old labels and label tables, they were never used anyway and just confuse things
query( "DROP TABLE label;" );
query( "DROP TABLE labels;" );
query( "DROP TABLE tags_labels;" );
//update for label support
TQString labelsAutoIncrement = "";
if ( getDbConnectionType() == DbConnection::postgresql )
{
query( TQString( "CREATE SEQUENCE labels_seq;" ) );
labelsAutoIncrement = TQString("DEFAULT nextval('labels_seq')");
}
else if ( getDbConnectionType() == DbConnection::mysql )
{
labelsAutoIncrement = "AUTO_INCREMENT";
}
query( TQString( "CREATE TABLE labels ("
"id INTEGER PRIMARY KEY " + labelsAutoIncrement + ", "
"name " + textColumnType() + ", "
"type INTEGER);" ) );
query( TQString( "CREATE TABLE tags_labels ("
"deviceid INTEGER,"
"url " + exactTextColumnType() + ", "
"uniqueid " + exactTextColumnType(32) + ", " //m:n relationship, DO NOT MAKE UNITQUE!
"labelid INTEGER REFERENCES labels( id ) ON DELETE CASCADE );" ) );
query( "CREATE UNITQUE INDEX labels_name ON labels( name, type );" );
query( "CREATE INDEX tags_labels_uniqueid ON tags_labels( uniqueid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
query( "CREATE INDEX tags_labels_url ON tags_labels( url, deviceid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
}
if ( PersistentVersion.toInt() < 18 )
{
query( "ALTER TABLE lyrics ADD uniqueid " + exactTextColumnType(32) + ';' );
query( "CREATE INDEX lyrics_uniqueid ON lyrics( uniqueid );" );
}
if ( PersistentVersion.toInt() < 19 )
{
query( "CREATE INDEX tags_labels_labelid ON tags_labels( labelid );" ); //m:n relationship, DO NOT MAKE UNITQUE!
}
//Up to date. Keep this number \/ in sync!
if ( PersistentVersion.toInt() > 19 || PersistentVersion.toInt() < 0 )
{
//Something is horribly wrong
if ( adminValue( "Database Persistent Tables Version" ).toInt() != DATABASE_PERSISTENT_TABLES_VERSION )
{
error() << "There is a bug in Amarok: instead of destroying your valuable"
<< " database tables, I'm quitting" << endl;
exit( 1 );
debug() << "Rebuilding persistent tables database!" << endl;
dropPersistentTables();
createPersistentTables();
}
}
}
}
void
CollectionDB::updatePodcastTables()
{
TQString PodcastVersion = adminValue( "Database Podcast Tables Version" );
if ( PodcastVersion.toInt() < 2 )
{
createPodcastTablesV2( true );
query( "INSERT INTO podcastchannels_fix SELECT url,title,weblink,image,comment,"
"copyright,parent,directory,autoscan,fetchtype,autotransfer,haspurge,"
"purgecount FROM podcastchannels;" );
query( "INSERT INTO podcastepisodes_fix SELECT id,url,localurl,parent,guid,title,"
"subtitle,composer,comment,filetype,createdate,length,size,isNew FROM "
"podcastepisodes;" );
query( "INSERT INTO podcastfolders_fix SELECT id,name,parent,isOpen FROM podcastfolders;" );
dropPodcastTablesV2();
createPodcastTablesV2( false );
query( "INSERT INTO podcastchannels SELECT * FROM podcastchannels_fix;" );
query( "INSERT INTO podcastepisodes SELECT * FROM podcastepisodes_fix;" );
query( "INSERT INTO podcastfolders SELECT * FROM podcastfolders_fix;" );
}
//Keep this number in sync \/
if ( PodcastVersion.toInt() > 2 )
{
error() << "Something is very wrong with the Podcast Tables. Aborting" << endl;
exit( 1 );
dropPodcastTables();
createPodcastTables();
}
}
void
CollectionDB::vacuum()
{
if ( DbConnection::sqlite == getDbConnectionType() ||
DbConnection::postgresql == getDbConnectionType() )
{
//Clean up DB and free unused space.
debug() << "Running VACUUM" << endl;
query( "VACUUM;" );
}
}
void
CollectionDB::destroy()
{
//do we need or want to delete the actual connection objects as well as clearing them from the TQMap?
//or does TQMap's clear function delete them?
//this situation is not at all likely to happen often, so leaving them might be okay to prevent a
//thread from having its connection torn out from under it...not likely, but possible
//and leaving it should not end up eating much memory at all
connectionMutex->lock();
threadConnections->clear();
delete m_dbConfig;
connectionMutex->unlock();
}
void
CollectionDB::scanModifiedDirs()
{
if ( !m_scanInProgress
&& ( !CollectionView::instance() || !CollectionView::instance()->isOrganizingFiles() )
&& ( !MediaBrowser::instance() || !MediaBrowser::instance()->isTranscoding() ) )
{
//we check if a job is pending because we don't want to abort incremental collection readings
if ( !ThreadManager::instance()->isJobPending( "CollectionScanner" ) && PlaylistBrowser::instance() )
{
m_scanInProgress = true;
m_rescanRequired = false;
emit scanStarted();
ThreadManager::instance()->onlyOneJob( new ScanController( this, true ) );
}
}
else
m_rescanRequired = true;
}
void
CollectionDB::customEvent( TQCustomEvent *e )
{
if ( e->type() == (int)ScanController::JobFinishedEvent )
{
ScanController* s = static_cast<ScanController*>( e );
m_scanInProgress = false;
if ( s->isIncremental() )
{
debug() << "JobFinishedEvent from Incremental ScanController received.\n";
emit scanDone( s->hasChanged() );
// check if something changed while we were scanning. in this case we should
// rescan again, now.
if ( m_rescanRequired )
TQTimer::singleShot( 0, CollectionDB::instance(), TQT_SLOT( scanMonitor() ) );
}
else
{
debug() << "JobFinishedEvent from ScanController received.\n";
emit scanDone( s->wasSuccessful() );
}
}
}
TQString
CollectionDB::loadHashFile( const TQCString& hash, uint width )
{
//debug() << "loadHashFile: " << hash << " - " << width << endl;
TQString full = tagCoverDir().filePath( hash );
if ( width == 0 ) {
if ( TQFileInfo( full ).isReadable() ) {
//debug() << "loadHashFile: fullsize: " << full << endl;
return full;
}
} else {
if ( width == 1 ) width = AmarokConfig::coverPreviewSize();
TQCString widthKey = makeWidthKey( width );
TQString path = cacheCoverDir().filePath( widthKey + hash );
if ( TQFileInfo( path ).isReadable() ) {
//debug() << "loadHashFile: scaled: " << path << endl;
return path;
} else if ( TQFileInfo( full ).isReadable() ) {
//debug() << "loadHashFile: scaling: " << full << endl;
TQImage image( full );
if ( image.smoothScale( width, width, TQ_ScaleMin ).save( path, "PNG" ) ) {
//debug() << "loadHashFile: scaled: " << path << endl;
return path;
}
}
}
return TQString();
}
bool
CollectionDB::extractEmbeddedImage( const MetaBundle &trackInformation, TQCString& hash )
{
//debug() << "extractEmbeddedImage: " << hash << " - " << trackInformation.url().path() << endl;
MetaBundle::EmbeddedImageList images;
trackInformation.embeddedImages( images );
foreachType ( MetaBundle::EmbeddedImageList, images ) {
if ( hash.isEmpty() || (*it).hash() == hash ) {
if ( (*it).save( tagCoverDir() ) ) {
//debug() << "extractEmbeddedImage: saved to " << tagCoverDir().path() << endl;
hash = (*it).hash();
return true;
}
}
}
return false;
}
TQStringList
CollectionDB::getLabels( const TQString &url, const uint type )
{
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, url );
return query( TQString( "SELECT labels.name FROM labels "
"LEFT JOIN tags_labels ON labels.id = tags_labels.labelid "
"WHERE labels.type = %1 AND tags_labels.deviceid = %2 AND tags_labels.url = '%3';" )
.tqarg( type ).tqarg( deviceid ).tqarg( escapeString( rpath ) ) );
}
void
CollectionDB::cleanLabels()
{
DEBUG_BLOCK
TQStringList labelIds = query( "select labels.id "
"from labels left join tags_labels on labels.id = tags_labels.labelid "
"where tags_labels.labelid is NULL;" );
if ( !labelIds.isEmpty() )
{
TQString ids;
foreach( labelIds )
{
if ( !ids.isEmpty() )
ids += ',';
ids += *it;
}
query( TQString( "DELETE FROM labels "
"WHERE labels.id IN ( %1 );" )
.tqarg( ids ) );
}
}
void
CollectionDB::setLabels( const TQString &url, const TQStringList &labels, const TQString &uid, const uint type )
{
DEBUG_BLOCK
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = escapeString( MountPointManager::instance()->getRelativePath( deviceid, url ) );
TQStringList labelIds = query( TQString( "SELECT id FROM labels WHERE type = %1;" ).tqarg( type ) );
TQString ids;
if ( !labelIds.isEmpty() )
{
foreach( labelIds )
{
if ( !ids.isEmpty() )
ids += ',';
ids += *it;
}
//TODO: max: add uniqueid handling
query( TQString( "DELETE FROM tags_labels "
"WHERE tags_labels.labelid IN (%1) AND tags_labels.deviceid = %2 AND tags_labels.url = '%3';" )
.tqarg( ids, TQString::number(deviceid), rpath ) );
}
foreach( labels )
{
int id = query( TQString( "SELECT id FROM labels WHERE type = %1 AND name = '%2';" )
.tqarg( type ).tqarg( escapeString( *it ) ) ).first().toInt();
if ( !id )
{
id = insert( TQString( "INSERT INTO labels( name, type ) VALUES ( '%2', %1 );" )
.tqarg( type ).tqarg( escapeString( *it ) ), "labels" );
}
insert( TQString( "INSERT INTO tags_labels( labelid, deviceid, url, uniqueid ) VALUES ( %1, %2, '%3', '%4' );" )
.tqarg( TQString::number(id), TQString::number(deviceid), rpath, escapeString( uid ) ), 0 );
}
emit labelsChanged( url );
}
void
CollectionDB::removeLabels( const TQString &url, const TQStringList &labels, const uint type )
{
DEBUG_BLOCK
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = escapeString( MountPointManager::instance()->getRelativePath( deviceid, url ) );
TQString sql = TQString( "DELETE FROM tags_labels "
"FROM tags_labels AS t LEFT JOIN labels AS l ON t.labelid = l.id "
"WHERE l.type = %1 AND t.deviceid = %2 AND t.url = '%3' AND ( 0" )
.tqarg( type ).tqarg( deviceid ).tqarg( rpath );
foreach( labels )
{
sql += TQString( " OR l.name = '%1'" ).tqarg( escapeString( *it ) );
}
sql += ");";
query( sql );
emit labelsChanged( url );
}
bool
CollectionDB::addLabel( const TQString &url, const TQString &label, const TQString &uid, const uint type )
{
DEBUG_BLOCK
int deviceid = MountPointManager::instance()->getIdForUrl( url );
TQString rpath = escapeString( MountPointManager::instance()->getRelativePath( deviceid, url ) );
int id = query( TQString( "SELECT id FROM labels WHERE type = %1 AND name = '%2';" )
.tqarg( type ).tqarg( escapeString( label ) ) ).first().toInt();
bool labelAlreadyExists = id > 0;
if ( !id )
{
id = insert( TQString( "INSERT INTO labels( name, type ) VALUES ( '%2', %1 );" )
.tqarg( type ).tqarg( escapeString( label ) ), "labels" );
}
if ( labelAlreadyExists )
{
//we can return if the link between the tags row and the labels row already exists
int count = query( TQString( "SELECT COUNT(*) FROM tags_labels WHERE labelid = %1 AND deviceid = %2 AND url = '%3';" )
.tqarg( id ).tqarg( deviceid ).tqarg( rpath ) ).first().toInt();
if ( count )
return false;
}
insert( TQString( "INSERT INTO tags_labels( labelid, deviceid, url, uniqueid ) VALUES ( %1, %2, '%3', '%4' );" )
.tqarg( TQString::number(id), TQString::number(deviceid), rpath, escapeString( uid ) ), "tags_labels" );
emit labelsChanged( url );
return true;
}
TQStringList
CollectionDB::favoriteLabels( int type, int count )
{
return query( TQString( "SELECT labels.name, count( tags_labels.labelid ) "
"FROM labels LEFT JOIN tags_labels ON labels.id = tags_labels.labelid "
"WHERE labels.type = %1 GROUP BY labels.name "
"ORDER BY count(tags_labels.labelid) DESC LIMIT %2;" )
.tqarg( TQString::number( type ), TQString::number( count ) ) );
}
TQDir
CollectionDB::largeCoverDir() //static
{
return TQDir( Amarok::saveLocation( "albumcovers/large/" ) );
}
TQDir
CollectionDB::tagCoverDir() //static
{
return TQDir( Amarok::saveLocation( "albumcovers/tagcover/" ) );
}
TQDir
CollectionDB::cacheCoverDir() //static
{
return TQDir( Amarok::saveLocation( "albumcovers/cache/" ) );
}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS DbConnection
//////////////////////////////////////////////////////////////////////////////////////////
DbConnection::DbConnection()
: m_initialized( false )
{}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS SqliteConnection
//////////////////////////////////////////////////////////////////////////////////////////
SqliteConnection::SqliteConnection( const SqliteConfig* config )
: DbConnection()
, m_db( 0 )
{
DEBUG_BLOCK
const TQCString path = TQFile::encodeName( config->dbFile() );
// Open database file and check for correctness
TQFile file( path );
if ( file.open( IO_ReadOnly ) )
{
TQString format;
file.readLine( format, 50 );
if ( !format.startsWith( "SQLite format 3" ) )
{
warning() << "Database versions incompatible. Removing and rebuilding database.\n";
}
else if ( sqlite3_open( path, &m_db ) != SQLITE_OK )
{
warning() << "Database file corrupt. Removing and rebuilding database.\n";
sqlite3_close( m_db );
}
else
m_initialized = true;
}
if ( !m_initialized )
{
// Remove old db file; create new
TQFile::remove( path );
if ( sqlite3_open( path, &m_db ) == SQLITE_OK )
{
m_initialized = true;
}
}
if ( m_initialized )
{
if( sqlite3_create_function(m_db, "rand", 0, SQLITE_UTF8, NULL, sqlite_rand, NULL, NULL) != SQLITE_OK )
m_initialized = false;
if( sqlite3_create_function(m_db, "power", 2, SQLITE_UTF8, NULL, sqlite_power, NULL, NULL) != SQLITE_OK )
m_initialized = false;
if ( sqlite3_create_function(m_db, "like", 2, SQLITE_UTF8, NULL, sqlite_like_new, NULL, NULL) != SQLITE_OK )
m_initialized = false;
if ( sqlite3_create_function(m_db, "like", 3, SQLITE_UTF8, NULL, sqlite_like_new, NULL, NULL) != SQLITE_OK )
m_initialized = false;
}
//optimization for speeding up SQLite
query( "PRAGMA default_synchronous = OFF;" );
}
SqliteConnection::~SqliteConnection()
{
if ( m_db ) sqlite3_close( m_db );
}
TQStringList SqliteConnection::query( const TQString& statement, bool /*suppressDebug*/ )
{
TQStringList values;
int error;
int rc = 0;
const char* tail;
sqlite3_stmt* stmt;
int busyCnt = 0;
int retryCnt = 0;
do {
//compile SQL program to virtual machine, reattempting if busy
do {
if ( busyCnt )
{
::usleep( 100000 ); // Sleep 100 msec
debug() << "sqlite3_prepare: BUSY counter: " << busyCnt << endl;
}
error = sqlite3_prepare( m_db, statement.utf8(), -1, &stmt, &tail );
}
while ( SQLITE_BUSY==error && busyCnt++ < 120 );
if ( error != SQLITE_OK )
{
if ( SQLITE_BUSY==error )
Debug::error() << "Gave up waiting for lock to clear" << endl;
Debug::error() << k_funcinfo << " sqlite3_compile error:" << endl;
Debug::error() << sqlite3_errmsg( m_db ) << endl;
Debug::error() << "on query: " << statement << endl;
values = TQStringList();
break;
}
else
{
busyCnt = 0;
int number = sqlite3_column_count( stmt );
//execute virtual machine by iterating over rows
while ( true )
{
error = sqlite3_step( stmt );
if ( error == SQLITE_BUSY )
{
if ( busyCnt++ > 120 ) {
Debug::error() << "Busy-counter has reached maximum. Aborting this sql statement!\n";
break;
}
::usleep( 100000 ); // Sleep 100 msec
debug() << "sqlite3_step: BUSY counter: " << busyCnt << endl;
continue;
}
if ( error == SQLITE_MISUSE )
debug() << "sqlite3_step: MISUSE" << endl;
if ( error == SQLITE_DONE || error == SQLITE_ERROR )
break;
//iterate over columns
for ( int i = 0; i < number; i++ )
{
values << TQString::fromUtf8( reinterpret_cast<const char*>( sqlite3_column_text( stmt, i ) ) );
}
}
//deallocate vm resources
rc = sqlite3_finalize( stmt );
if ( error != SQLITE_DONE && rc != SQLITE_SCHEMA )
{
Debug::error() << k_funcinfo << "sqlite_step error.\n";
Debug::error() << sqlite3_errmsg( m_db ) << endl;
Debug::error() << "on query: " << statement << endl;
values = TQStringList();
}
if ( rc == SQLITE_SCHEMA )
{
retryCnt++;
debug() << "SQLITE_SCHEMA error occurred on query: " << statement << endl;
if ( retryCnt < 10 )
debug() << "Retrying now." << endl;
else
{
Debug::error() << "Retry-Count has reached maximum. Aborting this SQL statement!" << endl;
Debug::error() << "SQL statement: " << statement << endl;
values = TQStringList();
}
}
}
}
while ( rc == SQLITE_SCHEMA && retryCnt < 10 );
return values;
}
int SqliteConnection::insert( const TQString& statement, const TQString& /* table */ )
{
int error;
int rc = 0;
const char* tail;
sqlite3_stmt* stmt;
int busyCnt = 0;
int retryCnt = 0;
do {
//compile SQL program to virtual machine, reattempting if busy
do {
if ( busyCnt )
{
::usleep( 100000 ); // Sleep 100 msec
debug() << "sqlite3_prepare: BUSY counter: " << busyCnt << endl;
}
error = sqlite3_prepare( m_db, statement.utf8(), -1, &stmt, &tail );
}
while ( SQLITE_BUSY==error && busyCnt++ < 120 );
if ( error != SQLITE_OK )
{
if ( SQLITE_BUSY==error )
Debug::error() << "Gave up waiting for lock to clear" << endl;
Debug::error() << k_funcinfo << " sqlite3_compile error:" << endl;
Debug::error() << sqlite3_errmsg( m_db ) << endl;
Debug::error() << "on insert: " << statement << endl;
break;
}
else
{
busyCnt = 0;
//execute virtual machine by iterating over rows
while ( true )
{
error = sqlite3_step( stmt );
if ( error == SQLITE_BUSY )
{
if ( busyCnt++ > 120 ) {
Debug::error() << "Busy-counter has reached maximum. Aborting this sql statement!\n";
break;
}
::usleep( 100000 ); // Sleep 100 msec
debug() << "sqlite3_step: BUSY counter: " << busyCnt << endl;
}
if ( error == SQLITE_MISUSE )
debug() << "sqlite3_step: MISUSE" << endl;
if ( error == SQLITE_DONE || error == SQLITE_ERROR )
break;
}
//deallocate vm resources
rc = sqlite3_finalize( stmt );
if ( error != SQLITE_DONE && rc != SQLITE_SCHEMA)
{
Debug::error() << k_funcinfo << "sqlite_step error.\n";
Debug::error() << sqlite3_errmsg( m_db ) << endl;
Debug::error() << "on insert: " << statement << endl;
}
if ( rc == SQLITE_SCHEMA )
{
retryCnt++;
debug() << "SQLITE_SCHEMA error occurred on insert: " << statement << endl;
if ( retryCnt < 10 )
debug() << "Retrying now." << endl;
else
{
Debug::error() << "Retry-Count has reached maximum. Aborting this SQL insert!" << endl;
Debug::error() << "SQL statement: " << statement << endl;
}
}
}
}
while ( SQLITE_SCHEMA == rc && retryCnt < 10 );
return sqlite3_last_insert_rowid( m_db );
}
// this implements a RAND() function compatible with the MySQL RAND() (0-param-form without seed)
void SqliteConnection::sqlite_rand(sqlite3_context *context, int /*argc*/, sqlite3_value ** /*argv*/)
{
sqlite3_result_double( context, static_cast<double>(KApplication::random()) / (RAND_MAX+1.0) );
}
// this implements a POWER() function compatible with the MySQL POWER()
void SqliteConnection::sqlite_power(sqlite3_context *context, int argc, sqlite3_value **argv)
{
Q_ASSERT( argc==2 );
if( sqlite3_value_type(argv[0])==SQLITE_NULL || sqlite3_value_type(argv[1])==SQLITE_NULL ) {
sqlite3_result_null(context);
return;
}
double a = sqlite3_value_double(argv[0]);
double b = sqlite3_value_double(argv[1]);
sqlite3_result_double( context, pow(a,b) );
}
// this implements a LIKE() function that overrides the default string comparison function
// Reason: default function is case-sensitive for utf8 strings (BUG: 116458, ...)
void SqliteConnection::sqlite_like_new( sqlite3_context *context, int argc, sqlite3_value **argv )
{
const unsigned char *zA = sqlite3_value_text( argv[0] );
const unsigned char *zB = sqlite3_value_text( argv[1] );
TQString pattern = TQString::fromUtf8( (const char*)zA );
TQString text = TQString::fromUtf8( (const char*)zB );
int begin = pattern.startsWith( "%" ), end = pattern.endsWith( "%" );
if (begin)
pattern = pattern.right( pattern.length() - 1 );
if (end)
pattern = pattern.left( pattern.length() - 1 );
if( argc == 3 ) // The function is given an escape character. In likeCondition() it defaults to '/'
pattern.replace( "/%", "%" ).replace( "/_", "_" ).replace( "//", "/" );
int result = 0;
if ( begin && end ) result = ( text.find( pattern, 0, 0 ) != -1);
else if ( begin ) result = text.tqendsWith( pattern, 0 );
else if ( end ) result = text.tqstartsWith( pattern, 0 );
else result = ( text.lower() == pattern.lower() );
sqlite3_result_int( context, result );
}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS MySqlConnection
//////////////////////////////////////////////////////////////////////////////////////////
#ifdef USE_MYSQL
MySqlConnection::MySqlConnection( const MySqlConfig* config )
: DbConnection()
, m_connected( false )
{
DEBUG_BLOCK
debug() << k_funcinfo << endl;
m_db = mysql_init(NULL);
if (m_db)
{
// if ( config->username().isEmpty() )
// pApp->slotConfigAmarok("MySql");
if ( mysql_real_connect( m_db, config->host().latin1(),
config->username().latin1(),
config->password().latin1(),
config->database().latin1(),
config->port(),
NULL, CLIENT_COMPRESS ) )
{
m_initialized = true;
#if MYSQL_VERSION_ID >= 40113
// now set the right charset for the connection
TQStringList my_qslist = query( "SHOW VARIABLES LIKE 'character_set_database'" );
if( !my_qslist.isEmpty() && !mysql_set_character_set( m_db, const_cast<char *>( my_qslist[1].latin1() ) ) )
//charset was updated
debug() << "Connection Charset is now: " << my_qslist[1].latin1() << endl;
else
error() << "Failed to set database charset\n";
#endif
m_db->reconnect = 1; //setting reconnect flag for newer mysqld
m_connected = true;
}
else
{
if ( mysql_real_connect(
m_db,
config->host().latin1(),
config->username().latin1(),
config->password().latin1(),
NULL,
config->port(),
NULL, CLIENT_COMPRESS))
{
if ( mysql_query(m_db,
TQString( "CREATE DATABASE " + config->database() ).latin1() ) )
{ m_connected = true; m_initialized = true; }
else
{ setMysqlError(); }
}
else
setMysqlError();
}
}
else
error() << "Failed to allocate/initialize MySql struct\n";
}
MySqlConnection::~MySqlConnection()
{
if ( m_db ) mysql_close( m_db );
}
TQStringList MySqlConnection::query( const TQString& statement, bool suppressDebug )
{
TQStringList values;
if ( !mysql_query( m_db, statement.utf8() ) )
{
MYSQL_RES* result;
if ( ( result = mysql_use_result( m_db ) ) )
{
int number = mysql_field_count( m_db );
MYSQL_ROW row;
while ( ( row = mysql_fetch_row( result ) ) )
{
for ( int i = 0; i < number; i++ )
{
values << TQString::fromUtf8( (const char*)row[i] );
}
}
}
else
{
if ( mysql_field_count( m_db ) != 0 )
{
if ( !suppressDebug )
debug() << "MYSQL TQUERY FAILED: " << mysql_error( m_db ) << "\n" << "FAILED TQUERY: " << statement << "\n";
values = TQStringList();
}
}
mysql_free_result( result );
}
else
{
if ( !suppressDebug )
debug() << "MYSQL TQUERY FAILED: " << mysql_error( m_db ) << "\n" << "FAILED TQUERY: " << statement << "\n";
values = TQStringList();
}
return values;
}
int MySqlConnection::insert( const TQString& statement, const TQString& /* table */ )
{
mysql_query( m_db, statement.utf8() );
if ( mysql_errno( m_db ) )
{
debug() << "MYSQL INSERT FAILED: " << mysql_error( m_db ) << "\n" << "FAILED INSERT: " << statement << endl;
}
return mysql_insert_id( m_db );
}
void
MySqlConnection::setMysqlError()
{
m_error = i18n("MySQL reported the following error:<br>") + mysql_error(m_db)
+ i18n("<p>You can configure MySQL in the Collection section under Settings->Configure Amarok</p>");
}
#endif
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS PostgresqlConnection
//////////////////////////////////////////////////////////////////////////////////////////
#ifdef USE_POSTGRESQL
PostgresqlConnection::PostgresqlConnection( const PostgresqlConfig* config )
: DbConnection()
, m_connected( false )
{
TQString conninfo;
debug() << k_funcinfo << endl;
// if ( config->username().isEmpty() )
// pApp->slotConfigAmarok("Postgresql");
conninfo = "host='" + config->host() +
"' port=" + TQString::number( config->port() ) +
" dbname='" + config->database() +
"' user='" + config->username() +
"' password='" + config->password() + '\'';
m_db = PQconnectdb( conninfo.utf8() );
if (!m_db)
{
debug() << "POSTGRESQL CONNECT FAILED: " << PQerrorMessage( m_db ) << "\n";
error() << "Failed to allocate/initialize Postgresql struct\n";
setPostgresqlError();
return;
}
if (PQstatus(m_db) != CONNECTION_OK)
{
debug() << "POSTGRESQL CONNECT FAILED: " << PQerrorMessage( m_db ) << "\n";
error() << "Failed to allocate/initialize Postgresql struct\n";
setPostgresqlError();
PQfinish(m_db);
m_db = NULL;
return;
}
m_initialized = true;
m_connected = true;
}
PostgresqlConnection::~PostgresqlConnection()
{
if ( m_db ) PQfinish( m_db );
}
TQStringList PostgresqlConnection::query( const TQString& statement, bool suppressDebug )
{
TQStringList values;
PGresult* result;
ExecStatusType status;
result = PQexec(m_db, statement.utf8());
if (result == NULL)
{
if ( !suppressDebug )
debug() << "POSTGRESQL TQUERY FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED TQUERY: " << statement << "\n";
return values;
}
status = PQresultStatus(result);
if ((status != PGRES_COMMAND_OK) && (status != PGRES_TUPLES_OK))
{
if ( !suppressDebug )
debug() << "POSTGRESQL TQUERY FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED TQUERY: " << statement << "\n";
PQclear(result);
return values;
}
int cols = PQnfields( result );
int rows = PQntuples( result );
TQMap<int, bool> discardCols;
for(int col=0; col< cols; col++) {
if (TQString(PQfname(result, col)) == TQString("__discard") || TQString(PQfname(result, col)) == TQString("__random"))
{
discardCols[col] = true;
}
}
for(int row=0; row< rows; row++)
{
for(int col=0; col< cols; col++)
{
if (discardCols[col]) continue;
values << TQString::fromUtf8(PQgetvalue(result, row, col));
}
}
PQclear(result);
return values;
}
int PostgresqlConnection::insert( const TQString& statement, const TQString& table )
{
PGresult* result;
ExecStatusType status;
TQString curvalSql;
int id;
result = PQexec(m_db, statement.utf8());
if (result == NULL)
{
debug() << "POSTGRESQL INSERT FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED SQL: " << statement << "\n";
return 0;
}
status = PQresultStatus(result);
if (status != PGRES_COMMAND_OK)
{
debug() << "POSTGRESQL INSERT FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED SQL: " << statement << "\n";
PQclear(result);
return 0;
}
PQclear(result);
if (table == NULL) return 0;
TQString _table = table;
if (table.find("_temp") > 0) _table = table.left(table.find("_temp"));
curvalSql = TQString("SELECT currval('%1_seq');").tqarg(_table);
result = PQexec(m_db, curvalSql.utf8());
if (result == NULL)
{
debug() << "POSTGRESQL INSERT FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED SQL: " << curvalSql << "\n";
return 0;
}
status = PQresultStatus(result);
if (status != PGRES_TUPLES_OK)
{
debug() << "POSTGRESQL INSERT FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED SQL: " << curvalSql << "\n";
PQclear(result);
return 0;
}
if ((PQnfields( result ) != 1) || (PQntuples( result ) != 1))
{
debug() << "POSTGRESQL INSERT FAILED: " << PQerrorMessage( m_db ) << "\n" << "FAILED SQL: " << curvalSql << "\n";
PQclear(result);
return 0;
}
id = TQString::fromUtf8(PQgetvalue(result, 0, 0)).toInt();
PQclear(result);
return id;
}
void PostgresqlConnection::setPostgresqlError()
{
m_error = i18n("Postgresql reported the following error:<br>") + PQerrorMessage(m_db)
+ i18n("<p>You can configure Postgresql in the Collection section under Settings->Configure Amarok</p>");
}
#endif
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS SqliteConfig
//////////////////////////////////////////////////////////////////////////////////////////
SqliteConfig::SqliteConfig( const TQString& dbfile )
: m_dbfile( dbfile )
{}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS MySqlConfig
//////////////////////////////////////////////////////////////////////////////////////////
MySqlConfig::MySqlConfig(
const TQString& host,
const int port,
const TQString& database,
const TQString& username,
const TQString& password )
: m_host( host ),
m_port( port ),
m_database( database ),
m_username( username ),
m_password( password )
{}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS PostgresqlConfig
//////////////////////////////////////////////////////////////////////////////////////////
PostgresqlConfig::PostgresqlConfig(
const TQString& host,
const int port,
const TQString& database,
const TQString& username,
const TQString& password )
: m_host( host ),
m_port( port ),
m_database( database ),
m_username( username ),
m_password( password )
{}
//////////////////////////////////////////////////////////////////////////////////////////
// CLASS QueryBuilder
//////////////////////////////////////////////////////////////////////////////////////////
QueryBuilder::QueryBuilder()
{
m_OR.push(false);
clear();
// there are a few string members with a large number of appends. to
// avoid reallocations, pre-reserve 1024 bytes and try never to assign
// it, instead doing setLength(0) and appends
// Note: unfortunately, QT3's setLength(), which is also called from append,
// squeezes the string if it's less than 4x the length. So this is useless.
// Uncomment after porting to QT4 if it's smarter about this, as the docs say.
// m_query.reserve(1024);
// m_values.reserve(1024);
// m_tables.reserve(1024);
}
void
QueryBuilder::linkTables( int tables )
{
m_tables.setLength(0);
m_tables += tableName( tabSong );
if ( !(tables & tabSong ) )
{
// check if only one table is selected (does somebody know a better way to check that?)
if (tables == tabAlbum || tables==tabArtist || tables==tabGenre || tables == tabYear || tables == tabStats || tables == tabPodcastEpisodes || tables == tabPodcastFolders || tables == tabPodcastChannels || tables == tabLabels) {
m_tables.setLength( 0 );
m_tables += tableName(tables);
}
else
tables |= tabSong;
}
if ( tables & tabSong )
{
if ( tables & tabAlbum )
((m_tables += " LEFT JOIN ") += tableName( tabAlbum)) += " ON album.id=tags.album";
if ( tables & tabArtist )
((m_tables += " LEFT JOIN ") += tableName( tabArtist)) += " ON artist.id=tags.artist";
if ( tables & tabComposer )
((m_tables += " LEFT JOIN ") += tableName( tabComposer)) += " ON composer.id=tags.composer";
if ( tables & tabGenre )
((m_tables += " LEFT JOIN ") += tableName( tabGenre)) += " ON genre.id=tags.genre";
if ( tables & tabYear )
((m_tables += " LEFT JOIN ") += tableName( tabYear)) += " ON year.id=tags.year";
if ( tables & tabStats )
{
((m_tables += " LEFT JOIN ") += tableName( tabStats))
+= " ON statistics.url=tags.url AND statistics.deviceid = tags.deviceid";
//if ( !m_url.isEmpty() ) {
// TQString url = TQString( '.' ) + m_url;
// m_tables += TQString( " OR statistics.deviceid = -1 AND statistics.url = '%1'" )
// .tqarg( CollectionDB::instance()->escapeString( url ) );
//}
}
if ( tables & tabLyrics )
((m_tables += " LEFT JOIN ") += tableName( tabLyrics))
+= " ON lyrics.url=tags.url AND lyrics.deviceid = tags.deviceid";
if ( tables & tabDevices )
((m_tables += " LEFT JOIN ") += tableName( tabDevices )) += " ON tags.deviceid = devices.id";
if ( tables & tabLabels )
( m_tables += " LEFT JOIN tags_labels ON tags.url = tags_labels.url AND tags.deviceid = tags_labels.deviceid" )
+= " LEFT JOIN labels ON tags_labels.labelid = labels.id";
}
}
void
QueryBuilder::addReturnValue( int table, TQ_INT64 value, bool caseSensitive /* = false, unless value refers to a string */ )
{
caseSensitive |= value == valName || value == valTitle || value == valComment;
if ( !m_values.isEmpty() && m_values != "DISTINCT " ) m_values += ',';
if ( value == valDummy )
m_values += "''";
else
{
if ( caseSensitive && CollectionDB::instance()->getType() == DbConnection::mysql )
m_values += "BINARY ";
m_values += tableName( table ) + '.';
m_values += valueName( value );
}
m_linkTables |= table;
m_returnValues++;
if ( value & valURL )
{
// make handling of deviceid transparent to calling code
m_deviceidPos = m_returnValues + 1; //the return value after the url is the deviceid
m_values += ',';
m_values += tableName( table );
m_values += '.';
m_values += valueName( valDeviceId );
}
}
void
QueryBuilder::addReturnFunctionValue( int function, int table, TQ_INT64 value)
{
// translate NULL and 0 values into the default value for percentage/rating
// First translate 0 to NULL via NULLIF, then NULL to default via COALESCE
bool defaults = function == funcAvg && ( value & valScore || value & valRating );
if ( !m_values.isEmpty() && m_values != "DISTINCT " ) m_values += ',';
m_values += functionName( function ) + '(';
if ( defaults )
m_values += "COALESCE(NULLIF(";
m_values += tableName( table ) + '.';
m_values += valueName( value );
if ( defaults )
{
m_values += ", 0), ";
if ( value & valScore )
m_values += "50";
else
m_values += '6';
m_values += ')';
}
m_values += ") AS ";
m_values += functionName( function )+tableName( table )+valueName( value );
m_linkTables |= table;
if ( !m_showAll ) m_linkTables |= tabSong;
m_returnValues++;
}
uint
QueryBuilder::countReturnValues()
{
return m_returnValues;
}
void
QueryBuilder::addURLFilters( const TQStringList& filter )
{
if ( !filter.isEmpty() )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolF() + ' ';
for ( uint i = 0; i < filter.count(); i++ )
{
int deviceid = MountPointManager::instance()->getIdForUrl( filter[i] );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid , filter[i] );
m_where += "OR (tags.url = '" + CollectionDB::instance()->escapeString( rpath ) + "' ";
m_where += TQString( "AND tags.deviceid = %1 ) " ).tqarg( TQString::number( deviceid ) );
//TODO MountPointManager fix this
}
m_where += " ) ";
}
m_linkTables |= tabSong;
}
void
QueryBuilder::setGoogleFilter( int defaultTables, TQString query )
{
//TODO MountPointManager fix google syntax
//no clue about what needs to be done atm
ParsedExpression parsed = ExpressionParser::parse( query );
for( uint i = 0, n = parsed.count(); i < n; ++i ) //check each part for matchiness
{
beginOR();
for( uint ii = 0, nn = parsed[i].count(); ii < nn; ++ii )
{
const expression_element &e = parsed[i][ii];
TQString s = e.text;
int mode;
switch( e.match )
{
case expression_element::More: mode = modeGreater; break;
case expression_element::Less: mode = modeLess; break;
case expression_element::Contains:
default: mode = modeNormal; break;
}
bool exact = false; // enable for numeric values
int table = -1;
TQ_INT64 value = -1;
if( e.field == "artist" )
table = tabArtist;
else if( e.field == "composer" )
table = tabComposer;
else if( e.field == "album" )
table = tabAlbum;
else if( e.field == "title" )
table = tabSong;
else if( e.field == "genre" )
table = tabGenre;
else if( e.field == "year" )
{
table = tabYear;
value = valName;
exact = true;
}
else if( e.field == "score" )
{
table = tabStats;
value = valScore;
exact = true;
}
else if( e.field == "rating" )
{
table = tabStats;
value = valRating;
exact = true;
s = TQString::number( int( s.toFloat() * 2 ) );
}
else if( e.field == "directory" )
{
table = tabSong;
value = valDirectory;
}
else if( e.field == "length" )
{
table = tabSong;
value = valLength;
exact = true;
}
else if( e.field == "playcount" )
{
table = tabStats;
value = valPlayCounter;
exact = true;
}
else if( e.field == "samplerate" )
{
table = tabSong;
value = valSamplerate;
exact = true;
}
else if( e.field == "track" )
{
table = tabSong;
value = valTrack;
exact = true;
}
else if( e.field == "disc" || e.field == "discnumber" )
{
table = tabSong;
value = valDiscNumber;
exact = true;
}
else if( e.field == "size" || e.field == "filesize" )
{
table = tabSong;
value = valFilesize;
exact = true;
if( s.lower().endsWith( "m" ) )
s = TQString::number( s.left( s.length()-1 ).toLong() * 1024 * 1024 );
else if( s.lower().endsWith( "k" ) )
s = TQString::number( s.left( s.length()-1 ).toLong() * 1024 );
}
else if( e.field == "filename" || e.field == "url" )
{
table = tabSong;
value = valURL;
}
else if( e.field == "filetype" || e.field == "type" )
{
table = tabSong;
value = valURL;
mode = modeEndMatch;
s.prepend( '.' );
}
else if( e.field == "bitrate" )
{
table = tabSong;
value = valBitrate;
exact = true;
}
else if( e.field == "comment" )
{
table = tabSong;
value = valComment;
}
else if( e.field == "bpm" )
{
table = tabSong;
value = valBPM;
exact = true;
}
else if( e.field == "lyrics" )
{
table = tabLyrics;
value = valLyrics;
}
else if( e.field == "device" )
{
table = tabDevices;
value = valDeviceLabel;
}
else if( e.field == "mountpoint" )
{
table = tabDevices;
value = valMountPoint;
}
else if( e.field == "label" )
{
table = tabLabels;
value = valName;
}
if( e.negate )
{
if( value >= 0 )
excludeFilter( table, value, s, mode, exact );
else
excludeFilter( table >= 0 ? table : defaultTables, s );
}
else
{
if( value >= 0 )
addFilter( table, value, s, mode, exact );
else
addFilter( table >= 0 ? table : defaultTables, s );
}
}
endOR();
}
}
void
QueryBuilder::addFilter( int tables, const TQString& filter )
{
if ( !filter.isEmpty() )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolF() + ' ';
if ( tables & tabAlbum )
m_where += "OR album.name " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabArtist )
m_where += "OR artist.name " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabComposer )
m_where += "OR composer.name " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabGenre )
m_where += "OR genre.name " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabYear )
m_where += "OR year.name " + CollectionDB::likeCondition( filter, false, false );
if ( tables & tabSong )
m_where += "OR tags.title " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabLabels )
m_where += "OR labels.name " + CollectionDB::likeCondition( filter, true, true );
if ( i18n( "Unknown" ).contains( filter, false ) )
{
if ( tables & tabAlbum )
m_where += "OR album.name = '' ";
if ( tables & tabArtist )
m_where += "OR artist.name = '' ";
if ( tables & tabComposer )
m_where += "OR composer.name = '' ";
if ( tables & tabGenre )
m_where += "OR genre.name = '' ";
if ( tables & tabYear )
m_where += "OR year.name = '' ";
if ( tables & tabSong )
m_where += "OR tags.title = '' ";
}
if ( ( tables & tabArtist ) && i18n( "Various Artists" ).contains( filter, false ) )
m_where += TQString( "OR tags.sampler = %1 " ).tqarg( CollectionDB::instance()->boolT() );
m_where += " ) ";
}
m_linkTables |= tables;
}
void
QueryBuilder::addFilter( int tables, TQ_INT64 value, const TQString& filter, int mode, bool exact )
{
//true for INTEGER fields (see comment of coalesceField(int, TQ_INT64)
bool useCoalesce = coalesceField( tables, value );
m_where += ANDslashOR() + " ( ";
TQString m, s;
if (mode == modeLess || mode == modeGreater)
{
TQString escapedFilter;
if (useCoalesce && DbConnection::sqlite == CollectionDB::instance()->getDbConnectionType())
escapedFilter = CollectionDB::instance()->escapeString( filter );
else
escapedFilter = "'" + CollectionDB::instance()->escapeString( filter ) + "' ";
s = ( mode == modeLess ? "< " : "> " ) + escapedFilter;
}
else
{
if (exact)
if (useCoalesce && DbConnection::sqlite == CollectionDB::instance()->getDbConnectionType())
s = " = " +CollectionDB::instance()->escapeString( filter ) + ' ';
else
s = " = '" + CollectionDB::instance()->escapeString( filter ) + "' ";
else
s = CollectionDB::likeCondition( filter, mode != modeBeginMatch, mode != modeEndMatch );
}
if( coalesceField( tables, value ) )
m_where += TQString( "COALESCE(%1.%2,0) " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) ) + s;
else
m_where += TQString( "%1.%2 " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) ) + s;
if ( !exact && (value & valName) && mode == modeNormal && i18n( "Unknown").contains( filter, false ) )
m_where += TQString( "OR %1.%2 = '' " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) );
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::addFilters( int tables, const TQStringList& filter )
{
if ( !filter.isEmpty() )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolT() + ' ';
for ( uint i = 0; i < filter.count(); i++ )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolF() + ' ';
if ( tables & tabAlbum )
m_where += "OR album.name " + CollectionDB::likeCondition( filter[i], true, true );
if ( tables & tabArtist )
m_where += "OR artist.name " + CollectionDB::likeCondition( filter[i], true, true );
if ( tables & tabComposer )
m_where += "OR composer.name " + CollectionDB::likeCondition( filter[i], true, true );
if ( tables & tabGenre )
m_where += "OR genre.name " + CollectionDB::likeCondition( filter[i], true, true );
if ( tables & tabYear )
m_where += "OR year.name " + CollectionDB::likeCondition( filter[i], false, false );
if ( tables & tabSong )
m_where += "OR tags.title " + CollectionDB::likeCondition( filter[i], true, true );
if ( tables & tabLabels )
m_where += "OR labels.name " + CollectionDB::likeCondition( filter[i], true, true );
if ( i18n( "Unknown" ).contains( filter[i], false ) )
{
if ( tables & tabAlbum )
m_where += "OR album.name = '' ";
if ( tables & tabArtist )
m_where += "OR artist.name = '' ";
if ( tables & tabComposer )
m_where += "OR composer.name = '' ";
if ( tables & tabGenre )
m_where += "OR genre.name = '' ";
if ( tables & tabYear )
m_where += "OR year.name = '' ";
if ( tables & tabSong )
m_where += "OR tags.title = '' ";
}
if ( i18n( "Various Artists" ).contains( filter[ i ], false ) && ( tables & tabArtist ) )
m_where += "OR tags.sampler = " + CollectionDB::instance()->boolT() + ' ';
m_where += " ) ";
}
m_where += " ) ";
}
m_linkTables |= tables;
}
void
QueryBuilder::excludeFilter( int tables, const TQString& filter )
{
if ( !filter.isEmpty() )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolT() + ' ';
if ( tables & tabAlbum )
m_where += "AND album.name NOT " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabArtist )
m_where += "AND artist.name NOT " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabComposer )
m_where += "AND composer.name NOT " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabGenre )
m_where += "AND genre.name NOT " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabYear )
m_where += "AND year.name NOT " + CollectionDB::likeCondition( filter, false, false );
if ( tables & tabSong )
m_where += "AND tags.title NOT " + CollectionDB::likeCondition( filter, true, true );
if ( tables & tabLabels )
m_where += "AND labels.name NOT " + CollectionDB::likeCondition( filter, true, true );
if ( i18n( "Unknown" ).contains( filter, false ) )
{
if ( tables & tabAlbum )
m_where += "AND album.name <> '' ";
if ( tables & tabArtist )
m_where += "AND artist.name <> '' ";
if ( tables & tabComposer )
m_where += "AND composer.name <> '' ";
if ( tables & tabGenre )
m_where += "AND genre.name <> '' ";
if ( tables & tabYear )
m_where += "AND year.name <> '' ";
if ( tables & tabSong )
m_where += "AND tags.title <> '' ";
}
if ( i18n( "Various Artists" ).contains( filter, false ) && ( tables & tabArtist ) )
m_where += "AND tags.sampler = " + CollectionDB::instance()->boolF() + ' ';
m_where += " ) ";
}
m_linkTables |= tables;
}
void
QueryBuilder::excludeFilter( int tables, TQ_INT64 value, const TQString& filter, int mode, bool exact )
{
m_where += ANDslashOR() + " ( ";
TQString m, s;
if (mode == modeLess || mode == modeGreater)
s = ( mode == modeLess ? ">= '" : "<= '" ) + CollectionDB::instance()->escapeString( filter ) + "' ";
else
{
if (exact)
{
bool isNumber;
filter.toInt( &isNumber );
if (isNumber)
s = " <> " + CollectionDB::instance()->escapeString( filter ) + " ";
else
s = " <> '" + CollectionDB::instance()->escapeString( filter ) + "' ";
}
else
s = "NOT " + CollectionDB::instance()->likeCondition( filter, mode != modeBeginMatch, mode != modeEndMatch ) + ' ';
}
if( coalesceField( tables, value ) )
m_where += TQString( "COALESCE(%1.%2,0) " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) ) + s;
else
m_where += TQString( "%1.%2 " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) ) + s;
if ( !exact && (value & valName) && mode == modeNormal && i18n( "Unknown").contains( filter, false ) )
m_where += TQString( "AND %1.%2 <> '' " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) );
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::addMatch( int tables, const TQString& match, bool interpretUnknown /* = true */, bool caseSensitive /* = true */ )
{
TQString matchCondition = caseSensitive ? CollectionDB::exactCondition( match ) : CollectionDB::likeCondition( match );
(((m_where += ANDslashOR()) += " ( ") += CollectionDB::instance()->boolF()) += ' ';
if ( tables & tabAlbum )
(m_where += "OR album.name ") += matchCondition;
if ( tables & tabArtist )
(m_where += "OR artist.name ") += matchCondition;
if ( tables & tabComposer )
(m_where += "OR composer.name ") += matchCondition;
if ( tables & tabGenre )
(m_where += "OR genre.name ") += matchCondition;
if ( tables & tabYear )
(m_where += "OR year.name ") += matchCondition;
if ( tables & tabSong )
(m_where += "OR tags.title ") += matchCondition;
if ( tables & tabLabels )
(m_where += "OR labels.name ") += matchCondition;
static TQString i18nUnknown = i18n("Unknown");
if ( interpretUnknown && match == i18nUnknown )
{
if ( tables & tabAlbum ) m_where += "OR album.name = '' ";
if ( tables & tabArtist ) m_where += "OR artist.name = '' ";
if ( tables & tabComposer ) m_where += "OR composer.name = '' ";
if ( tables & tabGenre ) m_where += "OR genre.name = '' ";
if ( tables & tabYear ) m_where += "OR year.name = '' ";
}
if ( tables & tabLabels && match.isEmpty() )
m_where += " OR labels.name IS NULL ";
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::addMatch( int tables, TQ_INT64 value, const TQString& match, bool interpretUnknown /* = true */, bool caseSensitive /* = true */ )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolF() + ' ';
if ( value & valURL )
m_url = match;
//FIXME max: doesn't work yet if we are querying for the mount point part of a directory
if ( value & valURL || value & valDirectory )
{
int deviceid = MountPointManager::instance()->getIdForUrl( match );
TQString rpath = MountPointManager::instance()->getRelativePath( deviceid, match );
//we are querying for a specific path, so we don't need the tags.deviceid IN (...) stuff
//which is automatially appended if m_showAll = false
m_showAll = true;
m_where += TQString( "OR %1.%2 " )
.tqarg( tableName( tables ) )
.tqarg( valueName( value ) );
m_where += caseSensitive ? CollectionDB::exactCondition( rpath ) : CollectionDB::likeCondition( rpath );
m_where += TQString( " AND %1.deviceid = %2 " ).tqarg( tableName( tables ) ).tqarg( deviceid );
if ( deviceid != -1 )
{
//handle corner case
TQString rpath2( '.' + match );
m_where += TQString( " OR %1.%2 " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) );
m_where += caseSensitive ? CollectionDB::exactCondition( rpath2 ) : CollectionDB::likeCondition( rpath2 );
m_where += TQString( " AND %1.deviceid = -1 " ).tqarg( tableName( tables ) );
}
}
else
{
m_where += TQString( "OR %1.%2 " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) );
m_where += caseSensitive ? CollectionDB::exactCondition( match ) : CollectionDB::likeCondition( match );
}
if ( ( value & valName ) && interpretUnknown && match == i18n( "Unknown" ) )
m_where += TQString( "OR %1.%2 = '' " ).tqarg( tableName( tables ) ).tqarg( valueName( value ) );
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::addMatches( int tables, const TQStringList& match, bool interpretUnknown /* = true */, bool caseSensitive /* = true */ )
{
TQStringList matchConditions;
for ( uint i = 0; i < match.count(); i++ )
matchConditions << ( caseSensitive ? CollectionDB::exactCondition( match[i] ) : CollectionDB::likeCondition( match[i] ) );
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolF() + ' ';
for ( uint i = 0; i < match.count(); i++ )
{
if ( tables & tabAlbum )
m_where += "OR album.name " + matchConditions[ i ];
if ( tables & tabArtist )
m_where += "OR artist.name " + matchConditions[ i ];
if ( tables & tabComposer )
m_where += "OR composer.name " + matchConditions[ i ];
if ( tables & tabGenre )
m_where += "OR genre.name " + matchConditions[ i ];
if ( tables & tabYear )
m_where += "OR year.name " + matchConditions[ i ];
if ( tables & tabSong )
m_where += "OR tags.title " + matchConditions[ i ];
if ( tables & tabStats )
m_where += "OR statistics.url " + matchConditions[ i ];
if ( tables & tabLabels )
(m_where += "OR labels.name ") += matchConditions[ i ];
if ( interpretUnknown && match[i] == i18n( "Unknown" ) )
{
if ( tables & tabAlbum ) m_where += "OR album.name = '' ";
if ( tables & tabArtist ) m_where += "OR artist.name = '' ";
if ( tables & tabComposer ) m_where += "OR composer.name = '' ";
if ( tables & tabGenre ) m_where += "OR genre.name = '' ";
if ( tables & tabYear ) m_where += "OR year.name = '' ";
}
if ( tables & tabLabels && match[i].isEmpty() )
m_where += " OR labels.name IS NULL ";
}
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::excludeMatch( int tables, const TQString& match )
{
m_where += ANDslashOR() + " ( " + CollectionDB::instance()->boolT() + ' ';
if ( tables & tabAlbum ) m_where += "AND album.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabArtist ) m_where += "AND artist.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabComposer ) m_where += "AND composer.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabGenre ) m_where += "AND genre.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabYear ) m_where += "AND year.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabSong ) m_where += "AND tags.title <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( tables & tabLabels ) m_where += "AND labels.name <> '" + CollectionDB::instance()->escapeString( match ) + "' ";
if ( match == i18n( "Unknown" ) )
{
if ( tables & tabAlbum ) m_where += "AND album.name <> '' ";
if ( tables & tabArtist ) m_where += "AND artist.name <> '' ";
if ( tables & tabComposer ) m_where += "AND composer.name <> '' ";
if ( tables & tabGenre ) m_where += "AND genre.name <> '' ";
if ( tables & tabYear ) m_where += "AND year.name <> '' ";
}
m_where += " ) ";
m_linkTables |= tables;
}
void
QueryBuilder::exclusiveFilter( int tableMatching, int tableNotMatching, TQ_INT64 value )
{
m_where += " AND ";
m_where += tableName( tableNotMatching ) + '.';
m_where += valueName( value );
m_where += " IS null ";
m_linkTables |= tableMatching;
m_linkTables |= tableNotMatching;
}
void
QueryBuilder::addNumericFilter(int tables, TQ_INT64 value, const TQString &n,
int mode /* = modeNormal */,
const TQString &endRange /* = TQString() */ )
{
m_where.append( ANDslashOR() ).append( " ( " );
if ( coalesceField( tables, value) )
m_where.append("COALESCE(");
m_where.append( tableName( tables ) ).append( '.' ).append( valueName( value ) );
if ( coalesceField( tables, value) )
m_where.append(",0)");
switch (mode) {
case modeNormal:
m_where.append( " = " ); break;
case modeLess:
m_where.append( " < " ); break;
case modeGreater:
m_where.append( " > " ); break;
case modeBetween:
m_where.append( " BETWEEN " ); break;
case modeNotBetween:
m_where.append(" NOT BETWEEN "); break;
default:
qWarning( "Unhandled mode in addNumericFilter, using equals: %d", mode );
m_where.append( " = " );
}
m_where.append( n );
if ( mode == modeBetween || mode == modeNotBetween )
m_where.append( " AND " ).append( endRange );
m_where.append( " ) " );
m_linkTables |= tables;
}
void
QueryBuilder::setOptions( int options )
{
if ( options & optNoCompilations || options & optOnlyCompilations )
m_linkTables |= tabSong;
if ( options & optNoCompilations ) m_where += TQString("AND tags.sampler = %1 ").tqarg(CollectionDB::instance()->boolF());
if ( options & optOnlyCompilations ) m_where += TQString("AND tags.sampler = %1 ").tqarg(CollectionDB::instance()->boolT());
if (CollectionDB::instance()->getType() == DbConnection::postgresql && options & optRemoveDuplicates && options & optRandomize)
{
m_values = "DISTINCT " + CollectionDB::instance()->randomFunc() + " AS __random "+ m_values;
if ( !m_sort.isEmpty() )
m_sort += ',';
m_sort += CollectionDB::instance()->randomFunc() + ' ';
}
else
{
if ( options & optRemoveDuplicates )
m_values = "DISTINCT " + m_values;
if ( options & optRandomize )
{
if ( !m_sort.isEmpty() ) m_sort += ',';
m_sort += CollectionDB::instance()->randomFunc() + ' ';
}
}
if ( options & optShowAll ) m_showAll = true;
}
void
QueryBuilder::sortBy( int table, TQ_INT64 value, bool descending )
{
//shall we sort case-sensitively? (not for integer columns!)
bool b = true;
if ( value & valID || value & valTrack || value & valScore || value & valRating || value & valLength || value & valBitrate ||
value & valSamplerate || value & valPlayCounter || value & valAccessDate || value & valCreateDate ||
value & valFilesize || value & valDiscNumber ||
table & tabYear )
b = false;
// only coalesce for certain columns
bool c = false;
if ( value & valScore || value & valRating || value & valPlayCounter || value & valAccessDate || value & valCreateDate )
c = true;
if ( !m_sort.isEmpty() ) m_sort += ',';
if ( b ) m_sort += "LOWER( ";
if ( c ) m_sort += "COALESCE( ";
m_sort += tableName( table ) + '.';
m_sort += valueName( value );
if ( c ) m_sort += ", 0 )";
if ( b ) m_sort += " ) ";
if ( descending ) m_sort += " DESC ";
if (CollectionDB::instance()->getType() == DbConnection::postgresql)
{
if (!m_values.isEmpty()) m_values += ',';
if ( b ) m_values += "LOWER( ";
m_values += tableName( table ) + '.';
m_values += valueName( value );
if ( b ) m_values += ')';
m_values += " as __discard ";
}
m_linkTables |= table;
}
void
QueryBuilder::sortByFunction( int function, int table, TQ_INT64 value, bool descending )
{
// This function should be used with the equivalent addReturnFunctionValue (with the same function on same values)
// since it uses the "func(table.value) AS functablevalue" definition.
// this column is already coalesced, but need to reconstruct for postgres
bool defaults = function == funcAvg && ( value & valScore || value & valRating );
//shall we sort case-sensitively? (not for integer columns!)
bool b = true;
if ( value & valID || value & valTrack || value & valScore || value & valRating || value & valLength || value & valBitrate ||
value & valSamplerate || value & valPlayCounter || value & valAccessDate || value & valCreateDate ||
value & valFilesize || value & valDiscNumber ||
table & tabYear )
b = false;
// only coalesce for certain columns
bool c = false;
if ( !defaults && ( value & valScore || value & valRating || value & valPlayCounter || value & valAccessDate || value & valCreateDate ) )
c = true;
if ( !m_sort.isEmpty() ) m_sort += ',';
//m_sort += functionName( function ) + '(';
if ( b ) m_sort += "LOWER( ";
if ( c && CollectionDB::instance()->getType() != DbConnection::mysql) m_sort += "COALESCE( ";
TQString columnName;
if (CollectionDB::instance()->getType() == DbConnection::postgresql)
{
columnName = functionName( function ) + '(';
if ( defaults )
columnName += "COALESCE(NULLIF(";
columnName += tableName( table )+'.'+valueName( value );
if ( defaults )
{
columnName += ", 0), ";
if ( value & valScore )
columnName += "50";
else
columnName += '6';
columnName += ')';
}
columnName += ')';
}
else
columnName = functionName( function )+tableName( table )+valueName( value );
m_sort += columnName;
if ( c && CollectionDB::instance()->getType() != DbConnection::mysql) m_sort += ", 0 )";
if ( b ) m_sort += " ) ";
//m_sort += " ) ";
if ( descending ) m_sort += " DESC ";
m_linkTables |= table;
}
void
QueryBuilder::groupBy( int table, TQ_INT64 value )
{
if ( !m_group.isEmpty() ) m_group += ',';
//Do case-sensitive comparisons for MySQL too. See also QueryBuilder::addReturnValue
if ( DbConnection::mysql == CollectionDB::instance()->getDbConnectionType() &&
( value == valName || value == valTitle || value == valComment ) )
{
m_group += "BINARY ";
}
m_group += tableName( table ) + '.';
m_group += valueName( value );
m_linkTables |= table;
}
void
QueryBuilder::having( int table, TQ_INT64 value, int function, int mode, const TQString& match )
{
if( !m_having.isEmpty() ) m_having += " AND ";
TQString fn = functionName( function );
fn.isEmpty() ?
m_having += tableName( table ) + '.' + valueName( value ) :
m_having += functionName( function )+'('+tableName( table )+'.'+valueName( value )+')';
switch( mode )
{
case modeNormal:
m_having += '=' + match;
break;
case modeLess:
m_having += '<' + match;
break;
case modeGreater:
m_having += '>' + match;
default:
break;
}
}
void
QueryBuilder::setLimit( int startPos, int length )
{
m_limit = TQString( " LIMIT %2 OFFSET %1 " ).tqarg( startPos ).tqarg( length );
}
void
QueryBuilder::shuffle( int table, TQ_INT64 value )
{
if ( !m_sort.isEmpty() ) m_sort += " , ";
if ( table == 0 || value == 0 ) {
// simple random
m_sort += CollectionDB::instance()->randomFunc();
} else {
// This is the score weighted random order.
// The RAND() function returns random values equally distributed between 0.0
// (inclusive) and 1.0 (exclusive). The obvious way to get this order is to
// put every track <score> times into a list, sort the list by RAND()
// (i.e. shuffle it) and discard every occurrence of every track but the very
// first of each. By putting every track into the list only once but applying
// a transfer function T_s(x) := 1-(1-x)^(1/s) where s is the score, to RAND()
// before sorting the list, exactly the same distribution of tracks can be
// achieved (for a proof write to Stefan Siegel <kde@sdas.de>)
// In the query below a simplified function is used: The score is incremented
// by one to prevent division by zero, RAND() is used instead of 1-RAND()
// because it doesn't matter if it becomes zero (the exponent is always
// non-zero), and finally POWER(...) is used instead of 1-POWER(...) because it
// only changes the order type.
m_sort += TQString("POWER( %1, 1.0 / (%2.%3 + 1) ) DESC")
.tqarg( CollectionDB::instance()->randomFunc() )
.tqarg( tableName( table ) )
.tqarg( valueName( value ) );
m_linkTables |= table;
}
}
/* NOTE: It's important to keep these two functions and the const in sync! */
/* NOTE: It's just as important to keep tags.url first! */
const int
QueryBuilder::dragFieldCount = 21;
TQString
QueryBuilder::dragSQLFields()
{
return "tags.url, tags.deviceid, album.name, artist.name, composer.name, "
"genre.name, tags.title, year.name, "
"tags.comment, tags.track, tags.bitrate, tags.discnumber, "
"tags.length, tags.samplerate, tags.filesize, "
"tags.sampler, tags.filetype, tags.bpm, "
"statistics.percentage, statistics.rating, statistics.playcounter, "
"statistics.accessdate";
}
void
QueryBuilder::initSQLDrag()
{
clear();
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valURL );
addReturnValue( QueryBuilder::tabAlbum, QueryBuilder::valName );
addReturnValue( QueryBuilder::tabArtist, QueryBuilder::valName );
addReturnValue( QueryBuilder::tabComposer, QueryBuilder::valName );
addReturnValue( QueryBuilder::tabGenre, QueryBuilder::valName );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTitle );
addReturnValue( QueryBuilder::tabYear, QueryBuilder::valName );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valComment );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valTrack );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valBitrate );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valDiscNumber );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valLength );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valSamplerate );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFilesize );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valIsCompilation );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valFileType );
addReturnValue( QueryBuilder::tabSong, QueryBuilder::valBPM );
addReturnValue( QueryBuilder::tabStats, QueryBuilder::valScore );
addReturnValue( QueryBuilder::tabStats, QueryBuilder::valRating );
addReturnValue( QueryBuilder::tabStats, QueryBuilder::valPlayCounter );
addReturnValue( QueryBuilder::tabStats, QueryBuilder::valAccessDate );
}
void
QueryBuilder::buildQuery( bool withDeviceidPlaceholder )
{
if ( m_query.isEmpty() )
{
linkTables( m_linkTables );
m_query += "SELECT ";
m_query += m_values;
m_query += " FROM ";
m_query += m_tables;
m_query += ' ';
m_query += m_join;
m_query += " WHERE ";
m_query += CollectionDB::instance()->boolT();
m_query += ' ';
m_query += m_where;
if ( !m_showAll && ( m_linkTables & tabSong || m_tables.contains( tableName( tabSong) ) ) ) //Only stuff on mounted devices, unless you use optShowAll
{
if ( withDeviceidPlaceholder )
m_query += "(*MountedDeviceSelection*)";
else
{
IdList list = MountPointManager::instance()->getMountedDeviceIds();
//debug() << "number of device ids " << list.count() << endl;
m_query += " AND tags.deviceid IN (";
foreachType( IdList, list )
{
if ( it != list.begin() ) m_query += ',';
m_query += TQString::number( *it );
}
m_query += ')';
}
}
// GROUP BY must be before ORDER BY for sqlite
// HAVING must be between GROUP BY and ORDER BY
if ( !m_group.isEmpty() ) { m_query += " GROUP BY "; m_query += m_group; }
if ( !m_having.isEmpty() ) { m_query += " HAVING "; m_query += m_having; }
if ( !m_sort.isEmpty() ) { m_query += " ORDER BY "; m_query += m_sort; }
m_query += m_limit;
m_query += ';';
}
}
// get the builded SQL-Query (used in smartplaylisteditor soon)
TQString
QueryBuilder::getQuery()
{
if ( m_query.isEmpty())
{
buildQuery();
}
return m_query;
}
TQStringList
QueryBuilder::run()
{
buildQuery();
//debug() << m_query << endl;
TQStringList rs = CollectionDB::instance()->query( m_query );
//calling code is unaware of the dynamic collection implementation, it simply expects an URL
if( m_deviceidPos > 0 )
return cleanURL( rs );
else
return rs;
}
void
QueryBuilder::clear()
{
m_query.setLength(0);
m_values.setLength(0);
m_tables.setLength(0);
m_join.setLength(0);
m_where.setLength(0);
m_sort.setLength(0);
m_group.setLength(0);
m_limit.setLength(0);
m_having.setLength(0);
m_linkTables = 0;
m_returnValues = 0;
m_showAll = false;
m_deviceidPos = 0;
}
TQ_INT64
QueryBuilder::valForFavoriteSorting() {
TQ_INT64 favSortBy = valRating;
if ( !AmarokConfig::useScores() && !AmarokConfig::useRatings() )
favSortBy = valPlayCounter;
else if( !AmarokConfig::useRatings() )
favSortBy = valScore;
return favSortBy;
}
void
QueryBuilder::sortByFavorite() {
if ( AmarokConfig::useRatings() )
sortBy(tabStats, valRating, true );
if ( AmarokConfig::useScores() )
sortBy(tabStats, valScore, true );
sortBy(tabStats, valPlayCounter, true );
}
void
QueryBuilder::sortByFavoriteAvg() {
// Due to MySQL4 weirdness, we need to add the function we're using to sort
// as return values as well.
if ( AmarokConfig::useRatings() ) {
sortByFunction(funcAvg, tabStats, valRating, true );
addReturnFunctionValue( funcAvg, tabStats, valRating );
}
if ( AmarokConfig::useScores() ) {
sortByFunction(funcAvg, tabStats, valScore, true );
addReturnFunctionValue( funcAvg, tabStats, valScore );
}
sortByFunction(funcAvg, tabStats, valPlayCounter, true );
addReturnFunctionValue( funcAvg, tabStats, valPlayCounter );
//exclude unrated and unplayed
if( !m_having.isEmpty() )
m_having += " AND ";
m_having += " (";
if (AmarokConfig::useRatings() )
m_having += TQString("%1(%2.%3) > 0 OR ")
.tqarg( functionName( funcAvg ), tableName(tabStats), valueName(valRating) );
m_having += TQString("%1(%2.%3) > 0")
.tqarg( functionName( funcAvg ), tableName(tabStats), valueName(valPlayCounter) );
m_having += ")";
}
// Helper method -- given a value, returns the index of the bit that is
// set, if only one, otherwise returns -1
// Binsearch seems appropriate since the values enum has 40 members
template<class ValueType>
static inline int
searchBit( ValueType value, int numBits ) {
int low = 0, high = numBits - 1;
while( low <= high ) {
int mid = (low + high) / 2;
ValueType compare = static_cast<ValueType>( 1 ) << mid;
if ( value == compare ) return mid;
else if ( value < compare ) high = mid - 1;
else low = mid + 1;
}
return -1;
}
TQString
QueryBuilder::tableName( int table )
{
// optimize for 1 table which is by far the most frequent case
static const char tabNames[][16] = {
"album",
"artist",
"composer",
"genre",
"year",
"<unused>", // 32 is missing from the enum
"tags",
"statistics",
"lyrics",
"podcastchannels",
"podcastepisodes",
"podcasttables",
"devices",
"labels"
};
int oneBit = searchBit( table, sizeof( tabNames ) / sizeof( TQString ) );
if ( oneBit >= 0 ) return tabNames[oneBit];
// slow path: multiple tables. This seems to be unneeded at the moment,
// but leaving it here since it appears to be intended usage
TQString tables;
if ( CollectionDB::instance()->getType() != DbConnection::postgresql )
{
if ( table & tabSong ) tables += ",tags";
}
if ( table & tabArtist ) tables += ",artist";
if ( table & tabComposer ) tables += ",composer";
if ( table & tabAlbum ) tables += ",album";
if ( table & tabGenre ) tables += ",genre";
if ( table & tabYear ) tables += ",year";
if ( table & tabStats ) tables += ",statistics";
if ( table & tabLyrics ) tables += ",lyrics";
if ( table & tabPodcastChannels ) tables += ",podcastchannels";
if ( table & tabPodcastEpisodes ) tables += ",podcastepisodes";
if ( table & tabPodcastFolders ) tables += ",podcasttables";
if ( CollectionDB::instance()->getType() == DbConnection::postgresql )
{
if ( table & tabSong ) tables += ",tags";
}
if ( table & tabDevices ) tables += ",devices";
if ( table & tabLabels ) tables += ",labels";
// when there are multiple tables involved, we always need table tags for linking them
return tables.mid( 1 );
}
const TQString &
QueryBuilder::valueName( TQ_INT64 value )
{
static const TQString values[] = {
"id",
"name",
"url",
"title",
"track",
"percentage",
"comment",
"bitrate",
"length",
"samplerate",
"playcounter",
"createdate",
"accessdate",
"percentage",
"artist",
"album",
"year",
"genre",
"dir",
"lyrics",
"rating",
"composer",
"discnumber",
"filesize",
"filetype",
"sampler",
"bpm",
"copyright",
"parent",
"weblink",
"autoscan",
"fetchtype",
"autotransfer",
"haspurge",
"purgeCount",
"isNew",
"deviceid",
"url",
"label",
"lastmountpoint",
"type"
};
int oneBit = searchBit( value, sizeof( values ) / sizeof( TQString ) );
if ( oneBit >= 0 ) return values[oneBit];
static const TQString error( "<ERROR valueName>" );
return error;
}
/*
* Return true if we should call COALESCE(..,0) for this DB field
* (field names sourced from the old smartplaylistbrowser.cpp code)
* Warning: addFilter( int, TQ_INT64, const TQString&, int bool )
* expects this method to return true for all statistics table clomuns of type INTEGER
* Sqlite doesn't like comparing strings to an INTEGER column.
*/
bool
QueryBuilder::coalesceField( int table, TQ_INT64 value )
{
if( tableName( table ) == "statistics" &&
( valueName( value ) == "playcounter" ||
valueName( value ) == "rating" ||
valueName( value ) == "percentage" ||
valueName( value ) == "accessdate" ||
valueName( value ) == "createdate"
)
)
return true;
return false;
}
TQString
QueryBuilder::functionName( int function )
{
TQString functions;
if ( function & funcCount ) functions += "Count";
if ( function & funcMax ) functions += "Max";
if ( function & funcMin ) functions += "Min";
if ( function & funcAvg ) functions += "Avg";
if ( function & funcSum ) functions += "Sum";
return functions;
}
// FIXME: the two functions below are inefficient, but this patch is getting too
// big already. They are not on any critical path right now. Ovy
int
QueryBuilder::getTableByName(const TQString &name)
{
for ( int i = 1; i <= tabLabels; i <<= 1 )
{
if (tableName(i) == name) return i;
}
return -1;
}
TQ_INT64
QueryBuilder::getValueByName(const TQString &name)
{
for ( TQ_INT64 i = 1; i <= valType; i <<= 1 ) {
if (valueName(i) == name) return i;
}
return -1;
}
bool
QueryBuilder::getField(const TQString &tableValue, int *table, TQ_INT64 *value)
{
int dotIndex = tableValue.find( '.' ) ;
if ( dotIndex < 0 ) return false;
int tmpTable = getTableByName( tableValue.left(dotIndex) );
TQ_UINT64 tmpValue = getValueByName( tableValue.mid( dotIndex + 1 ) );
if ( tmpTable >= 0 && value ) {
*table = tmpTable;
*value = tmpValue;
return true;
}
else
{
qFatal("invalid table.value: %s", tableValue.ascii());
return false;
}
}
TQStringList
QueryBuilder::cleanURL( TQStringList result )
{
//this method replaces the fields for relative path and devive/media id with a
//single field containing the absolute path for each row
//TODO Max improve this code
int count = 1;
for( TQStringList::Iterator it = result.begin(), end = result.end(); it != end; )
{
TQString rpath;
if ( (count % (m_returnValues + 1)) + 1== m_deviceidPos )
{
//this block is reached when the iterator points at the relative path
//deviceid is next
TQString rpath = *it;
int deviceid = (*(++it)).toInt();
TQString abspath = MountPointManager::instance()->getAbsolutePath( deviceid, rpath );
it = result.remove(--it);
result.insert( it, abspath );
it = result.remove( it );
//we advanced the iterator over two fields in this iteration
++count;
}
else
++it;
++count;
}
return result;
}
#include "collectiondb.moc"
View Git Blame Copy Permalink