try new build action

.github/workflows/docker-publish.yml

@@ -23,33 +23,57 @@ env:
   IMAGE_NAME: ${{ github.repository }}
 
 jobs:
-  build:
+  test:
     runs-on: ubuntu-latest
     permissions:
       contents: read
       packages: write
-      # This is used to complete the identity challenge
-      # with sigstore/fulcio when running outside of PRs.
       id-token: write
 
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Test
+        id: test
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          platforms: linux/amd64
+          cache-from: type=gha,scope=buildkit-x64
+          pull: true
+          cache-to: type=gha,mode=max,scope=buildkit-x64
+          target: test
+
+  build:
+    needs: test
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+      id-token: write
+
+    strategy:
+      matrix:
+        platform: [linux/amd64, linux/arm64]
+
     steps:
       - name: Checkout repository
         uses: actions/checkout@v4
 
       # Install the cosign tool except on PR
-      # https://github.com/sigstore/cosign-installer
       - name: Install cosign
         if: github.event_name != 'pull_request'
         uses: sigstore/cosign-installer@v3
 
-      # Set up BuildKit Docker container builder to be able to build
+      # Set up BuildKit Docker container builder
-      # multi-platform images and export cache
-      # https://github.com/docker/setup-buildx-action
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
       # Login against a Docker registry except on PR
-      # https://github.com/docker/login-action
       - name: Log into registry ${{ env.REGISTRY }}
         if: github.event_name != 'pull_request'
         uses: docker/login-action@v3
@@ -59,7 +83,6 @@ jobs:
           password: ${{ secrets.GITHUB_TOKEN }}
 
       # Extract metadata (tags, labels) for Docker
-      # https://github.com/docker/metadata-action
       - name: Extract Docker metadata
         id: meta
         uses: docker/metadata-action@v5
@@ -73,68 +96,94 @@ jobs:
             type=raw,enable=${{ github.ref_type == 'tag' }}, value=latest
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
 
-      - name: Test
+      # Get platform specific variables
-        id: test
+      - name: Set platform-specific variables
-        uses: docker/build-push-action@v6
+        id: platform-vars
-        with:
+        run: |
-          context: .
+          PLATFORM="${{ matrix.platform }}"
-          platforms: linux/amd64
+          PLATFORM_SUFFIX="${PLATFORM//\//-}"
-          cache-from: type=gha,scope=buildkit-x64
+          echo "platform_suffix=${PLATFORM_SUFFIX}" >> $GITHUB_OUTPUT
-          pull: true
+          echo "cache_scope=buildkit-${PLATFORM_SUFFIX}" >> $GITHUB_OUTPUT
-          cache-to: type=gha,mode=max,scope=buildkit-x64
-          target: test
 
-      - name: Build and push Docker image for arm64
+      # Build and push Docker image with platform-specific tag
-        id: build-and-push-arm64
+      - name: Build and push Docker image
+        id: build-and-push
         uses: docker/build-push-action@v6
         with:
           context: .
           pull: true
           push: ${{ github.event_name != 'pull_request' }}
-          tags: ${{ steps.meta.outputs.tags }}
+          tags: ${{ steps.meta.outputs.tags }}-${{ steps.platform-vars.outputs.platform_suffix }}
           labels: ${{ steps.meta.outputs.labels }}
-          platforms: linux/arm64
+          platforms: ${{ matrix.platform }}
-          cache-from: type=gha,scope=buildkit-arm64
+          cache-from: type=gha,scope=${{ steps.platform-vars.outputs.cache_scope }}
-          cache-to: type=gha,mode=max,scope=buildkit-arm64
+          cache-to: type=gha,mode=max,scope=${{ steps.platform-vars.outputs.cache_scope }}
           build-args: GITHUB_BUILD=true,VERSION=${{ github.ref_name }}
 
-      # Sign the resulting Docker image digest except on PRs.
+      # Sign the platform specific image
-      # This will only write to the public Rekor transparency log when the Docker
+      - name: Sign the published Docker image
-      # repository is public to avoid leaking data.  If you would like to publish
-      # transparency data even for private images, pass --force to cosign below.
-      # https://github.com/sigstore/cosign
-      - name: Sign the published Docker image for arm64
         if: ${{ github.event_name != 'pull_request' }}
         env:
