summaryrefslogtreecommitdiffstats
path: root/chrome/browser/extensions/extension_toolstrip_api.cc
blob: 248eedb247500c9149ad499e3dbab2025d2840b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/extensions/extension_toolstrip_api.h"

#include "chrome/browser/browser.h"
#include "chrome/browser/extensions/extension_host.h"
#include "chrome/browser/extensions/extension_shelf_model.h"

namespace extension_toolstrip_api_functions {
const char kExpandFunction[] = "toolstrip.expand";
const char kCollapseFunction[] = "toolstrip.collapse";
};  // namespace extension_toolstrip_api_functions

namespace {
// Errors.
const char kNotAToolstripError[] = "This page is not a toolstrip.";
const char kAlreadyExpandedError[] = "This toolstrip is already expanded.";
const char kAlreadyCollapsedError[] = "This toolstrip is already collapsed.";
const char kInvalidURLError[] = "Invalid URL";
const char kBadHeightError[] = "Bad height.";

// TODO(erikkay) what are good values here?
const int kMinHeight = 50;
const int kMaxHeight = 1000;
};  // namespace

bool ToolstripFunction::RunImpl() {
  ExtensionHost* host = dispatcher()->GetExtensionHost();
  if (!host) {
    error_ = kNotAToolstripError;
    return false;
  }
  Browser* browser = dispatcher()->GetBrowser();
  if (!browser) {
    error_ = kNotAToolstripError;
    return false;
  }
  model_ = browser->extension_shelf_model();
  if (!model_) {
    error_ = kNotAToolstripError;
    return false;
  }
  toolstrip_ = model_->ToolstripForHost(host);
  if (toolstrip_ == model_->end()) {
    error_ = kNotAToolstripError;
    return false;
  }
  return true;
}

bool ToolstripExpandFunction::RunImpl() {
  if (!ToolstripFunction::RunImpl())
    return false;
  if (toolstrip_->height != 0) {
    error_ = kAlreadyExpandedError;
    return false;
  }

  EXTENSION_FUNCTION_VALIDATE(args_->IsType(Value::TYPE_LIST));
  const ListValue* args = static_cast<const ListValue*>(args_);
  EXTENSION_FUNCTION_VALIDATE(args->GetSize() <= 2);

  int height;
  EXTENSION_FUNCTION_VALIDATE(args->GetInteger(0, &height));
  EXTENSION_FUNCTION_VALIDATE(height >= 0);
  if (height < kMinHeight || height > kMaxHeight) {
    error_ = kBadHeightError;
    return false;
  }

  GURL url;
  if (args->GetSize() == 2) {
    Value* url_val;
    EXTENSION_FUNCTION_VALIDATE(args->Get(1, &url_val));
    if (url_val->GetType() != Value::TYPE_NULL) {
      std::string url_str;
      EXTENSION_FUNCTION_VALIDATE(url_val->GetAsString(&url_str));
      url = GURL(url_str);
      if (!url.is_valid() && !url.is_empty()) {
        error_ = kInvalidURLError;
        return false;
      }
    }
  }

  model_->ExpandToolstrip(toolstrip_, url, height);
  return true;
}

bool ToolstripCollapseFunction::RunImpl() {
  if (!ToolstripFunction::RunImpl())
    return false;

  if (toolstrip_->height == 0) {
    error_ = kAlreadyCollapsedError;
    return false;
  }

  GURL url;
  if (args_->GetType() != Value::TYPE_NULL) {
    std::string url_str;
    EXTENSION_FUNCTION_VALIDATE(args_->GetAsString(&url_str));
    url = GURL(url_str);
    if (!url.is_valid() && !url.is_empty()) {
      error_ = kInvalidURLError;
      return false;
    }
  }

  model_->CollapseToolstrip(toolstrip_, url);
  return true;
}