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
19namespace {
20
21// Different platforms need slightly different constants to look good.
22#if defined(OS_LINUX) && !defined(TOOLKIT_VIEWS)
23const float kTextSize = 9.0;
24const int kBottomMargin = 0;
25const int kPadding = 2;
26const int kTopTextPadding = 0;
27#elif defined(OS_LINUX) && defined(TOOLKIT_VIEWS)
28const float kTextSize = 8.0;
29const int kBottomMargin = 5;
30const int kPadding = 2;
31const int kTopTextPadding = 1;
32#elif defined(OS_MACOSX)
33const float kTextSize = 9.0;
34const int kBottomMargin = 5;
35const int kPadding = 2;
36const int kTopTextPadding = 0;
37#else
38const float kTextSize = 10;
39const int kBottomMargin = 5;
40const int kPadding = 2;
41// The padding between the top of the badge and the top of the text.
42const int kTopTextPadding = -1;
43#endif
44
45const int kBadgeHeight = 11;
46const int kMaxTextWidth = 23;
47// The minimum width for center-aligning the badge.
48const int kCenterAlignThreshold = 20;
49
50
51}  // namespace
52
53const int ExtensionAction::kDefaultTabId = -1;
54
55ExtensionAction::ExtensionAction() {
56}
57
58ExtensionAction::~ExtensionAction() {
59}
60
61void 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
70bool ExtensionAction::HasPopup(int tab_id) {
71  return !GetPopupUrl(tab_id).is_empty();
72}
73
74GURL ExtensionAction::GetPopupUrl(int tab_id) {
75  return GetValue(&popup_url_, tab_id);
76}
77
78void ExtensionAction::SetIcon(int tab_id, const SkBitmap& bitmap) {
79  SetValue(&icon_, tab_id, bitmap);
80}
81
82SkBitmap ExtensionAction::GetIcon(int tab_id) {
83  return GetValue(&icon_, tab_id);
84}
85
86void 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
94void 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
105void 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