Feat: Add method to extract HTTP Bearer token from Authorization header.

pistacheio · Oct 27, 2024 · f0c77a3 · f0c77a3
1 parent 2d573ad
commit f0c77a3
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 0 deletions.
diff --git a/include/pistache/http_header.h b/include/pistache/http_header.h
@@ -479,11 +479,15 @@ namespace Pistache::Http::Header
         // Get decoded user ID and password if basic method was used...
         std::string getBasicUser() const;
         std::string getBasicPassword() const;
+        std::string getBearerToken() const;
 
         // Set encoded user ID and password for basic method...
         void setBasicUserPassword(const std::string& User,
                                   const std::string& Password);
 
+
+        void setBearertoken(const std::string& token);
+
         void parse(const std::string& data) override;
         void write(std::ostream& os) const override;
 

diff --git a/src/common/http_header.cc b/src/common/http_header.cc
@@ -450,6 +450,16 @@ namespace Pistache::Http::Header
         return true;
     }
 
+    std::string Authorization::getBearerToken() const
+    {
+        // Verify bearer authorization method was used...
+        if (!hasMethod<Authorization::Method::Bearer>())
+            throw std::runtime_error("Authorization header does not use Bearer method.");
+
+        const std::string token(value_.begin() + std::string("Bearer ").length(), value_.end());
+        return token;
+    }
+
     // Get decoded user ID if basic method was used...
     std::string Authorization::getBasicUser() const
     {

diff --git a/tests/headers_test.cc b/tests/headers_test.cc
@@ -427,6 +427,7 @@ TEST(headers_test, authorization_basic_test)
     // Verify it decoded correctly...
     ASSERT_EQ(au.getBasicUser(), "Aladdin");
     ASSERT_EQ(au.getBasicPassword(), "OpenSesame");
+    EXPECT_ANY_THROW( au.getBearerToken() );
 }
 
 TEST(headers_test, authorization_bearer_test)
@@ -458,6 +459,17 @@ TEST(headers_test, authorization_bearer_test)
                       "eyJleHAiOjE1NzA2MzA0MDcsImlhdCI6MTU3MDU0NDAwNywibmFtZSI6IkFkbWluIE5hbWUi"
                       "LCJzYW1wbGUiOiJUZXN0In0.zLTAAnBftlqccsU-4mL69P4tQl3VhcglMg-"
                       "d0131JxqX4xSZLlO5xMRrCPBgn_00OxKJ9CQdnpjpuzblNQd2-A");
+
+    EXPECT_ANY_THROW(au.getBasicUser() );
+    EXPECT_ANY_THROW(au.getBasicPassword() );
+
+    ASSERT_EQ(au.getBearerToken(),
+                      "eyJhbGciOiJIUzUxMiIsInR5cCI6IkpXUyJ9."
+                      "eyJleHAiOjE1NzA2MzA0MDcsImlhdCI6MTU3MDU0NDAwNywibmFtZSI6IkFkbWluIE5hbWUi"
+                      "LCJzYW1wbGUiOiJUZXN0In0.zLTAAnBftlqccsU-4mL69P4tQl3VhcglMg-"
+                      "d0131JxqX4xSZLlO5xMRrCPBgn_00OxKJ9CQdnpjpuzblNQd2-A");
+
+
 }
 
 TEST(headers_test, expect_test)