|
1 | 1 | import os |
2 | | - |
3 | 2 | from pathlib import Path |
4 | 3 |
|
5 | | -from invoke import task, Context # type: ignore |
| 4 | +from invoke import Context, task # type: ignore |
6 | 5 |
|
7 | 6 | # If no version is indicated, we will take the latest |
8 | 7 | VERSION = os.getenv("INFRAHUB_VERSION", None) |
9 | 8 | DOCKER_PROJECT = os.getenv("INFRAHUB_BUILD_NAME", "schema-library-ci") |
10 | 9 | COMPOSE_COMMAND = f"curl https://infrahub.opsmill.io/{VERSION if VERSION else ''} | sed 's/8000:8000/0:8000/' | docker compose -p {DOCKER_PROJECT} -f -" |
11 | 10 |
|
| 11 | + |
| 12 | +def _parse_and_load_extensions( |
| 13 | + context: Context, extensions_path: Path, allowed_to_fail: bool |
| 14 | +) -> None: |
| 15 | + # Looping over all entries in extensions dir |
| 16 | + for entry in os.listdir(extensions_path): |
| 17 | + # Make sure it's a dir |
| 18 | + # TODO: here if in extensions folder we have a dir without schema it will fail |
| 19 | + if os.path.isdir(extensions_path / entry): |
| 20 | + print(f"Loading `{entry}`") |
| 21 | + |
| 22 | + # Load extensions |
| 23 | + # TODO: Maybe improve what we return here... |
| 24 | + context.run( |
| 25 | + f"infrahubctl schema load {extensions_path / entry}", |
| 26 | + warn=allowed_to_fail, |
| 27 | + ) |
| 28 | + |
| 29 | + |
12 | 30 | @task |
13 | 31 | def start(context: Context) -> None: |
14 | 32 | context.run(f"{COMPOSE_COMMAND} up -d --wait") |
15 | 33 |
|
| 34 | + |
16 | 35 | @task |
17 | | -def load_schema_base(context: Context, schema: Path=Path("./base/*.yml")) -> None: |
18 | | - context.run(f"infrahubctl schema load {schema}") |
| 36 | +def load_schema_base(context: Context) -> None: |
| 37 | + base_path = Path("./base") |
| 38 | + context.run(f"infrahubctl schema load {base_path}") |
| 39 | + |
19 | 40 |
|
20 | 41 | @task |
21 | | -def load_schema_experimental(context: Context, schema: Path=Path("./experimental/**/*.yml")) -> None: |
22 | | - Base=Path("./base/*.yml") |
23 | | - context.run(f"infrahubctl schema load {Base}") |
24 | | - context.run(f"infrahubctl schema load {schema}") |
| 42 | +def load_schema_extensions(context: Context) -> None: |
| 43 | + extensions_path = Path("./extensions") |
| 44 | + |
| 45 | + # FIXME: Find a more efficient way to deal with dependencies that doesn't |
| 46 | + # require developer to explicitly maintain it |
| 47 | + |
| 48 | + # First/second loop: here we allow to fail so it will load as much as it can |
| 49 | + print("Loading all extensions once ...") |
| 50 | + _parse_and_load_extensions(context, extensions_path, True) |
| 51 | + |
| 52 | + print("Loading all extensions second time ...") |
| 53 | + _parse_and_load_extensions(context, extensions_path, True) |
| 54 | + |
| 55 | + # Third loop: all the dependencies are loaded it MUST work |
| 56 | + print("Loading all extensions third time ...") |
| 57 | + _parse_and_load_extensions(context, extensions_path, True) |
| 58 | + |
| 59 | + # FIXME: If we have 4 degrees of dependencies it won't work |
| 60 | + print("All good!") |
| 61 | + |
25 | 62 |
|
26 | 63 | @task |
27 | 64 | def destroy(context: Context) -> None: |
28 | 65 | context.run(f"{COMPOSE_COMMAND} down -v") |
29 | 66 |
|
| 67 | + |
30 | 68 | @task |
31 | 69 | def stop(context: Context) -> None: |
32 | 70 | context.run(f"{COMPOSE_COMMAND} down") |
33 | | - |
34 | | -@task |
35 | | -def restart(context: Context, component: str="")-> None: |
36 | | - if component: |
37 | | - context.run(f"{COMPOSE_COMMAND} restart {component}") |
38 | | - return |
39 | | - |
40 | | - context.run(f"{COMPOSE_COMMAND} restart") |
|
0 commit comments