mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-04-24 19:29:16 -05:00
abelanger5
d376b953aa
* feat: Adds a generated REST API client and exposes workflow methods for programmatic usage
81 lines
3.3 KiB
Python
81 lines
3.3 KiB
Python
import os
|
|
import yaml
|
|
from typing import Any, Optional, Dict
|
|
from .token import get_addresses_from_jwt, get_tenant_id_from_jwt
|
|
|
|
class ClientTLSConfig:
|
|
def __init__(self, tls_strategy: str, cert_file: str, key_file: str, ca_file: str, server_name: str):
|
|
self.tls_strategy = tls_strategy
|
|
self.cert_file = cert_file
|
|
self.key_file = key_file
|
|
self.ca_file = ca_file
|
|
self.server_name = server_name
|
|
|
|
class ClientConfig:
|
|
def __init__(self, tenant_id: str, tls_config: ClientTLSConfig, token: str, host_port: str="localhost:7070", server_url: str="https://app.dev.hatchet-tools.com"):
|
|
self.tenant_id = tenant_id
|
|
self.tls_config = tls_config
|
|
self.host_port = host_port
|
|
self.token = token
|
|
self.server_url = server_url
|
|
|
|
class ConfigLoader:
|
|
def __init__(self, directory: str):
|
|
self.directory = directory
|
|
|
|
def load_client_config(self) -> ClientConfig:
|
|
config_file_path = os.path.join(self.directory, "client.yaml")
|
|
|
|
config_data : Any = {
|
|
"tls": {},
|
|
}
|
|
|
|
# determine if client.yaml exists
|
|
if os.path.exists(config_file_path):
|
|
with open(config_file_path, 'r') as file:
|
|
config_data = yaml.safe_load(file)
|
|
|
|
tenant_id = config_data['tenantId'] if 'tenantId' in config_data else self._get_env_var('HATCHET_CLIENT_TENANT_ID')
|
|
token = config_data['token'] if 'token' in config_data else self._get_env_var('HATCHET_CLIENT_TOKEN')
|
|
|
|
if not token:
|
|
raise ValueError('Token must be set via HATCHET_CLIENT_TOKEN environment variable')
|
|
|
|
host_port = config_data['hostPort'] if 'hostPort' in config_data else self._get_env_var('HATCHET_CLIENT_HOST_PORT')
|
|
server_url : str | None = None
|
|
|
|
if not host_port:
|
|
# extract host and port from token
|
|
server_url, grpc_broadcast_address = get_addresses_from_jwt(token)
|
|
|
|
host_port = grpc_broadcast_address
|
|
|
|
if not tenant_id:
|
|
tenant_id = get_tenant_id_from_jwt(token)
|
|
|
|
tls_config = self._load_tls_config(config_data['tls'], host_port)
|
|
|
|
return ClientConfig(tenant_id, tls_config, token, host_port, server_url)
|
|
|
|
def _load_tls_config(self, tls_data: Dict, host_port) -> ClientTLSConfig:
|
|
tls_strategy = tls_data['tlsStrategy'] if 'tlsStrategy' in tls_data else self._get_env_var('HATCHET_CLIENT_TLS_STRATEGY')
|
|
|
|
if not tls_strategy:
|
|
tls_strategy = 'tls'
|
|
|
|
cert_file = tls_data['tlsCertFile'] if 'tlsCertFile' in tls_data else self._get_env_var('HATCHET_CLIENT_TLS_CERT_FILE')
|
|
key_file = tls_data['tlsKeyFile'] if 'tlsKeyFile' in tls_data else self._get_env_var('HATCHET_CLIENT_TLS_KEY_FILE')
|
|
ca_file = tls_data['tlsRootCAFile'] if 'tlsRootCAFile' in tls_data else self._get_env_var('HATCHET_CLIENT_TLS_ROOT_CA_FILE')
|
|
|
|
server_name = tls_data['tlsServerName'] if 'tlsServerName' in tls_data else self._get_env_var('HATCHET_CLIENT_TLS_SERVER_NAME')
|
|
|
|
# if server_name is not set, use the host from the host_port
|
|
if not server_name:
|
|
server_name = host_port.split(':')[0]
|
|
|
|
return ClientTLSConfig(tls_strategy, cert_file, key_file, ca_file, server_name)
|
|
|
|
@staticmethod
|
|
def _get_env_var(env_var: str, default: Optional[str] = None) -> str:
|
|
return os.environ.get(env_var, default)
|