diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..82f9275 --- /dev/null +++ b/.gitignore @@ -0,0 +1,162 @@ +# Byte-compiled / optimized / DLL files +__pycache__/ +*.py[cod] +*$py.class + +# C extensions +*.so + +# Distribution / packaging +.Python +build/ +develop-eggs/ +dist/ +downloads/ +eggs/ +.eggs/ +lib/ +lib64/ +parts/ +sdist/ +var/ +wheels/ +share/python-wheels/ +*.egg-info/ +.installed.cfg +*.egg +MANIFEST + +# PyInstaller +# Usually these files are written by a python script from a template +# before PyInstaller builds the exe, so as to inject date/other infos into it. +*.manifest +*.spec + +# Installer logs +pip-log.txt +pip-delete-this-directory.txt + +# Unit test / coverage reports +htmlcov/ +.tox/ +.nox/ +.coverage +.coverage.* +.cache +nosetests.xml +coverage.xml +*.cover +*.py,cover +.hypothesis/ +.pytest_cache/ +cover/ + +# Translations +*.mo +*.pot + +# Django stuff: +*.log +local_settings.py +db.sqlite3 +db.sqlite3-journal + +# Flask stuff: +instance/ +.webassets-cache + +# Scrapy stuff: +.scrapy + +# Sphinx documentation +docs/_build/ + +# PyBuilder +.pybuilder/ +target/ + +# Jupyter Notebook +.ipynb_checkpoints + +# IPython +profile_default/ +ipython_config.py + +# pyenv +# For a library or package, you might want to ignore these files since the code is +# intended to run in multiple environments; otherwise, check them in: +# .python-version + +# pipenv +# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control. +# However, in case of collaboration, if having platform-specific dependencies or dependencies +# having no cross-platform support, pipenv may install dependencies that don't work, or not +# install all needed dependencies. +#Pipfile.lock + +# poetry +# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control. +# This is especially recommended for binary packages to ensure reproducibility, and is more +# commonly ignored for libraries. +# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control +#poetry.lock + +# pdm +# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control. +#pdm.lock +# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it +# in version control. +# https://pdm.fming.dev/latest/usage/project/#working-with-version-control +.pdm.toml +.pdm-python +.pdm-build/ + +# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm +__pypackages__/ + +# Celery stuff +celerybeat-schedule +celerybeat.pid + +# SageMath parsed files +*.sage.py + +# Environments +.env +.venv +env/ +venv/ +ENV/ +env.bak/ +venv.bak/ + +# Spyder project settings +.spyderproject +.spyproject + +# Rope project settings +.ropeproject + +# mkdocs documentation +/site + +# mypy +.mypy_cache/ +.dmypy.json +dmypy.json + +# Pyre type checker +.pyre/ + +# pytype static type analyzer +.pytype/ + +# Cython debug symbols +cython_debug/ + +# PyCharm +# JetBrains specific template is maintained in a separate JetBrains.gitignore that can +# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore +# and can be added to the global gitignore or merged into this file. For a more nuclear +# option (not recommended) you can uncomment the following to ignore the entire idea folder. +#.idea/ diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..e89cd15 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,13 @@ +[project] +name = "synnax-shared" +version = "0.0.0" +description = "Synnax shared Python pacakges" +readme = "README.md" +requires-python = ">=3.8" + +dependencies = [] + +[project.optional-dependencies] +system-token-issuer = ["boto3", "boto3-stubs[lambda]", "PyJWT"] +http-bearer-token-client = ['requests'] +all = ["synnax-shared[system-token-issuer,http-bearer-token-client]"] diff --git a/src/synnax_shared/__init__.py b/src/synnax_shared/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/synnax_shared/http_bearer_token_client.py b/src/synnax_shared/http_bearer_token_client.py new file mode 100644 index 0000000..9372368 --- /dev/null +++ b/src/synnax_shared/http_bearer_token_client.py @@ -0,0 +1,18 @@ +from typing import Callable, Dict + +import requests + + +class HttpBearerTokenClient: + def __init__(self, api_url: str, token_provider: Callable[[], str]): + self.api_url = api_url + self.token_provider = token_provider + self.session = requests.Session() + + def post(self, endpoint: str, request_body: Dict) -> Dict: + response = self.session.post( + self.api_url + endpoint, + json=request_body, + headers={"Authorization": "Bearer " + self.token_provider()}, + ) + return response.json() diff --git a/src/synnax_shared/system_token_issuer.py b/src/synnax_shared/system_token_issuer.py new file mode 100644 index 0000000..aa5c465 --- /dev/null +++ b/src/synnax_shared/system_token_issuer.py @@ -0,0 +1,54 @@ +import json +import logging +import time +from typing import Dict, List + +import jwt +from mypy_boto3_lambda import LambdaClient + +logger = logging.getLogger() + + +class SystemTokenIssuer: + def __init__( + self, + lambda_client: LambdaClient, + function_name: str, + system_id: str, + permissions: List[Dict[str, str | List[str]]], + refresh_before_expiry_seconds: int, + ): + self.lambda_client = lambda_client + self.function_name = function_name + self.system_id = system_id + self.permissions = permissions + self.refresh_before_expiry_seconds = refresh_before_expiry_seconds + self.token = None + + def get_token(self) -> str: + if self.token is None: + self.refresh_token() + + if self.is_token_expiring(): + self.refresh_token() + + return self.token + + def refresh_token(self) -> None: + logging.info("Refreshing token") + response = self.lambda_client.invoke( + FunctionName=self.function_name, + InvocationType="RequestResponse", + Payload=json.dumps( + {"systemId": self.system_id, "permissions": self.permissions} + ), + ) + payload = json.loads(response["Payload"].read()) + self.token = payload["token"] + + def is_token_expiring(self) -> bool: + payload = jwt.decode(self.token, options={"verify_signature": False}) + epoch_time = int(time.time()) + if payload["exp"] - epoch_time < self.refresh_before_expiry_seconds: + return True + return False