mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2026-05-02 07:29:59 -05:00
abelanger5
d376b953aa
* feat: Adds a generated REST API client and exposes workflow methods for programmatic usage
129 lines
3.9 KiB
Python
129 lines
3.9 KiB
Python
# relative imports
|
|
import os
|
|
from typing import Any
|
|
from .clients.admin import AdminClientImpl, new_admin
|
|
from .clients.events import EventClientImpl, new_event
|
|
from .clients.dispatcher import DispatcherClientImpl, new_dispatcher
|
|
from .clients.listener import ListenerClientImpl, new_listener
|
|
|
|
from .loader import ConfigLoader, ClientConfig
|
|
import grpc
|
|
|
|
from .clients.rest.api_client import ApiClient
|
|
from .clients.rest.api.workflow_api import WorkflowApi
|
|
from .clients.rest.api.workflow_run_api import WorkflowRunApi
|
|
from .clients.rest.configuration import Configuration
|
|
from .clients.rest_client import RestApi
|
|
|
|
class Client:
|
|
def admin(self):
|
|
raise NotImplementedError
|
|
|
|
def dispatcher(self):
|
|
raise NotImplementedError
|
|
|
|
def event(self):
|
|
raise NotImplementedError
|
|
|
|
def listener(self):
|
|
raise NotImplementedError
|
|
|
|
def rest(self):
|
|
raise NotImplementedError
|
|
|
|
|
|
class ClientImpl(Client):
|
|
def __init__(
|
|
self,
|
|
event_client: EventClientImpl,
|
|
admin_client: AdminClientImpl,
|
|
dispatcher_client: DispatcherClientImpl,
|
|
listener_client: ListenerClientImpl,
|
|
rest_client: RestApi
|
|
):
|
|
# self.conn = conn
|
|
# self.tenant_id = tenant_id
|
|
# self.logger = logger
|
|
# self.validator = validator
|
|
self.admin = admin_client
|
|
self.dispatcher = dispatcher_client
|
|
self.event = event_client
|
|
self.listener = listener_client
|
|
self.rest_client = rest_client
|
|
|
|
def admin(self) -> ListenerClientImpl:
|
|
return self.admin
|
|
|
|
def dispatcher(self) -> DispatcherClientImpl:
|
|
return self.dispatcher
|
|
|
|
def event(self) -> EventClientImpl:
|
|
return self.event
|
|
|
|
def listener(self) -> ListenerClientImpl:
|
|
return self.listener
|
|
|
|
def rest(self) -> RestApi:
|
|
return self.rest_client
|
|
|
|
def with_host_port(host: str, port: int):
|
|
def with_host_port_impl(config: ClientConfig):
|
|
config.host = host
|
|
config.port = port
|
|
|
|
return with_host_port_impl
|
|
|
|
|
|
def new_client(*opts_functions):
|
|
config: ClientConfig = ConfigLoader(".").load_client_config()
|
|
|
|
for opt_function in opts_functions:
|
|
opt_function(config)
|
|
|
|
if config.tls_config is None:
|
|
raise ValueError("TLS config is required")
|
|
|
|
if config.host_port is None:
|
|
raise ValueError("Host and port are required")
|
|
|
|
credentials: grpc.ChannelCredentials | None = None
|
|
|
|
# load channel credentials
|
|
if config.tls_config.tls_strategy == 'tls':
|
|
root: Any | None = None
|
|
|
|
if config.tls_config.ca_file:
|
|
root = open(config.tls_config.ca_file, "rb").read()
|
|
|
|
credentials = grpc.ssl_channel_credentials(root_certificates=root)
|
|
elif config.tls_config.tls_strategy == 'mtls':
|
|
root = open(config.tls_config.ca_file, "rb").read()
|
|
private_key = open(config.tls_config.key_file, "rb").read()
|
|
certificate_chain = open(config.tls_config.cert_file, "rb").read()
|
|
|
|
credentials = grpc.ssl_channel_credentials(
|
|
root_certificates=root, private_key=private_key, certificate_chain=certificate_chain)
|
|
|
|
conn : grpc.Channel = None
|
|
|
|
if config.tls_config.tls_strategy == 'none':
|
|
conn = grpc.insecure_channel(
|
|
target=config.host_port,
|
|
)
|
|
else:
|
|
conn = grpc.secure_channel(
|
|
target=config.host_port,
|
|
credentials=credentials,
|
|
options=[('grpc.ssl_target_name_override',
|
|
config.tls_config.server_name)],
|
|
)
|
|
|
|
# Instantiate client implementations
|
|
event_client = new_event(conn, config)
|
|
admin_client = new_admin(conn, config)
|
|
dispatcher_client = new_dispatcher(conn, config)
|
|
listener_client = new_listener(conn, config)
|
|
rest_client = RestApi(config.server_url, config.token, config.tenant_id)
|
|
|
|
return ClientImpl(event_client, admin_client, dispatcher_client, listener_client, rest_client)
|