Merge pull request #448 from necusjz/support-aaz-hooks

Support aaz hooks

Author: kairu-ms

src/aaz_dev/app/run.py

@@ -1,4 +1,5 @@
 import os
+import subprocess
 import webbrowser
 
 import click
@@ -129,7 +130,7 @@ def is_port_in_use(host, port):
 )
 @pass_script_info
 def run_command(
-        info, host, port, reload, debugger, with_threads, extra_files, quiet
+        info, host, port, aaz_path, reload, debugger, with_threads, extra_files, quiet
 ):
     """Run a local development server.
 
@@ -139,6 +140,8 @@ def run_command(
     The reloader and debugger are enabled by default if
     FLASK_ENV=development or FLASK_DEBUG=1.
     """
+    _change_git_hooks_path(aaz_path)
+
     debug = get_debug_flag()
 
     if reload is None:
@@ -169,3 +172,11 @@ def run_command(
         threaded=with_threads,
         extra_files=extra_files,
     )
+
+
+def _change_git_hooks_path(repo_path):
+    # if .githooks folder exists in the repo folder, change the git config to use the .githooks folder in the repo
+    githooks_path = os.path.join(repo_path, '.githooks')
+    if os.path.exists(githooks_path):
+        subprocess.check_call(['git', 'config', 'core.hooksPath', githooks_path])
+        display(f"Changed git hooks path to {githooks_path}")

src/aaz_dev/command/api/_cmds.py

@@ -166,7 +166,11 @@ def generate_command_models_from_swagger(swagger_tag, workspace_path=None):
     expose_value=False,
     help="Path of `aaz` repository."
 )
-def verify():
+@click.option(
+    "--target", "-t",
+    help="Target module to focus on."
+)
+def verify(target):
     def verify_resource(model, path):
         if "commandGroups" not in model:
             return
@@ -196,9 +200,6 @@ def verify_command(file_path):
             json_path = os.path.join(Config.AAZ_PATH, os.path.splitext(path)[0][1:] + ".json")
             json_path = os.path.normpath(json_path)
 
-            if json_path in model_set:
-                continue
-
             if not os.path.exists(json_path):
                 raise Exception(f"{json_path} defined in {file_path} is missing.")
 
@@ -232,7 +233,13 @@ def verify_command(file_path):
 
     model_set = set()
     aaz = AAZSpecsManager()
-    stack = [aaz.commands_folder]
+
+    parent = aaz.commands_folder
+    stack = [os.path.join(parent, target)] if target else [
+        os.path.join(parent, i)
+        for i in os.listdir(parent)
+        if os.path.isdir(os.path.join(parent, i))
+    ]
 
     while stack:
         curr_path = stack.pop()
@@ -279,11 +286,12 @@ def verify_command(file_path):
                     for folder in folders:
                         stack.append(os.path.join(curr_path, folder))
 
-    for root, _, files in os.walk(aaz.resources_folder):
-        for file in files:
-            if not file.endswith(".json") or file.startswith("client"):  # support data-plane
-                continue
+    if not target:
+        for root, _, files in os.walk(aaz.resources_folder):
+            for file in files:
+                if not file.endswith(".json") or file.startswith("client"):  # support data-plane
+                    continue
 
-            file_path = os.path.join(root, file)
-            if file_path not in model_set:
-                raise Exception(f"{file_path} is redundant.")
+                file_path = os.path.join(root, file)
+                if file_path not in model_set:
+                    raise Exception(f"{file_path} is redundant.")