iTunes file util and data provider for media galleries

chrome/browser/media_galleries/fileapi/itunes_data_provider.cc

+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/media_galleries/fileapi/itunes_data_provider.h"
+
+#include "base/callback.h"
+#include "base/format_macros.h"
+#include "base/logging.h"
+#include "base/platform_file.h"
+#include "base/stl_util.h"
+#include "base/stringprintf.h"
+#include "base/threading/thread_restrictions.h"
+#include "chrome/browser/media_galleries/fileapi/itunes_library_parser.h"
+
+namespace itunes {
+
+namespace {
+
+// A "reasonable" artificial limit.
+// TODO(vandebo): Add a UMA to figure out what common values are.
+const int64 kMaxLibraryFileSize = 150 * 1024 * 1024;
+
+std::string ReadFile(const base::FilePath& path) {
+  base::ThreadRestrictions::AssertIOAllowed();
+
+  base::PlatformFile file = base::CreatePlatformFile(
+      path, base::PLATFORM_FILE_OPEN | base::PLATFORM_FILE_READ, NULL, NULL);
+  if (file == base::kInvalidPlatformFileValue)
+    return std::string();
+
+  base::PlatformFileInfo file_info;
+  if (!base::GetPlatformFileInfo(file, &file_info) ||
+      file_info.size > kMaxLibraryFileSize) {
+    base::ClosePlatformFile(file);
+    return std::string();
+  }
+
+  std::string result(file_info.size, 0);
+  if (base::ReadPlatformFile(file, 0, string_as_array(&result),
+        file_info.size) != file_info.size) {
+    result.clear();
+  }
+
+  base::ClosePlatformFile(file);
+  return result;
+}
+
+ITunesDataProvider::Album MakeUniqueTrackNames(
+    const ITunesLibraryParser::Album& album) {
+  // TODO(vandebo): It would be nice to ensure that names returned from here
+  // are stable, but aside from persisting every name returned, it's not
+  // obvious how to do that (without including the track id in every name).
+  typedef std::set<const ITunesLibraryParser::Track*> TrackRefs;
+  typedef std::map<ITunesDataProvider::TrackName, TrackRefs> AlbumInfo;
+
+  ITunesDataProvider::Album result;
+  AlbumInfo duped_tracks;
+
+  ITunesLibraryParser::Album::const_iterator album_it;
+  for (album_it = album.begin(); album_it != album.end(); ++album_it) {
+    const ITunesLibraryParser::Track& track = *album_it;
+    std::string name = track.location.BaseName().AsUTF8Unsafe();
+    duped_tracks[name].insert(&track);
+  }
+
+  for (AlbumInfo::const_iterator name_it = duped_tracks.begin();
+       name_it != duped_tracks.end();
+       ++name_it) {
+    const TrackRefs& track_refs = name_it->second;
+    if (track_refs.size() == 1) {
+      result[name_it->first] = (*track_refs.begin())->location;
+    } else {
+      for (TrackRefs::const_iterator track_it = track_refs.begin();
+           track_it != track_refs.end();
+           ++track_it) {
+        std::string id =
+            base::StringPrintf(" (%" PRId64 ")", (*track_it)->id);
+        base::FilePath unique_name =
+            (*track_it)->location.BaseName().InsertBeforeExtensionASCII(id);
+        result[unique_name.AsUTF8Unsafe()] = (*track_it)->location;
+      }
+    }
+  }
+
+  return result;
+}
+
+}  // namespace
+
+ITunesDataProvider::ITunesDataProvider(const base::FilePath& library_path)
+    : library_path_(library_path),
+      needs_refresh_(true) {
+}
+
+ITunesDataProvider::~ITunesDataProvider() {}
+
+// TODO(vandebo): add a file watch that resets |needs_refresh_| when the
+// file changes.
+void ITunesDataProvider::RefreshData(const base::Closure& ready_callback) {
+  if (needs_refresh_) {
+    ParseLibrary();
+    needs_refresh_ = false;
+  }
+  ready_callback.Run();
+}
+
+const base::FilePath& ITunesDataProvider::library_path() const {
+  return library_path_;
+}
+
+bool ITunesDataProvider::KnownArtist(const ArtistName& artist) const {
+  DCHECK(!needs_refresh_);
+  return ContainsKey(library_, artist);
+}
+
+bool ITunesDataProvider::KnownAlbum(const ArtistName& artist,
+                                    const AlbumName& album) const {
+  DCHECK(!needs_refresh_);
+  Library::const_iterator library_it = library_.find(artist);
+  if (library_it == library_.end())
+    return false;
+  return ContainsKey(library_it->second, album);
+
+}
+
+base::FilePath ITunesDataProvider::GetTrackLocation(
+    const ArtistName& artist, const AlbumName& album,
+    const TrackName& track) const {
+  DCHECK(!needs_refresh_);
+  Library::const_iterator library_it = library_.find(artist);
+  if (library_it == library_.end())
+    return base::FilePath();
+
+  Artist::const_iterator artist_it = library_it->second.find(album);
+  if (artist_it == library_it->second.end())
+    return base::FilePath();
+
+  Album::const_iterator album_it = artist_it->second.find(track);
+  if (album_it == artist_it->second.end())
+    return base::FilePath();
+  return album_it->second;
+}
+
+std::set<ITunesDataProvider::ArtistName>
+ITunesDataProvider::GetArtistNames() const {
+  DCHECK(!needs_refresh_);
+  std::set<ArtistName> result;
+  Library::const_iterator it;
+  for (it = library_.begin(); it != library_.end(); ++it) {
+    result.insert(it->first);
+  }
+  return result;
+}
+
+std::set<ITunesDataProvider::AlbumName> ITunesDataProvider::GetAlbumNames(
+    const ArtistName& artist) const {
+  DCHECK(!needs_refresh_);
+  std::set<AlbumName> result;
+  Library::const_iterator artist_lookup = library_.find(artist);
+  if (artist_lookup == library_.end())
+    return result;
+
+  const Artist& artist_entry = artist_lookup->second;
+  Artist::const_iterator it;
+  for (it = artist_entry.begin(); it != artist_entry.end(); ++it) {
+    result.insert(it->first);
+  }
+  return result;
+}
+
+ITunesDataProvider::Album ITunesDataProvider::GetAlbum(
+    const ArtistName& artist, const AlbumName& album) const {
+  DCHECK(!needs_refresh_);
+  Album empty_result;
+  Library::const_iterator artist_lookup = library_.find(artist);
+  if (artist_lookup == library_.end())
+    return empty_result;
+
+  Artist::const_iterator album_lookup = artist_lookup->second.find(album);
+  if (album_lookup == artist_lookup->second.end())
+    return empty_result;
+
+  return album_lookup->second;
+}
+
+void ITunesDataProvider::ParseLibrary() {
+  std::string xml = ReadFile(library_path_);
+
+  library_.clear();
+  ITunesLibraryParser parser;
+  if (!parser.Parse(xml))
+    return;
+
+  ITunesLibraryParser::Library::const_iterator artist_it;
+  ITunesLibraryParser::Albums::const_iterator album_it;
+  for (artist_it = parser.library().begin();
+       artist_it != parser.library().end();
+       ++artist_it) {
+    for (album_it = artist_it->second.begin();
+         album_it != artist_it->second.end();
+         ++album_it) {
+      library_[artist_it->first][album_it->first] =
+          MakeUniqueTrackNames(album_it->second);
+    }
+  }
+}
+
+}  // namespace itunes