++ publish contour slices

.gitlab-ci.yml

@@ -4,6 +4,7 @@ include:
 
 stages:
   - build
+  - slices
 
 variables:
   RELEASE_NAME: iac-offline
@@ -33,3 +34,18 @@ build_offline_mirror:
     - docker push ${IMAGE_NAME}
     - docker push ${CR}/${RELEASE_NAME}:latest
   timeout: 30m
+
+publish_contour_slices:
+  stage: slices
+  image: ${CR}/infra/slice-builder:1.0.0
+  tags:
+    - build
+  rules:
+    - if: $CI_COMMIT_TAG =~ /^v?(?:\d+\.){2}\d+$/
+  variables:
+    GIT_DEPTH: "1"
+  script:
+    - printf '%s' "$SLICES_PUSHER_KEY" | base64 -d > /tmp/slices-key.json
+    - python3 ci/publish-slices.py --repo . --release "$CI_COMMIT_TAG" --commit "$CI_COMMIT_SHA" --key-file /tmp/slices-key.json
+  after_script:
+    - rm -f /tmp/slices-key.json

ci/publish-slices.py

@@ -0,0 +1,178 @@
+#!/usr/bin/env python3
+import argparse
+import base64
+import hashlib
+import importlib.util
+import json
+import os
+import re
+import sys
+import tempfile
+import urllib.error
+import urllib.parse
+import urllib.request
+from datetime import datetime, timezone
+from pathlib import Path
+
+import yaml
+
+spec = importlib.util.spec_from_file_location("build_slice", Path(__file__).with_name("build-slice.py"))
+bs = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(bs)
+
+REGISTRY_HOST = "cr.yandex"
+SLICE_REGISTRY = "crpct2b2lsmao4nm0cjk"
+INSTALLER_GITEA_URL = "http://gitea.gitea.svc:3000/infra/iac.git"
+CONFIG_TYPE = "application/vnd.sarex.slice.config.v1+json"
+LAYER_TYPE = "application/vnd.oci.image.layer.v1.tar"
+MANIFEST_TYPE = "application/vnd.oci.image.manifest.v1+json"
+
+
+class PublishError(Exception):
+    pass
+
+
+def digest(data: bytes) -> str:
+    return "sha256:" + hashlib.sha256(data).hexdigest()
+
+
+def connected_contours(repo: Path) -> list[str]:
+    found = []
+    for sync in sorted(repo.glob("clusters/*/flux-system/gotk-sync.yaml")):
+        for doc in yaml.safe_load_all(sync.read_text()):
+            if isinstance(doc, dict) and doc.get("kind") == "GitRepository" and doc.get("spec", {}).get("url") == INSTALLER_GITEA_URL:
+                found.append(sync.parents[1].name)
+    return found
+
+
+def build_manifest(layer: bytes, annotations: dict[str, str]) -> tuple[bytes, bytes]:
+    config = b"{}"
+    manifest = {
+        "schemaVersion": 2,
+        "mediaType": MANIFEST_TYPE,
+        "config": {"mediaType": CONFIG_TYPE, "digest": digest(config), "size": len(config)},
+        "layers": [{
+            "mediaType": LAYER_TYPE,
+            "digest": digest(layer),
+            "size": len(layer),
+            "annotations": {"org.opencontainers.image.title": "slice.tar"},
+        }],
+        "annotations": dict(sorted(annotations.items())),
+    }
+    return config, json.dumps(manifest, sort_keys=True, separators=(",", ":")).encode()
+
+
+class Registry:
+    def __init__(self, username: str, password: str):
+        self.base = f"https://{REGISTRY_HOST}"
+        self.basic = "Basic " + base64.b64encode(f"{username}:{password}".encode()).decode()
+        self.tokens: dict[str, str] = {}
+
+    def _send(self, method, url, data=None, headers=None):
+        req = urllib.request.Request(url, data=data, method=method, headers=headers or {})
+        try:
+            with urllib.request.urlopen(req, timeout=60) as r:
+                return r.status, dict(r.headers), r.read()
+        except urllib.error.HTTPError as e:
+            return e.code, dict(e.headers), e.read()
+
+    def _token(self, challenge: str, scope: str) -> str:
+        params = dict(re.findall(r'(\w+)="([^"]*)"', challenge))
+        realm = params.pop("realm")
+        params["scope"] = scope
+        status, _, body = self._send("GET", f"{realm}?{urllib.parse.urlencode(params)}", headers={"Authorization": self.basic})
+        if status != 200:
+            raise PublishError(f"авторизация в реестре: HTTP {status}")
+        payload = json.loads(body)
+        return payload.get("token") or payload.get("access_token")
+
+    def request(self, method, repo, path, data=None, headers=None):
+        scope = f"repository:{repo}:pull,push"
+        url = path if path.startswith("http") else f"{self.base}/v2/{repo}/{path}"
+        headers = dict(headers or {})
+        if scope in self.tokens:
+            headers["Authorization"] = f"Bearer {self.tokens[scope]}"
+        status, h, body = self._send(method, url, data, headers)
+        if status == 401:
+            challenge = {k.lower(): v for k, v in h.items()}.get("www-authenticate", "")
+            self.tokens[scope] = self._token(challenge, scope)
+            headers["Authorization"] = f"Bearer {self.tokens[scope]}"
+            status, h, body = self._send(method, url, data, headers)
+        return status, {k.lower(): v for k, v in h.items()}, body
+
+    def current_layer(self, repo: str) -> str | None:
+        status, _, body = self.request("GET", repo, "manifests/current", headers={"Accept": MANIFEST_TYPE})
+        if status == 404:
+            return None
+        if status != 200:
+            raise PublishError(f"{repo}:current: HTTP {status}")
+        return json.loads(body)["layers"][0]["digest"]
+
+    def put_blob(self, repo: str, data: bytes):
+        d = digest(data)
+        if self.request("HEAD", repo, f"blobs/{d}")[0] == 200:
+            return
+        status, h, _ = self.request("POST", repo, "blobs/uploads/")
+        if status != 202:
+            raise PublishError(f"загрузка блоба в {repo}: HTTP {status}")
+        location = h["location"]
+        if location.startswith("/"):
+            location = self.base + location
+        sep = "&" if "?" in location else "?"
+        status, _, _ = self.request("PUT", repo, f"{location}{sep}digest={d}", data, {"Content-Type": "application/octet-stream"})
+        if status != 201:
+            raise PublishError(f"загрузка блоба в {repo}: HTTP {status}")
+
+    def push(self, repo: str, tags: list[str], layer: bytes, annotations: dict[str, str]):
+        config, manifest = build_manifest(layer, annotations)
+        self.put_blob(repo, config)
+        self.put_blob(repo, layer)
+        for tag in tags:
+            status, _, body = self.request("PUT", repo, f"manifests/{tag}", manifest, {"Content-Type": MANIFEST_TYPE})
+            if status != 201:
+                raise PublishError(f"{repo}:{tag}: HTTP {status} {body[:200]!r}")
+
+
+def main() -> int:
+    ap = argparse.ArgumentParser()
+    ap.add_argument("--repo", default=".")
+    ap.add_argument("--release", required=True)
+    ap.add_argument("--commit", required=True)
+    ap.add_argument("--branch", default="")
+    ap.add_argument("--key-file", required=True)
+    args = ap.parse_args()
+    repo = Path(args.repo).resolve()
+    registry = Registry("json_key", Path(args.key_file).read_text())
+    contours = connected_contours(repo)
+    print(f"подключённые контуры: {', '.join(contours) or 'нет'}")
+    failed = []
+    for contour in contours:
+        target = f"{SLICE_REGISTRY}/{contour}-iac"
+        try:
+            with tempfile.TemporaryDirectory() as tmp:
+                out = Path(tmp) / "slice.tar"
+                files, entries = bs.collect(repo, f"clusters/{contour}")
+                bs.write_tar(repo, files, out)
+                bs.verify(repo, out, entries)
+                layer = out.read_bytes()
+            changed = registry.current_layer(target) != digest(layer)
+            tags = [args.commit[:8], args.release] + (["current"] if changed else [])
+            registry.push(target, tags, layer, {
+                "org.opencontainers.image.source": "iac",
+                "org.opencontainers.image.revision": args.commit,
+                "org.opencontainers.image.version": args.release,
+                "org.opencontainers.image.created": datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ"),
+                "io.sarex.slice.branch": args.branch,
+                "io.sarex.slice.contour": contour,
+                "io.sarex.slice.kind": "iac",
+            })
+            state = "current переставлен" if changed else "содержимое как в current"
+            print(f"{target}: {len(files)} файлов, {digest(layer)[:19]}, теги {', '.join(tags)}, {state}")
+        except (bs.SliceError, PublishError) as e:
+            print(f"{target}: ОШИБКА {e}", file=sys.stderr)
+            failed.append(contour)
+    return 1 if failed else 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

