wip

crates/api_clients/grok_client/src/requests/listen_websocket/create_listen_websocket_raw_non_websocket.rs

@@ -1,14 +1,8 @@
 use crate::error::grok_client_error::GrokClientError;
 use crate::error::grok_error::GrokError;
-use crate::error::grok_generic_api_error::GrokGenericApiError;
-use log::{debug, error, info, trace, warn};
 use std::ops::Deref;
-use std::sync::Arc;
 use std::time::Duration;
 use tokio::io::AsyncReadExt;
-use wreq::header::{ACCEPT, ACCEPT_ENCODING, ACCEPT_LANGUAGE, CACHE_CONTROL, CONNECTION, COOKIE, ORIGIN, PRAGMA, UPGRADE, USER_AGENT};
-use wreq::http1::{Http1Options, Http1OptionsBuilder};
-use wreq::ws::message::Message;
 use wreq::Client;
 use wreq_util::Emulation;
 
@@ -31,7 +25,8 @@ pub async fn create_listen_websocket_raw_non_websocket() -> Result<(), GrokError
   let user_agent = "User-Agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/141.0.0.0 Safari/537.36";
 
   let builder = client
-      .get(WEBSOCKET_WSS_URL)
+      .get(WEBSOCKET_HTTPS_URL)
+      .header("Host", "grok.com")
       .header("Accept", "*/*")
       .header("Cookie", cookies)
       .header("Connection", "Upgrade")
@@ -56,7 +51,6 @@ pub async fn create_listen_websocket_raw_non_websocket() -> Result<(), GrokError
 #[cfg(test)]
 mod tests {
   use super::*;
-  use crate::test_utils::get_test_cookies::get_test_cookies;
   use crate::test_utils::setup_test_logging::setup_test_logging;
   use errors::AnyhowResult;
   use log::LevelFilter;
@@ -65,94 +59,9 @@ mod tests {
   #[ignore] // manually test
   async fn create() -> AnyhowResult<()> {
     setup_test_logging(LevelFilter::Trace);
-
-    let result = create_listen_websocket_raw_non_websocket().await;
-
-    match result {
-      Ok(ok) => {
-        println!("Okay");
-      }
-      Err(err) => {
-        println!("{:?}", err);
-      }
-    }
-
+    let _result = create_listen_websocket_raw_non_websocket().await;
     log::logger().flush();
-
     assert_eq!(1, 2);
-
     Ok(())
   }
 }
-
-
-
-
-
-/*
-let message = r#"
-  {"type":"conversation.item.create","timestamp":1760673207293,"item":{"type":"message","content":[{"requestId":"3cedf20e-f51f-da5d-a124-ccec05faedf1","text":"A pirannah","type":"input_text","properties":{"section_count":0,"is_kids_mode":false,"enable_nsfw":true,"skip_upsampler":false,"is_initial":false}}]}}
-"#.trim().to_string();
-
-let wire_message = Message::text(message);
-
-websocket.send(wire_message)
-    .await
-    .map_err(|err| GrokGenericApiError::WreqError(err))?;
-
-loop {
-  let maybe_message = websocket.recv().await;
-
-  if let Some(message) = maybe_message {
-    let message = message.map_err(|err| GrokGenericApiError::WreqError(err))?;
-
-    match message {
-      Message::Text(text) => {
-        println!("Received text message: {}", text);
-      }
-      Message::Binary(bin) => {
-        println!("Received binary message: {:?}", bin);
-      }
-      Message::Ping(ping) => {
-        println!("Received ping: {:?}", ping);
-      }
-      Message::Pong(pong) => {
-        println!("Received pong: {:?}", pong);
-      }
-      Message::Close(close_frame) => {
-        println!("Received close: {:?}", close_frame);
-        break;
-      }
-      _ => {
-        println!("Received other message: {:?}", message);
-      }
-    }
-
-    tokio::time::sleep(std::time::Duration::from_millis(15_000)).await;
-  }
-}
-*/
-
-//for (k, v) in response.headers().iter() {
-//  println!("Header: {}: {:?}", k, v);
-//}
-
-//println!("Upgrading...");
-//info!("Upgrading...");
-//let upgraded = response
-//    .upgrade()
-//    .await
-//    .map_err(|err| GrokClientError::WreqClientError(err))?;
-
-
-// println!("Into websocket...");
-// info!("Into websocket...");
-
-// // ApiGeneric(WreqError(wreq::Error { kind: Upgrade, source: "unexpected status code: 403 Forbidden" }))
-// let mut websocket = response.into_websocket()
-//     .await
-//     .map_err(|err| GrokGenericApiError::WreqError(err))?;
-
-// if let Some(protocol) = websocket.protocol() {
-//   println!("WebSocket subprotocol: {:?}", protocol);
-// }