airbyte-internal-ops 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

{airbyte_internal_ops-0.2.2.dist-info → airbyte_internal_ops-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airbyte-internal-ops
-Version: 0.2.2
+Version: 0.2.3
 Summary: MCP and API interfaces that let the agents do the admin work
 Author-email: Aaron Steers <aj@airbyte.io>
 Keywords: admin,airbyte,api,mcp

{airbyte_internal_ops-0.2.2.dist-info → airbyte_internal_ops-0.2.3.dist-info}/RECORD

@@ -2,7 +2,7 @@ airbyte_ops_mcp/__init__.py,sha256=tuzdlMkfnWBnsri5KGHM2M_xuNnzFk2u_aR79mmN7Yg,7
 airbyte_ops_mcp/_annotations.py,sha256=MO-SBDnbykxxHDESG7d8rviZZ4WlZgJKv0a8eBqcEzQ,1757
 airbyte_ops_mcp/constants.py,sha256=GeZ2_WWluMSrGkyqGvqUVFCy-5PD-lyzZbQ7eO-vyUo,5192
 airbyte_ops_mcp/gcp_auth.py,sha256=5k-k145ZoYhHLjyDES8nrA8f8BBihRI0ykrdD1IcfOs,3599
-airbyte_ops_mcp/github_actions.py,sha256=hcwwew98r0yetWsM7Qmdar3ATLBJQGIn3fJfJ_n59So,8599
+airbyte_ops_mcp/github_actions.py,sha256=wKnuIVmF4u1gMYNdSoryD_PUmvMz5SaHgOvbU0dsolA,9957
 airbyte_ops_mcp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/_legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airbyte_ops_mcp/_legacy/airbyte_ci/README.md,sha256=qEYx4geDR8AEDjrcA303h7Nol-CMDLojxUyiGzQprM8,236
@@ -354,7 +354,7 @@ airbyte_ops_mcp/cli/_shared.py,sha256=jg-xMyGzTCGPqKd8VTfE_3kGPIyO_3Kx5sQbG4rPc0
 airbyte_ops_mcp/cli/app.py,sha256=SEdBpqFUG2O8zGV5ifwptxrLGFph_dLr66-MX9d69gQ,789
 airbyte_ops_mcp/cli/cloud.py,sha256=JJusGl67ca41rvoS8BPl5Kmb7Kyu7iMH-tvbaJsKsPs,41359
 airbyte_ops_mcp/cli/gh.py,sha256=91b1AxFXvHQCFyXhrrym-756ZjnMCqvxFdmwCtma1zI,2046
-airbyte_ops_mcp/cli/registry.py,sha256=-yiLJWSslV_qGi6ImXZYfXOJSE4oJBO7yICkyA_RiUo,5792
+airbyte_ops_mcp/cli/registry.py,sha256=tcf_CDiUVJpSdBRNqlEL3zFKMqK53AhFpJjAETM4gLs,9781
 airbyte_ops_mcp/cli/repo.py,sha256=G1hoQpH0XYhUH3FFOsia9xabGB0LP9o3XcwBuqvFVo0,16331
 airbyte_ops_mcp/cloud_admin/__init__.py,sha256=cqE96Q10Kp6elhH9DAi6TVsIwSUy3sooDLLrxTaktGk,816
 airbyte_ops_mcp/cloud_admin/api_client.py,sha256=tx1kwGIKMPesibflQkFOlbNp0t0CfJD4Ab097ngsjHA,19126
@@ -375,7 +375,7 @@ airbyte_ops_mcp/mcp/connector_qa.py,sha256=aImpqdnqBPDrz10BS0owsV4kuIU2XdalzgbaG
 airbyte_ops_mcp/mcp/github.py,sha256=h3M3VJrq09y_F9ueQVCq3bUbVBNFuTNKprHtGU_ttio,8045
 airbyte_ops_mcp/mcp/github_repo_ops.py,sha256=PiERpt8abo20Gz4CfXhrDNlVM4o4FOt5sweZJND2a0s,5314
 airbyte_ops_mcp/mcp/metadata.py,sha256=fwGW97WknR5lfKcQnFtK6dU87aA6TmLj1NkKyqDAV9g,270
-airbyte_ops_mcp/mcp/prerelease.py,sha256=6G4zMo0KeCIYJPEIryHKHoZUiBHQMagPJU-uw-IzK94,8939
+airbyte_ops_mcp/mcp/prerelease.py,sha256=nc6VU03ADVHWM3OjGKxbS5XqY4VoyRyrZNU_fyAtaOI,10465
 airbyte_ops_mcp/mcp/prod_db_queries.py,sha256=FfGoq3aEj6ZUT4ysBIs1w7LzzwBeRXTaRvPGEx62RzI,25474
 airbyte_ops_mcp/mcp/prompts.py,sha256=mJld9mdPECXYZffWXGSvNs4Xevx3rxqUGNlzGKVC2_s,1599
 airbyte_ops_mcp/mcp/registry.py,sha256=PW-VYUj42qx2pQ_apUkVaoUFq7VgB9zEU7-aGrkSCCw,290
@@ -410,7 +410,7 @@ airbyte_ops_mcp/regression_tests/regression/comparators.py,sha256=MJkLZEKHivgrG0
 airbyte_ops_mcp/regression_tests/validation/__init__.py,sha256=MBEwGOoNuqT4_oCahtoK62OKWIjUCfWa7vZTxNj_0Ek,1532
 airbyte_ops_mcp/regression_tests/validation/catalog_validators.py,sha256=jqqVAMOk0mtdPgwu4d0hA0ZEjtsNh5gapvGydRv3_qk,12553
 airbyte_ops_mcp/regression_tests/validation/record_validators.py,sha256=RjauAhKWNwxMBTu0eNS2hMFNQVs5CLbQU51kp6FOVDk,7432
-airbyte_internal_ops-0.2.2.dist-info/METADATA,sha256=1ah9ZGR3rZ1676mhAJzJvrbfqOQsV_fuedRitpJG9h8,5679
-airbyte_internal_ops-0.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-airbyte_internal_ops-0.2.2.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
-airbyte_internal_ops-0.2.2.dist-info/RECORD,,
+airbyte_internal_ops-0.2.3.dist-info/METADATA,sha256=c-kIcUr44n9G4r4yImy7oEQuSrLRueORjI4SeW-2ZNI,5679
+airbyte_internal_ops-0.2.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+airbyte_internal_ops-0.2.3.dist-info/entry_points.txt,sha256=WxP0l7bRFss4Cr5uQqVj9mTEKwnRKouNuphXQF0lotA,171
+airbyte_internal_ops-0.2.3.dist-info/RECORD,,

airbyte_ops_mcp/cli/registry.py

@@ -5,6 +5,7 @@ This module provides CLI wrappers for registry operations. The core logic
 lives in the `airbyte_ops_mcp.registry` capability module.
 
 Commands:
+    airbyte-ops registry connector compute-prerelease-tag - Compute prerelease version tag
     airbyte-ops registry connector publish-prerelease - Publish connector prerelease
     airbyte-ops registry connector publish - Publish connector (apply/rollback version override)
     airbyte-ops registry image inspect - Inspect Docker image on DockerHub
@@ -12,9 +13,12 @@ Commands:
 
 from __future__ import annotations
 
+import contextlib
+import sys
 from pathlib import Path
 from typing import Annotated
 
+import yaml
 from cyclopts import App, Parameter
 
 from airbyte_ops_mcp.cli._base import app
@@ -24,8 +28,15 @@ from airbyte_ops_mcp.cli._shared import (
     print_json,
     print_success,
 )
+from airbyte_ops_mcp.github_actions import (
+    get_file_contents_at_ref,
+    resolve_github_token,
+)
 from airbyte_ops_mcp.mcp.github import get_docker_image_info
-from airbyte_ops_mcp.mcp.prerelease import publish_connector_to_airbyte_registry
+from airbyte_ops_mcp.mcp.prerelease import (
+    compute_prerelease_docker_image_tag,
+    publish_connector_to_airbyte_registry,
+)
 from airbyte_ops_mcp.registry import (
     ConnectorPublishResult,
     PublishAction,
@@ -47,6 +58,111 @@ image_app = App(name="image", help="Docker image operations.")
 registry_app.command(image_app)
 
 
+AIRBYTE_REPO_OWNER = "airbytehq"
+AIRBYTE_REPO_NAME = "airbyte"
+CONNECTOR_PATH_PREFIX = "airbyte-integrations/connectors"
+
+
+def _get_connector_version_from_github(
+    connector_name: str,
+    ref: str,
+    token: str | None = None,
+) -> str | None:
+    """Fetch connector version from metadata.yaml via GitHub API.
+
+    Args:
+        connector_name: Connector name (e.g., "source-github")
+        ref: Git ref (commit SHA, branch name, or tag)
+        token: GitHub API token (optional for public repos)
+
+    Returns:
+        Version string from metadata.yaml, or None if not found.
+    """
+    path = f"{CONNECTOR_PATH_PREFIX}/{connector_name}/metadata.yaml"
+    contents = get_file_contents_at_ref(
+        owner=AIRBYTE_REPO_OWNER,
+        repo=AIRBYTE_REPO_NAME,
+        path=path,
+        ref=ref,
+        token=token,
+    )
+    if contents is None:
+        return None
+
+    metadata = yaml.safe_load(contents)
+    return metadata.get("data", {}).get("dockerImageTag")
+
+
+@connector_app.command(name="compute-prerelease-tag")
+def compute_prerelease_tag(
+    connector_name: Annotated[
+        str,
+        Parameter(help="Connector name (e.g., 'source-github')."),
+    ],
+    sha: Annotated[
+        str,
+        Parameter(help="Git commit SHA (full or at least 7 characters)."),
+    ],
+    base_version: Annotated[
+        str | None,
+        Parameter(
+            help="Base version override. If not provided, fetched from metadata.yaml at the given SHA."
+        ),
+    ] = None,
+) -> None:
+    """Compute the pre-release docker image tag.
+
+    Outputs the version tag to stdout for easy capture in shell scripts.
+    This is the single source of truth for pre-release version format.
+
+    The command fetches the connector's metadata.yaml from GitHub at the given SHA
+    to determine the base version. It also compares against the master branch and
+    prints a warning to stderr if no version bump is detected.
+
+    If --base-version is provided, it is used directly instead of fetching from GitHub.
+
+    Example:
+        airbyte-ops registry connector compute-prerelease-tag --connector-name source-github --sha abcdef1234567
+        # Output: 1.2.3-preview.abcdef1
+
+        airbyte-ops registry connector compute-prerelease-tag --connector-name source-github --sha abcdef1234567 --base-version 1.2.3
+        # Output: 1.2.3-preview.abcdef1 (uses provided version, skips GitHub API)
+    """
+    # Try to get a GitHub token (optional, but helps avoid rate limiting)
+    # Token resolution may fail if no token is configured, which is fine for public repos
+    token: str | None = None
+    with contextlib.suppress(ValueError):
+        token = resolve_github_token()
+
+    # Determine base version
+    version: str
+    if base_version:
+        version = base_version
+    else:
+        # Fetch version from metadata.yaml at the given SHA
+        fetched_version = _get_connector_version_from_github(connector_name, sha, token)
+        if fetched_version is None:
+            print(
+                f"Error: Could not fetch metadata.yaml for {connector_name} at ref {sha}",
+                file=sys.stderr,
+            )
+            sys.exit(1)
+        version = fetched_version
+
+    # Compare with master branch version and warn if no bump detected
+    master_version = _get_connector_version_from_github(connector_name, "master", token)
+    if master_version and master_version == version:
+        print(
+            f"Warning: No version bump detected for {connector_name}. "
+            f"Version {version} matches master branch.",
+            file=sys.stderr,
+        )
+
+    # Compute and output the prerelease tag
+    tag = compute_prerelease_docker_image_tag(version, sha)
+    print(tag)
+
+
 @connector_app.command(name="publish-prerelease")
 def publish_prerelease(
     connector_name: Annotated[

airbyte_ops_mcp/github_actions.py

@@ -106,6 +106,51 @@ class WorkflowJobInfo:
     """ISO 8601 timestamp when the job completed"""
 
 
+def get_file_contents_at_ref(
+    owner: str,
+    repo: str,
+    path: str,
+    ref: str,
+    token: str | None = None,
+) -> str | None:
+    """Fetch file contents from GitHub at a specific ref.
+
+    Uses the GitHub Contents API to retrieve file contents at a specific
+    commit SHA, branch, or tag. This allows reading files without having
+    the repository checked out locally.
+
+    Args:
+        owner: Repository owner (e.g., "airbytehq")
+        repo: Repository name (e.g., "airbyte")
+        path: Path to the file within the repository
+        ref: Git ref (commit SHA, branch name, or tag)
+        token: GitHub API token (optional for public repos, but recommended
+            to avoid rate limiting)
+
+    Returns:
+        File contents as a string, or None if the file doesn't exist.
+
+    Raises:
+        requests.HTTPError: If API request fails (except 404).
+    """
+    url = f"{GITHUB_API_BASE}/repos/{owner}/{repo}/contents/{path}"
+    headers = {
+        "Accept": "application/vnd.github.raw+json",
+        "X-GitHub-Api-Version": "2022-11-28",
+    }
+    if token:
+        headers["Authorization"] = f"Bearer {token}"
+
+    params = {"ref": ref}
+
+    response = requests.get(url, headers=headers, params=params, timeout=30)
+    if response.status_code == 404:
+        return None
+    response.raise_for_status()
+
+    return response.text
+
+
 def get_workflow_jobs(
     owner: str,
     repo: str,

airbyte_ops_mcp/mcp/prerelease.py

@@ -31,6 +31,45 @@ PRERELEASE_TOKEN_ENV_VARS = [
     "GITHUB_TOKEN",
 ]
 
+# =============================================================================
+# Pre-release Version Tag Constants
+# =============================================================================
+
+PRERELEASE_TAG_PREFIX = "preview"
+"""The prefix used for pre-release version tags (e.g., '1.2.3-preview.abcde12')."""
+
+PRERELEASE_SHA_LENGTH = 7
+"""The number of characters from the git SHA to include in pre-release tags."""
+
+
+def compute_prerelease_docker_image_tag(base_version: str, sha: str) -> str:
+    """Compute the pre-release docker image tag.
+
+    This is the SINGLE SOURCE OF TRUTH for pre-release version format.
+    All other code should receive this value as a parameter, not recompute it.
+
+    The format is: {base_version}-preview.{short_sha}
+
+    Where:
+        - base_version: The base version from metadata.yaml (e.g., "1.2.3")
+        - short_sha: The first 7 characters of the git commit SHA
+
+    Examples:
+        >>> compute_prerelease_docker_image_tag("1.2.3", "abcdef1234567890")
+        '1.2.3-preview.abcdef1'
+        >>> compute_prerelease_docker_image_tag("0.1.0", "1234567")
+        '0.1.0-preview.1234567'
+
+    Args:
+        base_version: The base version from metadata.yaml (e.g., "1.2.3")
+        sha: The full git commit SHA (or at least 7 characters)
+
+    Returns:
+        Pre-release version tag (e.g., "1.2.3-preview.abcde12")
+    """
+    short_sha = sha[:PRERELEASE_SHA_LENGTH]
+    return f"{base_version}-{PRERELEASE_TAG_PREFIX}.{short_sha}"
+
 
 class PRHeadInfo(BaseModel):
     """Information about a PR's head commit."""
@@ -268,7 +307,9 @@ def publish_connector_to_airbyte_registry(
         docker_image = data.get("dockerRepository")
         base_version = data.get("dockerImageTag")
         if base_version:
-            docker_image_tag = f"{base_version}-preview.{head_info.short_sha}"
+            docker_image_tag = compute_prerelease_docker_image_tag(
+                base_version, head_info.sha
+            )
 
     return PrereleaseWorkflowResult(
         success=True,