83 lines
2.1 KiB
Rust
83 lines
2.1 KiB
Rust
|
use std::format;
|
||
|
|
|
||
|
|
use clap::Parser;
|
||
|
|
use futures_util::{pin_mut, StreamExt};
|
||
|
|
use tokio::io::{stdout, AsyncWriteExt};
|
||
|
|
use tokio_tungstenite::connect_async;
|
||
|
|
|
||
|
|
mod clash_conn_msg;
|
||
|
|
mod statistics;
|
||
|
|
|
||
|
|
#[derive(Parser)]
|
||
|
|
#[clap(
|
||
|
|
version = "0.1.0",
|
||
|
|
author = "Ivan Li",
|
||
|
|
about = "Watch Clash network traffic"
|
||
|
|
)]
|
||
|
|
struct Args {
|
||
|
|
#[clap(
|
||
|
|
short,
|
||
|
|
long,
|
||
|
|
default_value = "ws://192.168.31.1:9090/connections?token=123456"
|
||
|
|
)]
|
||
|
|
clash_url: String,
|
||
|
|
#[clap(short, long, default_value = "17890")]
|
||
|
|
listen_port: u16,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[tokio::main]
|
||
|
|
async fn main() {
|
||
|
|
println!("Hello, world!");
|
||
|
|
|
||
|
|
let args = Args::parse();
|
||
|
|
|
||
|
|
let connect_addr = args.clash_url;
|
||
|
|
|
||
|
|
let url = url::Url::parse(&connect_addr).unwrap();
|
||
|
|
|
||
|
|
let (ws_stream, _) = connect_async(url).await.expect("Failed to connect");
|
||
|
|
println!("WebSocket handshake has been successfully completed");
|
||
|
|
|
||
|
|
let (_, read) = ws_stream.split();
|
||
|
|
|
||
|
|
let ws_to_stdout = {
|
||
|
|
read.for_each(|message| async {
|
||
|
|
let data = message.unwrap().into_data();
|
||
|
|
|
||
|
|
let wrapper = serde_json::from_slice::<clash_conn_msg::ClashConnectionsWrapper>(&data);
|
||
|
|
|
||
|
|
if let Err(err) = wrapper {
|
||
|
|
stdout()
|
||
|
|
.write_all(format!("Error: {}\n", err).as_bytes())
|
||
|
|
.await
|
||
|
|
.unwrap();
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
let wrapper = wrapper.unwrap();
|
||
|
|
|
||
|
|
let statistics_manager = statistics::StatisticsManager::global().await;
|
||
|
|
|
||
|
|
statistics_manager.update(&wrapper).await;
|
||
|
|
|
||
|
|
let state = statistics_manager.get_state().await;
|
||
|
|
|
||
|
|
stdout()
|
||
|
|
.write_all(
|
||
|
|
format!(
|
||
|
|
"upload: {}, download: {}. speed_upload: {}, speed_download: {}\n",
|
||
|
|
wrapper.upload_total,
|
||
|
|
wrapper.download_total,
|
||
|
|
state.speed_upload,
|
||
|
|
state.speed_download
|
||
|
|
)
|
||
|
|
.as_bytes(),
|
||
|
|
)
|
||
|
|
.await
|
||
|
|
.unwrap();
|
||
|
|
})
|
||
|
|
};
|
||
|
|
|
||
|
|
pin_mut!(ws_to_stdout);
|
||
|
|
ws_to_stdout.await;
|
||
|
|
}
|