2 * API Reference: https://ethereum.github.io/beacon-APIs/
7 time
::{self, Duration
},
12 message
::header
::ContentType
, transport
::smtp
::authentication
::Credentials
, Message
,
13 SmtpTransport
, Transport
,
15 use serde
::Deserialize
;
17 use crate::config
::Config
;
22 const FILE_CONF
: &str = "config.ron";
23 const CHECK_PERIOD
: Duration
= Duration
::from_secs(10); // 10 s.
24 const EMAIL_RESEND_PERIOD
: Duration
= Duration
::from_secs(6 * 60 * 60); // 6 h.
25 const STATE_PRINT_PERIOD
: Duration
= Duration
::from_secs(15 * 60); // 15 min.
26 const BASE_URI
: &str = "http://localhost:5052/eth/v1/";
28 fn main() -> Result
<()> {
29 println!("Staking Watchdog");
31 let config
= Config
::read(FILE_CONF
)?
;
34 "Configuration: {:?}",
36 smtp_password
: "*****".to_string(),
41 let mut time_last_email_send
= time
::Instant
::now() - EMAIL_RESEND_PERIOD
;
42 let mut time_last_state_printed
= time
::Instant
::now() - STATE_PRINT_PERIOD
;
43 let mut error_state
= false;
46 let time_beginning_loop
= time
::Instant
::now();
48 if let Err(error
) = check_validators(&config
.pub_keys
) {
50 println!("Error: {:?}", error
);
51 if time
::Instant
::now() - time_last_email_send
>= EMAIL_RESEND_PERIOD
{
53 println!("Sending email...");
56 &format!("Error: {:?}", error
),
58 &config
.smtp_password
,
60 Err(email_error
) => println!("Error sending email: {:?}", email_error
),
62 println!("Email successfully sent");
63 time_last_email_send
= time
::Instant
::now();
70 println!("End of erroneous state");
73 if time
::Instant
::now() - time_last_state_printed
>= STATE_PRINT_PERIOD
{
74 println!("No error detected");
75 time_last_state_printed
= time
::Instant
::now();
79 let elapsed
= time
::Instant
::now() - time_beginning_loop
;
81 if elapsed
< CHECK_PERIOD
{
82 let to_wait
= CHECK_PERIOD
- elapsed
;
83 thread
::sleep(to_wait
);
94 UnknownCodeFromHealthCheck(u16),
95 ReqwestError(reqwest
::Error
),
96 ValidatorError
{ pub_key
: String
, message
: String
},
97 ValidatorStatusError
{ pub_key
: String
, message
: String
},
100 impl From
<reqwest
::Error
> for CheckError
{
101 fn from(value
: reqwest
::Error
) -> Self {
102 CheckError
::ReqwestError(value
)
106 #[derive(Deserialize, Debug)]
107 struct JsonValidatorState
{
108 data
: JsonValidatorStateData
,
111 #[derive(Deserialize, Debug)]
112 struct JsonValidatorStateData
{
116 #[derive(Deserialize, Debug)]
122 fn check_validators(pub_keys
: &[String
]) -> std
::result
::Result
<(), CheckError
> {
124 let client
= reqwest
::blocking
::Client
::new();
126 let request_health
= client
127 .get(format!("{url}node/health"))
128 .header("accept", "application/json");
129 match request_health
.send() {
131 // println!("{resp:?}"); // For debug.
132 match resp
.status().as_u16() {
134 206 => return Err(CheckError
::NotSync
),
135 400 => return Err(CheckError
::InvalidSyncStatus
),
136 503 => return Err(CheckError
::NodeHavingIssues
),
137 code
=> return Err(CheckError
::UnknownCodeFromHealthCheck(code
)),
141 println!("{error:?}");
142 return Err(CheckError
::HttpError(error
.to_string()));
146 for pub_key
in pub_keys
{
148 .get(format!("{url}beacon/states/head/validators/0x{pub_key}"))
149 .header("accept", "application/json");
150 match request
.send() {
152 // println!("{resp:?}"); // For debug.
153 match resp
.status().as_u16() {
155 let json
: JsonValidatorState
= resp
.json()?
;
156 if json
.data
.status
!= "active_ongoing" {
157 return Err(CheckError
::ValidatorStatusError
{
158 pub_key
: pub_key
.clone(),
159 message
: format!("Status: {}", json
.data
.status
),
164 let json
: JsonError
= resp
.json()?
;
165 return Err(CheckError
::ValidatorError
{
166 pub_key
: pub_key
.clone(),
168 "Http error code: {}, message: {}",
176 return Err(CheckError
::ValidatorError
{
177 pub_key
: pub_key
.clone(),
178 message
: error
.to_string(),
187 fn send_email(title
: &str, body
: &str, login
: &str, pass
: &str) -> Result
<()> {
188 let email
= Message
::builder()
190 .from("Staking Watchdog <redmine@d-lan.net>".parse()?
)
191 .to("Greg Burri <greg.burri@gmail.com>".parse()?
)
193 .header(ContentType
::TEXT_PLAIN
)
194 .body(body
.to_string())?
;
196 let creds
= Credentials
::new(login
.to_string(), pass
.to_string());
198 // Open a remote connection to gmail
199 let mailer
= SmtpTransport
::relay("mail.gandi.net")?
204 let response
= mailer
.send(&email
)?
;
206 println!("{:?}", response
);