from datadog_checks.dev.tooling.utils import get_valid_integrations, load_manifest, get_manifest_file
from datadog_checks.dev.tooling.constants import set_root

import json
import os

ROOT = '/Users/nicholasmuesch/git/integrations-core/'


def get_updated_manifest(manifest_data):
    manifest_data['assets']['metrics_metadata'] = 'metadata.csv'
    return manifest_data


if __name__ == "__main__":
    set_root(ROOT)
    checks = get_valid_integrations()
    for check in checks:
        print(f'Modifying {check}')
        current_dir = f'{ROOT}{check}'
        manifest_file = get_manifest_file(check)
        manifest = load_manifest(check)

        # Update the manifest file to include this new locations
        if os.path.exists(os.path.join(ROOT, check, 'metadata.csv')):
            manifest = get_updated_manifest(manifest)

            with open(manifest_file, 'w+') as f:
                f.write(f'{json.dumps(manifest, indent=2)}\n')