mascully/nu_plugin_ws
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Run cargo fmt

Browse source 
Signed-off-by: Alex Kattathra Johnson <alex.kattathra.johnson@gmail.com>
This commit is contained in:
Alex Kattathra Johnson 2024-12-06 17:07:09 -06:00
parent 5f35894619
commit e35542ddf3
No known key found for this signature in database
GPG key ID: 64BCC76905798553
3 changed files with 14 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/main.rs
View file

@ -1,8 +1,11 @@
use std::time::Duration;
use nu_plugin::{EvaluatedCall, JsonSerializer, serve_plugin};
use nu_plugin::{serve_plugin, EvaluatedCall, JsonSerializer};
use nu_plugin::{EngineInterface, Plugin, PluginCommand};
use nu_protocol::{ByteStream, ByteStreamType, Category, LabeledError, PipelineData, Signature, SyntaxShape, Type, Value};
use nu_protocol::{
ByteStream, ByteStreamType, Category, LabeledError, PipelineData, Signature, SyntaxShape, Type,
Value,
};
mod ws;
use ws::client::{connect, http_parse_url, request_headers};
@ -15,9 +18,7 @@ impl Plugin for WebSocketPlugin {
}
fn commands(&self) -> Vec<Box<dyn PluginCommand<Plugin = Self>>> {
vec![
Box::new(WebSocket),
]
vec![Box::new(WebSocket)]
}
}
@ -80,11 +81,7 @@ impl PluginCommand for WebSocket {
.expect("Timeout should be set to duration") as u64,
)
});
if let Some(cr) = connect(
requested_url,
timeout,
request_headers(headers)?,
) {
if let Some(cr) = connect(requested_url, timeout, request_headers(headers)?) {
let reader = Box::new(cr);
return Ok(PipelineData::ByteStream(
ByteStream::read(

12
src/ws/client.rs
View file

@ -100,17 +100,17 @@ pub fn connect(
tungstenite::Message::Text(msg) => {
if tx.send(msg.as_bytes().to_vec()).is_err() {
websocket.close(Some(tungstenite::protocol::CloseFrame{
code: tungstenite::protocol::frame::coding::CloseCode::Normal,
reason: std::borrow::Cow::Borrowed("byte stream closed"),
})).expect("Could not close connection")
code: tungstenite::protocol::frame::coding::CloseCode::Normal,
reason: std::borrow::Cow::Borrowed("byte stream closed"),
})).expect("Could not close connection")
}
}
tungstenite::Message::Binary(msg) => {
if tx.send(msg).is_err() {
websocket.close(Some(tungstenite::protocol::CloseFrame{
code: tungstenite::protocol::frame::coding::CloseCode::Normal,
reason: std::borrow::Cow::Borrowed("byte stream closed"),
})).expect("Could not close connection")
code: tungstenite::protocol::frame::coding::CloseCode::Normal,
reason: std::borrow::Cow::Borrowed("byte stream closed"),
})).expect("Could not close connection")
}
}
tungstenite::Message::Close(..) => {

2
src/ws/mod.rs
View file

@ -1 +1 @@
pub mod client;
pub mod client;