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
ExtensionLocalizationPeer(webkit_glue::ResourceLoaderBridge::Peer * peer,IPC::Message::Sender * message_sender,const GURL & request_url)17 ExtensionLocalizationPeer::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
~ExtensionLocalizationPeer()26 ExtensionLocalizationPeer::~ExtensionLocalizationPeer() {
27 }
28
29 // static
30 ExtensionLocalizationPeer*
CreateExtensionLocalizationPeer(webkit_glue::ResourceLoaderBridge::Peer * peer,IPC::Message::Sender * message_sender,const std::string & mime_type,const GURL & request_url)31 ExtensionLocalizationPeer::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
OnUploadProgress(uint64 position,uint64 size)43 void ExtensionLocalizationPeer::OnUploadProgress(
44 uint64 position, uint64 size) {
45 NOTREACHED();
46 }
47
OnReceivedRedirect(const GURL & new_url,const webkit_glue::ResourceResponseInfo & info,bool * has_new_first_party_for_cookies,GURL * new_first_party_for_cookies)48 bool 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
OnReceivedResponse(const webkit_glue::ResourceResponseInfo & info)57 void ExtensionLocalizationPeer::OnReceivedResponse(
58 const webkit_glue::ResourceResponseInfo& info) {
59 response_info_ = info;
60 }
61
OnReceivedData(const char * data,int data_length,int encoded_data_length)62 void ExtensionLocalizationPeer::OnReceivedData(const char* data,
63 int data_length,
64 int encoded_data_length) {
65 data_.append(data, data_length);
66 }
67
OnCompletedRequest(const net::URLRequestStatus & status,const std::string & security_info,const base::Time & completion_time)68 void 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
ReplaceMessages()95 void 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