load("ext://helm_remote", "helm_remote")
load("ext://secret", "secret_from_dict")
load("ext://configmap", "configmap_from_dict")

helm_remote("postgresql",
            repo_name="bitnami",
            repo_url="https://charts.bitnami.com/bitnami",
            set=["auth.postgresPassword=password", "auth.database=mev_inspect"],
)

helm_remote("redis",
            repo_name="bitnami",
            repo_url="https://charts.bitnami.com/bitnami",
            set=["global.redis.password=password"],
)

k8s_yaml(configmap_from_dict("mev-inspect-rpc", inputs = {
    "url" : os.environ["RPC_URL"],
}))

k8s_yaml(configmap_from_dict("mev-inspect-listener-healthcheck", inputs = {
    "url" : os.getenv("LISTENER_HEALTHCHECK_URL", default=""),
}))

k8s_yaml(secret_from_dict("mev-inspect-db-credentials", inputs = {
    "username" : "postgres",
    "password": "password",
    "host": "postgresql",
}))

# if using https://github.com/taarushv/trace-db
# k8s_yaml(secret_from_dict("trace-db-credentials", inputs = {
#     "username" : "username",
#     "password": "password",
#     "host": "trace-db-postgresql",
# }))

docker_build("mev-inspect-py", ".",
    live_update=[
        sync(".", "/app"),
        run("cd /app && poetry install",
            trigger="./pyproject.toml"),
    ],
)
k8s_yaml(helm(
    './k8s/mev-inspect',
    name='mev-inspect',
    set=[
        "extraEnv[0].name=AWS_ACCESS_KEY_ID",
        "extraEnv[0].value=foobar",
        "extraEnv[1].name=AWS_SECRET_ACCESS_KEY",
        "extraEnv[1].value=foobar",
        "extraEnv[2].name=AWS_REGION",
        "extraEnv[2].value=us-east-1",
        "extraEnv[3].name=AWS_ENDPOINT_URL",
        "extraEnv[3].value=http://localstack:4566",
    ],
))
k8s_resource(
    workload="mev-inspect",
    resource_deps=["postgresql", "redis-master"],
)

k8s_resource(
    workload="mev-inspect-workers",
    resource_deps=["postgresql", "redis-master"],
)

# uncomment to enable price monitor
# k8s_yaml(helm('./k8s/mev-inspect-prices', name='mev-inspect-prices'))
# k8s_resource(workload="mev-inspect-prices", resource_deps=["postgresql"])

local_resource(
    'pg-port-forward',
    serve_cmd='kubectl port-forward --namespace default svc/postgresql 5432:5432',
    resource_deps=["postgresql"]
)

# Uncomment to use local S3 exports
#
#k8s_yaml(configmap_from_dict("mev-inspect-export", inputs = {
#    "export-bucket-name" : "local-export",
#}))
#
#helm_remote("localstack",
#            repo_name="localstack-charts",
#            repo_url="https://localstack.github.io/helm-charts",
#)
#
#local_resource(
#    'localstack-port-forward',
#    serve_cmd='kubectl port-forward --namespace default svc/localstack 4566:4566',
#    resource_deps=["localstack"]
#)
#
#k8s_yaml(helm(
#    './k8s/mev-inspect-workers',
#    name='mev-inspect-workers',
#    set=[
#        "extraEnv[0].name=AWS_ACCESS_KEY_ID",
#        "extraEnv[0].value=foobar",
#        "extraEnv[1].name=AWS_SECRET_ACCESS_KEY",
#        "extraEnv[1].value=foobar",
#        "extraEnv[2].name=AWS_REGION",
#        "extraEnv[2].value=us-east-1",
#        "extraEnv[3].name=AWS_ENDPOINT_URL",
#        "extraEnv[3].value=http://localstack:4566",
#        "replicaCount=1",
#    ],
#))
#k8s_yaml(helm(
#    './k8s/mev-inspect',
#    name='mev-inspect',
#    set=[
#        "extraEnv[0].name=AWS_ACCESS_KEY_ID",
#        "extraEnv[0].value=foobar",
#        "extraEnv[1].name=AWS_SECRET_ACCESS_KEY",
#        "extraEnv[1].value=foobar",
#        "extraEnv[2].name=AWS_REGION",
#        "extraEnv[2].value=us-east-1",
#        "extraEnv[3].name=AWS_ENDPOINT_URL",
#        "extraEnv[3].value=http://localstack:4566",
#    ],
#))