1// Copyright (c) 2011 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/common/extensions/extension_localization_peer.h"
6
7#include "base/memory/scoped_ptr.h"
8#include "base/string_util.h"
9#include "chrome/common/extensions/extension_message_bundle.h"
10#include "chrome/common/extensions/extension_messages.h"
11#include "chrome/common/url_constants.h"
12#include "grit/generated_resources.h"
13#include "net/base/net_errors.h"
14#include "net/http/http_response_headers.h"
15#include "webkit/glue/webkit_glue.h"
16
17ExtensionLocalizationPeer::ExtensionLocalizationPeer(
18    webkit_glue::ResourceLoaderBridge::Peer* peer,
19    IPC::Message::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    webkit_glue::ResourceLoaderBridge::Peer* peer,
33    IPC::Message::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(chrome::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 GURL& new_url,
50    const webkit_glue::ResourceResponseInfo& info,
51    bool* has_new_first_party_for_cookies,
52    GURL* new_first_party_for_cookies) {
53  NOTREACHED();
54  return false;
55}
56
57void ExtensionLocalizationPeer::OnReceivedResponse(
58    const webkit_glue::ResourceResponseInfo& info) {
59  response_info_ = info;
60}
61
62void ExtensionLocalizationPeer::OnReceivedData(const char* data,
63                                               int data_length,
64                                               int encoded_data_length) {
65  data_.append(data, data_length);
66}
67
68void ExtensionLocalizationPeer::OnCompletedRequest(
69    const net::URLRequestStatus& status,
70    const std::string& security_info,
71    const base::Time& completion_time) {
72  // Make sure we delete ourselves at the end of this call.
73  scoped_ptr<ExtensionLocalizationPeer> this_deleter(this);
74
75  // Give sub-classes a chance at altering the data.
76  if (status.status() != net::URLRequestStatus::SUCCESS) {
77    // We failed to load the resource.
78    original_peer_->OnReceivedResponse(response_info_);
79    net::URLRequestStatus status(net::URLRequestStatus::CANCELED,
80                                 net::ERR_ABORTED);
81    original_peer_->OnCompletedRequest(status, security_info, completion_time);
82    return;
83  }
84
85  ReplaceMessages();
86
87  original_peer_->OnReceivedResponse(response_info_);
88  if (!data_.empty())
89    original_peer_->OnReceivedData(data_.data(),
90                                   static_cast<int>(data_.size()),
91                                   -1);
92  original_peer_->OnCompletedRequest(status, security_info, completion_time);
93}
94
95void ExtensionLocalizationPeer::ReplaceMessages() {
96  if (!message_sender_ || data_.empty())
97    return;
98
99  if (!request_url_.is_valid())
100    return;
101
102  std::string extension_id = request_url_.host();
103  L10nMessagesMap* l10n_messages = GetL10nMessagesMap(extension_id);
104  if (!l10n_messages) {
105    L10nMessagesMap messages;
106    message_sender_->Send(new ExtensionHostMsg_GetMessageBundle(
107        extension_id, &messages));
108
109    // Save messages we got, so we don't have to ask again.
110    // Messages map is never empty, it contains at least @@extension_id value.
111    ExtensionToL10nMessagesMap& l10n_messages_map =
112        *GetExtensionToL10nMessagesMap();
113    l10n_messages_map[extension_id] = messages;
114
115    l10n_messages = GetL10nMessagesMap(extension_id);
116  }
117
118  std::string error;
119  if (ExtensionMessageBundle::ReplaceMessagesWithExternalDictionary(
120          *l10n_messages, &data_, &error)) {
121    data_.resize(data_.size());
122  }
123}
124