diff --git a/addons/addons.json b/addons/addons.json new file mode 100644 index 00000000..96bf8d07 --- /dev/null +++ b/addons/addons.json @@ -0,0 +1,7 @@ +[ + { + "name":"cssLoader", + "file":"cssLoader.js", + "devMode":true + } +] \ No newline at end of file diff --git a/addons/cssLoader.js b/addons/cssLoader.js new file mode 100644 index 00000000..cbcb7d86 --- /dev/null +++ b/addons/cssLoader.js @@ -0,0 +1,25 @@ +var cssLoaderLiveReload={} +fetch("https://music.163.com/betterncm_api/read_file?stylesheets/all.json") +.then(function(resp){return resp.json()}) +.then(function(resp){ + for(var x=0;x, + ) -> RequestOrResponse { + let mut resp = String::from(""); + if req.uri().path().starts_with("/puzzle/puzzle") { + // resp = fs::read_to_string("I:\\betterNCM\\betterncm\\frontend\\inject.js").expect("No inject.js"); + resp = String::from_utf8_lossy(include_bytes!("../frontend/inject.js")).to_string(); + } + println!("{}", req.uri().path()); + { + let api = req.uri().path(); + if api == "/betterncm_api/addons" { + resp = fs::read_to_string("./addons/addons.json").expect("No addons.json"); + } + if api == "/betterncm_api/read_file" { + if req.uri().query().expect("wrong api call").contains("..") { + return RequestOrResponse::Request(req); + } + match fs::read_to_string(format!( + "./addons/{}", + req.uri().query().expect("wrong api call") + )) { + Ok(s) => resp = s, + Err(_) => resp = "".to_string(), + } + } + } + if !resp.is_empty() { + let res = Response::new(Body::from(resp.clone())); + println!("Handled request:{:?}\n\n\nReturned:{}", req, resp); + RequestOrResponse::Response(res) + } else { + RequestOrResponse::Request(req) + } + } + + async fn handle_response(&mut self, _ctx: &HttpContext, res: Response) -> Response { + res + } +} + +#[derive(Clone)] +struct WsLogHandler {} + +#[async_trait] +impl MessageHandler for WsLogHandler { + async fn handle_message(&mut self, _ctx: &MessageContext, msg: Message) -> Option { + println!("{:?}", msg); + Some(msg) + } +} + +fn elevate(){ + use is_elevated::is_elevated; + if !is_elevated() { + msgbox::create("警告", "初次运行，请右键管理员运行~", IconType::Error).unwrap(); + std::process::exit(0); + } +} + +fn write_assets(){ + if !fs::try_exists("cloudmusic.dll").unwrap() { + msgbox::create("用法", "1.关闭网易云音乐，将本程序拷贝到网易云音乐安装目录内，替换cloudmusic.exe\n2.在 网易云音乐-工具 上点击自定义代理，输入\n\t服务器：localhost\t端口：3000\n3.点击确定", IconType::Error).unwrap(); + std::process::exit(0); + } + + match fs::try_exists("cloudmusicn.exe"){ + Ok(true) => { + println!("cloudmusicn.exe exists"); + }, + Ok(false) => { + println!("cloudmusicn.exe not exists"); + elevate(); + fs::write("cloudmusicn.exe", include_bytes!("../cloudmusicn.exe")).expect("write cloudmusicn.exe failed"); + }, + Err(e) => { + println!("cloudmusicn.exe not exists"); + println!("{:?}", e); + } + } + + if !fs::try_exists("addons").unwrap() { + elevate(); + fs::create_dir("addons").expect("create addons failed"); + } + if !fs::try_exists("./addons/addons.json").unwrap(){ + elevate(); + fs::write("./addons/addons.json", include_bytes!("../../addons/addons.json")).expect("write addons.json failed"); + } + if !fs::try_exists("./addons/cssLoader.js").unwrap(){ + elevate(); + fs::write("./addons/cssLoader.js", include_bytes!("../../addons/cssLoader.js")).expect("write cssLoader.js failed"); + } + if !fs::try_exists("addons/stylesheets").unwrap() { + elevate(); + fs::create_dir("addons/stylesheets").expect("create addons/stylesheets failed"); + } + if !fs::try_exists("./addons/stylesheets/all.json").unwrap(){ + elevate(); + fs::write("./addons/stylesheets/all.json", include_bytes!("../../addons/stylesheets/all.json")).expect("write all.json failed"); + } + if !fs::try_exists("./addons/stylesheets/block.css").unwrap(){ + elevate(); + fs::write("./addons/stylesheets/block.css", include_bytes!("../../addons/stylesheets/block.css")).expect("write block.css failed"); + } +} + + +#[tokio::main] +async fn main() { + use std::process::Command; + + write_assets(); + + + Command::new("cloudmusicn.exe") + .spawn() + .expect("Failed to launch NCM"); + match Mutex::create_named("BetterNCM") { + Ok(_) => { + tracing_subscriber::fmt::init(); + let private_key_bytes: &[u8] = include_bytes!("../ca/key.pem"); + let ca_cert_bytes: &[u8] = include_bytes!("../ca/cert.pem"); + let private_key = + PKey::private_key_from_pem(private_key_bytes).expect("Failed to parse private key"); + let ca_cert = X509::from_pem(ca_cert_bytes).expect("Failed to parse CA certificate"); + + let ca = OpensslAuthority::new(private_key, ca_cert, MessageDigest::sha256(), 1_000); + + let proxy = ProxyBuilder::new() + .with_addr(SocketAddr::from(([127, 0, 0, 1], 3000))) + .with_rustls_client() + .with_ca(ca) + .with_http_handler(LogHandler {}) + .build(); + + if let Err(e) = proxy.start(shutdown_signal()).await { + error!("{}", e); + } + } + Err(CreateNamedError::AlreadyExists(_)) => { + println!("Process already running!"); + } + Err(error) => { + println!("An error occurred: {}", error); + } + } +}