1 // Copyright (c) 2010 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_action.h"
6
7 #include <algorithm>
8
9 #include "base/logging.h"
10 #include "chrome/common/badge_util.h"
11 #include "googleurl/src/gurl.h"
12 #include "grit/app_resources.h"
13 #include "third_party/skia/include/core/SkBitmap.h"
14 #include "third_party/skia/include/effects/SkGradientShader.h"
15 #include "ui/base/resource/resource_bundle.h"
16 #include "ui/gfx/canvas_skia.h"
17 #include "ui/gfx/rect.h"
18
19 namespace {
20
21 // Different platforms need slightly different constants to look good.
22 #if defined(OS_LINUX) && !defined(TOOLKIT_VIEWS)
23 const float kTextSize = 9.0;
24 const int kBottomMargin = 0;
25 const int kPadding = 2;
26 const int kTopTextPadding = 0;
27 #elif defined(OS_LINUX) && defined(TOOLKIT_VIEWS)
28 const float kTextSize = 8.0;
29 const int kBottomMargin = 5;
30 const int kPadding = 2;
31 const int kTopTextPadding = 1;
32 #elif defined(OS_MACOSX)
33 const float kTextSize = 9.0;
34 const int kBottomMargin = 5;
35 const int kPadding = 2;
36 const int kTopTextPadding = 0;
37 #else
38 const float kTextSize = 10;
39 const int kBottomMargin = 5;
40 const int kPadding = 2;
41 // The padding between the top of the badge and the top of the text.
42 const int kTopTextPadding = -1;
43 #endif
44
45 const int kBadgeHeight = 11;
46 const int kMaxTextWidth = 23;
47 // The minimum width for center-aligning the badge.
48 const int kCenterAlignThreshold = 20;
49
50
51 } // namespace
52
53 const int ExtensionAction::kDefaultTabId = -1;
54
ExtensionAction()55 ExtensionAction::ExtensionAction() {
56 }
57
~ExtensionAction()58 ExtensionAction::~ExtensionAction() {
59 }
60
SetPopupUrl(int tab_id,const GURL & url)61 void ExtensionAction::SetPopupUrl(int tab_id, const GURL& url) {
62 // We store |url| even if it is empty, rather than removing a URL from the
63 // map. If an extension has a default popup, and removes it for a tab via
64 // the API, we must remember that there is no popup for that specific tab.
65 // If we removed the tab's URL, GetPopupURL would incorrectly return the
66 // default URL.
67 SetValue(&popup_url_, tab_id, url);
68 }
69
HasPopup(int tab_id)70 bool ExtensionAction::HasPopup(int tab_id) {
71 return !GetPopupUrl(tab_id).is_empty();
72 }
73
GetPopupUrl(int tab_id)74 GURL ExtensionAction::GetPopupUrl(int tab_id) {
75 return GetValue(&popup_url_, tab_id);
76 }
77
SetIcon(int tab_id,const SkBitmap & bitmap)78 void ExtensionAction::SetIcon(int tab_id, const SkBitmap& bitmap) {
79 SetValue(&icon_, tab_id, bitmap);
80 }
81
GetIcon(int tab_id)82 SkBitmap ExtensionAction::GetIcon(int tab_id) {
83 return GetValue(&icon_, tab_id);
84 }
85
SetIconIndex(int tab_id,int index)86 void ExtensionAction::SetIconIndex(int tab_id, int index) {
87 if (static_cast<size_t>(index) >= icon_paths_.size()) {
88 NOTREACHED();
89 return;
90 }
91 SetValue(&icon_index_, tab_id, index);
92 }
93
ClearAllValuesForTab(int tab_id)94 void ExtensionAction::ClearAllValuesForTab(int tab_id) {
95 title_.erase(tab_id);
96 icon_.erase(tab_id);
97 icon_index_.erase(tab_id);
98 badge_text_.erase(tab_id);
99 badge_text_color_.erase(tab_id);
100 badge_background_color_.erase(tab_id);
101 visible_.erase(tab_id);
102 popup_url_.erase(tab_id);
103 }
104
PaintBadge(gfx::Canvas * canvas,const gfx::Rect & bounds,int tab_id)105 void ExtensionAction::PaintBadge(gfx::Canvas* canvas,
106 const gfx::Rect& bounds,
107 int tab_id) {
108 std::string text = GetBadgeText(tab_id);
109 if (text.empty())
110 return;
111
112 SkColor text_color = GetBadgeTextColor(tab_id);
113 SkColor background_color = GetBadgeBackgroundColor(tab_id);
114
115 if (SkColorGetA(text_color) == 0x00)
116 text_color = SK_ColorWHITE;
117
118 if (SkColorGetA(background_color) == 0x00)
119 background_color = SkColorSetARGB(255, 218, 0, 24); // Default badge color.
120
121 canvas->Save();
122
123 SkPaint* text_paint = badge_util::GetBadgeTextPaintSingleton();
124 text_paint->setTextSize(SkFloatToScalar(kTextSize));
125 text_paint->setColor(text_color);
126
127 // Calculate text width. We clamp it to a max size.
128 SkScalar text_width = text_paint->measureText(text.c_str(), text.size());
129 text_width = SkIntToScalar(
130 std::min(kMaxTextWidth, SkScalarFloor(text_width)));
131
132 // Calculate badge size. It is clamped to a min width just because it looks
133 // silly if it is too skinny.
134 int badge_width = SkScalarFloor(text_width) + kPadding * 2;
135 int icon_width = GetIcon(tab_id).width();
136 // Force the pixel width of badge to be either odd (if the icon width is odd)
137 // or even otherwise. If there is a mismatch you get http://crbug.com/26400.
138 if (icon_width != 0 && (badge_width % 2 != GetIcon(tab_id).width() % 2))
139 badge_width += 1;
140 badge_width = std::max(kBadgeHeight, badge_width);
141
142 // Paint the badge background color in the right location. It is usually
143 // right-aligned, but it can also be center-aligned if it is large.
144 SkRect rect;
145 rect.fBottom = SkIntToScalar(bounds.bottom() - kBottomMargin);
146 rect.fTop = rect.fBottom - SkIntToScalar(kBadgeHeight);
147 if (badge_width >= kCenterAlignThreshold) {
148 rect.fLeft = SkIntToScalar(
149 SkScalarFloor(SkIntToScalar(bounds.x()) +
150 SkIntToScalar(bounds.width() / 2.0) -
151 SkIntToScalar(badge_width / 2.0)));
152 rect.fRight = rect.fLeft + SkIntToScalar(badge_width);
153 } else {
154 rect.fRight = SkIntToScalar(bounds.right());
155 rect.fLeft = rect.fRight - badge_width;
156 }
157
158 SkPaint rect_paint;
159 rect_paint.setStyle(SkPaint::kFill_Style);
160 rect_paint.setAntiAlias(true);
161 rect_paint.setColor(background_color);
162 canvas->AsCanvasSkia()->drawRoundRect(rect, SkIntToScalar(2),
163 SkIntToScalar(2), rect_paint);
164
165 // Overlay the gradient. It is stretchy, so we do this in three parts.
166 ResourceBundle& resource_bundle = ResourceBundle::GetSharedInstance();
167 SkBitmap* gradient_left = resource_bundle.GetBitmapNamed(
168 IDR_BROWSER_ACTION_BADGE_LEFT);
169 SkBitmap* gradient_right = resource_bundle.GetBitmapNamed(
170 IDR_BROWSER_ACTION_BADGE_RIGHT);
171 SkBitmap* gradient_center = resource_bundle.GetBitmapNamed(
172 IDR_BROWSER_ACTION_BADGE_CENTER);
173
174 canvas->AsCanvasSkia()->drawBitmap(*gradient_left, rect.fLeft, rect.fTop);
175 canvas->TileImageInt(*gradient_center,
176 SkScalarFloor(rect.fLeft) + gradient_left->width(),
177 SkScalarFloor(rect.fTop),
178 SkScalarFloor(rect.width()) - gradient_left->width() -
179 gradient_right->width(),
180 SkScalarFloor(rect.height()));
181 canvas->AsCanvasSkia()->drawBitmap(*gradient_right,
182 rect.fRight - SkIntToScalar(gradient_right->width()), rect.fTop);
183
184 // Finally, draw the text centered within the badge. We set a clip in case the
185 // text was too large.
186 rect.fLeft += kPadding;
187 rect.fRight -= kPadding;
188 canvas->AsCanvasSkia()->clipRect(rect);
189 canvas->AsCanvasSkia()->drawText(text.c_str(), text.size(),
190 rect.fLeft + (rect.width() - text_width) / 2,
191 rect.fTop + kTextSize + kTopTextPadding,
192 *text_paint);
193 canvas->Restore();
194 }
195