#![allow(clippy::all, missing_docs, unused_imports)]
use serde::{Deserialize, Serialize};
use super::*;
use crate::playwright::connection::{
Binary, Channel, ChannelType, ObjectRef, ProtocolError, ProtocolEvent, Ref,
};
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteInitializer {
#[serde(rename = "url")]
pub url: String,
#[serde(rename = "protocols")]
pub protocols: Vec<String>,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteSendToPageParams {
#[serde(rename = "message")]
pub message: String,
#[serde(rename = "isBase64")]
pub is_base64: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteSendToServerParams {
#[serde(rename = "message")]
pub message: String,
#[serde(rename = "isBase64")]
pub is_base64: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteClosePageParams {
#[serde(rename = "code", default, skip_serializing_if = "Option::is_none")]
pub code: Option<i64>,
#[serde(rename = "reason", default, skip_serializing_if = "Option::is_none")]
pub reason: Option<String>,
#[serde(rename = "wasClean")]
pub was_clean: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteCloseServerParams {
#[serde(rename = "code", default, skip_serializing_if = "Option::is_none")]
pub code: Option<i64>,
#[serde(rename = "reason", default, skip_serializing_if = "Option::is_none")]
pub reason: Option<String>,
#[serde(rename = "wasClean")]
pub was_clean: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteMessageFromPageEventParams {
#[serde(rename = "message")]
pub message: String,
#[serde(rename = "isBase64")]
pub is_base64: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteMessageFromServerEventParams {
#[serde(rename = "message")]
pub message: String,
#[serde(rename = "isBase64")]
pub is_base64: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteClosePageEventParams {
#[serde(rename = "code", default, skip_serializing_if = "Option::is_none")]
pub code: Option<i64>,
#[serde(rename = "reason", default, skip_serializing_if = "Option::is_none")]
pub reason: Option<String>,
#[serde(rename = "wasClean")]
pub was_clean: bool,
}
#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct WebSocketRouteCloseServerEventParams {
#[serde(rename = "code", default, skip_serializing_if = "Option::is_none")]
pub code: Option<i64>,
#[serde(rename = "reason", default, skip_serializing_if = "Option::is_none")]
pub reason: Option<String>,
#[serde(rename = "wasClean")]
pub was_clean: bool,
}
#[derive(Debug, Clone, PartialEq)]
pub enum WebSocketRouteEvent {
MessageFromPage(WebSocketRouteMessageFromPageEventParams),
MessageFromServer(WebSocketRouteMessageFromServerEventParams),
ClosePage(WebSocketRouteClosePageEventParams),
CloseServer(WebSocketRouteCloseServerEventParams),
Unknown {
method: String,
params: serde_json::Value,
},
}
impl ProtocolEvent for WebSocketRouteEvent {
fn parse(method: &str, params: serde_json::Value) -> Result<Self, serde_json::Error> {
match method {
"messageFromPage" => Ok(Self::MessageFromPage(serde_json::from_value(params)?)),
"messageFromServer" => Ok(Self::MessageFromServer(serde_json::from_value(params)?)),
"closePage" => Ok(Self::ClosePage(serde_json::from_value(params)?)),
"closeServer" => Ok(Self::CloseServer(serde_json::from_value(params)?)),
_ => Ok(Self::Unknown { method: method.to_string(), params }),
}
}
}
#[derive(Debug, Clone)]
pub struct WebSocketRoute {
channel: Channel,
}
impl ChannelType for WebSocketRoute {
const INTERFACE: &'static str = "WebSocketRoute";
type Initializer = WebSocketRouteInitializer;
type Event = WebSocketRouteEvent;
fn accepts(interface: &str) -> bool {
matches!(interface, "WebSocketRoute")
}
fn from_channel(channel: Channel) -> Self {
Self { channel }
}
fn channel(&self) -> &Channel {
&self.channel
}
}
impl WebSocketRoute {
pub async fn connect(&self) -> Result<(), ProtocolError> {
self.channel.send_no_result("connect", &serde_json::json!({})).await
}
pub async fn ensure_opened(&self) -> Result<(), ProtocolError> {
self.channel.send_no_result("ensureOpened", &serde_json::json!({})).await
}
pub async fn send_to_page(&self, params: WebSocketRouteSendToPageParams) -> Result<(), ProtocolError> {
self.channel.send_no_result("sendToPage", ¶ms).await
}
pub async fn send_to_server(&self, params: WebSocketRouteSendToServerParams) -> Result<(), ProtocolError> {
self.channel.send_no_result("sendToServer", ¶ms).await
}
pub async fn close_page(&self, params: WebSocketRouteClosePageParams) -> Result<(), ProtocolError> {
self.channel.send_no_result("closePage", ¶ms).await
}
pub async fn close_server(&self, params: WebSocketRouteCloseServerParams) -> Result<(), ProtocolError> {
self.channel.send_no_result("closeServer", ¶ms).await
}
}