chrome/browser/ui/webui/app_list/start_page_ui.cc

   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/file_util.h"
   8 #include "base/files/file_path.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/profiles/profile.h"
  13 #include "chrome/browser/ui/webui/app_list/start_page_handler.h"
  14 #include "chrome/common/url_constants.h"
  15 #include "content/public/browser/browser_thread.h"
  16 #include "content/public/browser/web_ui.h"
  17 #include "content/public/browser/web_ui_data_source.h"
  18 #include "grit/browser_resources.h"
  19
  20 namespace app_list {
  21 namespace {
  22 #if defined(OS_CHROMEOS)
  23 const char* kHotwordFilenames[] = {
  24   "dnn",
  25   "ep_acoustic_model",
  26   "hclg_shotword",
  27   "hotword_arm.nexe",
  28   "hotword_classifier",
  29   "hotword_normalizer",
  30   "hotword_word_symbols",
  31   "hotword_x86_32.nexe",
  32   "hotword_x86_64.nexe",
  33   "okgoogle_hotword.config",
  34   "phonelist",
  35   "phone_state_map",
  36 };
  37
  38 const char kHotwordFilesDir[] = "/opt/google/hotword";
  39
  40 void LoadModelData(const std::string& path,
  41                    const content::WebUIDataSource::GotDataCallback& callback) {
  42   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::FILE));
  43   // Will be owned by |callback|.
  44   scoped_refptr<base::RefCountedString> data(new base::RefCountedString());
  45   base::FilePath base_dir(kHotwordFilesDir);
  46   base::ReadFileToString(base_dir.AppendASCII(path), &(data->data()));
  47   callback.Run(data.get());
  48 }
  49
  50 bool HandleHotwordFilesResourceFilter(
  51     const std::string& path,
  52     const content::WebUIDataSource::GotDataCallback& callback) {
  53   for (size_t i = 0; i < arraysize(kHotwordFilenames); ++i) {
  54     if (path == kHotwordFilenames[i]) {
  55       content::BrowserThread::PostTask(
  56           content::BrowserThread::FILE,
  57           FROM_HERE,
  58           base::Bind(&LoadModelData, path, callback));
  59       return true;
  60     }
  61   }
  62
  63   return false;
  64 }
  65 #endif  // OS_CHROMEOS
  66 }  // namespace
  67
  68 StartPageUI::StartPageUI(content::WebUI* web_ui)
  69     : content::WebUIController(web_ui) {
  70   web_ui->AddMessageHandler(new StartPageHandler);
  71   InitDataSource();
  72 }
  73
  74 StartPageUI::~StartPageUI() {}
  75
  76 void 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 #endif
  93
  94   content::WebUIDataSource::Add(Profile::FromWebUI(web_ui()), source.release());
  95 }
  96
  97 }  // namespace app_list