/* This file is part of Strawberry.
   Copyright 2011, David Sansome <me@davidsansome.com>
   Copyright 2018-2021, Jonas Kvinge <jonas@jkvinge.net>

   Strawberry 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.

   Strawberry 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 Strawberry.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "config.h"

#include <string>

#include <QCoreApplication>
#include <QObject>
#include <QIODevice>
#include <QByteArray>

#include "tagreaderworker.h"

TagReaderWorker::TagReaderWorker(QIODevice *socket, QObject *parent)
    : AbstractMessageHandler<spb::tagreader::Message>(socket, parent) {}

void TagReaderWorker::MessageArrived(const spb::tagreader::Message &message) {

  spb::tagreader::Message reply;

  bool success = HandleMessage(message, reply, &tag_reader_);
  if (!success) {
#if defined(USE_TAGLIB)
    HandleMessage(message, reply, &tag_reader_gme_);
#endif
  }

  SendReply(message, &reply);

}

void TagReaderWorker::DeviceClosed() {

  AbstractMessageHandler<spb::tagreader::Message>::DeviceClosed();

  QCoreApplication::exit();

}

bool TagReaderWorker::HandleMessage(const spb::tagreader::Message &message, spb::tagreader::Message &reply, TagReaderBase *reader) {

  if (message.has_is_media_file_request()) {
    const QString filename = QString::fromUtf8(message.is_media_file_request().filename().data(), message.is_media_file_request().filename().size());
    bool success = reader->IsMediaFile(filename);
    reply.mutable_is_media_file_response()->set_success(success);
    return success;
  }
  else if (message.has_read_file_request()) {
    const QString filename = QString::fromUtf8(message.read_file_request().filename().data(), message.read_file_request().filename().size());
    bool success = reader->ReadFile(filename, reply.mutable_read_file_response()->mutable_metadata());
    return success;
  }
  else if (message.has_save_file_request()) {
    bool success = reader->SaveFile(message.save_file_request());
    reply.mutable_save_file_response()->set_success(success);
    return success;
  }
  else if (message.has_load_embedded_art_request()) {
    const QString filename = QString::fromUtf8(message.load_embedded_art_request().filename().data(), message.load_embedded_art_request().filename().size());
    QByteArray data = reader->LoadEmbeddedArt(filename);
    reply.mutable_load_embedded_art_response()->set_data(data.constData(), data.size());
    return true;
  }
  else if (message.has_save_embedded_art_request()) {
    bool success = reader->SaveEmbeddedArt(message.save_embedded_art_request());
    reply.mutable_save_embedded_art_response()->set_success(success);
    return success;
  }
  else if (message.has_save_song_playcount_to_file_request()) {
    const QString filename = QString::fromUtf8(message.save_song_playcount_to_file_request().filename().data(), message.save_song_playcount_to_file_request().filename().size());
    bool success = reader->SaveSongPlaycountToFile(filename, message.save_song_playcount_to_file_request().metadata());
    reply.mutable_save_song_playcount_to_file_response()->set_success(success);
    return success;
  }
  else if (message.has_save_song_rating_to_file_request()) {
    const QString filename = QString::fromUtf8(message.save_song_rating_to_file_request().filename().data(), message.save_song_rating_to_file_request().filename().size());
    bool success = reader->SaveSongRatingToFile(filename, message.save_song_rating_to_file_request().metadata());
    reply.mutable_save_song_rating_to_file_response()->set_success(success);
    return success;
  }

  return false;

}