Clementine-audio-player-Mac-Linux-Windows/spotifyblob/common/spotifymessagehandler.cpp

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

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
*/

// Note: this file is licensed under the Apache License instead of GPL because
// it is used by the Spotify blob which links against libspotify and is not GPL
// compatible.


#include "spotifymessages.pb.h"
#include "spotifymessagehandler.h"
#include "core/logging.h"

#include <QBuffer>

SpotifyMessageHandler::SpotifyMessageHandler(QIODevice* device, QObject* parent)
  : QObject(parent),
    device_(device),
    reading_protobuf_(false),
    expected_length_(0) {
  buffer_.open(QIODevice::ReadWrite);

  connect(device, SIGNAL(readyRead()), SLOT(DeviceReadyRead()));
}

void SpotifyMessageHandler::DeviceReadyRead() {
  while (device_->bytesAvailable()) {
    if (!reading_protobuf_) {
      // Read the length of the next message
      QDataStream s(device_);
      s >> expected_length_;

      reading_protobuf_ = true;
    }

    // Read some of the message
    buffer_.write(device_->read(expected_length_ - buffer_.size()));

    // Did we get everything?
    if (buffer_.size() == expected_length_) {
      // Parse the message
      protobuf::SpotifyMessage message;
      if (!message.ParseFromArray(buffer_.data().constData(), buffer_.size())) {
        qLog(Error) << "Malformed protobuf message";
        device_->close();
        return;
      }

      emit MessageArrived(message);

      // Clear the buffer
      buffer_.close();
      buffer_.setData(QByteArray());
      buffer_.open(QIODevice::ReadWrite);
      reading_protobuf_ = false;
    }
  }
}

void SpotifyMessageHandler::SendMessage(const protobuf::SpotifyMessage& message) {
  std::string data(message.SerializeAsString());

  QDataStream s(device_);
  s << quint32(data.length());
  s.writeRawData(data.data(), data.length());
}