// Copyright 2014 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include "packager/app/packager_util.h"

#include <gflags/gflags.h>
#include <iostream>

#include "packager/app/fixed_key_encryption_flags.h"
#include "packager/app/mpd_flags.h"
#include "packager/app/muxer_flags.h"
#include "packager/app/widevine_encryption_flags.h"
#include "packager/base/logging.h"
#include "packager/base/strings/string_number_conversions.h"
#include "packager/media/base/fixed_key_source.h"
#include "packager/media/base/media_stream.h"
#include "packager/media/base/muxer.h"
#include "packager/media/base/muxer_options.h"
#include "packager/media/base/request_signer.h"
#include "packager/media/base/stream_info.h"
#include "packager/media/base/widevine_key_source.h"
#include "packager/media/file/file.h"
#include "packager/mpd/base/mpd_builder.h"

DEFINE_bool(dump_stream_info, false, "Dump demuxed stream info.");

namespace shaka {
namespace media {

void DumpStreamInfo(const std::vector<MediaStream*>& streams) {
  printf("Found %zu stream(s).\n", streams.size());
  for (size_t i = 0; i < streams.size(); ++i)
    printf("Stream [%zu] %s\n", i, streams[i]->info()->ToString().c_str());
}

scoped_ptr<RequestSigner> CreateSigner() {
  scoped_ptr<RequestSigner> signer;

  if (!FLAGS_aes_signing_key.empty()) {
    signer.reset(AesRequestSigner::CreateSigner(
        FLAGS_signer, FLAGS_aes_signing_key, FLAGS_aes_signing_iv));
    if (!signer) {
      LOG(ERROR) << "Cannot create an AES signer object from '"
                 << FLAGS_aes_signing_key << "':'" << FLAGS_aes_signing_iv
                 << "'.";
      return scoped_ptr<RequestSigner>();
    }
  } else if (!FLAGS_rsa_signing_key_path.empty()) {
    std::string rsa_private_key;
    if (!File::ReadFileToString(FLAGS_rsa_signing_key_path.c_str(),
                                &rsa_private_key)) {
      LOG(ERROR) << "Failed to read from '" << FLAGS_rsa_signing_key_path
                 << "'.";
      return scoped_ptr<RequestSigner>();
    }
    signer.reset(RsaRequestSigner::CreateSigner(FLAGS_signer, rsa_private_key));
    if (!signer) {
      LOG(ERROR) << "Cannot create a RSA signer object from '"
                 << FLAGS_rsa_signing_key_path << "'.";
      return scoped_ptr<RequestSigner>();
    }
  }
  return signer.Pass();
}

scoped_ptr<KeySource> CreateEncryptionKeySource() {
  scoped_ptr<KeySource> encryption_key_source;
  if (FLAGS_enable_widevine_encryption) {
    scoped_ptr<WidevineKeySource> widevine_key_source(
        new WidevineKeySource(FLAGS_key_server_url, FLAGS_include_common_pssh));
    if (!FLAGS_signer.empty()) {
      scoped_ptr<RequestSigner> request_signer(CreateSigner());
      if (!request_signer)
        return scoped_ptr<KeySource>();
      widevine_key_source->set_signer(request_signer.Pass());
    }

    std::vector<uint8_t> content_id;
    if (!base::HexStringToBytes(FLAGS_content_id, &content_id)) {
      LOG(ERROR) << "Invalid content_id hex string specified.";
      return scoped_ptr<KeySource>();
    }
    Status status = widevine_key_source->FetchKeys(content_id, FLAGS_policy);
    if (!status.ok()) {
      LOG(ERROR) << "Widevine encryption key source failed to fetch keys: "
                 << status.ToString();
      return scoped_ptr<KeySource>();
    }
    encryption_key_source = widevine_key_source.Pass();
  } else if (FLAGS_enable_fixed_key_encryption) {
    encryption_key_source = FixedKeySource::CreateFromHexStrings(
        FLAGS_key_id, FLAGS_key, FLAGS_pssh, FLAGS_iv);
  }
  return encryption_key_source.Pass();
}

scoped_ptr<KeySource> CreateDecryptionKeySource() {
  scoped_ptr<KeySource> decryption_key_source;
  if (FLAGS_enable_widevine_decryption) {
    scoped_ptr<WidevineKeySource> widevine_key_source(
        new WidevineKeySource(FLAGS_key_server_url, FLAGS_include_common_pssh));
    if (!FLAGS_signer.empty()) {
      scoped_ptr<RequestSigner> request_signer(CreateSigner());
      if (!request_signer)
        return scoped_ptr<KeySource>();
      widevine_key_source->set_signer(request_signer.Pass());
    }

    decryption_key_source = widevine_key_source.Pass();
  } else if (FLAGS_enable_fixed_key_decryption) {
    const char kNoPssh[] = "";
    const char kNoIv[] = "";
    decryption_key_source = FixedKeySource::CreateFromHexStrings(
        FLAGS_key_id, FLAGS_key, kNoPssh, kNoIv);
  }
  return decryption_key_source.Pass();
}

bool AssignFlagsFromProfile() {
  bool single_segment = FLAGS_single_segment;
  if (FLAGS_profile == "on-demand") {
    single_segment = true;
  } else if (FLAGS_profile == "live") {
    single_segment = false;
  } else if (FLAGS_profile != "") {
    fprintf(stderr, "ERROR: --profile '%s' is not supported.\n",
            FLAGS_profile.c_str());
    return false;
  }

  if (FLAGS_single_segment != single_segment) {
    FLAGS_single_segment = single_segment;
    fprintf(stdout, "Profile %s: set --single_segment to %s.\n",
            FLAGS_profile.c_str(), single_segment ? "true" : "false");
  }
  return true;
}

bool GetMuxerOptions(MuxerOptions* muxer_options) {
  DCHECK(muxer_options);

  muxer_options->single_segment = FLAGS_single_segment;
  muxer_options->segment_duration = FLAGS_segment_duration;
  muxer_options->fragment_duration = FLAGS_fragment_duration;
  muxer_options->segment_sap_aligned = FLAGS_segment_sap_aligned;
  muxer_options->fragment_sap_aligned = FLAGS_fragment_sap_aligned;
  muxer_options->num_subsegments_per_sidx = FLAGS_num_subsegments_per_sidx;
  muxer_options->temp_dir = FLAGS_temp_dir;
  return true;
}

bool GetMpdOptions(MpdOptions* mpd_options) {
  DCHECK(mpd_options);

  mpd_options->availability_time_offset = FLAGS_availability_time_offset;
  mpd_options->minimum_update_period = FLAGS_minimum_update_period;
  mpd_options->min_buffer_time = FLAGS_min_buffer_time;
  mpd_options->time_shift_buffer_depth = FLAGS_time_shift_buffer_depth;
  mpd_options->suggested_presentation_delay =
      FLAGS_suggested_presentation_delay;
  return true;
}

MediaStream* FindFirstStreamOfType(const std::vector<MediaStream*>& streams,
                                   StreamType stream_type) {
  typedef std::vector<MediaStream*>::const_iterator StreamIterator;
  for (StreamIterator it = streams.begin(); it != streams.end(); ++it) {
    if ((*it)->info()->stream_type() == stream_type)
      return *it;
  }
  return NULL;
}
MediaStream* FindFirstVideoStream(const std::vector<MediaStream*>& streams) {
  return FindFirstStreamOfType(streams, kStreamVideo);
}
MediaStream* FindFirstAudioStream(const std::vector<MediaStream*>& streams) {
  return FindFirstStreamOfType(streams, kStreamAudio);
}

bool AddStreamToMuxer(const std::vector<MediaStream*>& streams,
                      const std::string& stream_selector,
                      const std::string& language_override,
                      Muxer* muxer) {
  DCHECK(muxer);

  MediaStream* stream = NULL;
  if (stream_selector == "video") {
    stream = FindFirstVideoStream(streams);
  } else if (stream_selector == "audio") {
    stream = FindFirstAudioStream(streams);
  } else {
    // Expect stream_selector to be a zero based stream id.
    size_t stream_id;
    if (!base::StringToSizeT(stream_selector, &stream_id) ||
        stream_id >= streams.size()) {
      LOG(ERROR) << "Invalid argument --stream=" << stream_selector << "; "
                 << "should be 'audio', 'video', or a number within [0, "
                 << streams.size() - 1 << "].";
      return false;
    }
    stream = streams[stream_id];
    DCHECK(stream);
  }

  // This could occur only if stream_selector=audio|video and the corresponding
  // stream does not exist in the input.
  if (!stream) {
    LOG(ERROR) << "No " << stream_selector << " stream found in the input.";
    return false;
  }

  if (!language_override.empty()) {
    stream->info()->set_language(language_override);
  }

  muxer->AddStream(stream);
  return true;
}

}  // namespace media
}  // namespace shaka