#!/usr/bin/env python3

from __future__ import annotations

import importlib.metadata

import typing as t

from check_jsonschema.catalog import SCHEMA_CATALOG

version = importlib.metadata.version("check_jsonschema")

def iter_catalog_hooks():

for name in SCHEMA_CATALOG:

# copy config (new dict)

config = dict(SCHEMA_CATALOG[name]["hook_config"])

# set computed attributes

config["schema_name"] = name

config["id"] = f"check-{name}"

config["description"] = (

config.get("description")

or f"{config['name']} against the schema provided by SchemaStore"

)

if "types" in config and isinstance(config["types"], str):

config["types"] = [config["types"]]

yield config

def update_hook_config(new_config: str) -> None:

print("updating .pre-commit-hooks.yaml")

with open(".pre-commit-hooks.yaml") as fp:

content = fp.read()

autogen_marker = "# --AUTOGEN_HOOKS_START-- #"

# get the first part of the content (and assert that it's OK)

content = content.split(autogen_marker)[0]

assert "check-jsonschema" in content

content = content + autogen_marker + "\n\n" + new_config

with open(".pre-commit-hooks.yaml", "w") as fp:

fp.write(content)

def generate_hook_lines(config) -> t.Iterator[str]:

yield "# this hook is autogenerated from a script"

yield "# to modify this hook, update `src/check_jsonschema/catalog.py`"

yield "# and run `just generate-hooks` or `tox run -e generate-hooks-config`"

yield f"- id: {config['id']}"

yield f" name: {config['name']}"

yield f" description: '{config['description']}'"

add_args = " ".join(config.get("add_args", []))

if add_args:

add_args = " " + add_args

yield (

" entry: check-jsonschema --builtin-schema "

f"vendor.{config['schema_name']}{add_args}"

)

yield " language: python"

if isinstance(config["files"], list):

config["files"] = r""">

(?x)^(

{}

)$""".format("|\n ".join(config["files"]))

yield f" files: {config['files']}"

if "types" in config:

yield f" types: [{','.join(config['types'])}]"

if "types_or" in config:

yield f" types_or: [{','.join(config['types_or'])}]"

yield ""

def generate_all_hook_config_lines() -> t.Iterator[str]:

for hook_config in iter_catalog_hooks():

yield from generate_hook_lines(hook_config)

def format_all_hook_config() -> str:

return "\n".join(generate_all_hook_config_lines())

def update_usage_list_schemas() -> None:

print("updating docs/usage.rst -- list schemas")

with open("docs/usage.rst") as fp:

content = fp.read()

vendored_list_start = ".. vendored-schema-list-start\n"

vendored_list_end = "\n.. vendored-schema-list-end"

content_head = content.split(vendored_list_start)[0]

content_tail = content.split(vendored_list_end)[-1]

generated_list = "\n".join(

[vendored_list_start]

+ [f"- ``vendor.{n}``" for n in SCHEMA_CATALOG]

+ [vendored_list_end]

)

content = content_head + generated_list + content_tail

with open("docs/usage.rst", "w") as fp:

fp.write(content)

def update_precommit_usage_supported_hooks() -> None:

print("updating docs/precommit_usage.rst -- generated hooks")

with open("docs/precommit_usage.rst") as fp:

content = fp.read()

generated_list_start = ".. generated-hook-list-start\n"

generated_list_end = ".. generated-hook-list-end"

content_head = content.split(generated_list_start)[0]

content_tail = content.split(generated_list_end)[-1]

generated_list = "\n\n".join([generated_list_start] + [f"""\

``{config["id"]}``

{"~" * (len(config["id"]) + 4)}

{config["description"]}

.. code-block:: yaml

:caption: example config

- repo: https://github.com/python-jsonschema/check-jsonschema

rev: {version}

hooks:

- id: {config["id"]}

""" for config in iter_catalog_hooks()] + [generated_list_end])

content = content_head + generated_list + content_tail

with open("docs/precommit_usage.rst", "w") as fp:

fp.write(content)

def update_docs() -> None:

update_usage_list_schemas()

update_precommit_usage_supported_hooks()

def main() -> None:

update_hook_config(format_all_hook_config())

update_docs()

if __name__ == "__main__":

main()