import os
from datahub.api.entities.dataproduct.dataproduct import DataProduct
from datahub.ingestion.source.metadata.business_glossary import BusinessGlossaryConfig
import click

@click.command()
@click.option("--out-dir", type=str, required=True)
def generate_specs(
    out_dir: str
) -> None:
    print(out_dir)
    schemas_dir = f"{out_dir}/schemas"
    os.makedirs(schemas_dir, exist_ok=True)
    concept_class_map = {
        "dataproduct": DataProduct,
        "businessglossary": BusinessGlossaryConfig
    }
    for concept, concept_class in concept_class_map.items():
        with open(f"{schemas_dir}/{concept}_schema.json", "w") as f:
            f.write(concept_class.schema_json(indent=2))


if __name__ == "__main__":
    generate_specs()