Separated bot

src/bot.rs

@@ -0,0 +1,59 @@
+use futures_util::{SinkExt, StreamExt};
+use serde_json::json;
+use tokio::net::TcpStream;
+use tokio_tungstenite::{MaybeTlsStream, WebSocketStream, connect_async};
+
+use crate::types::NewToken;
+
+pub struct Bot {
+    pub ws: WebSocketStream<MaybeTlsStream<TcpStream>>,
+}
+
+impl Bot {
+    pub async fn on_new_coin(&mut self, token: NewToken) -> anyhow::Result<()> {
+        Ok(())
+    }
+}
+
+impl Bot {
+    pub async fn new() -> anyhow::Result<Self> {
+        Ok(Self {
+            ws: connect_async("wss://pumpdev.io/ws").await?.0,
+        })
+    }
+
+    pub async fn initialize_websocket_subscribe(&mut self) -> anyhow::Result<()> {
+        self.ws
+            .send(tokio_tungstenite::tungstenite::Message::Text(
+                json!({ "method": "subscribeNewToken" }).to_string().into(),
+            ))
+            .await?;
+
+        Ok(())
+    }
+
+    pub async fn start(&mut self) -> anyhow::Result<()> {
+        self.initialize_websocket_subscribe().await?;
+
+        while let Some(msg) = self.ws.next().await {
+            let msg = msg?;
+            if let tokio_tungstenite::tungstenite::Message::Text(text) = msg {
+                match serde_json::from_str::<crate::types::PumpDevEvent>(&text) {
+                    Ok(crate::types::PumpDevEvent::Create(token)) => {
+                        self.on_new_coin(token).await?;
+                    }
+
+                    Ok(event) => {
+                        println!("{:?}", event);
+                    }
+
+                    Err(err) => {
+                        log::error!("{err}");
+                    }
+                }
+            }
+        }
+
+        Ok(())
+    }
+}

src/main.rs

@@ -1,10 +1,9 @@
 pub mod account;
+pub mod bot;
 pub mod executor;
 pub mod types;
 
-use futures_util::{SinkExt, StreamExt};
+use crate::bot::Bot;
-use serde_json::json;
-use tokio_tungstenite::connect_async;
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
@@ -12,26 +11,7 @@ async fn main() -> anyhow::Result<()> {
     builder.filter_level(log::LevelFilter::Info);
     builder.init();
 
-    let (mut ws, _) = connect_async("wss://pumpdev.io/ws").await?;
+    let mut bot = Bot::new().await?;
 
-    ws.send(tokio_tungstenite::tungstenite::Message::Text(
+    bot.start().await
-        json!({ "method": "subscribeNewToken" }).to_string().into(),
-    ))
-    .await?;
-
-    while let Some(msg) = ws.next().await {
-        let msg = msg?;
-        if let tokio_tungstenite::tungstenite::Message::Text(text) = msg {
-            match serde_json::from_str::<types::PumpDevEvent>(&text) {
-                Ok(event) => {
-                    println!("{:?}", event);
-                }
-
-                Err(err) => {
-                    log::error!("{err}");
-                }
-            }
-        }
-    }
-    Ok(())
 }