#!/bin/bash
set -e

OUTPUT_FILE=".gitlab-ci.generated.yml"

yaml_escape() {
  echo "$1" | sed 's/"/\\"/g'
}

cat > "$OUTPUT_FILE" << 'EOF'

workflow:
  rules:
    - if: '$CI_PIPELINE_SOURCE == "merge_request_event"'
    - if: '$CI_COMMIT_BRANCH == "master" || $CI_COMMIT_BRANCH == "main"'
    - if: '$CI_COMMIT_BRANCH =~ /^feature\/.*/'
    - if: '$CI_PIPELINE_SOURCE == "pipeline"'
    - when: always

stages:
  - validate
  - plan
  - apply

EOF

find live -name "terragrunt.hcl" -not -path "*/.terragrunt-cache/*" | sort | while read -r config_file; do
  component_dir=$(dirname "$config_file")
  relative_path=$(echo "$component_dir" | sed 's|^live/||')
  
  # live/{env}/{component}/terragrunt.hcl
  env=$(echo "$relative_path" | cut -d'/' -f1)
  component=$(echo "$relative_path" | cut -d'/' -f2)
  
  if [ -z "$component" ] || [ "$component" = "$env" ]; then
    continue
  fi
  
  job_prefix="${env}-${component}"
  
  case "$env" in
    stage)
      folder_var="YC_STAGE_FOLDER_ID"
      ;;
    prod)
      folder_var="YC_PROD_FOLDER_ID"
      ;;
    preprod)
      folder_var="YC_PREPROD_FOLDER_ID"
      ;;
    *)
      folder_var="YC_FOLDER_ID"
      ;;
  esac
  
  cat >> "$OUTPUT_FILE" << EOF
validate-${job_prefix}:
  stage: validate
  variables:
    TG_ROOT: "${component_dir}"
    ENVIRONMENT: "${env}"
    YC_FOLDER_ID: "\${${folder_var}}"
  before_script:
    - apk add --no-cache curl unzip
    - |
      if [ ! -f /usr/local/bin/terragrunt ]; then
        TERRAFORM_VERSION=1.6.0
        TERRAGRUNT_VERSION=0.50.0
        curl -fsSL https://releases.hashicorp.com/terraform/\${TERRAFORM_VERSION}/terraform_\${TERRAFORM_VERSION}_linux_amd64.zip -o terraform.zip
        unzip terraform.zip -d /usr/local/bin/
        chmod +x /usr/local/bin/terraform
        curl -fsSL https://github.com/gruntwork-io/terragrunt/releases/download/v\${TERRAGRUNT_VERSION}/terragrunt_linux_amd64 -o /usr/local/bin/terragrunt
        chmod +x /usr/local/bin/terragrunt
        rm -f terraform.zip
      fi
    - cd \${TG_ROOT}
  script:
    - terragrunt init -reconfigure -input=false --terragrunt-non-interactive
    - terragrunt validate-inputs --terragrunt-non-interactive
    - terragrunt validate --terragrunt-non-interactive
  rules:
    - if: '\$CI_PIPELINE_SOURCE == "merge_request_event"'
    - if: '\$CI_COMMIT_BRANCH == "master" || \$CI_COMMIT_BRANCH == "main"'
    - if: '\$CI_COMMIT_BRANCH =~ /^feature\/.*/'
    - if: '\$CI_PIPELINE_SOURCE == "pipeline"'
  interruptible: true

EOF

  cat >> "$OUTPUT_FILE" << EOF
plan-${job_prefix}:
  stage: plan
  variables:
    TG_ROOT: "${component_dir}"
    ENVIRONMENT: "${env}"
    YC_FOLDER_ID: "\${${folder_var}}"
  before_script:
    - apk add --no-cache curl unzip
    - |
      if [ ! -f /usr/local/bin/terragrunt ]; then
        TERRAFORM_VERSION=1.6.0
        TERRAGRUNT_VERSION=0.50.0
        curl -fsSL https://releases.hashicorp.com/terraform/\${TERRAFORM_VERSION}/terraform_\${TERRAFORM_VERSION}_linux_amd64.zip -o terraform.zip
        unzip terraform.zip -d /usr/local/bin/
        chmod +x /usr/local/bin/terraform
        curl -fsSL https://github.com/gruntwork-io/terragrunt/releases/download/v\${TERRAGRUNT_VERSION}/terragrunt_linux_amd64 -o /usr/local/bin/terragrunt
        chmod +x /usr/local/bin/terragrunt
        rm -f terraform.zip
      fi
    - cd \${TG_ROOT}
  script:
    - terragrunt init -reconfigure -input=false --terragrunt-non-interactive
    - terragrunt plan -input=false --terragrunt-non-interactive -out=tfplan
  rules:
    - if: '\$CI_COMMIT_BRANCH == "master" || \$CI_COMMIT_BRANCH == "main"'
    - if: '\$CI_PIPELINE_SOURCE == "merge_request_event"'
    - if: '\$CI_COMMIT_BRANCH =~ /^feature\/.*/'
    - if: '\$CI_PIPELINE_SOURCE == "pipeline"'
  interruptible: true
  artifacts:
    paths:
      - "\${TG_ROOT}/tfplan"
      - "\${TG_ROOT}/.terragrunt-cache/**/*"
    expire_in: 1 week
    when: always

EOF

  if [ "$env" = "prod" ]; then
    when_clause="manual"
  else
    when_clause="on_success"
  fi
  
  cat >> "$OUTPUT_FILE" << EOF
apply-${job_prefix}:
  stage: apply
  variables:
    TG_ROOT: "${component_dir}"
    ENVIRONMENT: "${env}"
    YC_FOLDER_ID: "\${${folder_var}}"
  before_script:
    - apk add --no-cache curl unzip
    - |
      if [ ! -f /usr/local/bin/terragrunt ]; then
        TERRAFORM_VERSION=1.6.0
        TERRAGRUNT_VERSION=0.50.0
        curl -fsSL https://releases.hashicorp.com/terraform/\${TERRAFORM_VERSION}/terraform_\${TERRAFORM_VERSION}_linux_amd64.zip -o terraform.zip
        unzip terraform.zip -d /usr/local/bin/
        chmod +x /usr/local/bin/terraform
        curl -fsSL https://github.com/gruntwork-io/terragrunt/releases/download/v\${TERRAGRUNT_VERSION}/terragrunt_linux_amd64 -o /usr/local/bin/terragrunt
        chmod +x /usr/local/bin/terragrunt
        rm -f terraform.zip
      fi
    - cd \${TG_ROOT}
  script:
    - terragrunt init -reconfigure -input=false --terragrunt-non-interactive
    - terragrunt apply -input=false --terragrunt-non-interactive tfplan
  rules:
    - if: '\$CI_COMMIT_BRANCH == "master" || \$CI_COMMIT_BRANCH == "main"'
      when: ${when_clause}
    - if: '\$CI_COMMIT_BRANCH =~ /^feature\/.*/'
      when: ${when_clause}
    - if: '\$CI_PIPELINE_SOURCE == "pipeline"'
      when: ${when_clause}
  interruptible: true
  dependencies:
    - plan-${job_prefix}

EOF

done

echo "Pipeline generated successfully: $OUTPUT_FILE"
echo "Found components:"
find live -name "terragrunt.hcl" -not -path "*/.terragrunt-cache/*" | sed 's|live/||; s|/terragrunt.hcl||' | sort