1 files changed, 84 insertions, 0 deletions
diff --git a/testing/web-platform/tests/tools/manifest/spec.py b/testing/web-platform/tests/tools/manifest/spec.py
new file mode 100644
index 0000000000..5148fceb3e
--- /dev/null
+++ b/testing/web-platform/tests/tools/manifest/spec.py
@@ -0,0 +1,84 @@
+#!/usr/bin/env python3
+import argparse
+import os
+from typing import Any, Optional, Text
+
+from . import vcs
+from .manifest import compute_manifest_spec_items, InvalidCacheError, Manifest, write
+from .log import get_logger, enable_debug_logging
+
+
+here = os.path.dirname(__file__)
+
+wpt_root = os.path.abspath(os.path.join(here, os.pardir, os.pardir))
+
+logger = get_logger()
+
+
+def update_spec(tests_root: Text,
+                manifest_path: Text,
+                url_base: Text,
+                cache_root: Optional[Text] = None,
+                working_copy: bool = True,
+                parallel: bool = True
+                ) -> None:
+
+    manifest = Manifest(tests_root, url_base)
+
+    logger.info("Updating SPEC_MANIFEST")
+    try:
+        tree = vcs.get_tree(tests_root, manifest, manifest_path, cache_root,
+                            working_copy, True)
+        changed = manifest.update(tree, parallel, compute_manifest_spec_items)
+    except InvalidCacheError:
+        logger.error("Manifest cache in spec.py was invalid.")
+        return
+
+    if changed:
+        write(manifest, manifest_path)
+    tree.dump_caches()
+
+
+def update_from_cli(**kwargs: Any) -> None:
+    tests_root = kwargs["tests_root"]
+    path = kwargs["path"]
+    assert tests_root is not None
+
+    update_spec(tests_root,
+                path,
+                kwargs["url_base"],
+                cache_root=kwargs["cache_root"],
+                parallel=kwargs["parallel"])
+
+
+def abs_path(path: str) -> str:
+    return os.path.abspath(os.path.expanduser(path))
+
+
+def create_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-v", "--verbose", dest="verbose", action="store_true", default=False,
+        help="Turn on verbose logging")
+    parser.add_argument(
+        "-p", "--path", type=abs_path, help="Path to manifest file.")
+    parser.add_argument(
+        "--tests-root", type=abs_path, default=wpt_root, help="Path to root of tests.")
+    parser.add_argument(
+        "--url-base", action="store", default="/",
+        help="Base url to use as the mount point for tests in this manifest.")
+    parser.add_argument(
+        "--cache-root", action="store", default=os.path.join(wpt_root, ".wptcache"),
+        help="Path in which to store any caches (default <tests_root>/.wptcache/)")
+    parser.add_argument(
+        "--no-parallel", dest="parallel", action="store_false", default=True,
+        help="Do not parallelize building the manifest")
+    return parser
+
+
+def run(*args: Any, **kwargs: Any) -> None:
+    if kwargs["path"] is None:
+        kwargs["path"] = os.path.join(kwargs["tests_root"], "SPEC_MANIFEST.json")
+    if kwargs["verbose"]:
+        enable_debug_logging()
+    update_from_cli(**kwargs)