Skip to content

Commit 86fed1a

Browse files
authored
refactor: remove extension api from request builder (#787)
1 parent 82964c2 commit 86fed1a

File tree

3 files changed

+10
-19
lines changed

3 files changed

+10
-19
lines changed

crates/core/src/fixtures.rs

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -199,18 +199,21 @@ pub fn request_fixture(
199199
)
200200
.unwrap();
201201

202-
let request_config = RequestConfig::default();
202+
let mut builder = RequestConfig::builder();
203+
204+
for extension in extensions {
205+
builder.extension(extension);
206+
}
207+
208+
let request_config = builder.build().unwrap();
209+
203210
let mut request_builder = Request::builder(&request_config);
204211
request_builder
205212
.server_name(server_name)
206213
.server_cert_data(server_cert_data)
207214
.transcript(transcript)
208215
.encoding_tree(encoding_tree.clone());
209216

210-
for extension in extensions {
211-
request_builder.extension(extension);
212-
}
213-
214217
let (request, _) = request_builder.build(&provider).unwrap();
215218

216219
RequestFixture {

crates/core/src/request/builder.rs

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
use crate::{
2-
attestation::Extension,
32
connection::{ServerCertData, ServerCertOpening, ServerName},
43
index::Index,
54
request::{Request, RequestConfig},
@@ -15,7 +14,6 @@ pub struct RequestBuilder<'a> {
1514
server_cert_data: Option<ServerCertData>,
1615
encoding_tree: Option<EncodingTree>,
1716
transcript: Option<Transcript>,
18-
extensions: Vec<Extension>,
1917
}
2018

2119
impl<'a> RequestBuilder<'a> {
@@ -27,7 +25,6 @@ impl<'a> RequestBuilder<'a> {
2725
server_cert_data: None,
2826
encoding_tree: None,
2927
transcript: None,
30-
extensions: Vec::new(),
3128
}
3229
}
3330

@@ -55,12 +52,6 @@ impl<'a> RequestBuilder<'a> {
5552
self
5653
}
5754

58-
/// Adds an extension to the request.
59-
pub fn extension(&mut self, extension: Extension) -> &mut Self {
60-
self.extensions.push(extension);
61-
self
62-
}
63-
6455
/// Builds the attestation request and returns the corresponding secrets.
6556
pub fn build(
6657
self,
@@ -72,7 +63,6 @@ impl<'a> RequestBuilder<'a> {
7263
server_cert_data,
7364
encoding_tree,
7465
transcript,
75-
extensions,
7666
} = self;
7767

7868
let signature_alg = *config.signature_alg();
@@ -97,6 +87,8 @@ impl<'a> RequestBuilder<'a> {
9787

9888
let encoding_commitment_root = encoding_tree.as_ref().map(|tree| tree.root());
9989

90+
let extensions = config.extensions().to_vec();
91+
10092
let request = Request {
10193
signature_alg,
10294
hash_alg,

crates/prover/src/notarize.rs

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -87,10 +87,6 @@ impl Prover<Notarize> {
8787
}
8888
}
8989

90-
for extension in config.extensions() {
91-
builder.extension(extension.clone());
92-
}
93-
9490
let (request, secrets) = builder.build(provider).map_err(ProverError::attestation)?;
9591

9692
let attestation = mux_fut

0 commit comments

Comments
 (0)