From 7ec4c69fd763749a7c510bb5ee50121473933ab4 Mon Sep 17 00:00:00 2001
From: Bryan Stitt <bryan@llamanodes.com>
Date: Tue, 24 Jan 2023 00:05:41 -0800
Subject: [PATCH] actually add the new file

---
 .../src/bin/web3_proxy_cli/pagerduty.rs       | 89 +++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 web3_proxy/src/bin/web3_proxy_cli/pagerduty.rs

diff --git a/web3_proxy/src/bin/web3_proxy_cli/pagerduty.rs b/web3_proxy/src/bin/web3_proxy_cli/pagerduty.rs
new file mode 100644
index 00000000..2097068d
--- /dev/null
+++ b/web3_proxy/src/bin/web3_proxy_cli/pagerduty.rs
@@ -0,0 +1,89 @@
+use argh::FromArgs;
+use gethostname::gethostname;
+use log::{error, info};
+use pagerduty_rs::{
+    eventsv2async::EventsV2 as PagerdutyAsyncEventsV2,
+    types::{AlertTrigger, AlertTriggerPayload, Event},
+};
+use web3_proxy::config::TopConfig;
+
+#[derive(FromArgs, PartialEq, Debug, Eq)]
+/// Quickly create a pagerduty alert
+#[argh(subcommand, name = "pagerduty")]
+pub struct PagerdutySubCommand {
+    #[argh(positional)]
+    /// short description of the alert
+    summary: String,
+
+    #[argh(option)]
+    /// the class/type of the event
+    class: Option<String>,
+
+    #[argh(option)]
+    /// deduplicate alerts based on this key.
+    /// If there are no open incidents with this key, a new incident will be created.
+    /// If there is an open incident with a matching key, the new event will be appended to that incident's Alerts log as an additional Trigger log entry.
+    dedup_key: Option<String>,
+
+    #[argh(option, default = "\"web3-proxy\".to_string()")]
+    /// a cluster or grouping of sources.
+    /// For example, sources "ethereum-proxy" and "polygon-proxy" might both be part of "web3-proxy".
+    group: String,
+}
+
+impl PagerdutySubCommand {
+    pub async fn main(
+        &self,
+        pagerduty_async: Option<PagerdutyAsyncEventsV2>,
+        top_config: Option<TopConfig>,
+    ) -> anyhow::Result<()> {
+        let client = top_config
+            .as_ref()
+            .map(|top_config| format!("web3-proxy chain #{}", top_config.app.chain_id))
+            .unwrap_or_else(|| format!("web3-proxy w/o chain"));
+
+        let client_url = top_config
+            .as_ref()
+            .and_then(|x| x.app.redirect_public_url.clone());
+
+        let hostname = gethostname().into_string().unwrap_or("unknown".to_string());
+
+        let payload = AlertTriggerPayload {
+            severity: pagerduty_rs::types::Severity::Error,
+            summary: self.summary.clone(),
+            source: hostname,
+            timestamp: None,
+            component: None,
+            group: Some(self.group.clone()),
+            class: self.class.clone(),
+            custom_details: None::<()>,
+        };
+
+        let event = AlertTrigger {
+            payload,
+            dedup_key: None,
+            images: None,
+            links: None,
+            client: Some(client),
+            client_url: client_url,
+        };
+
+        if let Some(pagerduty_async) = pagerduty_async {
+            info!(
+                "sending to pagerduty: {}",
+                serde_json::to_string_pretty(&event)?
+            );
+
+            if let Err(err) = pagerduty_async.event(Event::AlertTrigger(event)).await {
+                error!("Failed sending to pagerduty: {}", err);
+            }
+        } else {
+            info!(
+                "would send to pagerduty if PAGERDUTY_INTEGRATION_KEY were set: {}",
+                serde_json::to_string_pretty(&event)?
+            );
+        }
+
+        Ok(())
+    }
+}