-          # https://docs.github.com/en/actions/security-guides/security-hardening-for-github-actions#using-an-intermediate-environment-variable
+          TAGS: ${{ steps.meta.outputs.tags }}-${{ steps.platform-vars.outputs.platform_suffix }}
-          TAGS: ${{ steps.meta.outputs.tags }}
+          DIGEST: ${{ steps.build-and-push.outputs.digest }}
-          DIGEST: ${{ steps.build-and-push-arm64.outputs.digest }}
-        # This step uses the identity token to provision an ephemeral certificate
-        # against the sigstore community Fulcio instance.
         run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST}
 
-      # Build and push Docker image with Buildx (don't push on PR)
+  merge:
-      # https://github.com/docker/build-push-action
+    needs: build
-      - name: Build and push Docker image for x64
+    runs-on: ubuntu-latest
-        id: build-and-push-x64
+    if: github.event_name != 'pull_request'
-        uses: docker/build-push-action@v6
+    permissions:
+      contents: read
+      packages: write
+      id-token: write
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Log into registry ${{ env.REGISTRY }}
+        uses: docker/login-action@v3
         with:
-          context: .
+          registry: ${{ env.REGISTRY }}
-          pull: true
+          username: ${{ github.actor }}
-          push: ${{ github.event_name != 'pull_request' }}
+          password: ${{ secrets.GITHUB_TOKEN }}
-          tags: ${{ steps.meta.outputs.tags }}
-          labels: ${{ steps.meta.outputs.labels }}
-          platforms: linux/amd64
-          cache-from: type=gha,scope=buildkit-x64
-          cache-to: type=gha,mode=max,scope=buildkit-x64
-          build-args: GITHUB_BUILD=true,VERSION=${{ github.ref_name }}
 
-      - name: Sign the published Docker image for x64
+      - name: Extract Docker metadata
-        if: ${{ github.event_name != 'pull_request' }}
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          tags: |
+            type=ref,event=branch
+            type=ref,event=pr
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=semver,pattern={{major}}
+            type=raw,enable=${{ github.ref_type == 'tag' }}, value=latest
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+
+      - name: Install cosign
+        uses: sigstore/cosign-installer@v3
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Create and push manifest
+        run: |
+          TAGS="${{ steps.meta.outputs.tags }}"
+          for TAG in $TAGS; do
+            # Create manifest list and push it
+            docker buildx imagetools create -t $TAG \
+              $TAG-linux-amd64 \
+              $TAG-linux-arm64
+          done
+
+      - name: Install cosign
+        uses: sigstore/cosign-installer@v3
+
+      - name: Sign the manifest
         env:
-          # https://docs.github.com/en/actions/security-guides/security-hardening-for-github-actions#using-an-intermediate-environment-variable
           TAGS: ${{ steps.meta.outputs.tags }}
-          DIGEST: ${{ steps.build-and-push-x64.outputs.digest }}
+        run: |
-        # This step uses the identity token to provision an ephemeral certificate
+          for TAG in $TAGS; do
-        # against the sigstore community Fulcio instance.
+            cosign sign --yes $TAG
-        run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST}
+          done

src/endpoints.py

@@ -57,10 +57,7 @@ def read_item(request: LinkRequest, sb: SeleniumDep) -> LinkResponse:
         sb.uc_gui_click_captcha()
         logger.info("Clicked captcha")
 
-    source_bs = sb.get_beautiful_soup()
+    if sb.get_title() in CHALLENGE_TITLES:
-    title_tag = source_bs.title
-
-    if title_tag and title_tag.string in CHALLENGE_TITLES:
         save_screenshot(sb)
         raise HTTPException(status_code=500, detail="Could not bypass challenge")
 
@@ -72,7 +69,7 @@ def read_item(request: LinkRequest, sb: SeleniumDep) -> LinkResponse:
             status=200,
             cookies=sb.get_cookies(),
             headers={},
-            response=str(source_bs),
+            response=str(sb.get_beautiful_soup()),
         ),
         start_timestamp=start_time,
     )

src/middlewares.py

@@ -1,4 +1,5 @@
 import time
+from http import HTTPStatus
 
 from starlette.middleware.base import BaseHTTPMiddleware
 
@@ -20,5 +21,9 @@ class LogRequest(BaseHTTPMiddleware):
         response = await call_next(request)
         process_time = time.perf_counter() - start_time
 
-        logger.info(f"Done {request_body.url} in {process_time:.2f}s")
+        if response.status_code == HTTPStatus.OK:
+            logger.info(f"Done {request_body.url} in {process_time:.2f}s")
+        else:
+            logger.info(f"Failed {request_body.url} in {process_time:.2f}s")
+
         return response