feat(cli): add standalone script to inject AIP-66 Git DAG Bundles into compose

2026-10-21 18:39:05 +07:00
parent bfd0df3f28
commit eca0fd135c
1 changed files with 69 additions and 0 deletions
@@ -0,0 +1,69 @@
+#!/usr/bin/env bash
+""":"
+# Python wrapper for execution
+exec python3 "$0" "$@"
+"""
+import sys
+import os
+import shutil
+
+DEFAULT_FILE = "/home/broto/sicabe-core/deploy/generated/host-control/docker-compose.yml"
+file_path = sys.argv[1] if len(sys.argv) > 1 else DEFAULT_FILE
+
+print("==================================================================")
+print("🚀 MERICA -> Airflow 3 AIP-66 Git DAG Bundles Auto-Injector")
+print("==================================================================")
+print(f"Target compose: {file_path}")
+
+if not os.path.exists(file_path):
+    print(f"❌ Error: File tidak ditemukan di {file_path}")
+    print("Penggunaan: python3 scripts/inject_airflow_bundle.py [path_ke_docker-compose.yml]")
+    sys.exit(1)
+
+# 1. Backup file
+backup_path = file_path + ".bak"
+shutil.copyfile(file_path, backup_path)
+print(f"🛡️ [1/3] Backup file tersimpan di: {backup_path}")
+
+with open(file_path, "r", encoding="utf-8") as f:
+    lines = f.readlines()
+
+# Check if already present
+full_content = "".join(lines)
+if "AIRFLOW__DAG_PROCESSOR__DAG_BUNDLE_CONFIG_LIST" in full_content:
+    print("ℹ️ Konfigurasi Git DAG Bundles sudah ada di dalam file. Lewati penyuntikan.")
+    sys.exit(0)
+
+in_dag_processor = False
+in_env = False
+insert_idx = None
+indent = "      "
+
+for i, line in enumerate(lines):
+    if "airflow-dag-processor:" in line:
+        in_dag_processor = True
+        continue
+    if in_dag_processor:
+        # Berhenti jika berpindah ke service berikutnya (indentasi level service: 2 spasi)
+        if line.startswith("  ") and not line.startswith("    ") and not line.strip().startswith("#"):
+            break
+        if line.strip().startswith("environment:"):
+            in_env = True
+            continue
+        if in_env and "AIRFLOW__METRICS__STATSD_PREFIX:" in line:
+            insert_idx = i + 1
+            break
+
+if insert_idx is not None:
+    lines.insert(insert_idx, f'{indent}AIRFLOW__DAG_PROCESSOR__DAG_BUNDLE_STORAGE_PATH: /opt/airflow/dag_bundles\n')
+    lines.insert(insert_idx + 1, f'{indent}AIRFLOW__DAG_PROCESSOR__DAG_BUNDLE_CONFIG_LIST: \'[{{"name":"merica-pipeline","classpath":"airflow.providers.git.bundles.git.GitDagBundle","kwargs":{{"repo_url":"https://repo.sicabe.com/merica-dlh/merica-pipelines.git","tracking_ref":"main","subdir":"dags","refresh_interval":30}}}}]\'\n')
+    
+    with open(file_path, "w", encoding="utf-8") as f:
+        f.writelines(lines)
+    print("🎉 [2/3] SUKSES: Variabel Git DAG Bundles berhasil disuntikkan ke airflow-dag-processor!")
+    print("\n[3/3] Langkah Terakhir — Restart HANYA DAG Processor:")
+    print("   👉 docker restart sicabe-airflow-dag-processor")
+    print("   👉 docker logs --tail 30 -f sicabe-airflow-dag-processor\n")
+else:
+    print("❌ Gagal: Blok 'airflow-dag-processor:' dengan 'environment:' tidak ditemukan.")
+    sys.exit(1)