ci/tests/test_build_slice.py

@@ -68,3 +68,18 @@ def test_tar_is_byte_identical_across_runs(repo, tmp_path_factory):
     assert out1.read_bytes() == out2.read_bytes()
     with tarfile.open(out1) as t:
         assert all(m.mtime == 0 for m in t.getmembers())
+
+
+pspec = importlib.util.spec_from_file_location("publish_slices", Path(__file__).parents[1] / "publish-slices.py")
+ps = importlib.util.module_from_spec(pspec)
+pspec.loader.exec_module(ps)
+
+
+def test_only_installer_contours_are_connected(tmp_path):
+    put(tmp_path, "clusters/stand/flux-system/gotk-sync.yaml",
+        f"apiVersion: source.toolkit.fluxcd.io/v1\nkind: GitRepository\nspec:\n  url: {ps.INSTALLER_GITEA_URL}\n")
+    put(tmp_path, "clusters/vad/flux-system/gotk-sync.yaml",
+        "apiVersion: source.toolkit.fluxcd.io/v1\nkind: GitRepository\nspec:\n  url: http://192.168.8.141:31361/infra/iac.git\n")
+    put(tmp_path, "clusters/old/flux-system/gotk-sync.yaml",
+        "apiVersion: source.toolkit.fluxcd.io/v1\nkind: GitRepository\nspec:\n  url: https://gitlab.sarex.io/infra/iac.git\n")
+    assert ps.connected_contours(tmp_path) == ["stand"]