electron-injector 1.0.2

electron-injector is an open source command-line tool written in Rust that enables you to inject JavaScript code into Electron apps at runtime. It is inspired by the Python project electron-inject and offers similar functionality.
Documentation 
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

use std::net::TcpStream;

use log::{debug, info};
use tungstenite::{stream::MaybeTlsStream, Message};

pub struct WebSocket {
    address: String,
    socket: tungstenite::WebSocket<MaybeTlsStream<TcpStream>>,
}

impl WebSocket {
    pub fn connect(address: &str) -> Result<Self, tungstenite::Error> {
        let url = url::Url::parse(address).expect("Should be a valid address");

        let (socket, response) = tungstenite::connect(url)?;
        info!("WebSocket connected (status: {})", response.status());

        debug!("Response headers: {:#?}", response.headers());

        Ok(WebSocket {
            address: String::from(address),
            socket,
        })
    }

    pub fn send(&mut self, msg: &str) -> Result<(), tungstenite::Error> {
        self.socket
            .write_message(Message::Text(String::from(msg)))?;

        Ok(())
    }

    pub fn receive(&mut self) -> Result<Message, tungstenite::Error> {
        let msg = self.socket.read_message()?;

        Ok(msg)
    }

    pub fn send_and_receive(&mut self, msg: &str) -> Result<Message, tungstenite::Error> {
        self.send(msg)?;
        self.receive()
    }

    pub fn close(&mut self) -> Result<(), tungstenite::Error> {
        self.socket.close(None)
    }
}

impl Drop for WebSocket {
    fn drop(&mut self) {
        debug!("Closing WebSocket (address: {:?})", self.address);
        _ = self.close();
    }
}