#!/bin/bash

echo "Final dependency cleanup..."

# Remove all problematic dependencies
PROBLEM_DEPS=(
    "pulumi"
    "terraform-sdk"
    "kubernetes-sdk"
    "helm-sdk"
    "ansible-sdk"
    "cloudformation-sdk"
    "cdk-sdk"
    "dataflow-sdk"
    "beam-sdk"
    "airflow-sdk"
    "mlflow-sdk"
    "kubeflow-sdk"
    "metaflow-sdk"
    "dagster-sdk"
    "prefect-sdk"
    "kedro-sdk"
    "flyte-sdk"
    "azkaban-sdk"
    "oozie-sdk"
    "vertex-ai"
    "sagemaker-sdk"
    "databricks-sdk"
    "snowflake-sdk"
    "redshift-sdk"
    "bigquery-sdk"
    "athena-sdk"
    "datadog-sdk"
    "newrelic-sdk"
    "sumologic-sdk"
    "elastic-apm"
    "prometheus-pushgateway"
    "grafana-sdk"
    "pagerduty-sdk"
    "opsgenie-sdk"
    "statuspage-sdk"
    "gpu-cluster-manager"
    "ray-sdk"
    "horovod-sdk"
    "deepspeed-sdk"
    "fairscale-sdk"
    "megatron-sdk"
    "tensorflow-serving"
    "triton-inference-server"
    "onnx-runtime-server"
    "seldon-core"
    "kserve-sdk"
    "bentoml-sdk"
    "cortex-sdk"
    "algorithmia-sdk"
    "clearml-sdk"
    "comet-ml"
    "wandb-sdk"
    "neptune-sdk"
    "sacred-sdk"
    "optuna-sdk"
    "hyperopt-sdk"
    "custom-gpu-scheduler"
    "custom-memory-allocator"
)

for dep in "${PROBLEM_DEPS[@]}"; do
    echo "Removing: $dep"
    find crates -name "Cargo.toml" -exec sed -i "/$dep/d" {} \; 2>/dev/null
done

# Also fix features that might reference these
find crates -name "Cargo.toml" -exec sed -i \
    -e 's/, "dep:[^"]*-sdk"//g' \
    -e 's/"dep:[^"]*-sdk", //g' \
    {} \;

echo "Cleanup complete!"