From 4f365ab0a6cb783dbe3615e0804e8b7ceb0f563b Mon Sep 17 00:00:00 2001
From: Quentin Gliech <quenting@element.io>
Date: Fri, 19 Aug 2022 12:18:38 +0200
Subject: [PATCH] Add the evaluation context policy generic parameter

---
 crates/policy/src/lib.rs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/crates/policy/src/lib.rs b/crates/policy/src/lib.rs
index b2cfb38dc..21e942b24 100644
--- a/crates/policy/src/lib.rs
+++ b/crates/policy/src/lib.rs
@@ -170,17 +170,16 @@ impl EvaluationResult {
     }
 }
 
-#[derive(Debug)]
 pub struct Policy {
     store: Store<()>,
-    instance: opa_wasm::Policy,
+    instance: opa_wasm::Policy<opa_wasm::DefaultContext>,
     register_entrypoint: String,
     client_registration_entrypoint: String,
     authorization_grant_endpoint: String,
 }
 
 impl Policy {
-    #[tracing::instrument]
+    #[tracing::instrument(skip(self, password))]
     pub async fn evaluate_register(
         &mut self,
         username: &str,
@@ -203,7 +202,7 @@ impl Policy {
         Ok(res)
     }
 
-    #[tracing::instrument]
+    #[tracing::instrument(skip(self))]
     pub async fn evaluate_client_registration(
         &mut self,
         client_metadata: &ClientMetadata,
@@ -225,7 +224,7 @@ impl Policy {
         Ok(res)
     }
 
-    #[tracing::instrument]
+    #[tracing::instrument(skip(self))]
     pub async fn evaluate_authorization_grant<T: StorageBackend + std::fmt::Debug>(
         &mut self,
         authorization_grant: &AuthorizationGrant<T>,