use crate::pow::err::Error;
use crate::pow::v1::challenge::Challenge;
use crate::pow::v1::{err::RuntimeErrorV1, err::SolutionErrorV1, types::Instance, types::Solution};
use equix::{EquiXBuilder, HashError, RuntimeOption};
use super::Seed;
pub struct Verifier {
instance: Instance,
equix: EquiXBuilder,
}
impl Verifier {
pub fn new(instance: Instance) -> Self {
Self {
instance,
equix: Default::default(),
}
}
pub fn new_with_equix(instance: Instance, equix: EquiXBuilder) -> Self {
Self { instance, equix }
}
pub fn runtime(&mut self, option: RuntimeOption) -> &mut Self {
self.equix.runtime(option);
self
}
pub fn check(&self, solution: &Solution) -> Result<(), Error> {
match self.check_seed(solution) {
Err(e) => Err(Error::BadSolution(e.into())),
Ok(()) => {
let challenge = Challenge::new(&self.instance, solution.effort(), solution.nonce());
match challenge.check_effort(&solution.proof_to_bytes()) {
Err(e) => Err(Error::BadSolution(e.into())),
Ok(()) => match self.equix.verify(challenge.as_ref(), solution.proof()) {
Ok(()) => Ok(()),
Err(equix::Error::HashSum) => {
Err(Error::BadSolution(SolutionErrorV1::HashSum.into()))
}
Err(equix::Error::Hash(HashError::ProgramConstraints)) => Err(
Error::BadSolution(SolutionErrorV1::ChallengeConstraints.into()),
),
Err(e) => Err(Error::VerifyRuntime(RuntimeErrorV1::EquiX(e).into())),
},
}
}
}
}
fn check_seed(&self, solution: &Solution) -> Result<(), SolutionErrorV1> {
if solution.seed_head() == self.instance.seed().head() {
Ok(())
} else {
Err(SolutionErrorV1::Seed)
}
}
pub fn seed(&self) -> &Seed {
self.instance.seed()
}
}