Skip to content

Update main.rs #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kodsher wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update main.rs #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
131 changes: 66 additions & 65 deletions src/main.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,90 +8,87 @@ use std::borrow::Cow;

async fn send_to_discord(path: String, data: Value) -> Result<impl Reply, Rejection> {
let discord_webhooks_env = env::var("DISCORD_WEBHOOKS").unwrap_or_else(|_| String::from("[]"));
info!("DISCORD_WEBHOOKS (encoded): {}", discord_webhooks_env); // Debugging line to log the content of DISCORD_WEBHOOKS
info!("DISCORD_WEBHOOKS (encoded): {}", discord_webhooks_env);

let decoded_webhooks: Cow<str> = decode(&discord_webhooks_env).unwrap_or_else(|_| String::from("[]").into());
info!("DISCORD_WEBHOOKS (decoded): {}", decoded_webhooks); // Debugging line to log the decoded content of DISCORD_WEBHOOKS
info!("DISCORD_WEBHOOKS (decoded): {}", decoded_webhooks);

// Ensure that the decoded JSON is properly formatted
let webhooks: Value = match serde_json::from_str(&decoded_webhooks) {
Ok(value) => value,
Err(err) => {
let webhooks: Value = serde_json::from_str(&decoded_webhooks)
.unwrap_or_else(|err| {
error!("Failed to parse DISCORD_WEBHOOKS: {}", err);
error!("Decoded DISCORD_WEBHOOKS value: {}", decoded_webhooks);
json!([])
}
};
});

info!("Parsed webhooks: {:?}", webhooks); // Debugging line to log parsed webhooks
info!("Parsed webhooks: {:?}", webhooks);

let webhook_url = webhooks
.as_array()
.and_then(|arr| arr.iter().find(|obj| obj["path"] == path))
.and_then(|obj| obj["url"].as_str());

match webhook_url {
Some(url) => {
info!("Using webhook URL: {}", url);
let client = reqwest::Client::new();

info!("Received data: {}", data);

let payload = if let Some(map) = data.as_object() {
let exchange = map.get("exchange").and_then(Value::as_str).unwrap_or("");
let ticker = map.get("ticker").and_then(Value::as_str).unwrap_or("");
let close = map.get("close").and_then(Value::as_str).unwrap_or("");
let open = map.get("open").and_then(Value::as_str).unwrap_or("");
let volume = map.get("volume").and_then(Value::as_str).unwrap_or("");
let event = map.get("event").and_then(Value::as_str).unwrap_or("");
let interval = map.get("interval").and_then(Value::as_str).unwrap_or("");

let color = if close < open {
16711680 // Red color in decimal (0xFF0000)
} else {
65280 // Green color in decimal (0x00FF00)
};

json!({
"embeds": [{
"author": {
"name": format!("Whistle: {} {} at {}", ticker, event, exchange),
"url": "https://github.com/coinchimp/whistle",
"icon_url": "https://raw.githubusercontent.com/coinchimp/whistle/main/assets/images/whistle.png"
},
"description": format!("Open: {}\nClose: {}\nInterval: {}\nVolume: {}\n", open, close, interval, volume),
"color": color
}]
})
if let Some(url) = webhook_url {
info!("Using webhook URL: {}", url);
let client = reqwest::Client::new();

info!("Received data: {}", data);

let payload = if let Some(map) = data.as_object() {
let exchange = map.get("exchange").and_then(Value::as_str).unwrap_or("");
let ticker = map.get("ticker").and_then(Value::as_str).unwrap_or("");
let close = map.get("close").and_then(Value::as_str).unwrap_or("");
let open = map.get("open").and_then(Value::as_str).unwrap_or("");
let volume = map.get("volume").and_then(Value::as_str).unwrap_or("");
let event = map.get("event").and_then(Value::as_str).unwrap_or("");
let interval = map.get("interval").and_then(Value::as_str).unwrap_or("");

// Determine color based on closing and opening prices
let color = if close < open {
0xFF0000 // Red
} else {
json!({
"embeds": [{
"author": {
"name": "Whistle: Text Notification",
"url": "https://github.com/coinchimp/whistle",
"icon_url": "https://raw.githubusercontent.com/coinchimp/whistle/main/assets/images/whistle.png"
},
"description": format!("Event: {}", data),
"color": 16761035 // Pink color in decimal (0xFFC0CB)
}]
})
0x00FF00 // Green
};

match client.post(url).json(&payload).send().await {
Ok(_) => {
info!("Message successfully sent to Discord.");
Ok(reply::with_status("Content sent to Discord", StatusCode::OK))
},
Err(e) => {
error!("Failed to send message to Discord: {:?}", e);
Err(warp::reject::reject())
}
json!({
"embeds": [{
"author": {
"name": format!("Whistle: {} {} at {}", ticker, event, exchange),
"url": "https://github.com/coinchimp/whistle",
"icon_url": "https://raw.githubusercontent.com/coinchimp/whistle/main/assets/images/whistle.png"
},
"description": format!("Open: {}\nClose: {}\nInterval: {}\nVolume: {}\n", open, close, interval, volume),
"color": color
}]
})
} else {
json!({
"embeds": [{
"author": {
"name": "Whistle: Text Notification",
"url": "https://github.com/coinchimp/whistle",
"icon_url": "https://raw.githubusercontent.com/coinchimp/whistle/main/assets/images/whistle.png"
},
"description": format!("Event: {}", data),
"color": 0xFFC0CB // Pink
}]
})
};

// Send payload to Discord
match client.post(url).json(&payload).send().await {
Ok(_) => {
info!("Message successfully sent to Discord.");
Ok(reply::with_status("Content sent to Discord", StatusCode::OK))
},
Err(e) => {
error!("Failed to send message to Discord: {:?}", e);
Err(warp::reject::reject())
}
},
None => {
error!("No valid webhook URL found for path: {}", path);
Err(warp::reject::reject())
}
} else {
error!("No valid webhook URL found for path: {}", path);
Err(warp::reject::reject())
}
}

Expand All @@ -106,19 +103,23 @@ async fn main() {

let port: u16 = env::var("PORT").unwrap_or_else(|_| "8080".to_string()).parse().unwrap();

// Define the webhook route
let webhook_route = warp::post()
.and(warp::path("webhook"))
.and(warp::path::param())
.and(warp::body::json::<Value>())
.and_then(|path: String, data: Value| send_to_discord(path, data));

// Define the health check route
let health_route = warp::get()
.and(warp::path::end())
.map(|| "Healthy");

// Combine the routes and set a rejection handler
let routes = webhook_route.or(health_route)
.recover(handle_rejection);

// Start the server
warp::serve(routes)
.run(([0, 0, 0, 0], port))
.await;
Expand Down