1// Copyright 2013 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/browser/ui/webui/app_list/start_page_ui.h"
6
7#include "base/files/file_path.h"
8#include "base/files/file_util.h"
9#include "base/memory/ref_counted_memory.h"
10#include "base/memory/scoped_ptr.h"
11#include "base/sys_info.h"
12#include "chrome/browser/extensions/extension_service.h"
13#include "chrome/browser/profiles/profile.h"
14#include "chrome/browser/ui/webui/app_list/start_page_handler.h"
15#include "chrome/common/extensions/extension_constants.h"
16#include "chrome/common/url_constants.h"
17#include "content/public/browser/browser_thread.h"
18#include "content/public/browser/web_ui.h"
19#include "content/public/browser/web_ui_data_source.h"
20#include "extensions/browser/extension_system.h"
21#include "extensions/common/extension.h"
22#include "grit/browser_resources.h"
23
24namespace app_list {
25namespace {
26#if defined(OS_CHROMEOS)
27const char* kHotwordFilePrefixes[] = {
28  "hotword_",
29  "_platform_specific/",
30};
31
32void LoadModelData(const base::FilePath& base_dir,
33                   const std::string& path,
34                   const content::WebUIDataSource::GotDataCallback& callback) {
35  DCHECK_CURRENTLY_ON(content::BrowserThread::FILE);
36  // Will be owned by |callback|.
37  scoped_refptr<base::RefCountedString> data(new base::RefCountedString());
38  base::ReadFileToString(base_dir.AppendASCII(path), &(data->data()));
39  callback.Run(data.get());
40}
41
42bool HandleHotwordFilesResourceFilter(
43    Profile* profile,
44    const std::string& path,
45    const content::WebUIDataSource::GotDataCallback& callback) {
46  ExtensionService* service =
47      extensions::ExtensionSystem::Get(profile)->extension_service();
48  const extensions::Extension* extension =
49      service->GetExtensionById(extension_misc::kHotwordExtensionId, false);
50  if (!extension)
51    return false;
52
53  for (size_t i = 0; i < arraysize(kHotwordFilePrefixes); ++i) {
54    if (path.find(kHotwordFilePrefixes[i]) == 0) {
55      content::BrowserThread::PostTask(
56          content::BrowserThread::FILE,
57          FROM_HERE,
58          base::Bind(&LoadModelData, extension->path(), path, callback));
59      return true;
60    }
61  }
62
63  return false;
64}
65#endif  // OS_CHROMEOS
66}  // namespace
67
68StartPageUI::StartPageUI(content::WebUI* web_ui)
69    : content::WebUIController(web_ui) {
70  web_ui->AddMessageHandler(new StartPageHandler);
71  InitDataSource();
72}
73
74StartPageUI::~StartPageUI() {}
75
76void StartPageUI::InitDataSource() {
77  scoped_ptr<content::WebUIDataSource> source(
78      content::WebUIDataSource::Create(chrome::kChromeUIAppListStartPageHost));
79
80  source->SetUseJsonJSFormatV2();
81  source->SetJsonPath("strings.js");
82
83  source->AddResourcePath("start_page.css", IDR_APP_LIST_START_PAGE_CSS);
84  source->AddResourcePath("start_page.js", IDR_APP_LIST_START_PAGE_JS);
85  source->AddResourcePath("hotword_nacl.nmf", IDR_APP_LIST_HOTWORD_NACL_NMF);
86  source->SetDefaultResource(IDR_APP_LIST_START_PAGE_HTML);
87
88#if defined(OS_CHROMEOS)
89  source->OverrideContentSecurityPolicyObjectSrc("object-src 'self' data:;");
90  if (base::SysInfo::IsRunningOnChromeOS())
91    source->SetRequestFilter(base::Bind(&HandleHotwordFilesResourceFilter,
92                                        Profile::FromWebUI(web_ui())));
93#endif
94
95  content::WebUIDataSource::Add(Profile::FromWebUI(web_ui()), source.release());
96}
97
98}  // namespace app_list
99