// Copyright 2015 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/renderer/plugins/plugin_preroller.h"

#include "base/base64.h"
#include "chrome/grit/renderer_resources.h"
#include "chrome/renderer/plugins/chrome_plugin_placeholder.h"
#include "chrome/renderer/plugins/power_saver_info.h"
#include "third_party/blink/public/platform/web_rect.h"
#include "third_party/blink/public/web/web_element.h"
#include "third_party/blink/public/web/web_plugin.h"
#include "third_party/blink/public/web/web_plugin_container.h"
#include "ui/gfx/codec/png_codec.h"

PluginPreroller::PluginPreroller(content::RenderFrame* render_frame,
                                 const blink::WebPluginParams& params,
                                 const content::WebPluginInfo& info,
                                 const std::string& identifier,
                                 const base::string16& name,
                                 const base::string16& message,
                                 content::PluginInstanceThrottler* throttler)
    : RenderFrameObserver(render_frame),
      params_(params),
      info_(info),
      identifier_(identifier),
      name_(name),
      message_(message),
      throttler_(throttler) {
  DCHECK(throttler);
  throttler_->AddObserver(this);
}

PluginPreroller::~PluginPreroller() {
  if (throttler_)
    throttler_->RemoveObserver(this);
}

void PluginPreroller::OnKeyframeExtracted(const SkBitmap* bitmap) {
  std::vector<unsigned char> png_data;
  if (!gfx::PNGCodec::EncodeBGRASkBitmap(*bitmap, false, &png_data)) {
    DLOG(ERROR) << "Provided keyframe could not be encoded as PNG.";
    return;
  }

  base::StringPiece png_as_string(reinterpret_cast<char*>(&png_data[0]),
                                  png_data.size());

  std::string data_url_header = "data:image/png;base64,";
  std::string data_url_body;
  base::Base64Encode(png_as_string, &data_url_body);
  keyframe_data_url_ = GURL(data_url_header + data_url_body);
}

void PluginPreroller::OnThrottleStateChange() {
  if (!throttler_->IsThrottled())
    return;

  PowerSaverInfo power_saver_info;
  power_saver_info.power_saver_enabled = true;
  power_saver_info.poster_attribute = keyframe_data_url_.spec();
  power_saver_info.custom_poster_size = throttler_->GetSize();

  ChromePluginPlaceholder* placeholder =
      ChromePluginPlaceholder::CreateBlockedPlugin(
          render_frame(), params_, info_, identifier_, name_,
          IDR_PLUGIN_POSTER_HTML, message_, power_saver_info);
  placeholder->SetPremadePlugin(throttler_);
  placeholder->AllowLoading();

  blink::WebPluginContainer* container =
      throttler_->GetWebPlugin()->Container();
  container->SetPlugin(placeholder->plugin());

  bool success = placeholder->plugin()->Initialize(container);
  DCHECK(success);

  container->Invalidate();
  container->ReportGeometry();

  delete this;
}

void PluginPreroller::OnThrottlerDestroyed() {
  throttler_ = nullptr;
  delete this;
}

void PluginPreroller::OnDestruct() {
  delete this;
}