import os
import argparse
import json
def load_json_file(filepath):
with open(filepath, 'r') as f:
return json.load(f)
def find_json_by_id(ref_value, folder):
for root, _, files in os.walk(folder):
for fname in files:
if fname.endswith(".json"):
path = os.path.join(root, fname)
try:
with open(path, "r") as f:
content = json.load(f)
if isinstance(content, dict) and content.get("listId") == ref_value:
return content
except Exception:
continue
return {}
def find_app_template_by_id(appref, folder):
for root, _, files in os.walk(folder):
for fname in files:
if fname.endswith(".json"):
path = os.path.join(root, fname)
try:
with open(path, "r") as f:
content = json.load(f)
if content.get("appId") == appref or content.get("id") == appref:
return content.get("serverNames", [])
except Exception:
continue
return []
def process_policy(input_json, base_dir):
output = []
lines = json.dumps(input_json, indent=2).splitlines()
i = 0
while i < len(lines):
line = lines[i]
output.append(line)
if '"ref"' in line:
ref_value = line.split(":")[1].strip().strip('",')
prev_line = lines[i - 1] if i > 0 else ""
if "DataPrefixList" in prev_line:
data = find_json_by_id(ref_value, os.path.join(base_dir, "policy_lists", "DataPrefix"))
name = data.get("name")
if name:
output.append(' ' * 10 + f'"name": "{name}",')
for entry in data.get("entries", []):
if "ipPrefix" in entry:
output.append(' ' * 10 + f'"ipPrefix": "{entry["ipPrefix"]}",')
elif "appList" in prev_line:
data = find_json_by_id(ref_value, os.path.join(base_dir, "policy_lists", "App"))
name = data.get("name")
if name:
output.append(' ' * 10 + f'"name": "{name}",')
for entry in data.get("entries", []):
if "app" in entry:
output.append(' ' * 10 + f'"app": "{entry["app"]}",')
if "appRef" in entry:
#output.append(' ' * 10 + f'"appRef": "{entry["appRef"]}",')
servers = find_app_template_by_id(entry["appRef"], os.path.join(base_dir, "policy_templates", "CustomApp"))
for server in servers:
output.append(' ' * 10 + f'"serverName": "{server}",')
i += 1
return output
def main():
parser = argparse.ArgumentParser(description="Expand Cisco SD-WAN policy file with inlined entries.")
parser.add_argument("base_dir", help="Base directory (extracted from policy.tar)")
parser.add_argument("input_file", help="Filename of the input policy (e.g. CHINA-DATA-POLICY-S1.json)")
args = parser.parse_args()
input_path = os.path.join(args.base_dir, "policy_definitions", "Data", args.input_file)
input_data = load_json_file(input_path)
output_lines = process_policy(input_data, args.base_dir)
output_path = os.path.splitext(args.input_file)[0] + "_expanded.json"
with open(output_path, "w") as f:
f.write("\n".join(output_lines))
print(f"✅ Expanded file written to: {output_path}")
if __name__ == "__main__":
main()
