404 lines
12 KiB
C++
404 lines
12 KiB
C++
|
#include "api__ml_impex.h"
|
||
|
|
#include "../xml/obj_xml.h"
|
||
|
|
#include <map>
|
||
|
|
#include <bfc/string/url.h>
|
||
|
|
#include "importer.h"
|
||
|
|
#include "resource.h"
|
||
|
|
#include "../plist/loader.h"
|
||
|
|
#include "../playlist/ifc_playlist.h"
|
||
|
|
#include "../Winamp/wa_ipc.h"
|
||
|
|
#include <shlwapi.h>
|
||
|
|
|
||
|
|
struct iTunesFileInfo
|
||
|
|
{
|
||
|
|
iTunesFileInfo(const wchar_t *_filename, uint64_t _length)
|
||
|
|
{
|
||
|
|
filename = _wcsdup(_filename);
|
||
|
|
length = _length;
|
||
|
|
}
|
||
|
|
~iTunesFileInfo()
|
||
|
|
{
|
||
|
|
free(filename);
|
||
|
|
}
|
||
|
|
wchar_t *filename;
|
||
|
|
uint64_t length;
|
||
|
|
};
|
||
|
|
typedef std::map<int64_t, iTunesFileInfo*> FilesList;
|
||
|
|
extern winampMediaLibraryPlugin plugin;
|
||
|
|
int Load(const wchar_t *filename, obj_xml *parser);
|
||
|
|
|
||
|
|
class PlistPlaylist : public ifc_playlist
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
PlistPlaylist(const plistArray *_items, FilesList &_files) : items(_items), files(_files)
|
||
|
|
{
|
||
|
|
length_sum = 0;
|
||
|
|
}
|
||
|
|
size_t GetNumItems();
|
||
|
|
size_t GetItem(size_t item, wchar_t *filename, size_t filenameCch);
|
||
|
|
int GetItemLengthMilliseconds(size_t item);
|
||
|
|
uint64_t length_sum;
|
||
|
|
protected:
|
||
|
|
RECVS_DISPATCH;
|
||
|
|
const plistArray *items;
|
||
|
|
FilesList &files;
|
||
|
|
};
|
||
|
|
|
||
|
|
size_t PlistPlaylist::GetNumItems()
|
||
|
|
{
|
||
|
|
return items->getNumItems();
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t PlistPlaylist::GetItem(size_t item, wchar_t *filename, size_t filenameCch)
|
||
|
|
{
|
||
|
|
plistDict *item_dict = (plistDict *)items->enumItem((int)item);
|
||
|
|
if (item_dict)
|
||
|
|
{
|
||
|
|
plistKey *id_key = item_dict->getKey(L"Track ID");
|
||
|
|
if (id_key)
|
||
|
|
{
|
||
|
|
plistInteger *id_data = (plistInteger *)id_key->getData();
|
||
|
|
if (id_data)
|
||
|
|
{
|
||
|
|
int64_t key = id_data->getValue();
|
||
|
|
iTunesFileInfo *info = files[key];
|
||
|
|
if (info)
|
||
|
|
{
|
||
|
|
const wchar_t *track_name = info->filename;
|
||
|
|
if (track_name)
|
||
|
|
{
|
||
|
|
length_sum += info->length;
|
||
|
|
StringCchCopyW(filename, filenameCch, track_name);
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int PlistPlaylist::GetItemLengthMilliseconds(size_t item)
|
||
|
|
{
|
||
|
|
plistDict *item_dict = (plistDict *)items->enumItem((int)item);
|
||
|
|
if (item_dict)
|
||
|
|
{
|
||
|
|
plistKey *id_key = item_dict->getKey(L"Track ID");
|
||
|
|
if (id_key)
|
||
|
|
{
|
||
|
|
plistInteger *id_data = (plistInteger *)id_key->getData();
|
||
|
|
if (id_data)
|
||
|
|
{
|
||
|
|
int64_t key = id_data->getValue();
|
||
|
|
iTunesFileInfo *info = files[key];
|
||
|
|
if (info)
|
||
|
|
{
|
||
|
|
return (int)info->length;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#define CBCLASS PlistPlaylist
|
||
|
|
START_DISPATCH;
|
||
|
|
CB(IFC_PLAYLIST_GETNUMITEMS, GetNumItems)
|
||
|
|
CB(IFC_PLAYLIST_GETITEM, GetItem)
|
||
|
|
CB(IFC_PLAYLIST_GETITEMLENGTHMILLISECONDS, GetItemLengthMilliseconds)
|
||
|
|
END_DISPATCH;
|
||
|
|
#undef CBCLASS
|
||
|
|
|
||
|
|
static bool GetInteger(const plistDict *dict, const wchar_t *key_name, int64_t *int_val)
|
||
|
|
{
|
||
|
|
plistKey *key = dict->getKey(key_name);
|
||
|
|
if (!key)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
plistData *data = key->getData();
|
||
|
|
if (!data)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (data->getType() != PLISTDATA_INTEGER)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
plistInteger *data_int = static_cast<plistInteger *>(data);
|
||
|
|
*int_val = data_int->getValue();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool GetString(const plistDict *dict, const wchar_t *key_name, const wchar_t **str_val)
|
||
|
|
{
|
||
|
|
plistKey *key = dict->getKey(key_name);
|
||
|
|
if (!key)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
plistData *data = key->getData();
|
||
|
|
if (!data)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (data->getType() != PLISTDATA_STRING)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
plistString *data_str = static_cast<plistString *>(data);
|
||
|
|
*str_val = data_str->getString();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool GetArray(const plistDict *dict, const wchar_t *key_name, plistArray **array_val)
|
||
|
|
{
|
||
|
|
plistKey *key = dict->getKey(key_name);
|
||
|
|
if (!key)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
plistData *data = key->getData();
|
||
|
|
if (!data)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
if (data->getType() != PLISTDATA_ARRAY)
|
||
|
|
return false;
|
||
|
|
|
||
|
|
*array_val = static_cast<plistArray *>(data);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool CheckDuplicatePlaylist(uint64_t playlist_id64, GUID &dup_guid)
|
||
|
|
{
|
||
|
|
AGAVE_API_PLAYLISTS->Lock();
|
||
|
|
size_t numPlaylists = AGAVE_API_PLAYLISTS->GetCount();
|
||
|
|
uint64_t compare_id64=0;
|
||
|
|
for (size_t i=0;i!=numPlaylists;i++)
|
||
|
|
{
|
||
|
|
if (AGAVE_API_PLAYLISTS->GetInfo(i, api_playlists_iTunesID, &compare_id64, sizeof(compare_id64)) == API_PLAYLISTS_SUCCESS)
|
||
|
|
{
|
||
|
|
if (compare_id64 == playlist_id64)
|
||
|
|
{
|
||
|
|
dup_guid = AGAVE_API_PLAYLISTS->GetGUID(i);
|
||
|
|
AGAVE_API_PLAYLISTS->Unlock();
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
AGAVE_API_PLAYLISTS->Unlock();
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
enum
|
||
|
|
{
|
||
|
|
DUPLICATE_PLAYLIST_SKIP,
|
||
|
|
DUPLICATE_PLAYLIST_REPLACE,
|
||
|
|
DUPLICATE_PLAYLIST_NEW,
|
||
|
|
};
|
||
|
|
|
||
|
|
static int PromptReplaceSkipNew(GUID &dup_guid)
|
||
|
|
{
|
||
|
|
/* TODO:
|
||
|
|
* get name and stuff from api_playlists
|
||
|
|
* we'll need an HWND for the UI
|
||
|
|
* we'll need some passed-in state variable to remember "do for all" choice
|
||
|
|
*/
|
||
|
|
return DUPLICATE_PLAYLIST_SKIP;
|
||
|
|
}
|
||
|
|
|
||
|
|
HINSTANCE cloud_hinst = 0;
|
||
|
|
int IPC_GET_CLOUD_HINST = -1, IPC_GET_CLOUD_ACTIVE = -1;
|
||
|
|
int cloudAvailable()
|
||
|
|
{
|
||
|
|
if (IPC_GET_CLOUD_HINST == -1) IPC_GET_CLOUD_HINST = (INT)SendMessage(plugin.hwndWinampParent, WM_WA_IPC, (WPARAM)&"WinampCloud", IPC_REGISTER_WINAMP_IPCMESSAGE);
|
||
|
|
if (IPC_GET_CLOUD_ACTIVE == -1) IPC_GET_CLOUD_ACTIVE = (INT)SendMessage(plugin.hwndWinampParent, WM_WA_IPC, (WPARAM)&"WinampCloudActive", IPC_REGISTER_WINAMP_IPCMESSAGE);
|
||
|
|
if (!cloud_hinst) cloud_hinst = (HINSTANCE)SendMessage(plugin.hwndWinampParent, WM_WA_IPC, 0, IPC_GET_CLOUD_HINST);
|
||
|
|
|
||
|
|
return (/*0/*/!(!cloud_hinst || cloud_hinst == (HINSTANCE)1 || !SendMessage(plugin.hwndWinampParent, WM_WA_IPC, 0, IPC_GET_CLOUD_ACTIVE))/**/);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void AddPlaylist(plistDict *playlist, FilesList &files)
|
||
|
|
{
|
||
|
|
const wchar_t *name;
|
||
|
|
const wchar_t *playlist_persistent_id;
|
||
|
|
int64_t playlist_id;
|
||
|
|
int64_t visible;
|
||
|
|
plistArray *items;
|
||
|
|
uint64_t playlist_id64;
|
||
|
|
|
||
|
|
if (GetString(playlist, L"Name", &name)
|
||
|
|
&& GetArray(playlist, L"Playlist Items", &items)
|
||
|
|
&& GetInteger(playlist, L"Playlist ID", &playlist_id)
|
||
|
|
&& GetString(playlist, L"Playlist Persistent ID", &playlist_persistent_id)
|
||
|
|
&& (!GetInteger(playlist, L"Visible", &visible) || visible))
|
||
|
|
{
|
||
|
|
playlist_id64 =_wcstoui64(playlist_persistent_id, 0, 16);
|
||
|
|
// see if it's already in the database
|
||
|
|
|
||
|
|
GUID dup_guid; // so we know the GUID we clash with, in case we want to replace it instead of skip it
|
||
|
|
if (playlist_id64 && CheckDuplicatePlaylist(playlist_id64, dup_guid))
|
||
|
|
{
|
||
|
|
switch(PromptReplaceSkipNew(dup_guid))
|
||
|
|
{
|
||
|
|
case DUPLICATE_PLAYLIST_SKIP:
|
||
|
|
break;
|
||
|
|
case DUPLICATE_PLAYLIST_REPLACE:
|
||
|
|
// TODO
|
||
|
|
break;
|
||
|
|
case DUPLICATE_PLAYLIST_NEW:
|
||
|
|
// TODO
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
PlistPlaylist plist_playlist(items, files);
|
||
|
|
|
||
|
|
const wchar_t *user_folder = WASABI_API_APP->path_getUserSettingsPath();
|
||
|
|
wchar_t destination[MAX_PATH] = {0};
|
||
|
|
PathCombineW(destination, user_folder, L"plugins\\ml\\playlists");
|
||
|
|
|
||
|
|
wchar_t playlist_filename[MAX_PATH] = {0};
|
||
|
|
StringCbPrintfW(playlist_filename, sizeof(playlist_filename), L"i_%I64u.m3u8", playlist_id);
|
||
|
|
PathAppendW(destination, playlist_filename);
|
||
|
|
|
||
|
|
static wchar_t ml_ini_file[MAX_PATH] = {0};
|
||
|
|
if (!ml_ini_file[0]) lstrcpynW(ml_ini_file, (const wchar_t*)SendMessage(plugin.hwndWinampParent, WM_WA_IPC, 0, IPC_GETMLINIFILEW), MAX_PATH);
|
||
|
|
size_t cloud = (cloudAvailable() ? GetPrivateProfileIntW(L"gen_ml_config", L"cloud_always", 1, ml_ini_file) : 0);
|
||
|
|
|
||
|
|
AGAVE_API_PLAYLISTMANAGER->Save(destination, &plist_playlist);
|
||
|
|
AGAVE_API_PLAYLISTS->Lock();
|
||
|
|
int new_index = (!cloud ? (int)AGAVE_API_PLAYLISTS->AddPlaylist(destination, name) : (int)AGAVE_API_PLAYLISTS->AddCloudPlaylist(destination, name));
|
||
|
|
if (new_index >= 0)
|
||
|
|
{
|
||
|
|
uint32_t numItems = (uint32_t)plist_playlist.GetNumItems();
|
||
|
|
uint64_t totalLength = plist_playlist.length_sum/1000;
|
||
|
|
AGAVE_API_PLAYLISTS->SetInfo(new_index, api_playlists_totalTime, &totalLength, sizeof(totalLength));
|
||
|
|
AGAVE_API_PLAYLISTS->SetInfo(new_index, api_playlists_itemCount, &numItems, sizeof(numItems));
|
||
|
|
if (cloud) AGAVE_API_PLAYLISTS->SetInfo(new_index, api_playlists_cloud, &cloud, sizeof(cloud));
|
||
|
|
AGAVE_API_PLAYLISTS->SetInfo(new_index, api_playlists_iTunesID, &playlist_id64, sizeof(playlist_id64));
|
||
|
|
}
|
||
|
|
AGAVE_API_PLAYLISTS->Unlock();
|
||
|
|
|
||
|
|
PostMessage(plugin.hwndWinampParent, WM_WA_IPC, 0, SendMessage(plugin.hwndWinampParent, WM_WA_IPC, (WPARAM)&"ml_playlist_refresh", IPC_REGISTER_WINAMP_IPCMESSAGE));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void FixPath(const wchar_t *strdata, StringW &f);
|
||
|
|
|
||
|
|
int ImportPlaylists(HWND parent, const wchar_t *library_file)
|
||
|
|
{
|
||
|
|
FilesList files;
|
||
|
|
// create an XML parser
|
||
|
|
obj_xml *parser=0;
|
||
|
|
waServiceFactory *factory = plugin.service->service_getServiceByGuid(obj_xmlGUID);
|
||
|
|
if (factory)
|
||
|
|
parser = (obj_xml *)factory->getInterface();
|
||
|
|
|
||
|
|
if (parser)
|
||
|
|
{
|
||
|
|
// create status window
|
||
|
|
//HWND hwndDlg = WASABI_API_CREATEDIALOGW(IDD_INFODIALOG,plugin.hwndLibraryParent,import_dlgproc);
|
||
|
|
|
||
|
|
// init it
|
||
|
|
//ShowWindow(hwndDlg, SW_NORMAL);
|
||
|
|
//UpdateWindow(hwndDlg);
|
||
|
|
|
||
|
|
// create an iTunes XML library reader
|
||
|
|
plistLoader it;
|
||
|
|
|
||
|
|
// load the XML, this creates an iTunes DB in memory, and returns the root key
|
||
|
|
parser->xmlreader_open();
|
||
|
|
parser->xmlreader_registerCallback(L"plist\f*", &it);
|
||
|
|
Load(library_file, parser);
|
||
|
|
parser->xmlreader_unregisterCallback(&it);
|
||
|
|
parser->xmlreader_close();
|
||
|
|
plistKey *root_key = ⁢
|
||
|
|
|
||
|
|
// show import progress controls
|
||
|
|
//ShowWindow(GetDlgItem(hwndDlg, IDC_PROCESSING_STATE), SW_HIDE);
|
||
|
|
//ShowWindow(GetDlgItem(hwndDlg, IDC_PROGRESS_PERCENT), SW_SHOWNORMAL);
|
||
|
|
//ShowWindow(GetDlgItem(hwndDlg, IDC_TRACKS), SW_SHOWNORMAL);
|
||
|
|
//UpdateWindow(hwndDlg);
|
||
|
|
|
||
|
|
// we start at the root key
|
||
|
|
if (root_key)
|
||
|
|
{
|
||
|
|
// the root key contains a dictionary
|
||
|
|
plistData *root_dict = root_key->getData();
|
||
|
|
if (root_dict && root_dict->getType() == PLISTDATA_DICT)
|
||
|
|
{
|
||
|
|
// that dictionary contains a number of keys, one of which contains a dictionary of tracks
|
||
|
|
plistKey *tracks_key = ((plistDict*)root_dict)->getKey(L"Tracks");
|
||
|
|
plistData *tracks_dict = tracks_key?tracks_key->getData():0;
|
||
|
|
if (tracks_dict && tracks_dict->getType() == PLISTDATA_DICT)
|
||
|
|
{
|
||
|
|
// we have the tracks dictionary ...
|
||
|
|
plistDict *tracks = (plistDict *)tracks_dict;
|
||
|
|
int n =tracks?tracks->getNumKeys():0;
|
||
|
|
// ... now enumerate tracks
|
||
|
|
for (int i=0;i<n;i++)
|
||
|
|
{
|
||
|
|
// each track is a key in the tracks dictionary, and contains a dictionary of properties
|
||
|
|
plistKey *track_key = tracks->enumKey(i);
|
||
|
|
plistData *track_dict = track_key->getData();
|
||
|
|
// prepare an item record
|
||
|
|
|
||
|
|
if (track_dict->getType() == PLISTDATA_DICT)
|
||
|
|
{
|
||
|
|
// we have the track's dictionary of properties...
|
||
|
|
plistDict *track = (plistDict *)track_dict;
|
||
|
|
int64_t id = 0;
|
||
|
|
const wchar_t *location = 0;
|
||
|
|
if (GetInteger(track, L"Track ID", &id) && GetString(track, L"Location", &location))
|
||
|
|
{
|
||
|
|
StringW f;
|
||
|
|
FixPath(location, f);
|
||
|
|
|
||
|
|
int64_t length = 0;
|
||
|
|
GetInteger(track, L"Total Time", &length);
|
||
|
|
// done
|
||
|
|
wchar_t *filename = _wcsdup(f);
|
||
|
|
files[id] = new iTunesFileInfo(filename, length);
|
||
|
|
|
||
|
|
// show progress
|
||
|
|
//SetDlgItemText(hwndDlg, IDC_TRACKS, StringPrintfW(WASABI_API_LNGSTRINGW(IDS_TRACKS_IMPORTED_X), ++count));
|
||
|
|
//SendDlgItemMessage(hwndDlg, IDC_PROGRESS_PERCENT, PBM_SETPOS, (int)((double)count/n*100.0), 0);
|
||
|
|
//if (count % 10 == 0 || count == n)
|
||
|
|
// UpdateWindow(hwndDlg);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// ok we're done building the track list, now let's enumerate the playlists
|
||
|
|
plistKey *playlists_key = ((plistDict*)root_dict)->getKey(L"Playlists");
|
||
|
|
plistData *playlists_dict = playlists_key?playlists_key->getData():0;
|
||
|
|
if (playlists_dict && playlists_dict->getType() == PLISTDATA_ARRAY)
|
||
|
|
{
|
||
|
|
plistArray *playlists = (plistArray *)playlists_dict;
|
||
|
|
int n =playlists?playlists->getNumItems():0;
|
||
|
|
// ... now enumerate playlists
|
||
|
|
for (int i=0;i<n;i++)
|
||
|
|
{
|
||
|
|
// each playlist is a key in the playlists dictionary, and contains a dictionary of properties
|
||
|
|
plistData *playlist_dict = playlists->enumItem(i);
|
||
|
|
if (playlist_dict->getType() == PLISTDATA_DICT)
|
||
|
|
{
|
||
|
|
// we have the playlist's dictionary of properties...
|
||
|
|
plistDict *playlist = (plistDict *)playlist_dict;
|
||
|
|
AddPlaylist(playlist, files);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
//DestroyWindow(hwndDlg);
|
||
|
|
factory->releaseInterface(parser);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
return DISPATCH_FAILURE;
|
||
|
|
|
||
|
|
FilesList::iterator itr;
|
||
|
|
for (itr = files.begin(); itr!= files.end(); itr++)
|
||
|
|
{
|
||
|
|
iTunesFileInfo *info = itr->second;
|
||
|
|
delete info;
|
||
|
|
}
|
||
|
|
return DISPATCH_SUCCESS;
|
||
|
|
}
|