config.py

#!/usr/bin/env python3
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License.
import base64
import hashlib
import os

import dotenv
from botframework.connector.auth import AppCredentials
from botframework.connector.auth import CertificateAppCredentials
from botframework.connector.auth import MicrosoftAppCredentials

dotenv.load_dotenv()
""" Bot Configuration """

__all__ = ["DefaultConfig", "config"]


class DefaultConfig:
    """Bot Configuration"""

    PORT = int(os.environ.get("PORT", "3980"))
    APP_ID = os.environ.get("MICROSOFT_APP_ID", "")
    APP_PASSWORD = os.environ.get("MICROSOFT_APP_PASSWORD", "")
    APP_CERTIFICATE = os.environ.get("MICROSOFT_APP_CERTIFICATE", "")
    APP_PRIVATEKEY = os.environ.get("MICROSOFT_APP_PRIVATEKEY", "")
    APP_TYPE = os.environ.get("MICROSOFT_APP_TYPE", "MultiTenant")
    APP_TENANTID = os.environ.get("MICROSOFT_APP_TENANT_ID", "")
    DATABASE_URL = os.environ.get("DATABASE_URL", "")
    DATABASE_POOL_MIN_SIZE = int(os.environ.get("DATABASE_POOL_MIN_SIZE", "1"))
    DATABASE_POOL_MAX_SIZE = int(os.environ.get("DATABASE_POOL_MAX_SIZE", "10"))

    def get_credentials(self) -> AppCredentials:
        if self.APP_PASSWORD:
            return MicrosoftAppCredentials(self.APP_ID, self.APP_PASSWORD)

        if not self.APP_CERTIFICATE or not self.APP_PRIVATEKEY:
            raise ValueError(
                "missing either MICROSOFT_APP_PASSWORD or "
                "MICROSOFT_APP_CERTIFICATE and MICROSOFT_APP_PRIVATEKEY"
            )

        certificate = base64.b64decode(self.APP_CERTIFICATE).decode("ascii")
        cert_thumbprint = hashlib.sha1(
            base64.b64decode(
                certificate.split("-----BEGIN CERTIFICATE-----\n")[1].split("-----END CERTIFICATE-----\n")[0]
            )
        ).hexdigest()
        privkey = base64.b64decode(self.APP_PRIVATEKEY).decode("ascii")

        return CertificateAppCredentials(
            app_id=self.APP_ID,
            certificate_thumbprint=cert_thumbprint,
            certificate_private_key=privkey,
        )


config = DefaultConfig()