import os
from os import path

import aws_cdk as cdk
from aws_cdk import Stack
from aws_cdk import aws_glue_alpha as glue
from constructs import Construct


class RayStack(Stack):
    def __init__(self, scope: Construct, id: str, **kwargs) -> None:
        super().__init__(scope, id, **kwargs)

        # Create a Glue job with Ray engine
        glue_job = glue.Job(
            self,
            "ray-job",
            job_name='ray-job',
            worker_count=2,
            worker_type=glue.WorkerType.Z_2_X,
            executable=glue.JobExecutable.python_ray(
                glue_version=glue.GlueVersion.V4_0,
                python_version=glue.PythonVersion.THREE_NINE,
                script=glue.Code.from_asset(path.join(path.dirname(__file__), 'hello_world.py'))
            )
        )

app = cdk.App()

env = cdk.Environment(
    account=os.getenv('CDK_DEFAULT_ACCOUNT'),
    region=os.getenv('CDK_DEFAULT_REGION')
)

RayStack(app, "rayStack", env=env)
app.synth()