forked from mirrors/gecko-dev
1aea46318e
<!-- Please describe your changes on the following line: --> --- <!-- Thank you for contributing to Servo! Please replace each `[ ]` by `[X]` when the step is complete, and replace `__` with appropriate data: --> - [x] `./mach build -d` does not report any errors - [x] `./mach test-tidy` does not report any errors - [x] These changes fix #12520 (github issue number if applicable). <!-- Either: --> - [ ] There are tests for these changes OR - [x] These changes do not require tests because it's a refactoring <!-- Pull requests that do not address these steps are welcome, but they will require additional verification as part of the review process. --> Source-Repo: https://github.com/servo/servo Source-Revision: a39bd7db5ba26fa3c95e4bbbb5f1a358bc01965f
54 lines
1.9 KiB
Rust
54 lines
1.9 KiB
Rust
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
use dom::bindings::codegen::Bindings::DocumentBinding::DocumentMethods;
|
|
use dom::bindings::codegen::Bindings::NodeBinding::NodeMethods;
|
|
use dom::bindings::inheritance::Castable;
|
|
use dom::bindings::js::RootedReference;
|
|
use dom::bindings::str::DOMString;
|
|
use dom::htmlheadelement::HTMLHeadElement;
|
|
use dom::node::Node;
|
|
use std::borrow::ToOwned;
|
|
use std::fs::read_dir;
|
|
use std::path::PathBuf;
|
|
use util::opts;
|
|
use util::resource_files::resources_dir_path;
|
|
|
|
|
|
pub fn load_script(head: &HTMLHeadElement) {
|
|
if let Some(ref path_str) = opts::get().userscripts {
|
|
let node = head.upcast::<Node>();
|
|
let first_child = node.GetFirstChild();
|
|
let doc = node.owner_doc();
|
|
let doc = doc.r();
|
|
|
|
let path = if &**path_str == "" {
|
|
if let Ok(mut p) = resources_dir_path() {
|
|
p.push("user-agent-js");
|
|
p
|
|
} else {
|
|
return
|
|
}
|
|
} else {
|
|
PathBuf::from(path_str)
|
|
};
|
|
|
|
let mut files = read_dir(&path).expect("Bad path passed to --userscripts")
|
|
.filter_map(|e| e.ok())
|
|
.map(|e| e.path()).collect::<Vec<_>>();
|
|
|
|
files.sort();
|
|
|
|
for file in files {
|
|
let name = match file.into_os_string().into_string() {
|
|
Ok(ref s) if s.ends_with(".js") => "file://".to_owned() + &s[..],
|
|
_ => continue
|
|
};
|
|
let new_script = doc.CreateElement(DOMString::from("script")).unwrap();
|
|
let new_script = new_script.r();
|
|
new_script.set_string_attribute(&atom!("src"), DOMString::from(name));
|
|
node.InsertBefore(new_script.upcast(), first_child.r()).unwrap();
|
|
}
|
|
}
|
|
}
|