use chrono::{DateTime, Utc};
use exonum::runtime::{CommonError, ExecutionContext, ExecutionError};
use exonum_derive::{exonum_interface, interface_method, BinaryValue, ExecutionFail, ObjectHash};
use exonum_proto::ProtobufConvert;
use serde::{Deserialize, Serialize};
use crate::{proto, schema::TimeSchema, TimeService};
#[derive(Debug, ExecutionFail)]
pub enum Error {
ValidatorTimeIsGreater = 0,
}
#[derive(Clone, Debug)]
#[derive(Serialize, Deserialize)]
#[derive(ProtobufConvert, BinaryValue, ObjectHash)]
#[protobuf_convert(source = "proto::TxTime")]
pub struct TxTime {
pub time: DateTime<Utc>,
}
impl TxTime {
pub fn new(time: DateTime<Utc>) -> Self {
Self { time }
}
}
#[exonum_interface]
pub trait TimeOracleInterface<Ctx> {
type Output;
#[interface_method(id = 0)]
fn report_time(&self, ctx: Ctx, arg: TxTime) -> Self::Output;
}
impl TimeOracleInterface<ExecutionContext<'_>> for TimeService {
type Output = Result<(), ExecutionError>;
fn report_time(&self, context: ExecutionContext<'_>, arg: TxTime) -> Self::Output {
let author = context
.caller()
.author()
.ok_or(CommonError::UnauthorizedCaller)?;
let core_schema = context.data().for_core();
core_schema
.validator_id(author)
.ok_or(CommonError::UnauthorizedCaller)?;
let mut schema = TimeSchema::new(context.service_data());
schema
.update_validator_time(author, arg.time)
.map_err(|()| Error::ValidatorTimeIsGreater)?;
let validator_keys = core_schema.consensus_config().validator_keys;
schema.update_consolidated_time(&validator_keys);
Ok(())
}
}