Clementine-audio-player-Mac-Linux-Windows/src/radio/spotifyserver.cpp

/* This file is part of Clementine.
   Copyright 2010, David Sansome <me@davidsansome.com>

   Clementine is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   Clementine is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Clementine.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "spotifyserver.h"
#include "core/logging.h"

#include "spotifyblob/common/spotifymessages.pb.h"
#include "spotifyblob/common/spotifymessagehandler.h"

#include <QTcpServer>
#include <QTcpSocket>

SpotifyServer::SpotifyServer(QObject* parent)
  : QObject(parent),
    server_(new QTcpServer(this)),
    protocol_socket_(NULL),
    handler_(NULL),
    logged_in_(false)
{
  connect(server_, SIGNAL(newConnection()), SLOT(NewConnection()));
}

void SpotifyServer::Init() {
  if (!server_->listen(QHostAddress::LocalHost)) {
    qLog(Error) << "Couldn't open server socket" << server_->errorString();
  }
}

int SpotifyServer::server_port() const {
  return server_->serverPort();
}

void SpotifyServer::NewConnection() {
  delete protocol_socket_;
  delete handler_;

  protocol_socket_ = server_->nextPendingConnection();
  handler_ = new SpotifyMessageHandler(protocol_socket_, this);
  connect(handler_, SIGNAL(MessageArrived(protobuf::SpotifyMessage)),
          SLOT(HandleMessage(protobuf::SpotifyMessage)));

  qLog(Info) << "Connection from port" << protocol_socket_->peerPort();

  // Send any login messages that were queued before the client connected
  foreach (const protobuf::SpotifyMessage& message, queued_login_messages_) {
    SendMessage(message);
  }
  queued_login_messages_.clear();
}

void SpotifyServer::SendMessage(const protobuf::SpotifyMessage& message) {
  const bool is_login_message = message.has_login_request();

  QList<protobuf::SpotifyMessage>* queue =
      is_login_message ? &queued_login_messages_ : &queued_messages_;

  if (!protocol_socket_ || (!is_login_message && !logged_in_)) {
    queue->append(message);
  } else {
    handler_->SendMessage(message);
  }
}

void SpotifyServer::Login(const QString& username, const QString& password) {
  protobuf::SpotifyMessage message;

  protobuf::LoginRequest* request = message.mutable_login_request();
  request->set_username(DataCommaSizeFromQString(username));
  request->set_password(DataCommaSizeFromQString(password));

  SendMessage(message);
}

void SpotifyServer::HandleMessage(const protobuf::SpotifyMessage& message) {
  if (message.has_login_response()) {
    const protobuf::LoginResponse& response = message.login_response();
    logged_in_ = response.success();

    if (!response.success()) {
      qLog(Info) << QStringFromStdString(response.error());
    } else {
      // Send any messages that were queued before the client logged in
      foreach (const protobuf::SpotifyMessage& message, queued_messages_) {
        SendMessage(message);
      }
      queued_messages_.clear();
    }

    emit LoginCompleted(response.success());
  } else if (message.has_playlists_updated()) {
    emit PlaylistsUpdated(message.playlists_updated());
  } else if (message.has_load_playlist_response()) {
    const protobuf::LoadPlaylistResponse& response = message.load_playlist_response();

    switch (response.request().type()) {
      case protobuf::LoadPlaylistRequest_Type_Inbox:
        emit InboxLoaded(response);
        break;

      case protobuf::LoadPlaylistRequest_Type_Starred:
        emit StarredLoaded(response);
        break;

      case protobuf::LoadPlaylistRequest_Type_UserPlaylist:
        emit UserPlaylistLoaded(response);
        break;
    }
  } else if (message.has_playback_error()) {
    emit PlaybackError(QStringFromStdString(message.playback_error().error()));
  } else if (message.has_search_response()) {
    emit SearchResults(message.search_response());
  } else if (message.has_image_response()) {
    const protobuf::ImageResponse& response = message.image_response();
    const QString id = QStringFromStdString(response.id());

    if (response.has_data()) {
      emit ImageLoaded(id, QImage::fromData(QByteArray(
          response.data().data(), response.data().size())));
    } else {
      emit ImageLoaded(id, QImage());
    }
  }
}

void SpotifyServer::LoadPlaylist(protobuf::LoadPlaylistRequest_Type type, int index) {
  protobuf::SpotifyMessage message;
  protobuf::LoadPlaylistRequest* req = message.mutable_load_playlist_request();

  req->set_type(type);
  if (index != -1) {
    req->set_user_playlist_index(index);
  }

  SendMessage(message);
}

void SpotifyServer::LoadInbox() {
  LoadPlaylist(protobuf::LoadPlaylistRequest_Type_Inbox);
}

void SpotifyServer::LoadStarred() {
  LoadPlaylist(protobuf::LoadPlaylistRequest_Type_Starred);
}

void SpotifyServer::LoadUserPlaylist(int index) {
  LoadPlaylist(protobuf::LoadPlaylistRequest_Type_UserPlaylist, index);
}

void SpotifyServer::StartPlayback(const QString& uri, quint16 port) {
  protobuf::SpotifyMessage message;
  protobuf::PlaybackRequest* req = message.mutable_playback_request();

  req->set_track_uri(DataCommaSizeFromQString(uri));
  req->set_media_port(port);
  SendMessage(message);
}

void SpotifyServer::Search(const QString& text, int limit) {
  protobuf::SpotifyMessage message;
  protobuf::SearchRequest* req = message.mutable_search_request();

  req->set_query(DataCommaSizeFromQString(text));
  req->set_limit(limit);
  SendMessage(message);
}

void SpotifyServer::LoadImage(const QString& id) {
  protobuf::SpotifyMessage message;
  protobuf::ImageRequest* req = message.mutable_image_request();

  req->set_id(DataCommaSizeFromQString(id));
  SendMessage(message);
}