1// Copyright 2013 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/renderer/extensions/extension_localization_peer.h"
6
7#include "base/memory/scoped_ptr.h"
8#include "base/strings/string_util.h"
9#include "chrome/common/url_constants.h"
10#include "extensions/common/constants.h"
11#include "extensions/common/extension_messages.h"
12#include "extensions/common/message_bundle.h"
13#include "ipc/ipc_sender.h"
14#include "net/base/net_errors.h"
15#include "net/http/http_response_headers.h"
16
17ExtensionLocalizationPeer::ExtensionLocalizationPeer(
18    content::RequestPeer* peer,
19    IPC::Sender* message_sender,
20    const GURL& request_url)
21    : original_peer_(peer),
22      message_sender_(message_sender),
23      request_url_(request_url) {
24}
25
26ExtensionLocalizationPeer::~ExtensionLocalizationPeer() {
27}
28
29// static
30ExtensionLocalizationPeer*
31ExtensionLocalizationPeer::CreateExtensionLocalizationPeer(
32    content::RequestPeer* peer,
33    IPC::Sender* message_sender,
34    const std::string& mime_type,
35    const GURL& request_url) {
36  // Return NULL if content is not text/css or it doesn't belong to extension
37  // scheme.
38  return (request_url.SchemeIs(extensions::kExtensionScheme) &&
39          StartsWithASCII(mime_type, "text/css", false)) ?
40      new ExtensionLocalizationPeer(peer, message_sender, request_url) : NULL;
41}
42
43void ExtensionLocalizationPeer::OnUploadProgress(
44    uint64 position, uint64 size) {
45  NOTREACHED();
46}
47
48bool ExtensionLocalizationPeer::OnReceivedRedirect(
49    const net::RedirectInfo& redirect_info,
50    const content::ResourceResponseInfo& info) {
51  NOTREACHED();
52  return false;
53}
54
55void ExtensionLocalizationPeer::OnReceivedResponse(
56    const content::ResourceResponseInfo& info) {
57  response_info_ = info;
58}
59
60void ExtensionLocalizationPeer::OnReceivedData(const char* data,
61                                               int data_length,
62                                               int encoded_data_length) {
63  data_.append(data, data_length);
64}
65
66void ExtensionLocalizationPeer::OnCompletedRequest(
67    int error_code,
68    bool was_ignored_by_handler,
69    bool stale_copy_in_cache,
70    const std::string& security_info,
71    const base::TimeTicks& completion_time,
72    int64 total_transfer_size) {
73  // Make sure we delete ourselves at the end of this call.
74  scoped_ptr<ExtensionLocalizationPeer> this_deleter(this);
75
76  // Give sub-classes a chance at altering the data.
77  if (error_code != net::OK) {
78    // We failed to load the resource.
79    original_peer_->OnReceivedResponse(response_info_);
80    original_peer_->OnCompletedRequest(net::ERR_ABORTED, false,
81                                       stale_copy_in_cache, security_info,
82                                       completion_time,
83                                       total_transfer_size);
84    return;
85  }
86
87  ReplaceMessages();
88
89  original_peer_->OnReceivedResponse(response_info_);
90  if (!data_.empty())
91    original_peer_->OnReceivedData(data_.data(),
92                                   static_cast<int>(data_.size()),
93                                   -1);
94  original_peer_->OnCompletedRequest(error_code, was_ignored_by_handler,
95                                     stale_copy_in_cache,
96                                     security_info, completion_time,
97                                     total_transfer_size);
98}
99
100void ExtensionLocalizationPeer::ReplaceMessages() {
101  if (!message_sender_ || data_.empty())
102    return;
103
104  if (!request_url_.is_valid())
105    return;
106
107  std::string extension_id = request_url_.host();
108  extensions::L10nMessagesMap* l10n_messages =
109      extensions::GetL10nMessagesMap(extension_id);
110  if (!l10n_messages) {
111    extensions::L10nMessagesMap messages;
112    message_sender_->Send(new ExtensionHostMsg_GetMessageBundle(
113        extension_id, &messages));
114
115    // Save messages we got, so we don't have to ask again.
116    // Messages map is never empty, it contains at least @@extension_id value.
117    extensions::ExtensionToL10nMessagesMap& l10n_messages_map =
118        *extensions::GetExtensionToL10nMessagesMap();
119    l10n_messages_map[extension_id] = messages;
120
121    l10n_messages = extensions::GetL10nMessagesMap(extension_id);
122  }
123
124  std::string error;
125  if (extensions::MessageBundle::ReplaceMessagesWithExternalDictionary(
126          *l10n_messages, &data_, &error)) {
127    data_.resize(data_.size());
128  }
129}
130