Include request in get_headers

Author: wvanlint

src/client.rs

@@ -130,7 +130,7 @@ impl<R: RetryPolicy<E = VssError>> VssClient<R> {
 		let request_body = request.encode_to_vec();
 		let headers = self
 			.header_provider
-			.get_headers()
+			.get_headers(&request_body)
 			.await
 			.map_err(|e| VssError::InternalError(e.to_string()))?;
 		let headermap = get_headermap(&headers).map_err(VssError::InternalError)?;

src/headers/lnurl_auth_jwt.rs

@@ -139,7 +139,7 @@ impl LnurlAuthJwt {
 
 #[async_trait]
 impl VssHeaderProvider for LnurlAuthJwt {
-	async fn get_headers(&self) -> Result<HashMap<String, String>, VssError> {
+	async fn get_headers(&self, _request: &[u8]) -> Result<HashMap<String, String>, VssError> {
 		let jwt_token = self.get_jwt_token(false).await?;
 		let mut headers = self.default_headers.clone();
 		headers.insert(AUTHORIZATION.to_string(), format!("Bearer {}", jwt_token));

src/headers/mod.rs

@@ -15,7 +15,7 @@ pub use lnurl_auth_jwt::LnurlAuthJwt;
 pub trait VssHeaderProvider {
 	/// Returns the HTTP headers to be used for a VSS request.
 	/// This method is called on each request, and should likely perform some form of caching.
-	async fn get_headers(&self) -> Result<HashMap<String, String>, VssError>;
+	async fn get_headers(&self, request: &[u8]) -> Result<HashMap<String, String>, VssError>;
 }
 
 /// A header provider returning an given, fixed set of headers.
@@ -32,7 +32,7 @@ impl FixedHeaders {
 
 #[async_trait]
 impl VssHeaderProvider for FixedHeaders {
-	async fn get_headers(&self) -> Result<HashMap<String, String>, VssError> {
+	async fn get_headers(&self, _request: &[u8]) -> Result<HashMap<String, String>, VssError> {
 		Ok(self.headers.clone())
 	}
 }

tests/lnurl_auth_jwt_tests.rs

@@ -56,7 +56,7 @@ mod lnurl_auth_jwt_tests {
 				.with_body(lnurl_auth_response(&expired_jwt))
 				.create();
 			assert_eq!(
-				lnurl_auth_jwt.get_headers().await.unwrap().get("authorization").unwrap(),
+				lnurl_auth_jwt.get_headers(&[]).await.unwrap().get("authorization").unwrap(),
 				&format!("Bearer {}", expired_jwt),
 			);
 			lnurl.assert();
@@ -86,11 +86,11 @@ mod lnurl_auth_jwt_tests {
 				.with_body(lnurl_auth_response(&valid_jwt))
 				.create();
 			assert_eq!(
-				lnurl_auth_jwt.get_headers().await.unwrap().get("authorization").unwrap(),
+				lnurl_auth_jwt.get_headers(&[]).await.unwrap().get("authorization").unwrap(),
 				&format!("Bearer {}", valid_jwt),
 			);
 			assert_eq!(
-				lnurl_auth_jwt.get_headers().await.unwrap().get("authorization").unwrap(),
+				lnurl_auth_jwt.get_headers(&[]).await.unwrap().get("authorization").unwrap(),
 				&format!("Bearer {}", valid_jwt),
 			);
 			lnurl.assert();