mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-22 03:14:06 -05:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Seed a dev license blob into the DB for CI test environments.
|
|
|
|
Usage (docker):
|
|
docker exec -e ONYX_DEV_LICENSE onyx-api_server-1 \
|
|
python -m scripts.seed_dev_license
|
|
|
|
Reads ONYX_DEV_LICENSE from the environment. Empty values no-op so the
|
|
script can be invoked unconditionally (e.g. local dev runs without a
|
|
license to hand). Accepts both PEM-armored and raw base64 license blobs;
|
|
verifies the RSA-4096 signature before persisting.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
sys.path.append(parent_dir)
|
|
|
|
from ee.onyx.db.license import upsert_license # noqa: E402
|
|
from ee.onyx.utils.license import verify_license_signature # noqa: E402
|
|
from onyx.db.engine.sql_engine import get_session_with_current_tenant # noqa: E402
|
|
from onyx.db.engine.sql_engine import SqlEngine # noqa: E402
|
|
|
|
_PEM_BEGIN = "-----BEGIN ONYX LICENSE-----"
|
|
_PEM_END = "-----END ONYX LICENSE-----"
|
|
|
|
|
|
def _strip_pem_delimiters(content: str) -> str:
|
|
content = content.strip()
|
|
if content.startswith(_PEM_BEGIN) and content.endswith(_PEM_END):
|
|
return "\n".join(content.split("\n")[1:-1]).strip()
|
|
return content
|
|
|
|
|
|
def main() -> None:
|
|
blob = os.environ.get("ONYX_DEV_LICENSE", "").strip()
|
|
if not blob:
|
|
print("ONYX_DEV_LICENSE empty: skipping license seed")
|
|
return
|
|
|
|
license_data = _strip_pem_delimiters(blob)
|
|
verify_license_signature(license_data)
|
|
|
|
SqlEngine.init_engine(pool_size=1, max_overflow=0)
|
|
with get_session_with_current_tenant() as db_session:
|
|
upsert_license(db_session, license_data)
|
|
|
|
print("Dev license seeded")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|