fix: custom token generator ( #1 )

1️⃣  updates package dependencies
2️⃣  test cases updated

Author: AsifArmanRahman

firebase/auth/__init__.py

@@ -15,7 +15,7 @@
 import json
 import datetime
 import python_jwt as jwt
-from Crypto.PublicKey import RSA
+import jwcrypto.jwk as jwk
 
 from firebase._exception import raise_detailed_error
 
@@ -121,7 +121,7 @@ def create_custom_token(self, uid, additional_claims=None, expiry_minutes=60):
 		"""
 
 		service_account_email = self.credentials.service_account_email
-		private_key = RSA.importKey(self.credentials._private_key_pkcs8_pem)
+		private_key = jwk.JWK.from_pem(self.credentials._private_key_pkcs8_pem.encode('utf-8'))
 
 		payload = {
 			"iss": service_account_email,
@@ -135,7 +135,7 @@ def create_custom_token(self, uid, additional_claims=None, expiry_minutes=60):
 
 		exp = datetime.timedelta(minutes=expiry_minutes)
 
-		return jwt.generate_jwt(payload, private_key, "RS256", exp)
+		return jwt.generate_jwt(payload, private_key, "RS256", exp, other_headers={'kid': self.credentials._private_key_id})
 
 	def sign_in_with_custom_token(self, token):
 		""" Exchange custom token for an ID and refresh token.

pyproject.toml

@@ -79,7 +79,6 @@ classifiers = [
 dependencies = [
 	'gcloud>=0.18.3',
 	'oauth2client>=4.1.3',
-	'pycryptodome>=3.14.1',
 	'python_jwt>=3.3.2',
 	'requests>=2.27.1',
 	'six>=1.16.0',

requirements.txt

@@ -1,6 +1,5 @@
 gcloud>=0.18.3
 oauth2client>=4.1.3
-pycryptodome>=3.15.0
 python_jwt>=3.3.2
 requests>=2.27.1
 six>=1.16.0

tests/test_auth.py

@@ -42,19 +42,16 @@ def test_sign_in_anonymous(self, auth):
 		self.__class__.anonymous_user = user
 		assert user
 
-	@pytest.mark.xfail
 	def test_create_custom_token(self, auth):
-		token = auth.create_custom_token('CreateCustomToken')
+		token = auth.create_custom_token('CreateCustomToken1')
 		self.__class__.custom_token = token
 		assert token
 
-	@pytest.mark.xfail
 	def test_create_custom_token_with_claims(self, auth):
-		token = auth.create_custom_token('CreateCustomToken', {'premium': True})
+		token = auth.create_custom_token('CreateCustomToken2', {'premium': True})
 		self.__class__.custom_token_with_claims = token
 		assert token
 
-	@pytest.mark.xfail
 	def test_sign_in_with_custom_token(self, auth):
 		user1 = auth.sign_in_with_custom_token(self.__class__.custom_token)
 		user2 = auth.sign_in_with_custom_token(self.__class__.custom_token_with_claims)
@@ -87,8 +84,5 @@ def test_update_profile_display_name(self, auth):
 	def test_delete_user_account(self, auth):
 		assert auth.delete_user_account(self.__class__.user.get('idToken'))
 		assert auth.delete_user_account(self.__class__.anonymous_user.get('idToken'))
-
-	@pytest.mark.xfail
-	def test_delete_custom_user_account(self, auth):
 		assert auth.delete_user_account(self.__class__.custom_user.get('idToken'))
 		assert auth.delete_user_account(self.__class__.custom_user_with_claims.get('idToken'))