import os
from collections.abc import Sequence
from typing import Any

from temporalio.client import Client
from temporalio.worker import PollerBehaviorAutoscaling, Worker

parameters = [
    ('MAX_CONCURRENT_WORKFLOW_TASKS', '200'),
    ('MAX_CONCURRENT_ACTIVITIES', '200'),
    ('MAX_CONCURRENT_LOCAL_ACTIVITIES', '200'),
    ('MAX_CACHED_WORKFLOWS', '200'),
    ('WORKFLOW_POLLER_BEHAVIUR_MINIMUM', '10'),
    ('WORKFLOW_POLLER_BEHAVIUR_INITIAL', '100'),
    ('WORKFLOW_POLLER_BEHAVIUR_MAXIMUM', '200'),
    ('ACTIVITY_POLLER_BEHAVIUR_MINIMUM', '10'),
    ('ACTIVITY_POLLER_BEHAVIUR_INITIAL', '100'),
    ('ACTIVITY_POLLER_BEHAVIUR_MAXIMUM', '200'),
]


def prepare_worker(
    main_workflow: type,
    other_workflows: Sequence[type],
    activities: Sequence[Any],
    temporal_client: Client,
) -> Worker:
    main_workflow_name = main_workflow.__name__.upper()

    local_workflow_parameters = {}

    for parameter in parameters:
        local_workflow_parameters[parameter[0]] = int(
            os.getenv(main_workflow_name + '_' + parameter[0], parameter[1])
        )

    return Worker(
        temporal_client,
        task_queue='scouter-queue',
        workflows=[main_workflow, *other_workflows],
        activities=[*activities],
        max_concurrent_workflow_tasks=local_workflow_parameters['MAX_CONCURRENT_WORKFLOW_TASKS'],
        max_concurrent_activities=local_workflow_parameters['MAX_CONCURRENT_ACTIVITIES'],
        max_concurrent_local_activities=local_workflow_parameters[
            'MAX_CONCURRENT_LOCAL_ACTIVITIES'
        ],
        max_cached_workflows=local_workflow_parameters['MAX_CACHED_WORKFLOWS'],
        workflow_task_poller_behavior=PollerBehaviorAutoscaling(
            minimum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MINIMUM'],
            initial=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_INITIAL'],
            maximum=local_workflow_parameters['WORKFLOW_POLLER_BEHAVIUR_MAXIMUM'],
        ),
        activity_task_poller_behavior=PollerBehaviorAutoscaling(
            minimum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MINIMUM'],
            initial=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_INITIAL'],
            maximum=local_workflow_parameters['ACTIVITY_POLLER_BEHAVIUR_MAXIMUM'],
        ),
    )