dvirlabs/apps-gitops
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add auth options

Browse Source
This commit is contained in:
dvirlabs 2025-06-26 05:43:47 +03:00
parent 3e6582b326
commit 71d489e2b1
1 changed files with 29 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
automation/generate-scrape-config.py
View File

@ -17,11 +17,10 @@ OUTPUT_FILE = os.path.join(
) )
os.makedirs(TMP_DIR, exist_ok=True) os.makedirs(TMP_DIR, exist_ok=True)
yaml = YAML()
yaml.default_flow_style = False
def collect_targets():
jobs = {} def collect_jobs():
jobs = []
for name, url in REPOS.items(): for name, url in REPOS.items():
repo_path = os.path.join(TMP_DIR, name) repo_path = os.path.join(TMP_DIR, name)
@ -30,33 +29,39 @@ def collect_targets():
for path in Path(repo_path, "manifests").glob("*/monitoring.yaml"): for path in Path(repo_path, "manifests").glob("*/monitoring.yaml"):
with open(path) as f: with open(path) as f:
data = yaml.load(f) data = YAML().load(f)
if not data.get("enabled") or "targets" not in data:
continue
if data.get("enabled"):
job = { job = {
"job_name": path.parent.name, # app name "job_name": path.parent.name, # Use app folder name
"static_configs": [{"targets": data.get("targets", [])}] "static_configs": [{"targets": data["targets"]}]
} }
# Optional auth fields
if "basic_auth" in data:
job["basic_auth"] = data["basic_auth"]
if "bearer_token" in data:
job["bearer_token"] = data["bearer_token"]
if "bearer_token_file" in data:
job["bearer_token_file"] = data["bearer_token_file"]
if "metrics_path" in data: if "metrics_path" in data:
job["metrics_path"] = data["metrics_path"] job["metrics_path"] = data["metrics_path"]
if "scheme" in data: if "scheme" in data:
job["scheme"] = data["scheme"] job["scheme"] = data["scheme"]
if "basic_auth" in data:
job["basic_auth"] = data["basic_auth"]
jobs.setdefault(name, []).append(job) jobs.append(job)
return jobs return jobs
def write_scrape_config(jobs, output_file): def write_scrape_config(jobs, output_file):
job_list = []
for repo_jobs in jobs.values():
job_list.extend(repo_jobs)
stream = StringIO() stream = StringIO()
yaml.dump(job_list, stream) yaml_writer = YAML()
yaml_writer.default_flow_style = False
yaml_writer.dump(jobs, stream)
scrape_yaml = "# This content will be auto-updated by the pipeline\n" + stream.getvalue() scrape_yaml = "# This content will be auto-updated by the pipeline\n" + stream.getvalue()
secret = { secret = {
@ -77,10 +82,10 @@ def write_scrape_config(jobs, output_file):
os.makedirs(os.path.dirname(output_file), exist_ok=True) os.makedirs(os.path.dirname(output_file), exist_ok=True)
with open(output_file, "w") as f: with open(output_file, "w") as f:
yaml.dump(secret, f) yaml_writer.dump(secret, f)
if __name__ == "__main__": if __name__ == "__main__":
jobs = collect_targets() jobs = collect_jobs()
write_scrape_config(jobs, OUTPUT_FILE) write_scrape_config(jobs, OUTPUT_FILE)
print(f"✅ Generated: {OUTPUT_FILE}") print(f"✅ Generated: {OUTPUT_FILE}")