GoogleCloudPlatform · parthea · Jun 8, 2023 · May 27, 2022 · May 27, 2022 · Jul 16, 2022
@@ -103,6 +103,7 @@
 /retail/**/*                           @GoogleCloudPlatform/cloud-retail-team @GoogleCloudPlatform/python-samples-reviewers
 /billing/**/*                          @GoogleCloudPlatform/billing-samples-maintainers @GoogleCloudPlatform/python-samples-reviewers
 /iot/**/*                              @GoogleCloudPlatform/api-iot @GoogleCloudPlatform/python-samples-reviewers
+/video/stitcher/*                      @GoogleCloudPlatform/cloud-media-team @GoogleCloudPlatform/python-samples-reviewers
 
 # Does not have owner
 /blog/**/*                             @GoogleCloudPlatform/python-samples-reviewers

@@ -304,6 +304,7 @@ assign_prs_by:
       - GoogleCloudPlatform/cloud-retail-team
   - labels:
       - "api: cloudmedia"
+      - "api: videostitcher"
     to:
       - GoogleCloudPlatform/cloud-media-team
 

@@ -0,0 +1,174 @@
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import uuid
+
+from google.protobuf import timestamp_pb2
+import pytest
+
+import create_cdn_key
+import create_cdn_key_akamai
+import delete_cdn_key
+import get_cdn_key
+import list_cdn_keys
+import update_cdn_key
+import update_cdn_key_akamai
+import utils
+
+location = "us-central1"
+project_id = os.environ["GOOGLE_CLOUD_PROJECT"]
+now = timestamp_pb2.Timestamp()
+now.GetCurrentTime()
+
+media_cdn_key_id = f"python-test-media-key-{uuid.uuid4().hex[:5]}-{now.seconds}"
+cloud_cdn_key_id = f"python-test-cloud-key-{uuid.uuid4().hex[:5]}-{now.seconds}"
+akamai_cdn_key_id = f"python-test-akamai-key-{uuid.uuid4().hex[:5]}-{now.seconds}"
+
+hostname = "cdn.example.com"
+updated_hostname = "updated.example.com"
+key_name = "my-key"
+
+media_cdn_private_key = "MTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0NTY3ODkwMTIzNDU2Nzg5MDEyMzQ1Njc4OTAxMjM0NTY3ODkwMTIzNA"
+updated_media_cdn_private_key = (
+    "ZZZzNDU2Nzg5MDEyMzQ1Njc4OTAxzg5MDEyMzQ1Njc4OTAxMjM0NTY3DkwMTIZZZ"
+)
+cloud_cdn_private_key = "VGhpcyBpcyBhIHRlc3Qgc3RyaW5nLg=="
+updated_cloud_cdn_private_key = "VGhpcyBpcyBhbiB1cGRhdGVkIHRlc3Qgc3RyaW5nLg=="
+akamai_key = cloud_cdn_private_key
+updated_akamai_key = updated_cloud_cdn_private_key
+
+
+def test_cdn_key_operations(capsys: pytest.fixture) -> None:
+
+    utils.delete_stale_cdn_keys(project_id, location)
+
+    # Media CDN key tests
+
+    create_cdn_key.create_cdn_key(
+        project_id,
+        location,
+        media_cdn_key_id,
+        hostname,
+        key_name,
+        media_cdn_private_key,
+        False,
+    )
+    out, _ = capsys.readouterr()
+    assert media_cdn_key_id in out
+
+    list_cdn_keys.list_cdn_keys(project_id, location)
+    out, _ = capsys.readouterr()
+    assert media_cdn_key_id in out
+
+    # Update the hostname and private key; the private key value
+    # is not returned by the client
+    response = update_cdn_key.update_cdn_key(
+        project_id,
+        location,
+        media_cdn_key_id,
+        updated_hostname,
+        key_name,
+        updated_media_cdn_private_key,
+        False,
+    )
+    out, _ = capsys.readouterr()
+    assert media_cdn_key_id in out
+    assert updated_hostname in response.hostname
+
+    get_cdn_key.get_cdn_key(project_id, location, media_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert media_cdn_key_id in out
+
+    delete_cdn_key.delete_cdn_key(project_id, location, media_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert "Deleted CDN key" in out
+
+    # Cloud CDN key tests
+
+    create_cdn_key.create_cdn_key(
+        project_id,
+        location,
+        cloud_cdn_key_id,
+        hostname,
+        key_name,
+        cloud_cdn_private_key,
+        True,
+    )
+    out, _ = capsys.readouterr()
+    assert cloud_cdn_key_id in out
+
+    list_cdn_keys.list_cdn_keys(project_id, location)
+    out, _ = capsys.readouterr()
+    assert cloud_cdn_key_id in out
+
+    # Update the hostname and private key; the private key value
+    # is not returned by the client
+    response = update_cdn_key.update_cdn_key(
+        project_id,
+        location,
+        cloud_cdn_key_id,
+        updated_hostname,
+        key_name,
+        updated_cloud_cdn_private_key,
+        True,
+    )
+    out, _ = capsys.readouterr()
+    assert cloud_cdn_key_id in out
+    assert updated_hostname in response.hostname
+
+    get_cdn_key.get_cdn_key(project_id, location, cloud_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert cloud_cdn_key_id in out
+
+    delete_cdn_key.delete_cdn_key(project_id, location, cloud_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert "Deleted CDN key" in out
+
+    # Akamai CDN key tests
+
+    create_cdn_key_akamai.create_cdn_key_akamai(
+        project_id,
+        location,
+        akamai_cdn_key_id,
+        hostname,
+        akamai_key,
+    )
+    out, _ = capsys.readouterr()
+    assert akamai_cdn_key_id in out
+
+    list_cdn_keys.list_cdn_keys(project_id, location)
+    out, _ = capsys.readouterr()
+    assert akamai_cdn_key_id in out
+
+    # Update the hostname and private key; the private key value
+    # is not returned by the client
+    response = update_cdn_key_akamai.update_cdn_key_akamai(
+        project_id,
+        location,
+        akamai_cdn_key_id,
+        updated_hostname,
+        updated_akamai_key,
+    )
+    out, _ = capsys.readouterr()
+    assert akamai_cdn_key_id in out
+    assert updated_hostname in response.hostname
+
+    get_cdn_key.get_cdn_key(project_id, location, akamai_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert akamai_cdn_key_id in out
+
+    delete_cdn_key.delete_cdn_key(project_id, location, akamai_cdn_key_id)
+    out, _ = capsys.readouterr()
+    assert "Deleted CDN key" in out
@@ -0,0 +1,135 @@
+#!/usr/bin/env python
+
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Google Cloud Video Stitcher sample for creating a Media CDN key
+or a Cloud CDN key. A CDN key is used to retrieve protected media.
+Example usage:
+    python create_cdn_key.py --project_id <project-id> --location <location> \
+        --cdn_key_id <cdn_key_id> --hostname <hostname> \
+        --key_name <name> --private_key <key> [--is_cloud_cdn]
+"""
+
+# [START videostitcher_create_cdn_key]
+
+import argparse
+
+from google.cloud.video import stitcher_v1
+from google.cloud.video.stitcher_v1.services.video_stitcher_service import (
+    VideoStitcherServiceClient,
+)
+
+
+def create_cdn_key(
+    project_id: str,
+    location: str,
+    cdn_key_id: str,
+    hostname: str,
+    key_name: str,
+    private_key: str,
+    is_cloud_cdn: bool,
+) -> str:
+    """Creates a Cloud CDN or Media CDN key.
+    Args:
+        project_id: The GCP project ID.
+        location: The location in which to create the CDN key.
+        cdn_key_id: The user-defined CDN key ID.
+        hostname: The hostname to which this CDN key applies.
+        key_name: For a Media CDN key, this is the keyset name.
+                  For a Cloud CDN key, this is the public name of the CDN key.
+        private_key: For a Media CDN key, this is a 64-byte Ed25519 private
+                     key encoded as a base64-encoded string.
+                     See https://cloud.google.com/video-stitcher/docs/how-to/managing-cdn-keys#create-private-key-media-cdn
+                     for more information. For a Cloud CDN key, this is a base64-encoded string secret.
+        is_cloud_cdn: If true, create a Cloud CDN key. If false, create a Media CDN key."""
+
+    client = VideoStitcherServiceClient()
+
+    parent = f"projects/{project_id}/locations/{location}"
+
+    cdn_key = stitcher_v1.types.CdnKey(
+        name=cdn_key_id,
+        hostname=hostname,
+    )
+
+    if is_cloud_cdn:
+        cdn_key.google_cdn_key = stitcher_v1.types.GoogleCdnKey(
+            key_name=key_name,
+            private_key=private_key,
+        )
+    else:
+        cdn_key.media_cdn_key = stitcher_v1.types.MediaCdnKey(
+            key_name=key_name,
+            private_key=private_key,
+        )
+
+    operation = client.create_cdn_key(
+        parent=parent, cdn_key_id=cdn_key_id, cdn_key=cdn_key
+    )
+    response = operation.result()
+    print(f"CDN key: {response.name}")
+    return response
+
+
+# [END videostitcher_create_cdn_key]
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--project_id", help="Your Cloud project ID.", required=True)
+    parser.add_argument(
+        "--location",
+        help="The location in which to create the CDN key.",
+        default="us-central1",
+    )
+    parser.add_argument(
+        "--cdn_key_id",
+        help="The user-defined CDN key ID.",
+        required=True,
+    )
+    parser.add_argument(
+        "--hostname",
+        help="The hostname to which this CDN key applies.",
+        required=True,
+    )
+    parser.add_argument(
+        "--key_name",
+        help="For a Media CDN key, this is the keyset name. For a Cloud CDN"
+        + " key, this is the public name of the CDN key.",
+        required=True,
+    )
+    parser.add_argument(
+        "--private_key",
+        help="For a Media CDN key, this is a 64-byte Ed25519 private key"
+        + "encoded as a base64-encoded string. See"
+        + " https://cloud.google.com/video-stitcher/docs/how-to/managing-cdn-keys#create-private-key-media-cdn"
+        + " for more information. For a Cloud CDN key, this is a"
+        + " base64-encoded string secret.",
+        required=True,
+    )
+    parser.add_argument(
+        "--is_cloud_cdn",
+        action="store_true",
+        help="If included, create a Cloud CDN key. If absent, create a Media CDN key.",
+    )
+    args = parser.parse_args()
+    create_cdn_key(
+        args.project_id,
+        args.location,
+        args.cdn_key_id,
+        args.hostname,
+        args.key_name,
+        args.private_key,
+        args.is_cloud_cdn,
+    )