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
|
// Copyright (c) 2010 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/user_style_sheet_watcher.h"
#include "base/base64.h"
#include "base/file_util.h"
#include "chrome/common/notification_service.h"
#include "chrome/common/notification_type.h"
namespace {
// The subdirectory of the profile that contains the style sheet.
const char kStyleSheetDir[] = "User StyleSheets";
// The filename of the stylesheet.
const char kUserStyleSheetFile[] = "Custom.css";
} // namespace
UserStyleSheetWatcher::UserStyleSheetWatcher(const FilePath& profile_path)
: profile_path_(profile_path),
has_loaded_(false) {
// Listen for when the first render view host is created. If we load
// too fast, the first tab won't hear the notification and won't get
// the user style sheet.
registrar_.Add(this, NotificationType::RENDER_VIEW_HOST_CREATED_FOR_TAB,
NotificationService::AllSources());
}
void UserStyleSheetWatcher::Observe(NotificationType type,
const NotificationSource& source, const NotificationDetails& details) {
DCHECK(type == NotificationType::RENDER_VIEW_HOST_CREATED_FOR_TAB);
if (has_loaded_) {
NotificationService::current()->Notify(
NotificationType::USER_STYLE_SHEET_UPDATED,
Source<UserStyleSheetWatcher>(this),
NotificationService::NoDetails());
}
registrar_.RemoveAll();
}
void UserStyleSheetWatcher::OnFileChanged(const FilePath& path) {
ChromeThread::PostTask(ChromeThread::FILE, FROM_HERE,
NewRunnableMethod(this, &UserStyleSheetWatcher::LoadStyleSheet,
profile_path_));
}
void UserStyleSheetWatcher::Init() {
ChromeThread::PostTask(ChromeThread::FILE, FROM_HERE,
NewRunnableMethod(this, &UserStyleSheetWatcher::LoadStyleSheet,
profile_path_));
}
void UserStyleSheetWatcher::LoadStyleSheet(const FilePath& profile_path) {
DCHECK(ChromeThread::CurrentlyOn(ChromeThread::FILE));
// We keep the user style sheet in a subdir so we can watch for changes
// to the file.
FilePath style_sheet_dir = profile_path.AppendASCII(kStyleSheetDir);
if (!file_util::DirectoryExists(style_sheet_dir)) {
if (!file_util::CreateDirectory(style_sheet_dir))
return;
}
// Create the file if it doesn't exist.
FilePath css_file = style_sheet_dir.AppendASCII(kUserStyleSheetFile);
if (!file_util::PathExists(css_file))
file_util::WriteFile(css_file, "", 0);
std::string css;
bool rv = file_util::ReadFileToString(css_file, &css);
GURL style_sheet_url;
if (rv && !css.empty()) {
std::string css_base64;
rv = base::Base64Encode(css, &css_base64);
if (rv) {
// WebKit knows about data urls, so convert the file to a data url.
const char kDataUrlPrefix[] = "data:text/css;charset=utf-8;base64,";
style_sheet_url = GURL(kDataUrlPrefix + css_base64);
}
}
ChromeThread::PostTask(ChromeThread::UI, FROM_HERE,
NewRunnableMethod(this, &UserStyleSheetWatcher::SetStyleSheet,
style_sheet_url));
if (!file_watcher_.get()) {
file_watcher_.reset(new FileWatcher);
file_watcher_->Watch(profile_path_.AppendASCII(kStyleSheetDir)
.AppendASCII(kUserStyleSheetFile), this);
}
}
void UserStyleSheetWatcher::SetStyleSheet(const GURL& url) {
DCHECK(ChromeThread::CurrentlyOn(ChromeThread::UI));
has_loaded_ = true;
user_style_sheet_ = url;
NotificationService::current()->Notify(
NotificationType::USER_STYLE_SHEET_UPDATED,
Source<UserStyleSheetWatcher>(this),
NotificationService::NoDetails());
}
|
