refactor: remove extension api from request builder (#787)

This commit is contained in:
sinu.eth
2025-04-18 13:01:28 +07:00
committed by GitHub
parent 82964c273b
commit 86fed1a90c
3 changed files with 10 additions and 19 deletions

View File

@@ -199,7 +199,14 @@ pub fn request_fixture(
)
.unwrap();
let request_config = RequestConfig::default();
let mut builder = RequestConfig::builder();
for extension in extensions {
builder.extension(extension);
}
let request_config = builder.build().unwrap();
let mut request_builder = Request::builder(&request_config);
request_builder
.server_name(server_name)
@@ -207,10 +214,6 @@ pub fn request_fixture(
.transcript(transcript)
.encoding_tree(encoding_tree.clone());
for extension in extensions {
request_builder.extension(extension);
}
let (request, _) = request_builder.build(&provider).unwrap();
RequestFixture {

View File

@@ -1,5 +1,4 @@
use crate::{
attestation::Extension,
connection::{ServerCertData, ServerCertOpening, ServerName},
index::Index,
request::{Request, RequestConfig},
@@ -15,7 +14,6 @@ pub struct RequestBuilder<'a> {
server_cert_data: Option<ServerCertData>,
encoding_tree: Option<EncodingTree>,
transcript: Option<Transcript>,
extensions: Vec<Extension>,
}
impl<'a> RequestBuilder<'a> {
@@ -27,7 +25,6 @@ impl<'a> RequestBuilder<'a> {
server_cert_data: None,
encoding_tree: None,
transcript: None,
extensions: Vec::new(),
}
}
@@ -55,12 +52,6 @@ impl<'a> RequestBuilder<'a> {
self
}
/// Adds an extension to the request.
pub fn extension(&mut self, extension: Extension) -> &mut Self {
self.extensions.push(extension);
self
}
/// Builds the attestation request and returns the corresponding secrets.
pub fn build(
self,
@@ -72,7 +63,6 @@ impl<'a> RequestBuilder<'a> {
server_cert_data,
encoding_tree,
transcript,
extensions,
} = self;
let signature_alg = *config.signature_alg();
@@ -97,6 +87,8 @@ impl<'a> RequestBuilder<'a> {
let encoding_commitment_root = encoding_tree.as_ref().map(|tree| tree.root());
let extensions = config.extensions().to_vec();
let request = Request {
signature_alg,
hash_alg,

View File

@@ -87,10 +87,6 @@ impl Prover<Notarize> {
}
}
for extension in config.extensions() {
builder.extension(extension.clone());
}
let (request, secrets) = builder.build(provider).map_err(ProverError::attestation)?;
let attestation = mux_fut