Extract a registered client type

1 files changed, 14 insertions, 1 deletions

diff --git a/src/http/mod.rs b/src/http/mod.rs
index 4523d3b..7b1b983 100644
--- a/src/http/mod.rs
+++ b/src/http/mod.rs
@@ -166,8 +166,11 @@ impl Server {
     fn handle_token(&self, stream: &mut TcpStream, request: &str) {
         let body = self.extract_body(request);
         let form_params = self.parse_form_data(&body);
+        
+        // Extract Authorization header from request
+        let auth_header = self.extract_auth_header(request);
 
-        match self.oauth_server.handle_token(&form_params) {
+        match self.oauth_server.handle_token(&form_params, auth_header.as_deref()) {
             Ok(token_response) => {
                 self.send_json_response(stream, 200, "OK", &token_response);
             }
@@ -200,4 +203,14 @@ impl Server {
             })
             .collect()
     }
+
+    fn extract_auth_header(&self, request: &str) -> Option<String> {
+        let lines: Vec<&str> = request.lines().collect();
+        for line in lines.iter().skip(1) { // Skip the request line
+            if line.to_lowercase().starts_with("authorization:") {
+                return Some(line[14..].trim().to_string()); // Skip "Authorization: "
+            }
+        }
+        None
+    }
 }
\ No newline at end of file