mirror of
https://github.com/tlsnotary/tlsn.git
synced 2026-01-09 14:48:13 -05:00
refactor: remove extension api from request builder (#787)
This commit is contained in:
@@ -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 {
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user