Merge pull request #258 from crazy-max/update-toolkit

bump @docker/actions-toolkit from 0.7.1 to 0.10.0
ci: check lab releases
2025-06-26 12:41:09 +02:00 · 2023-08-27 18:39:52 -07:00 · 2023-08-26 14:03:23 +02:00 · 2023-08-26 14:03:22 +02:00 · 2023-08-26 14:03:22 +02:00 · 2023-08-26 14:03:22 +02:00
48 changed files with 3499 additions and 2852 deletions
--- a/.eslintrc.json
+++ b/.eslintrc.json
@ -2,7 +2,7 @@
  "env": {
    "node": true,
    "es2021": true,
-    "jest/globals": true
+    "jest": true
  },
  "extends": [
    "eslint:recommended",
--- a/.github/CODEOWNERS
+++ b/.github/CODEOWNERS
@ -1 +0,0 @@
-*	@crazy-max
--- a/.github/buildkit-container-logs.png
+++ b/.github/buildkit-container-logs.png
--- a/.github/dependabot.yml
+++ b/.github/dependabot.yml
@ -11,6 +11,7 @@ updates:
    directory: "/"
    schedule:
      interval: "daily"
+    versioning-strategy: "increase"
    allow:
      - dependency-type: "production"
    labels:
--- a/.github/setup-buildx-action.png
+++ b/.github/setup-buildx-action.png
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@ -2,7 +2,7 @@ name: ci

 on:
  schedule:
-    - cron: '0 10 * * *' # everyday at 10am
+    - cron: '0 10 * * *'
  push:
    branches:
      - 'master'
@ -10,9 +10,6 @@ on:
    tags:
      - 'v*'
  pull_request:
-    branches:
-      - 'master'
-      - 'releases/v*'

 jobs:
  main:
@ -21,9 +18,11 @@ jobs:
      fail-fast: false
      matrix:
        buildx-version:
-          - latest
-          - v0.4.1
          - ""
+          - "latest"
+          - "v0.4.1"
+          - "lab:latest"
+          - "lab:v0.11.2-desktop.2"
    steps:
      -
        name: Checkout
@ -63,7 +62,7 @@ jobs:
      -
        name: Stop docker
        run: |
-          sudo systemctl stop docker
+          sudo systemctl stop docker docker.socket
      -
        name: Set up Docker Buildx
        id: buildx
@ -80,7 +79,7 @@ jobs:
      -
        name: Dump context
        if: always()
-        uses: crazy-max/ghaction-dump-context@v1
+        uses: crazy-max/ghaction-dump-context@v2

  debug:
    runs-on: ubuntu-latest
@ -105,7 +104,7 @@ jobs:
          buildkitd-flags: --debug
      -
        name: Build
-        uses: docker/build-push-action@v3
+        uses: docker/build-push-action@v4
        with:
          context: .
          platforms: linux/amd64,linux/arm64,linux/ppc64le
@ -251,7 +250,7 @@ jobs:
          config: /tmp/buildkitd.toml
      -
        name: Build
-        uses: docker/build-push-action@v3
+        uses: docker/build-push-action@v4
        with:
          context: .

@ -278,7 +277,7 @@ jobs:
              mirrors = ["mirror.gcr.io"]
      -
        name: Build
-        uses: docker/build-push-action@v3
+        uses: docker/build-push-action@v4
        with:
          context: .

@ -288,8 +287,7 @@ jobs:
      fail-fast: false
      matrix:
        buildx-version:
-          - latest
-          - v0.4.1
+          - v0.9.1
          - ""
        qemu-platforms:
          - all
@ -353,7 +351,11 @@ jobs:
      -
        name: Uninstall docker cli
        run: |
+          if dpkg -s "docker-ce" >/dev/null 2>&1; then
+            sudo dpkg -r --force-depends docker-ce-cli docker-buildx-plugin
+          else
            sudo apt-get purge -y moby-cli moby-buildx
+          fi
      -
        name: Set up Docker Buildx
        uses: ./
@ -377,7 +379,11 @@ jobs:
      -
        name: Uninstall docker cli
        run: |
+          if dpkg -s "docker-ce" >/dev/null 2>&1; then
+            sudo dpkg -r --force-depends docker-ce-cli docker-buildx-plugin
+          else
            sudo apt-get purge -y moby-cli moby-buildx
+          fi
      -
        name: Set up Docker Buildx
        uses: ./
@ -404,7 +410,11 @@ jobs:
      -
        name: Uninstall docker cli
        run: |
+          if dpkg -s "docker-ce" >/dev/null 2>&1; then
+            sudo dpkg -r --force-depends docker-ce-cli docker-buildx-plugin
+          else
            sudo apt-get purge -y moby-cli moby-buildx
+          fi
      -
        name: Set up Docker Buildx
        id: buildx
@ -464,3 +474,88 @@ jobs:
        uses: ./
        with:
          platforms: linux/amd64
+
+  docker-context:
+    runs-on: ubuntu-latest
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v3
+      -
+        name: Create Docker context
+        run: |
+          docker context create foo
+          docker context use foo
+      -
+        name: Set up Docker Buildx
+        uses: ./
+        with:
+          driver: docker
+
+  cleanup:
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        cleanup:
+          - true
+          - false
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v3
+      -
+        name: Set up Docker Buildx
+        uses: ./
+        with:
+          cleanup: ${{ matrix.cleanup }}
+
+  k3s:
+    runs-on: ubuntu-20.04
+    strategy:
+      fail-fast: false
+      matrix:
+        version:
+          - v0.11.0
+          - v0.10.5
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v3
+      -
+        name: Install k3s
+        uses: crazy-max/.github/.github/actions/install-k3s@605d30d5ae97f6680578ace4b56645af79343e60
+        with:
+          version: "v1.21.2-k3s1"
+      -
+        name: Set up Docker Buildx
+        id: buildx
+        uses: ./
+        with:
+          version: ${{ matrix.version }}
+          driver: kubernetes
+          platforms: linux/amd64
+          driver-opts: |
+            image=moby/buildkit:buildx-stable-1
+            qemu.install=true
+          append: |
+            - platforms: linux/arm64
+              driver-opts:
+                - image=moby/buildkit:buildx-stable-1
+                - qemu.install=true
+            - platforms: linux/s390x
+              driver-opts:
+                - image=moby/buildkit:buildx-stable-1
+                - qemu.install=true
+      -
+        name: Create Dockerfile
+        run: |
+          cat > ./Dockerfile <<EOL
+          FROM alpine
+          RUN uname -a
+          EOL
+      -
+        name: Build
+        uses: docker/build-push-action@master
+        with:
+          context: .
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@ -6,9 +6,6 @@ on:
      - 'master'
      - 'releases/v*'
  pull_request:
-    branches:
-      - 'master'
-      - 'releases/v*'

 jobs:
  test:
@ -17,17 +14,12 @@ jobs:
      -
        name: Checkout
        uses: actions/checkout@v3
-      -
-        name: Validate
-        uses: docker/bake-action@v2
-        with:
-          targets: validate
      -
        name: Set up Docker Buildx
        uses: ./
      -
        name: Test
-        uses: docker/bake-action@v2
+        uses: docker/bake-action@v3
        with:
          targets: test
      -
--- a/.github/workflows/validate.yml
+++ b/.github/workflows/validate.yml
@ -0,0 +1,41 @@
+name: validate
+
+on:
+  push:
+    branches:
+      - 'master'
+      - 'releases/v*'
+  pull_request:
+
+jobs:
+  prepare:
+    runs-on: ubuntu-latest
+    outputs:
+      targets: ${{ steps.targets.outputs.matrix }}
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v3
+      -
+        name: Targets matrix
+        id: targets
+        run: |
+          echo "matrix=$(docker buildx bake validate --print | jq -cr '.group.validate.targets')" >> $GITHUB_OUTPUT
+
+  validate:
+    runs-on: ubuntu-latest
+    needs:
+      - prepare
+    strategy:
+      fail-fast: false
+      matrix:
+        target: ${{ fromJson(needs.prepare.outputs.targets) }}
+    steps:
+      -
+        name: Checkout
+        uses: actions/checkout@v3
+      -
+        name: Validate
+        uses: docker/bake-action@v3
+        with:
+          targets: ${{ matrix.target }}
--- a/README.md
+++ b/README.md
@ -1,7 +1,7 @@
 [![GitHub release](https://img.shields.io/github/release/docker/setup-buildx-action.svg?style=flat-square)](https://github.com/docker/setup-buildx-action/releases/latest)
 [![GitHub marketplace](https://img.shields.io/badge/marketplace-docker--setup--buildx-blue?logo=github&style=flat-square)](https://github.com/marketplace/actions/docker-setup-buildx)
-[![CI workflow](https://img.shields.io/github/workflow/status/docker/setup-buildx-action/ci?label=ci&logo=github&style=flat-square)](https://github.com/docker/setup-buildx-action/actions?workflow=ci)
-[![Test workflow](https://img.shields.io/github/workflow/status/docker/setup-buildx-action/test?label=test&logo=github&style=flat-square)](https://github.com/docker/setup-buildx-action/actions?workflow=test)
+[![CI workflow](https://img.shields.io/github/actions/workflow/status/docker/setup-buildx-action/ci.yml?branch=master&label=ci&logo=github&style=flat-square)](https://github.com/docker/setup-buildx-action/actions?workflow=ci)
+[![Test workflow](https://img.shields.io/github/actions/workflow/status/docker/setup-buildx-action/test.yml?branch=master&label=test&logo=github&style=flat-square)](https://github.com/docker/setup-buildx-action/actions?workflow=test)
 [![Codecov](https://img.shields.io/codecov/c/github/docker/setup-buildx-action?logo=codecov&style=flat-square)](https://codecov.io/gh/docker/setup-buildx-action)

 ## About
@ -19,20 +19,14 @@ a [BuildKit](https://github.com/moby/buildkit) container.
 ___

 * [Usage](#usage)
-* [Advanced usage](#advanced-usage)
-  * [Authentication support](docs/advanced/auth.md)
-  * [Append additional nodes to the builder](docs/advanced/append-nodes.md)
-  * [Install by default](docs/advanced/install-default.md)
-  * [BuildKit daemon configuration](docs/advanced/buildkit-config.md)
-  * [Standalone mode](docs/advanced/standalone.md)
+* [Configuring your builder](#configuring-your-builder)
 * [Customizing](#customizing)
  * [inputs](#inputs)
  * [outputs](#outputs)
  * [environment variables](#environment-variables)
 * [Notes](#notes)
  * [`nodes` output](#nodes-output)
-  * [BuildKit container logs](#buildkit-container-logs)
-* [Keep up-to-date with GitHub Dependabot](#keep-up-to-date-with-github-dependabot)
+* [Contributing](#contributing)

 ## Usage

@ -59,13 +53,19 @@ jobs:
        uses: docker/setup-buildx-action@v2
 ```

-## Advanced usage
+## Configuring your builder

-* [Authentication support](docs/advanced/auth.md)
-* [Append additional nodes to the builder](docs/advanced/append-nodes.md)
-* [Install by default](docs/advanced/install-default.md)
-* [BuildKit daemon configuration](docs/advanced/buildkit-config.md)
-* [Standalone mode](docs/advanced/standalone.md)
+* [Version pinning](https://docs.docker.com/build/ci/github-actions/configure-builder/#version-pinning): Pin to a specific Buildx or BuildKit version
+* [BuildKit container logs](https://docs.docker.com/build/ci/github-actions/configure-builder/#buildkit-container-logs): Enable BuildKit container logs for debugging purposes
+* [BuildKit Daemon configuration](https://docs.docker.com/build/ci/github-actions/configure-builder/#buildkit-daemon-configuration)
+  * [Registry mirror](https://docs.docker.com/build/ci/github-actions/configure-builder/#registry-mirror): Configure a registry mirror for your builds
+  * [Max parallelism](https://docs.docker.com/build/ci/github-actions/configure-builder/#max-parallelism): Configure the maximum parallelism for your builds
+* [Append additional nodes to the builder](https://docs.docker.com/build/ci/github-actions/configure-builder/#append-additional-nodes-to-the-builder): Create additional nodes for your builder
+* [Authentication for remote builders](https://docs.docker.com/build/ci/github-actions/configure-builder/#authentication-for-remote-builders)
+  * [SSH authentication](https://docs.docker.com/build/ci/github-actions/configure-builder/#ssh-authentication): Authenticate to a remote builder using SSH
+  * [TLS authentication](https://docs.docker.com/build/ci/github-actions/configure-builder/#tls-authentication): Authenticate to a remote builder using TLS
+* [Standalone mode](https://docs.docker.com/build/ci/github-actions/configure-builder/#standalone-mode): Use Buildx as a standalone binary (without the Docker CLI)
+* [Isolated builders](https://docs.docker.com/build/ci/github-actions/configure-builder/#isolated-builders): Create isolated builders for your builds

 ## Customizing

@ -80,24 +80,25 @@ Following inputs can be used as `step.with` keys:
 >   network=host
 > ```

-> `CSV` type must be a newline-delimited string
+> `CSV` type must be a comma-delimited string
 > ```yaml
 > platforms: linux/amd64,linux/arm64
 > ```

-| Name              | Type     | Description                                                                                                                                                                                     |
-|-------------------|----------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| `version`         | String   | [Buildx](https://github.com/docker/buildx) version. (eg. `v0.3.0`, `latest`, `https://github.com/docker/buildx.git#master`)                                                                     |
-| `driver`          | String   | Sets the [builder driver](https://docs.docker.com/engine/reference/commandline/buildx_create/#driver) to be used (default `docker-container`)                                                   |
-| `driver-opts`     | List     | List of additional [driver-specific options](https://docs.docker.com/engine/reference/commandline/buildx_create/#driver-opt) (eg. `image=moby/buildkit:master`)                                 |
-| `buildkitd-flags` | String   | [Flags for buildkitd](https://docs.docker.com/engine/reference/commandline/buildx_create/#buildkitd-flags) daemon (since [buildx v0.3.0](https://github.com/docker/buildx/releases/tag/v0.3.0)) |
-| `install`         | Bool     | Sets up `docker build` command as an alias to `docker buildx` (default `false`)                                                                                                                 |
-| `use`             | Bool     | Switch to this builder instance (default `true`)                                                                                                                                                |
-| `endpoint`        | String   | [Optional address for docker socket](https://docs.docker.com/engine/reference/commandline/buildx_create/#description) or context from `docker context ls`                                       |
-| `platforms`       | List/CSV | Fixed [platforms](https://docs.docker.com/engine/reference/commandline/buildx_create/#platform) for current node. If not empty, values take priority over the detected ones.                    |
-| `config`¹         | String   | [BuildKit config file](https://docs.docker.com/engine/reference/commandline/buildx_create/#config)                                                                                              |
-| `config-inline`¹  | String   | Same as `config` but inline                                                                                                                                                                     |
-| `append`          | YAML     | [Append additional nodes](docs/advanced/append-nodes.md) to the builder                                                                                                                         |
+| Name              | Type     | Default            | Description                                                                                                                                                                  |
+|-------------------|----------|--------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| `version`         | String   |                    | [Buildx](https://github.com/docker/buildx) version. (eg. `v0.3.0`, `latest`, `https://github.com/docker/buildx.git#master`)                                                  |
+| `driver`          | String   | `docker-container` | Sets the [builder driver](https://docs.docker.com/engine/reference/commandline/buildx_create/#driver) to be used                                                             |
+| `driver-opts`     | List     |                    | List of additional [driver-specific options](https://docs.docker.com/engine/reference/commandline/buildx_create/#driver-opt) (eg. `image=moby/buildkit:master`)              |
+| `buildkitd-flags` | String   |                    | [Flags for buildkitd](https://docs.docker.com/engine/reference/commandline/buildx_create/#buildkitd-flags) daemon                                                            |
+| `install`         | Bool     | `false`            | Sets up `docker build` command as an alias to `docker buildx`                                                                                                                |
+| `use`             | Bool     | `true`             | Switch to this builder instance                                                                                                                                              |
+| `endpoint`        | String   |                    | [Optional address for docker socket](https://docs.docker.com/engine/reference/commandline/buildx_create/#description) or context from `docker context ls`                    |
+| `platforms`       | List/CSV |                    | Fixed [platforms](https://docs.docker.com/engine/reference/commandline/buildx_create/#platform) for current node. If not empty, values take priority over the detected ones. |
+| `config`¹         | String   |                    | [BuildKit config file](https://docs.docker.com/engine/reference/commandline/buildx_create/#config)                                                                           |
+| `config-inline`¹  | String   |                    | Same as `config` but inline                                                                                                                                                  |
+| `append`          | YAML     |                    | [Append additional nodes](docs/advanced/append-nodes.md) to the builder                                                                                                      |
+| `cleanup`         | Bool     | `true`             | Cleanup temp files and remove builder at the end of a job                                                                                                                    |

 > * ¹ `config` and `config-inline` are mutually exclusive

@ -151,35 +152,7 @@ The following [official docker environment variables](https://docs.docker.com/en
 | `buildkit`        | String | BuildKit version           |
 | `platforms`       | String | Platforms available        |

-### BuildKit container logs
+## Contributing

-To display BuildKit container logs (when `docker-container` driver is used) you have to [enable step debug logging](https://docs.github.com/en/actions/managing-workflow-runs/enabling-debug-logging#enabling-step-debug-logging),
-or you can also enable debugging in the [setup-buildx action step](https://github.com/docker/setup-buildx-action):
-
-```yaml
-  -
-    name: Set up Docker Buildx
-    uses: docker/setup-buildx-action@v2
-    with:
-      buildkitd-flags: --debug
-```
-
-Logs will be available at the end of a job:
-
-![BuildKit container logs](.github/buildkit-container-logs.png)
-
-## Keep up-to-date with GitHub Dependabot
-
-Since [Dependabot](https://docs.github.com/en/github/administering-a-repository/keeping-your-actions-up-to-date-with-github-dependabot)
-has [native GitHub Actions support](https://docs.github.com/en/github/administering-a-repository/configuration-options-for-dependency-updates#package-ecosystem),
-to enable it on your GitHub repo all you need to do is add the `.github/dependabot.yml` file:
-
-```yaml
-version: 2
-updates:
-  # Maintain dependencies for GitHub Actions
-  - package-ecosystem: "github-actions"
-    directory: "/"
-    schedule:
-      interval: "daily"
-```
+Want to contribute? Awesome! You can find information about contributing to
+this project in the [CONTRIBUTING.md](/.github/CONTRIBUTING.md)
--- a/tests/auth.test.ts
+++ b/tests/auth.test.ts
@ -1,88 +0,0 @@
-import {describe, expect, test, beforeEach} from '@jest/globals';
-import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
-import * as auth from '../src/auth';
-
-const tmpdir = fs.mkdtempSync(path.join(os.tmpdir(), 'docker-setup-buildx-jest')).split(path.sep).join(path.posix.sep);
-const dockerConfigHome = path.join(tmpdir, '.docker');
-const credsdir = path.join(dockerConfigHome, 'buildx', 'creds');
-
-describe('setCredentials', () => {
-  beforeEach(() => {
-    process.env = Object.keys(process.env).reduce((object, key) => {
-      if (!key.startsWith(auth.envPrefix)) {
-        object[key] = process.env[key];
-      }
-      return object;
-    }, {});
-  });
-
-  // prettier-ignore
-  test.each([
-    [
-      'mycontext',
-      'docker-container',
-      {},
-      [],
-      []
-    ],
-    [
-      'docker-container://mycontainer',
-      'docker-container',
-      {},
-      [],
-      []
-    ],
-    [
-      'tcp://graviton2:1234',
-      'remote',
-      {},
-      [],
-      []
-    ],
-    [
-      'tcp://graviton2:1234',
-      'remote',
-      {
-        'BUILDER_NODE_0_AUTH_TLS_CACERT': 'foo',
-        'BUILDER_NODE_0_AUTH_TLS_CERT': 'foo',
-        'BUILDER_NODE_0_AUTH_TLS_KEY': 'foo'
-      },
-      [
-        path.join(credsdir, 'cacert_graviton2-1234.pem'),
-        path.join(credsdir, 'cert_graviton2-1234.pem'),
-        path.join(credsdir, 'key_graviton2-1234.pem')
-      ],
-      [
-        `cacert=${path.join(credsdir, 'cacert_graviton2-1234.pem')}`,
-        `cert=${path.join(credsdir, 'cert_graviton2-1234.pem')}`,
-        `key=${path.join(credsdir, 'key_graviton2-1234.pem')}`
-      ]
-    ],
-    [
-      'tcp://graviton2:1234',
-      'docker-container',
-      {
-        'BUILDER_NODE_0_AUTH_TLS_CACERT': 'foo',
-        'BUILDER_NODE_0_AUTH_TLS_CERT': 'foo',
-        'BUILDER_NODE_0_AUTH_TLS_KEY': 'foo'
-      },
-      [
-        path.join(credsdir, 'cacert_graviton2-1234.pem'),
-        path.join(credsdir, 'cert_graviton2-1234.pem'),
-        path.join(credsdir, 'key_graviton2-1234.pem')
-      ],
-      []
-    ],
-  ])('given %p endpoint', async (endpoint: string, driver: string, envs: Record<string, string>, expectedFiles: Array<string>, expectedOpts: Array<string>) => {
-    fs.mkdirSync(credsdir, {recursive: true});
-    for (const [key, value] of Object.entries(envs)) {
-      process.env[key] = value;
-    }
-    expect(auth.setCredentials(credsdir, 0, driver, endpoint)).toEqual(expectedOpts);
-    expectedFiles.forEach( (file) => {
-      expect(fs.existsSync(file)).toBe(true);
-    });
-  });
-});
--- a/tests/buildx.test.ts
+++ b/tests/buildx.test.ts
@ -1,261 +0,0 @@
-import {describe, expect, it, jest, test} from '@jest/globals';
-import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
-import * as buildx from '../src/buildx';
-import * as context from '../src/context';
-import * as semver from 'semver';
-import * as exec from '@actions/exec';
-
-const tmpdir = fs.mkdtempSync(path.join(os.tmpdir(), 'docker-setup-buildx-')).split(path.sep).join(path.posix.sep);
-jest.spyOn(context, 'tmpDir').mockImplementation((): string => {
-  return tmpdir;
-});
-
-const tmpname = path.join(tmpdir, '.tmpname').split(path.sep).join(path.posix.sep);
-jest.spyOn(context, 'tmpNameSync').mockImplementation((): string => {
-  return tmpname;
-});
-
-describe('isAvailable', () => {
-  const execSpy = jest.spyOn(exec, 'getExecOutput');
-  buildx.isAvailable();
-
-  // eslint-disable-next-line jest/no-standalone-expect
-  expect(execSpy).toHaveBeenCalledWith(`docker`, ['buildx'], {
-    silent: true,
-    ignoreReturnCode: true
-  });
-});
-
-describe('isAvailable standalone', () => {
-  const execSpy = jest.spyOn(exec, 'getExecOutput');
-  buildx.isAvailable(true);
-
-  // eslint-disable-next-line jest/no-standalone-expect
-  expect(execSpy).toHaveBeenCalledWith(`buildx`, [], {
-    silent: true,
-    ignoreReturnCode: true
-  });
-});
-
-describe('getVersion', () => {
-  it('valid', async () => {
-    const version = await buildx.getVersion();
-    expect(semver.valid(version)).not.toBeNull();
-  });
-});
-
-describe('parseVersion', () => {
-  test.each([
-    ['github.com/docker/buildx 0.4.1+azure bda4882a65349ca359216b135896bddc1d92461c', '0.4.1'],
-    ['github.com/docker/buildx v0.4.1 bda4882a65349ca359216b135896bddc1d92461c', '0.4.1'],
-    ['github.com/docker/buildx v0.4.2 fb7b670b764764dc4716df3eba07ffdae4cc47b2', '0.4.2'],
-    ['github.com/docker/buildx f117971 f11797113e5a9b86bd976329c5dbb8a8bfdfadfa', 'f117971']
-  ])('given %p', async (stdout, expected) => {
-    expect(buildx.parseVersion(stdout)).toEqual(expected);
-  });
-});
-
-describe('satisfies', () => {
-  test.each([
-    ['0.4.1', '>=0.3.2', true],
-    ['bda4882a65349ca359216b135896bddc1d92461c', '>0.1.0', false],
-    ['f117971', '>0.6.0', true]
-  ])('given %p', async (version, range, expected) => {
-    expect(buildx.satisfies(version, range)).toBe(expected);
-  });
-});
-
-describe('inspect', () => {
-  it('valid', async () => {
-    const builder = await buildx.inspect('');
-    expect(builder).not.toBeUndefined();
-    expect(builder.name).not.toEqual('');
-    expect(builder.driver).not.toEqual('');
-    expect(builder.nodes).not.toEqual({});
-  }, 100000);
-});
-
-describe('parseInspect', () => {
-  // prettier-ignore
-  test.each([
-    [
-     'inspect1.txt',
-     {
-       "nodes": [
-         {
-           "name": "builder-5cb467f7-0940-47e1-b94b-d51f54054d620",
-           "endpoint": "unix:///var/run/docker.sock",
-           "status": "running",
-           "buildkitd-flags": "--allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host",
-           "buildkit": "v0.10.4",
-           "platforms": "linux/amd64,linux/amd64/v2,linux/amd64/v3,linux/amd64/v4,linux/arm64,linux/riscv64,linux/386,linux/arm/v7,linux/arm/v6"
-         }
-       ],
-       "name": "builder-5cb467f7-0940-47e1-b94b-d51f54054d62",
-       "driver": "docker-container"
-     }
-    ],
-    [
-     'inspect2.txt',
-     {
-       "nodes": [
-         {
-           "name": "builder-5f449644-ff29-48af-8344-abb0292d06730",
-           "endpoint": "unix:///var/run/docker.sock",
-           "driver-opts": [
-             "image=moby/buildkit:latest"
-           ],
-           "status": "running",
-           "buildkitd-flags": "--allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host",
-           "buildkit": "v0.10.4",
-           "platforms": "linux/amd64,linux/amd64/v2,linux/amd64/v3,linux/amd64/v4,linux/386"
-         }
-       ],
-       "name": "builder-5f449644-ff29-48af-8344-abb0292d0673",
-       "driver": "docker-container"
-     }
-    ],
-    [
-     'inspect3.txt',
-     {
-       "nodes": [
-         {
-           "name": "builder-9929e463-7954-4dc3-89cd-514cca29ff800",
-           "endpoint": "unix:///var/run/docker.sock",
-           "driver-opts": [
-             "image=moby/buildkit:master",
-             "network=host"
-           ],
-           "status": "running",
-           "buildkitd-flags": "--allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host",
-           "buildkit": "3fab389",
-           "platforms": "linux/amd64,linux/amd64/v2,linux/amd64/v3,linux/amd64/v4,linux/386"
-         }
-       ],
-       "name": "builder-9929e463-7954-4dc3-89cd-514cca29ff80",
-       "driver": "docker-container"
-     }
-    ],
-    [
-     'inspect4.txt',
-     {
-       "nodes": [
-         {
-           "name": "default",
-           "endpoint": "default",
-           "status": "running",
-           "buildkit": "20.10.17",
-           "platforms": "linux/amd64,linux/arm64,linux/riscv64,linux/ppc64le,linux/s390x,linux/386,linux/arm/v7,linux/arm/v6"
-         }
-       ],
-       "name": "default",
-       "driver": "docker"
-     }
-    ],
-    [
-     'inspect5.txt',
-     {
-       "nodes": [
-         {
-           "name": "aws_graviton2",
-           "endpoint": "tcp://1.23.45.67:1234",
-           "driver-opts": [
-             "cert=/home/user/.certs/aws_graviton2/cert.pem",
-             "key=/home/user/.certs/aws_graviton2/key.pem",
-             "cacert=/home/user/.certs/aws_graviton2/ca.pem"
-           ],
-           "status": "running",
-           "platforms": "darwin/arm64,linux/arm64,linux/arm/v5,linux/arm/v6,linux/arm/v7,windows/arm64"
-         }
-       ],
-       "name": "remote-builder",
-       "driver": "remote"
-     }
-    ],
-    [
-     'inspect6.txt',
-     {
-       "nodes": [
-         {
-           "name": "builder-17cfff01-48d9-4c3d-9332-9992e308a5100",
-           "endpoint": "unix:///var/run/docker.sock",
-           "status": "running",
-           "buildkitd-flags": "--allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host",
-           "platforms": "linux/amd64,linux/amd64/v2,linux/amd64/v3,linux/386"
-         }
-       ],
-       "name": "builder-17cfff01-48d9-4c3d-9332-9992e308a510",
-       "driver": "docker-container"
-     }
-    ],
-  ])('given %p', async (inspectFile, expected) => {
-    expect(await buildx.parseInspect(fs.readFileSync(path.join(__dirname, 'fixtures', inspectFile)).toString())).toEqual(expected);
-  });
-});
-
-describe('build', () => {
-  const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'setup-buildx-'));
-
-  // eslint-disable-next-line jest/no-disabled-tests
-  it.skip('builds refs/pull/648/head', async () => {
-    const buildxBin = await buildx.build('https://github.com/docker/buildx.git#refs/pull/648/head', tmpDir, false);
-    expect(fs.existsSync(buildxBin)).toBe(true);
-  }, 100000);
-
-  // eslint-disable-next-line jest/no-disabled-tests
-  it.skip('builds 67bd6f4dc82a9cd96f34133dab3f6f7af803bb14', async () => {
-    const buildxBin = await buildx.build('https://github.com/docker/buildx.git#67bd6f4dc82a9cd96f34133dab3f6f7af803bb14', tmpDir, false);
-    expect(fs.existsSync(buildxBin)).toBe(true);
-  }, 100000);
-});
-
-describe('install', () => {
-  const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'setup-buildx-'));
-  test.each([
-    ['v0.4.1', false],
-    ['latest', false],
-    ['v0.4.1', true],
-    ['latest', true]
-  ])(
-    'acquires %p of buildx (standalone: %p)',
-    async (version, standalone) => {
-      const buildxBin = await buildx.install(version, tmpDir, standalone);
-      expect(fs.existsSync(buildxBin)).toBe(true);
-    },
-    100000
-  );
-});
-
-describe('getConfig', () => {
-  test.each([
-    ['debug = true', false, 'debug = true', false],
-    [`notfound.toml`, true, '', true],
-    [
-      `${path.join(__dirname, 'fixtures', 'buildkitd.toml').split(path.sep).join(path.posix.sep)}`,
-      true,
-      `debug = true
-[registry."docker.io"]
-  mirrors = ["mirror.gcr.io"]
-`,
-      false
-    ]
-  ])('given %p config', async (val, file, exValue, invalid) => {
-    try {
-      let config: string;
-      if (file) {
-        config = await buildx.getConfigFile(val);
-      } else {
-        config = await buildx.getConfigInline(val);
-      }
-      expect(true).toBe(!invalid);
-      expect(config).toEqual(tmpname);
-      const configValue = fs.readFileSync(tmpname, 'utf-8');
-      expect(configValue).toEqual(exValue);
-    } catch (err) {
-      // eslint-disable-next-line jest/no-conditional-expect
-      expect(true).toBe(invalid);
-    }
-  });
-});
--- a/tests/context.test.ts
+++ b/tests/context.test.ts
@ -1,23 +1,19 @@
-import {beforeEach, describe, expect, it, jest, test} from '@jest/globals';
-import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
+import {beforeEach, describe, expect, jest, test} from '@jest/globals';
 import * as uuid from 'uuid';
+import {Buildx} from '@docker/actions-toolkit/lib/buildx/buildx';
+import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
+import {Toolkit} from '@docker/actions-toolkit/lib/toolkit';
+import {Node} from '@docker/actions-toolkit/lib/types/builder';
+
 import * as context from '../src/context';
-import * as nodes from '../src/nodes';
-
-const tmpdir = fs.mkdtempSync(path.join(os.tmpdir(), 'docker-setup-buildx-')).split(path.sep).join(path.posix.sep);
-jest.spyOn(context, 'tmpDir').mockImplementation((): string => {
-  return tmpdir;
-});
-
-jest.spyOn(context, 'tmpNameSync').mockImplementation((): string => {
-  return path.join(tmpdir, '.tmpname').split(path.sep).join(path.posix.sep);
-});

 jest.mock('uuid');
 jest.spyOn(uuid, 'v4').mockReturnValue('9b1deb4d-3b7d-4bad-9bdd-2b0d7b3dcb6d');

+jest.spyOn(Docker, 'context').mockImplementation((): Promise<string> => {
+  return Promise.resolve('default');
+});
+
 describe('getCreateArgs', () => {
  beforeEach(() => {
    process.env = Object.keys(process.env).reduce((object, key) => {
@ -32,9 +28,11 @@ describe('getCreateArgs', () => {
  test.each([
    [
      0,
+      'v0.10.3',
      new Map<string, string>([
        ['install', 'false'],
        ['use', 'true'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -46,10 +44,12 @@ describe('getCreateArgs', () => {
    ],
    [
      1,
+      'v0.10.3',
      new Map<string, string>([
        ['driver', 'docker'],
        ['install', 'false'],
        ['use', 'true'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -61,10 +61,12 @@ describe('getCreateArgs', () => {
    ],
    [
      2,
+      'v0.10.3',
      new Map<string, string>([
        ['install', 'false'],
        ['use', 'false'],
        ['driver-opts', 'image=moby/buildkit:master\nnetwork=host'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -77,11 +79,13 @@ describe('getCreateArgs', () => {
    ],
    [
      3,
+      'v0.10.3',
      new Map<string, string>([
        ['driver', 'remote'],
        ['endpoint', 'tls://foo:1234'],
        ['install', 'false'],
        ['use', 'true'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -93,12 +97,14 @@ describe('getCreateArgs', () => {
    ],
    [
      4,
+      'v0.10.3',
      new Map<string, string>([
        ['driver', 'remote'],
        ['platforms', 'linux/arm64,linux/arm/v7'],
        ['endpoint', 'tls://foo:1234'],
        ['install', 'false'],
        ['use', 'true'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -111,10 +117,12 @@ describe('getCreateArgs', () => {
    ],
    [
      5,
+      'v0.10.3',
      new Map<string, string>([
        ['install', 'false'],
        ['use', 'false'],
        ['driver-opts', `"env.no_proxy=localhost,127.0.0.1,.mydomain"`],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -126,10 +134,12 @@ describe('getCreateArgs', () => {
    ],
    [
      6,
+      'v0.10.3',
      new Map<string, string>([
        ['install', 'false'],
        ['use', 'false'],
        ['platforms', 'linux/amd64\n"linux/arm64,linux/arm/v7"'],
+        ['cleanup', 'true'],
      ]),
      [
        'create',
@ -139,14 +149,33 @@ describe('getCreateArgs', () => {
        '--platform', 'linux/amd64,linux/arm64,linux/arm/v7'
      ]
    ],
+    [
+      7,
+      'v0.10.3',
+      new Map<string, string>([
+        ['install', 'false'],
+        ['use', 'false'],
+        ['driver', 'unknown'],
+        ['cleanup', 'true'],
+      ]),
+      [
+        'create',
+        '--name', 'builder-9b1deb4d-3b7d-4bad-9bdd-2b0d7b3dcb6d',
+        '--driver', 'unknown',
+      ]
+    ]
  ])(
-    '[%d] given %p as inputs, returns %p',
-    async (num: number, inputs: Map<string, string>, expected: Array<string>) => {
+    '[%d] given buildx %s and %p as inputs, returns %p',
+    async (num: number, buildxVersion: string, inputs: Map<string, string>, expected: Array<string>) => {
      inputs.forEach((value: string, name: string) => {
        setInput(name, value);
      });
+      const toolkit = new Toolkit();
+      jest.spyOn(Buildx.prototype, 'version').mockImplementation(async (): Promise<string> => {
+        return buildxVersion;
+      });
      const inp = await context.getInputs();
-      const res = await context.getCreateArgs(inp, '0.9.0');
+      const res = await context.getCreateArgs(inp, toolkit);
      expect(res).toEqual(expected);
    }
  );
@ -166,9 +195,11 @@ describe('getAppendArgs', () => {
  test.each([
    [
      0,
+      'v0.10.3',
      new Map<string, string>([
        ['install', 'false'],
        ['use', 'true'],
+        ['cleanup', 'true'],
      ]),
      {
        "name": "aws_graviton2",
@ -191,87 +222,22 @@ describe('getAppendArgs', () => {
      ]
    ]
  ])(
-    '[%d] given %p as inputs, returns %p',
-    async (num: number, inputs: Map<string, string>, node: nodes.Node, expected: Array<string>) => {
+    '[%d] given buildx %s and %p as inputs, returns %p',
+    async (num: number, buildxVersion: string, inputs: Map<string, string>, node: Node, expected: Array<string>) => {
      inputs.forEach((value: string, name: string) => {
        setInput(name, value);
      });
+      const toolkit = new Toolkit();
+      jest.spyOn(Buildx.prototype, 'version').mockImplementation(async (): Promise<string> => {
+        return buildxVersion;
+      });
      const inp = await context.getInputs();
-      const res = await context.getAppendArgs(inp, node, '0.9.0');
+      const res = await context.getAppendArgs(inp, node, toolkit);
      expect(res).toEqual(expected);
    }
  );
 });

-describe('getInputList', () => {
-  it('handles single line correctly', async () => {
-    await setInput('foo', 'bar');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar']);
-  });
-
-  it('handles multiple lines correctly', async () => {
-    setInput('foo', 'bar\nbaz');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz']);
-  });
-
-  it('remove empty lines correctly', async () => {
-    setInput('foo', 'bar\n\nbaz');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz']);
-  });
-
-  it('handles comma correctly', async () => {
-    setInput('foo', 'bar,baz');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz']);
-  });
-
-  it('remove empty result correctly', async () => {
-    setInput('foo', 'bar,baz,');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz']);
-  });
-
-  it('handles different new lines correctly', async () => {
-    setInput('foo', 'bar\r\nbaz');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz']);
-  });
-
-  it('handles different new lines and comma correctly', async () => {
-    setInput('foo', 'bar\r\nbaz,bat');
-    const res = await context.getInputList('foo');
-    expect(res).toEqual(['bar', 'baz', 'bat']);
-  });
-
-  it('handles multiple lines and ignoring comma correctly', async () => {
-    setInput('driver-opts', 'image=moby/buildkit:master\nnetwork=host');
-    const res = await context.getInputList('driver-opts', true);
-    expect(res).toEqual(['image=moby/buildkit:master', 'network=host']);
-  });
-
-  it('handles different new lines and ignoring comma correctly', async () => {
-    setInput('driver-opts', 'image=moby/buildkit:master\r\nnetwork=host');
-    const res = await context.getInputList('driver-opts', true);
-    expect(res).toEqual(['image=moby/buildkit:master', 'network=host']);
-  });
-});
-
-describe('asyncForEach', () => {
-  it('executes async tasks sequentially', async () => {
-    const testValues = [1, 2, 3, 4, 5];
-    const results: number[] = [];
-
-    await context.asyncForEach(testValues, async value => {
-      results.push(value);
-    });
-
-    expect(results).toEqual(testValues);
-  });
-});
-
 // See: https://github.com/actions/toolkit/blob/master/packages/core/src/core.ts#L67
 function getInputName(name: string): string {
  return `INPUT_${name.replace(/ /g, '_').toUpperCase()}`;
--- a/tests/docker.test.ts
+++ b/tests/docker.test.ts
@ -1,16 +0,0 @@
-import {describe, expect, it, jest} from '@jest/globals';
-import * as docker from '../src/docker';
-import * as exec from '@actions/exec';
-
-describe('isAvailable', () => {
-  it('cli', () => {
-    const execSpy = jest.spyOn(exec, 'getExecOutput');
-    docker.isAvailable();
-
-    // eslint-disable-next-line jest/no-standalone-expect
-    expect(execSpy).toHaveBeenCalledWith(`docker`, undefined, {
-      silent: true,
-      ignoreReturnCode: true
-    });
-  });
-});
--- a/tests/fixtures/buildkitd.toml
+++ b/tests/fixtures/buildkitd.toml
@ -1,3 +0,0 @@
-debug = true
-[registry."docker.io"]
-  mirrors = ["mirror.gcr.io"]
--- a/tests/fixtures/inspect1.txt
+++ b/tests/fixtures/inspect1.txt
@ -1,10 +0,0 @@
-Name:   builder-5cb467f7-0940-47e1-b94b-d51f54054d62
-Driver: docker-container
-
-Nodes:
-Name:      builder-5cb467f7-0940-47e1-b94b-d51f54054d620
-Endpoint:  unix:///var/run/docker.sock
-Status:    running
-Flags:     --allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host
-Buildkit:  v0.10.4
-Platforms: linux/amd64, linux/amd64/v2, linux/amd64/v3, linux/amd64/v4, linux/arm64, linux/riscv64, linux/386, linux/arm/v7, linux/arm/v6
--- a/tests/fixtures/inspect2.txt
+++ b/tests/fixtures/inspect2.txt
@ -1,11 +0,0 @@
-Name:   builder-5f449644-ff29-48af-8344-abb0292d0673
-Driver: docker-container
-
-Nodes:
-Name:           builder-5f449644-ff29-48af-8344-abb0292d06730
-Endpoint:       unix:///var/run/docker.sock
-Driver Options: image="moby/buildkit:latest"
-Status:         running
-Flags:          --allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host
-Buildkit:       v0.10.4
-Platforms:      linux/amd64, linux/amd64/v2, linux/amd64/v3, linux/amd64/v4, linux/386
--- a/tests/fixtures/inspect3.txt
+++ b/tests/fixtures/inspect3.txt
@ -1,11 +0,0 @@
-Name:   builder-9929e463-7954-4dc3-89cd-514cca29ff80
-Driver: docker-container
-
-Nodes:
-Name:           builder-9929e463-7954-4dc3-89cd-514cca29ff800
-Endpoint:       unix:///var/run/docker.sock
-Driver Options: image="moby/buildkit:master" network="host"
-Status:         running
-Flags:          --allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host
-Buildkit:       3fab389
-Platforms:      linux/amd64, linux/amd64/v2, linux/amd64/v3, linux/amd64/v4, linux/386
--- a/tests/fixtures/inspect4.txt
+++ b/tests/fixtures/inspect4.txt
@ -1,9 +0,0 @@
-Name:   default
-Driver: docker
-
-Nodes:
-Name:      default
-Endpoint:  default
-Status:    running
-Buildkit:  20.10.17
-Platforms: linux/amd64, linux/arm64, linux/riscv64, linux/ppc64le, linux/s390x, linux/386, linux/arm/v7, linux/arm/v6
--- a/tests/fixtures/inspect5.txt
+++ b/tests/fixtures/inspect5.txt
@ -1,9 +0,0 @@
-Name:   remote-builder
-Driver: remote
-
-Nodes:
-Name:           aws_graviton2
-Endpoint:       tcp://1.23.45.67:1234
-Driver Options: cert="/home/user/.certs/aws_graviton2/cert.pem" key="/home/user/.certs/aws_graviton2/key.pem" cacert="/home/user/.certs/aws_graviton2/ca.pem"
-Status:         running
-Platforms:      darwin/arm64*, linux/arm64*, linux/arm/v5*, linux/arm/v6*, linux/arm/v7*, windows/arm64*, linux/amd64, linux/amd64/v2, linux/riscv64, linux/ppc64le, linux/s390x, linux/mips64le, linux/mips64
--- a/tests/fixtures/inspect6.txt
+++ b/tests/fixtures/inspect6.txt
@ -1,9 +0,0 @@
-Name:   builder-17cfff01-48d9-4c3d-9332-9992e308a510
-Driver: docker-container
-
-Nodes:
-Name:      builder-17cfff01-48d9-4c3d-9332-9992e308a5100
-Endpoint:  unix:///var/run/docker.sock
-Status:    running
-Flags:     --allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host
-Platforms: linux/amd64, linux/amd64/v2, linux/amd64/v3, linux/386
--- a/tests/git.test.ts
+++ b/tests/git.test.ts
@ -1,9 +0,0 @@
-import {describe, expect, it} from '@jest/globals';
-import * as git from '../src/git';
-
-describe('git', () => {
-  it('returns git remote ref', async () => {
-    const ref: string = await git.getRemoteSha('https://github.com/docker/buildx.git', 'refs/pull/648/head');
-    expect(ref).toEqual('f11797113e5a9b86bd976329c5dbb8a8bfdfadfa');
-  });
-});
--- a/tests/github.test.ts
+++ b/tests/github.test.ts
@ -1,16 +0,0 @@
-import {describe, expect, it} from '@jest/globals';
-import * as github from '../src/github';
-
-describe('github', () => {
-  it('returns latest buildx GitHub release', async () => {
-    const release = await github.getRelease('latest');
-    expect(release).not.toBeNull();
-    expect(release?.tag_name).not.toEqual('');
-  });
-
-  it('returns v0.2.2 buildx GitHub release', async () => {
-    const release = await github.getRelease('v0.2.2');
-    expect(release).not.toBeNull();
-    expect(release?.tag_name).toEqual('v0.2.2');
-  });
-});
--- a/tests/util.test.ts
+++ b/tests/util.test.ts
@ -1,12 +0,0 @@
-import {describe, expect, test} from '@jest/globals';
-import * as util from '../src/util';
-
-describe('isValidUrl', () => {
-  test.each([
-    ['https://github.com/docker/buildx.git', true],
-    ['https://github.com/docker/buildx.git#refs/pull/648/head', true],
-    ['v0.4.1', false]
-  ])('given %p', async (url, expected) => {
-    expect(util.isValidUrl(url)).toEqual(expected);
-  });
-});
--- a/action.yml
+++ b/action.yml
@ -44,6 +44,10 @@ inputs:
  append:
    description: 'Append additional nodes to the builder'
    required: false
+  cleanup:
+    description: 'Cleanup temp files and remove builder at the end of a job'
+    default: 'true'
+    required: false

 outputs:
  name:
--- a/dev.Dockerfile
+++ b/dev.Dockerfile
@ -71,6 +71,7 @@ ENV RUNNER_TOOL_CACHE=/tmp/github_tool_cache
 RUN --mount=type=bind,target=.,rw \
  --mount=type=cache,target=/src/node_modules \
  --mount=type=bind,from=docker,source=/usr/local/bin/docker,target=/usr/bin/docker \
+  --mount=type=bind,from=buildx,source=/buildx,target=/usr/bin/buildx \
  --mount=type=bind,from=buildx,source=/buildx,target=/usr/libexec/docker/cli-plugins/docker-buildx \
  yarn run test --coverageDirectory=/tmp/coverage

--- a/dist/index.js
+++ b/dist/index.js
--- a/dist/index.js.map
+++ b/dist/index.js.map
--- a/dist/licenses.txt
+++ b/dist/licenses.txt
--- a/dist/sourcemap-register.js
+++ b/dist/sourcemap-register.js
--- a/docs/advanced/append-nodes.md
+++ b/docs/advanced/append-nodes.md
@ -1,56 +1,3 @@
 # Append additional nodes to the builder

-Buildx also supports running builds on multiple machines. This is useful for
-building [multi-platform images](https://docs.docker.com/build/building/multi-platform/)
-on native nodes for more complicated cases that are not handled by QEMU and
-generally have better performance or for distributing the build across multiple
-machines.
-
-You can append nodes to the builder that is going to be created with the
-`append` input in the form of a YAML string document to remove limitations
-intrinsically linked to GitHub Actions (only string format is handled in the
-input fields):
-
-| Name              | Type   | Description                                                                                                                                                                                                                                                                           |
-|-------------------|--------|---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| `name`            | String | [Name of the node](https://docs.docker.com/engine/reference/commandline/buildx_create/#node). If empty, it is the name of the builder it belongs to, with an index number suffix. This is useful to set it if you want to modify/remove a node in an underlying step of you workflow. |
-| `endpoint`        | String | [Docker context or endpoint](https://docs.docker.com/engine/reference/commandline/buildx_create/#description) of the node to add to the builder                                                                                                                                       |
-| `driver-opts`     | List   | List of additional [driver-specific options](https://docs.docker.com/engine/reference/commandline/buildx_create/#driver-opt)                                                                                                                                                          |
-| `buildkitd-flags` | String | [Flags for buildkitd](https://docs.docker.com/engine/reference/commandline/buildx_create/#buildkitd-flags) daemon                                                                                                                                                                     |
-| `platforms`       | String | Fixed [platforms](https://docs.docker.com/engine/reference/commandline/buildx_create/#platform) for the node. If not empty, values take priority over the detected ones.                                                                                                              |
-
-Here is an example using remote nodes with the [`remote` driver](https://docs.docker.com/build/building/drivers/remote/)
-and [TLS authentication](auth.md#tls-authentication):
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          driver: remote
-          endpoint: tcp://oneprovider:1234
-          append: |
-            - endpoint: tcp://graviton2:1234
-              platforms: linux/arm64
-            - endpoint: tcp://linuxone:1234
-              platforms: linux/s390x
-        env:
-          BUILDER_NODE_0_AUTH_TLS_CACERT: ${{ secrets.ONEPROVIDER_CA }}
-          BUILDER_NODE_0_AUTH_TLS_CERT: ${{ secrets.ONEPROVIDER_CERT }}
-          BUILDER_NODE_0_AUTH_TLS_KEY: ${{ secrets.ONEPROVIDER_KEY }}
-          BUILDER_NODE_1_AUTH_TLS_CACERT: ${{ secrets.GRAVITON2_CA }}
-          BUILDER_NODE_1_AUTH_TLS_CERT: ${{ secrets.GRAVITON2_CERT }}
-          BUILDER_NODE_1_AUTH_TLS_KEY: ${{ secrets.GRAVITON2_KEY }}
-          BUILDER_NODE_2_AUTH_TLS_CACERT: ${{ secrets.LINUXONE_CA }}
-          BUILDER_NODE_2_AUTH_TLS_CERT: ${{ secrets.LINUXONE_CERT }}
-          BUILDER_NODE_2_AUTH_TLS_KEY: ${{ secrets.LINUXONE_KEY }}
-```
+This page has moved to [Docker Docs website](https://docs.docker.com/build/ci/github-actions/configure-builder/#append-additional-nodes-to-the-builder)
--- a/docs/advanced/auth.md
+++ b/docs/advanced/auth.md
@ -1,64 +1,3 @@
 # Authentication support

-## SSH authentication
-
-To be able to connect to an SSH endpoint using the [`docker-container` driver](https://docs.docker.com/build/building/drivers/docker-container/),
-you have to set up the SSH private key and configuration on the GitHub Runner:
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Set up SSH
-        uses: MrSquaare/ssh-setup-action@523473d91581ccbf89565e12b40faba93f2708bd # v1.1.0
-        with:
-          host: graviton2
-          private-key: ${{ secrets.SSH_PRIVATE_KEY }}
-          private-key-name: aws_graviton2
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          endpoint: ssh://me@graviton2
-```
-
-## TLS authentication
-
-You can also [set up a remote BuildKit instance](https://docs.docker.com/build/building/drivers/remote/#remote-buildkit-in-docker-container)
-using the remote driver. To ease the integration in your workflow, we put in
-place environment variables that will set up authentication using the BuildKit
-client certificates for the `tcp://` endpoint where `<idx>` is the position of
-the node in the list of nodes:
-
-* `BUILDER_NODE_<idx>_AUTH_TLS_CACERT`
-* `BUILDER_NODE_<idx>_AUTH_TLS_CERT`
-* `BUILDER_NODE_<idx>_AUTH_TLS_KEY`
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          driver: remote
-          endpoint: tcp://graviton2:1234
-        env:
-          BUILDER_NODE_0_AUTH_TLS_CACERT: ${{ secrets.GRAVITON2_CA }}
-          BUILDER_NODE_0_AUTH_TLS_CERT: ${{ secrets.GRAVITON2_CERT }}
-          BUILDER_NODE_0_AUTH_TLS_KEY: ${{ secrets.GRAVITON2_KEY }}
-```
+This page has moved to [Docker Docs website](https://docs.docker.com/build/ci/github-actions/configure-builder/#authentication-for-remote-builders)
--- a/docs/advanced/buildkit-config.md
+++ b/docs/advanced/buildkit-config.md
@ -1,67 +1,3 @@
 # BuildKit daemon configuration

-You can provide a [BuildKit configuration](https://github.com/moby/buildkit/blob/master/docs/buildkitd.toml.md)
-to your builder if you're using the [`docker-container` driver](https://docs.docker.com/build/building/drivers/docker-container/)
-(default) with the `config` or `config-inline` inputs:
-
-## Registry mirror
-
-You can configure a registry mirror using an inline block directly in your
-workflow with the `config-inline` input:
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Checkout
-        uses: actions/checkout@v3
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          config-inline: |
-            [registry."docker.io"]
-              mirrors = ["mirror.gcr.io"]
-```
-
-## Max parallelism
-
-You can limit the parallelism of the BuildKit solver which is particularly
-useful for low-powered machines.
-
-You can use the `config-inline` input like the previous example, or you can use
-a dedicated BuildKit config file from your repo if you want with the
-`config` input:
-
-```toml
-# .github/buildkitd.toml
-[worker.oci]
-  max-parallelism = 4
-```
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Checkout
-        uses: actions/checkout@v3
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          config: .github/buildkitd.toml
-```
+This page has moved to [Docker Docs website](https://docs.docker.com/build/ci/github-actions/configure-builder/#buildkit-daemon-configuration)
--- a/docs/advanced/install-default.md
+++ b/docs/advanced/install-default.md
@ -1,28 +0,0 @@
-# Install by default
-
-If you want set up the `docker build` command as an alias to
-`docker buildx build`:
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Checkout
-        uses: actions/checkout@v3
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          install: true
-      -
-        name: Build
-        run: |
-          docker build . # will run buildx
-```
--- a/docs/advanced/standalone.md
+++ b/docs/advanced/standalone.md
@ -1,29 +1,3 @@
 # Standalone mode

-If you don't have the Docker CLI installed on the GitHub Runner, Buildx binary
-is invoked directly, instead of calling it as a docker plugin. This can be
-useful if you want to use the `kubernetes` driver in your self-hosted runner:
-
-```yaml
-name: ci
-
-on:
-  push:
-
-jobs:
-  buildx:
-    runs-on: ubuntu-latest
-    steps:
-      -
-        name: Checkout
-        uses: actions/checkout@v3
-      -
-        name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v2
-        with:
-          driver: kubernetes
-      -
-        name: Build
-        run: |
-          buildx build .
-```
+This page has moved to [Docker Docs website](https://docs.docker.com/build/ci/github-actions/configure-builder/#standalone-mode)
--- a/jest.config.ts
+++ b/jest.config.ts
@ -1,7 +1,21 @@
+import fs from 'fs';
+import os from 'os';
+import path from 'path';
+
+const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'docker-setup-buildx-action-')).split(path.sep).join(path.posix.sep);
+
+process.env = Object.assign({}, process.env, {
+  TEMP: tmpDir,
+  GITHUB_REPOSITORY: 'docker/setup-buildx-action',
+  RUNNER_TEMP: path.join(tmpDir, 'runner-temp').split(path.sep).join(path.posix.sep),
+  RUNNER_TOOL_CACHE: path.join(tmpDir, 'runner-tool-cache').split(path.sep).join(path.posix.sep)
+}) as {
+  [key: string]: string;
+};
+
 module.exports = {
  clearMocks: true,
  moduleFileExtensions: ['js', 'ts'],
-  setupFiles: ['dotenv/config'],
  testMatch: ['**/*.test.ts'],
  transform: {
    '^.+\\.ts$': 'ts-jest'
@ -9,5 +23,7 @@ module.exports = {
  moduleNameMapper: {
    '^csv-parse/sync': '<rootDir>/node_modules/csv-parse/dist/cjs/sync.cjs'
  },
+  collectCoverageFrom: ['src/**/{!(main.ts),}.ts'],
+  coveragePathIgnorePatterns: ['lib/', 'node_modules/', '__tests__/'],
  verbose: true
 };
--- a/package.json
+++ b/package.json
@ -29,30 +29,23 @@
  "dependencies": {
    "@actions/core": "^1.10.0",
    "@actions/exec": "^1.1.1",
-    "@actions/http-client": "^2.0.1",
-    "@actions/tool-cache": "^2.0.1",
-    "csv-parse": "^5.3.1",
+    "@docker/actions-toolkit": "^0.10.0",
    "js-yaml": "^4.1.0",
-    "semver": "^7.3.7",
-    "tmp": "^0.2.1",
    "uuid": "^9.0.0"
  },
  "devDependencies": {
-    "@types/node": "^16.11.26",
-    "@types/semver": "^7.3.9",
-    "@types/tmp": "^0.2.3",
-    "@typescript-eslint/eslint-plugin": "^5.14.0",
-    "@typescript-eslint/parser": "^5.14.0",
-    "@vercel/ncc": "^0.33.3",
-    "dotenv": "^16.0.0",
-    "eslint": "^8.11.0",
-    "eslint-config-prettier": "^8.5.0",
-    "eslint-plugin-jest": "^26.1.1",
-    "eslint-plugin-prettier": "^4.0.0",
-    "jest": "^27.2.5",
-    "prettier": "^2.3.1",
-    "ts-jest": "^27.1.2",
-    "ts-node": "^10.7.0",
-    "typescript": "^4.4.4"
+    "@types/node": "^16.18.21",
+    "@typescript-eslint/eslint-plugin": "^5.56.0",
+    "@typescript-eslint/parser": "^5.56.0",
+    "@vercel/ncc": "^0.36.1",
+    "eslint": "^8.36.0",
+    "eslint-config-prettier": "^8.8.0",
+    "eslint-plugin-jest": "^27.2.1",
+    "eslint-plugin-prettier": "^4.2.1",
+    "jest": "^29.5.0",
+    "prettier": "^2.8.7",
+    "ts-jest": "^29.0.5",
+    "ts-node": "^10.9.1",
+    "typescript": "^4.9.5"
  }
 }
--- a/src/auth.ts
+++ b/src/auth.ts
@ -1,51 +0,0 @@
-import * as fs from 'fs';
-
-export const envPrefix = 'BUILDER_NODE';
-
-export function setCredentials(credsdir: string, index: number, driver: string, endpoint: string): Array<string> {
-  let url: URL;
-  try {
-    url = new URL(endpoint);
-  } catch (e) {
-    return [];
-  }
-  switch (url.protocol) {
-    case 'tcp:': {
-      return setBuildKitClientCerts(credsdir, index, driver, url);
-    }
-  }
-  return [];
-}
-
-function setBuildKitClientCerts(credsdir: string, index: number, driver: string, endpoint: URL): Array<string> {
-  const driverOpts: Array<string> = [];
-  const buildkitCacert = process.env[`${envPrefix}_${index}_AUTH_TLS_CACERT`] || '';
-  const buildkitCert = process.env[`${envPrefix}_${index}_AUTH_TLS_CERT`] || '';
-  const buildkitKey = process.env[`${envPrefix}_${index}_AUTH_TLS_KEY`] || '';
-  if (buildkitCacert.length == 0 && buildkitCert.length == 0 && buildkitKey.length == 0) {
-    return driverOpts;
-  }
-  let host = endpoint.hostname;
-  if (endpoint.port.length > 0) {
-    host += `-${endpoint.port}`;
-  }
-  if (buildkitCacert.length > 0) {
-    const cacertpath = `${credsdir}/cacert_${host}.pem`;
-    fs.writeFileSync(cacertpath, buildkitCacert);
-    driverOpts.push(`cacert=${cacertpath}`);
-  }
-  if (buildkitCert.length > 0) {
-    const certpath = `${credsdir}/cert_${host}.pem`;
-    fs.writeFileSync(certpath, buildkitCert);
-    driverOpts.push(`cert=${certpath}`);
-  }
-  if (buildkitKey.length > 0) {
-    const keypath = `${credsdir}/key_${host}.pem`;
-    fs.writeFileSync(keypath, buildkitKey);
-    driverOpts.push(`key=${keypath}`);
-  }
-  if (driver != 'remote') {
-    return [];
-  }
-  return driverOpts;
-}
--- a/src/buildx.ts
+++ b/src/buildx.ts
@ -1,378 +0,0 @@
-import * as fs from 'fs';
-import * as path from 'path';
-import * as semver from 'semver';
-import * as util from 'util';
-import * as context from './context';
-import * as git from './git';
-import * as github from './github';
-import * as core from '@actions/core';
-import * as exec from '@actions/exec';
-import * as tc from '@actions/tool-cache';
-
-export type Builder = {
-  name?: string;
-  driver?: string;
-  nodes: Node[];
-};
-
-export type Node = {
-  name?: string;
-  endpoint?: string;
-  'driver-opts'?: Array<string>;
-  status?: string;
-  'buildkitd-flags'?: string;
-  buildkit?: string;
-  platforms?: string;
-};
-
-export async function getConfigInline(s: string): Promise<string> {
-  return getConfig(s, false);
-}
-
-export async function getConfigFile(s: string): Promise<string> {
-  return getConfig(s, true);
-}
-
-export async function getConfig(s: string, file: boolean): Promise<string> {
-  if (file) {
-    if (!fs.existsSync(s)) {
-      throw new Error(`config file ${s} not found`);
-    }
-    s = fs.readFileSync(s, {encoding: 'utf-8'});
-  }
-  const configFile = context.tmpNameSync({
-    tmpdir: context.tmpDir()
-  });
-  fs.writeFileSync(configFile, s);
-  return configFile;
-}
-
-export async function isAvailable(standalone?: boolean): Promise<boolean> {
-  const cmd = getCommand([], standalone);
-  return await exec
-    .getExecOutput(cmd.commandLine, cmd.args, {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(res => {
-      if (res.stderr.length > 0 && res.exitCode != 0) {
-        return false;
-      }
-      return res.exitCode == 0;
-    })
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    .catch(error => {
-      return false;
-    });
-}
-
-export async function getVersion(standalone?: boolean): Promise<string> {
-  const cmd = getCommand(['version'], standalone);
-  return await exec
-    .getExecOutput(cmd.commandLine, cmd.args, {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(res => {
-      if (res.stderr.length > 0 && res.exitCode != 0) {
-        throw new Error(res.stderr.trim());
-      }
-      return parseVersion(res.stdout.trim());
-    });
-}
-
-export function parseVersion(stdout: string): string {
-  const matches = /\sv?([0-9a-f]{7}|[0-9.]+)/.exec(stdout);
-  if (!matches) {
-    throw new Error(`Cannot parse buildx version`);
-  }
-  return matches[1];
-}
-
-export function satisfies(version: string, range: string): boolean {
-  return semver.satisfies(version, range) || /^[0-9a-f]{7}$/.exec(version) !== null;
-}
-
-export async function inspect(name: string, standalone?: boolean): Promise<Builder> {
-  const cmd = getCommand(['inspect', name], standalone);
-  return await exec
-    .getExecOutput(cmd.commandLine, cmd.args, {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(res => {
-      if (res.stderr.length > 0 && res.exitCode != 0) {
-        throw new Error(res.stderr.trim());
-      }
-      return parseInspect(res.stdout);
-    });
-}
-
-export async function parseInspect(data: string): Promise<Builder> {
-  const builder: Builder = {
-    nodes: []
-  };
-  let node: Node = {};
-  for (const line of data.trim().split(`\n`)) {
-    const [key, ...rest] = line.split(':');
-    const value = rest.map(v => v.trim()).join(':');
-    if (key.length == 0 || value.length == 0) {
-      continue;
-    }
-    switch (key.toLowerCase()) {
-      case 'name': {
-        if (builder.name == undefined) {
-          builder.name = value;
-        } else {
-          if (Object.keys(node).length > 0) {
-            builder.nodes.push(node);
-            node = {};
-          }
-          node.name = value;
-        }
-        break;
-      }
-      case 'driver': {
-        builder.driver = value;
-        break;
-      }
-      case 'endpoint': {
-        node.endpoint = value;
-        break;
-      }
-      case 'driver options': {
-        node['driver-opts'] = (value.match(/(\w+)="([^"]*)"/g) || []).map(v => v.replace(/^(.*)="(.*)"$/g, '$1=$2'));
-        break;
-      }
-      case 'status': {
-        node.status = value;
-        break;
-      }
-      case 'flags': {
-        node['buildkitd-flags'] = value;
-        break;
-      }
-      case 'buildkit': {
-        node.buildkit = value;
-        break;
-      }
-      case 'platforms': {
-        let platforms: Array<string> = [];
-        // if a preferred platform is being set then use only these
-        // https://docs.docker.com/engine/reference/commandline/buildx_inspect/#get-information-about-a-builder-instance
-        if (value.includes('*')) {
-          for (const platform of value.split(', ')) {
-            if (platform.includes('*')) {
-              platforms.push(platform.replace('*', ''));
-            }
-          }
-        } else {
-          // otherwise set all platforms available
-          platforms = value.split(', ');
-        }
-        node.platforms = platforms.join(',');
-        break;
-      }
-    }
-  }
-  if (Object.keys(node).length > 0) {
-    builder.nodes.push(node);
-  }
-  return builder;
-}
-
-export async function build(inputBuildRef: string, dest: string, standalone: boolean): Promise<string> {
-  // eslint-disable-next-line prefer-const
-  let [repo, ref] = inputBuildRef.split('#');
-  if (ref.length == 0) {
-    ref = 'master';
-  }
-
-  let vspec: string;
-  if (ref.match(/^[0-9a-fA-F]{40}$/)) {
-    vspec = ref;
-  } else {
-    vspec = await git.getRemoteSha(repo, ref);
-  }
-  core.debug(`Tool version spec ${vspec}`);
-
-  let toolPath: string;
-  toolPath = tc.find('buildx', vspec);
-  if (!toolPath) {
-    const outFolder = path.join(context.tmpDir(), 'out').split(path.sep).join(path.posix.sep);
-    let buildWithStandalone = false;
-    const standaloneFound = await isAvailable(true);
-    const pluginFound = await isAvailable(false);
-    if (standalone && standaloneFound) {
-      core.debug(`Buildx standalone found, build with it`);
-      buildWithStandalone = true;
-    } else if (!standalone && pluginFound) {
-      core.debug(`Buildx plugin found, build with it`);
-      buildWithStandalone = false;
-    } else if (standaloneFound) {
-      core.debug(`Buildx plugin not found, but standalone found so trying to build with it`);
-      buildWithStandalone = true;
-    } else if (pluginFound) {
-      core.debug(`Buildx standalone not found, but plugin found so trying to build with it`);
-      buildWithStandalone = false;
-    } else {
-      throw new Error(`Neither buildx standalone or plugin have been found to build from ref`);
-    }
-    const buildCmd = getCommand(['build', '--target', 'binaries', '--build-arg', 'BUILDKIT_CONTEXT_KEEP_GIT_DIR=1', '--output', `type=local,dest=${outFolder}`, inputBuildRef], buildWithStandalone);
-    toolPath = await exec
-      .getExecOutput(buildCmd.commandLine, buildCmd.args, {
-        ignoreReturnCode: true
-      })
-      .then(res => {
-        if (res.stderr.length > 0 && res.exitCode != 0) {
-          core.warning(res.stderr.trim());
-        }
-        return tc.cacheFile(`${outFolder}/buildx`, context.osPlat == 'win32' ? 'docker-buildx.exe' : 'docker-buildx', 'buildx', vspec);
-      });
-  }
-
-  if (standalone) {
-    return setStandalone(toolPath, dest);
-  }
-  return setPlugin(toolPath, dest);
-}
-
-export async function install(inputVersion: string, dest: string, standalone: boolean): Promise<string> {
-  const release: github.GitHubRelease | null = await github.getRelease(inputVersion);
-  if (!release) {
-    throw new Error(`Cannot find buildx ${inputVersion} release`);
-  }
-  core.debug(`Release ${release.tag_name} found`);
-  const version = release.tag_name.replace(/^v+|v+$/g, '');
-
-  let toolPath: string;
-  toolPath = tc.find('buildx', version);
-  if (!toolPath) {
-    const c = semver.clean(version) || '';
-    if (!semver.valid(c)) {
-      throw new Error(`Invalid Buildx version "${version}".`);
-    }
-    toolPath = await download(version);
-  }
-
-  if (standalone) {
-    return setStandalone(toolPath, dest);
-  }
-  return setPlugin(toolPath, dest);
-}
-
-async function setStandalone(toolPath: string, dest: string): Promise<string> {
-  core.info('Standalone mode');
-  const toolBinPath = path.join(toolPath, context.osPlat == 'win32' ? 'docker-buildx.exe' : 'docker-buildx');
-
-  const binDir = path.join(dest, 'bin');
-  core.debug(`Bin dir is ${binDir}`);
-  if (!fs.existsSync(binDir)) {
-    fs.mkdirSync(binDir, {recursive: true});
-  }
-
-  const filename: string = context.osPlat == 'win32' ? 'buildx.exe' : 'buildx';
-  const buildxPath: string = path.join(binDir, filename);
-  core.debug(`Bin path is ${buildxPath}`);
-  fs.copyFileSync(toolBinPath, buildxPath);
-
-  core.info('Fixing perms');
-  fs.chmodSync(buildxPath, '0755');
-
-  core.addPath(binDir);
-  core.info('Added buildx to the path');
-
-  return buildxPath;
-}
-
-async function setPlugin(toolPath: string, dockerConfigHome: string): Promise<string> {
-  core.info('Docker plugin mode');
-  const toolBinPath = path.join(toolPath, context.osPlat == 'win32' ? 'docker-buildx.exe' : 'docker-buildx');
-
-  const pluginsDir: string = path.join(dockerConfigHome, 'cli-plugins');
-  core.debug(`Plugins dir is ${pluginsDir}`);
-  if (!fs.existsSync(pluginsDir)) {
-    fs.mkdirSync(pluginsDir, {recursive: true});
-  }
-
-  const filename: string = context.osPlat == 'win32' ? 'docker-buildx.exe' : 'docker-buildx';
-  const pluginPath: string = path.join(pluginsDir, filename);
-  core.debug(`Plugin path is ${pluginPath}`);
-  fs.copyFileSync(toolBinPath, pluginPath);
-
-  core.info('Fixing perms');
-  fs.chmodSync(pluginPath, '0755');
-
-  return pluginPath;
-}
-
-async function download(version: string): Promise<string> {
-  const targetFile: string = context.osPlat == 'win32' ? 'docker-buildx.exe' : 'docker-buildx';
-  const downloadUrl = util.format('https://github.com/docker/buildx/releases/download/v%s/%s', version, await filename(version));
-  core.info(`Downloading ${downloadUrl}`);
-  const downloadPath = await tc.downloadTool(downloadUrl);
-  core.debug(`Downloaded to ${downloadPath}`);
-  return await tc.cacheFile(downloadPath, targetFile, 'buildx', version);
-}
-
-async function filename(version: string): Promise<string> {
-  let arch: string;
-  switch (context.osArch) {
-    case 'x64': {
-      arch = 'amd64';
-      break;
-    }
-    case 'ppc64': {
-      arch = 'ppc64le';
-      break;
-    }
-    case 'arm': {
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      const arm_version = (process.config.variables as any).arm_version;
-      arch = arm_version ? 'arm-v' + arm_version : 'arm';
-      break;
-    }
-    default: {
-      arch = context.osArch;
-      break;
-    }
-  }
-  const platform: string = context.osPlat == 'win32' ? 'windows' : context.osPlat;
-  const ext: string = context.osPlat == 'win32' ? '.exe' : '';
-  return util.format('buildx-v%s.%s-%s%s', version, platform, arch, ext);
-}
-
-export async function getBuildKitVersion(containerID: string): Promise<string> {
-  return exec
-    .getExecOutput(`docker`, ['inspect', '--format', '{{.Config.Image}}', containerID], {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(bkitimage => {
-      if (bkitimage.exitCode == 0 && bkitimage.stdout.length > 0) {
-        return exec
-          .getExecOutput(`docker`, ['run', '--rm', bkitimage.stdout.trim(), '--version'], {
-            ignoreReturnCode: true,
-            silent: true
-          })
-          .then(bkitversion => {
-            if (bkitversion.exitCode == 0 && bkitversion.stdout.length > 0) {
-              return `${bkitimage.stdout.trim()} => ${bkitversion.stdout.trim()}`;
-            } else if (bkitversion.stderr.length > 0) {
-              core.warning(bkitversion.stderr.trim());
-            }
-            return bkitversion.stdout.trim();
-          });
-      } else if (bkitimage.stderr.length > 0) {
-        core.warning(bkitimage.stderr.trim());
-      }
-      return bkitimage.stdout.trim();
-    });
-}
-
-export function getCommand(args: Array<string>, standalone?: boolean) {
-  return {
-    commandLine: standalone ? 'buildx' : 'docker',
-    args: standalone ? args : ['buildx', ...args]
-  };
-}
--- a/src/context.ts
+++ b/src/context.ts
@ -1,27 +1,11 @@
-import fs from 'fs';
-import * as os from 'os';
-import path from 'path';
-import * as tmp from 'tmp';
 import * as uuid from 'uuid';
-import {parse} from 'csv-parse/sync';
-import * as buildx from './buildx';
-import * as nodes from './nodes';
 import * as core from '@actions/core';
+import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
+import {Util} from '@docker/actions-toolkit/lib/util';
+import {Toolkit} from '@docker/actions-toolkit/lib/toolkit';
+import {Node} from '@docker/actions-toolkit/lib/types/builder';

-let _tmpDir: string;
-export const osPlat: string = os.platform();
-export const osArch: string = os.arch();
-
-export function tmpDir(): string {
-  if (!_tmpDir) {
-    _tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'docker-setup-buildx-')).split(path.sep).join(path.posix.sep);
-  }
-  return _tmpDir;
-}
-
-export function tmpNameSync(options?: tmp.TmpNameOptions): string {
-  return tmp.tmpNameSync(options);
-}
+export const builderNodeEnvPrefix = 'BUILDER_NODE';

 export interface Inputs {
  version: string;
@ -36,36 +20,38 @@ export interface Inputs {
  config: string;
  configInline: string;
  append: string;
+  cleanup: boolean;
 }

 export async function getInputs(): Promise<Inputs> {
  return {
    version: core.getInput('version'),
-    name: getBuilderName(core.getInput('driver') || 'docker-container'),
+    name: await getBuilderName(core.getInput('driver') || 'docker-container'),
    driver: core.getInput('driver') || 'docker-container',
-    driverOpts: await getInputList('driver-opts', true),
+    driverOpts: Util.getInputList('driver-opts', {ignoreComma: true, quote: false}),
    buildkitdFlags: core.getInput('buildkitd-flags') || '--allow-insecure-entitlement security.insecure --allow-insecure-entitlement network.host',
-    platforms: await getInputList('platforms', false, true),
+    platforms: Util.getInputList('platforms'),
    install: core.getBooleanInput('install'),
    use: core.getBooleanInput('use'),
    endpoint: core.getInput('endpoint'),
    config: core.getInput('config'),
    configInline: core.getInput('config-inline'),
-    append: core.getInput('append')
+    append: core.getInput('append'),
+    cleanup: core.getBooleanInput('cleanup')
  };
 }

-export function getBuilderName(driver: string): string {
-  return driver == 'docker' ? 'default' : `builder-${uuid.v4()}`;
+export async function getBuilderName(driver: string): Promise<string> {
+  return driver == 'docker' ? await Docker.context() : `builder-${uuid.v4()}`;
 }

-export async function getCreateArgs(inputs: Inputs, buildxVersion: string): Promise<Array<string>> {
+export async function getCreateArgs(inputs: Inputs, toolkit: Toolkit): Promise<Array<string>> {
  const args: Array<string> = ['create', '--name', inputs.name, '--driver', inputs.driver];
-  if (buildx.satisfies(buildxVersion, '>=0.3.0')) {
-    await asyncForEach(inputs.driverOpts, async driverOpt => {
+  if (await toolkit.buildx.versionSatisfies('>=0.3.0')) {
+    await Util.asyncForEach(inputs.driverOpts, async driverOpt => {
      args.push('--driver-opt', driverOpt);
    });
-    if (inputs.driver != 'remote' && inputs.buildkitdFlags) {
+    if (driverSupportsFlags(inputs.driver) && inputs.buildkitdFlags) {
      args.push('--buildkitd-flags', inputs.buildkitdFlags);
    }
  }
@ -75,11 +61,11 @@ export async function getCreateArgs(inputs: Inputs, buildxVersion: string): Prom
  if (inputs.use) {
    args.push('--use');
  }
-  if (inputs.driver != 'remote') {
+  if (driverSupportsFlags(inputs.driver)) {
    if (inputs.config) {
-      args.push('--config', await buildx.getConfigFile(inputs.config));
+      args.push('--config', toolkit.buildkit.config.resolveFromFile(inputs.config));
    } else if (inputs.configInline) {
-      args.push('--config', await buildx.getConfigInline(inputs.configInline));
+      args.push('--config', toolkit.buildkit.config.resolveFromString(inputs.configInline));
    }
  }
  if (inputs.endpoint) {
@ -88,16 +74,18 @@ export async function getCreateArgs(inputs: Inputs, buildxVersion: string): Prom
  return args;
 }

-export async function getAppendArgs(inputs: Inputs, node: nodes.Node, buildxVersion: string): Promise<Array<string>> {
+export async function getAppendArgs(inputs: Inputs, node: Node, toolkit: Toolkit): Promise<Array<string>> {
  const args: Array<string> = ['create', '--name', inputs.name, '--append'];
  if (node.name) {
    args.push('--node', node.name);
+  } else if (inputs.driver == 'kubernetes' && (await toolkit.buildx.versionSatisfies('<0.11.0'))) {
+    args.push('--node', `node-${uuid.v4()}`);
  }
-  if (node['driver-opts'] && buildx.satisfies(buildxVersion, '>=0.3.0')) {
-    await asyncForEach(node['driver-opts'], async driverOpt => {
+  if (node['driver-opts'] && (await toolkit.buildx.versionSatisfies('>=0.3.0'))) {
+    await Util.asyncForEach(node['driver-opts'], async driverOpt => {
      args.push('--driver-opt', driverOpt);
    });
-    if (inputs.driver != 'remote' && node['buildkitd-flags']) {
+    if (driverSupportsFlags(inputs.driver) && node['buildkitd-flags']) {
      args.push('--buildkitd-flags', node['buildkitd-flags']);
    }
  }
@ -110,47 +98,14 @@ export async function getAppendArgs(inputs: Inputs, node: nodes.Node, buildxVers
  return args;
 }

-export async function getInspectArgs(inputs: Inputs, buildxVersion: string): Promise<Array<string>> {
+export async function getInspectArgs(inputs: Inputs, toolkit: Toolkit): Promise<Array<string>> {
  const args: Array<string> = ['inspect', '--bootstrap'];
-  if (buildx.satisfies(buildxVersion, '>=0.4.0')) {
+  if (await toolkit.buildx.versionSatisfies('>=0.4.0')) {
    args.push('--builder', inputs.name);
  }
  return args;
 }

-export async function getInputList(name: string, ignoreComma?: boolean, escapeQuotes?: boolean): Promise<string[]> {
-  const res: Array<string> = [];
-
-  const items = core.getInput(name);
-  if (items == '') {
-    return res;
-  }
-
-  const records = parse(items, {
-    columns: false,
-    relaxQuotes: true,
-    comment: '#',
-    relaxColumnCount: true,
-    skipEmptyLines: true,
-    quote: escapeQuotes ? `"` : false
-  });
-
-  for (const record of records as Array<string[]>) {
-    if (record.length == 1) {
-      res.push(record[0]);
-      continue;
-    } else if (!ignoreComma) {
-      res.push(...record);
-      continue;
-    }
-    res.push(record.join(','));
-  }
-
-  return res.filter(item => item).map(pat => pat.trim());
+function driverSupportsFlags(driver: string): boolean {
+  return driver == '' || driver == 'docker-container' || driver == 'docker' || driver == 'kubernetes';
 }
-
-export const asyncForEach = async (array, callback) => {
-  for (let index = 0; index < array.length; index++) {
-    await callback(array[index], index, array);
-  }
-};
--- a/src/docker.ts
+++ b/src/docker.ts
@ -1,19 +0,0 @@
-import * as exec from '@actions/exec';
-
-export async function isAvailable(): Promise<boolean> {
-  return await exec
-    .getExecOutput('docker', undefined, {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(res => {
-      if (res.stderr.length > 0 && res.exitCode != 0) {
-        return false;
-      }
-      return res.exitCode == 0;
-    })
-    // eslint-disable-next-line @typescript-eslint/no-unused-vars
-    .catch(error => {
-      return false;
-    });
-}
--- a/src/git.ts
+++ b/src/git.ts
@ -1,19 +0,0 @@
-import * as exec from '@actions/exec';
-
-export async function getRemoteSha(repo: string, ref: string): Promise<string> {
-  return await exec
-    .getExecOutput(`git`, ['ls-remote', repo, ref], {
-      ignoreReturnCode: true,
-      silent: true
-    })
-    .then(res => {
-      if (res.stderr.length > 0 && res.exitCode != 0) {
-        throw new Error(res.stderr);
-      }
-      const [rsha] = res.stdout.trim().split(/[\s\t]/);
-      if (rsha.length == 0) {
-        throw new Error(`Cannot find remote ref for ${repo}#${ref}`);
-      }
-      return rsha;
-    });
-}
--- a/src/github.ts
+++ b/src/github.ts
@ -1,12 +0,0 @@
-import * as httpm from '@actions/http-client';
-
-export interface GitHubRelease {
-  id: number;
-  tag_name: string;
-}
-
-export const getRelease = async (version: string): Promise<GitHubRelease | null> => {
-  const url = `https://github.com/docker/buildx/releases/${version}`;
-  const http: httpm.HttpClient = new httpm.HttpClient('setup-buildx');
-  return (await http.getJson<GitHubRelease>(url)).result;
-};
--- a/src/main.ts
+++ b/src/main.ts
@ -1,111 +1,127 @@
 import * as fs from 'fs';
-import * as os from 'os';
-import * as path from 'path';
-import * as auth from './auth';
-import * as buildx from './buildx';
-import * as context from './context';
-import * as docker from './docker';
-import * as nodes from './nodes';
-import * as stateHelper from './state-helper';
-import * as util from './util';
+import * as yaml from 'js-yaml';
 import * as core from '@actions/core';
 import * as exec from '@actions/exec';
+import * as actionsToolkit from '@docker/actions-toolkit';
+import {Buildx} from '@docker/actions-toolkit/lib/buildx/buildx';
+import {Builder} from '@docker/actions-toolkit/lib/buildx/builder';
+import {Docker} from '@docker/actions-toolkit/lib/docker/docker';
+import {Toolkit} from '@docker/actions-toolkit/lib/toolkit';
+import {Util} from '@docker/actions-toolkit/lib/util';
+import {Node} from '@docker/actions-toolkit/lib/types/builder';

-async function run(): Promise<void> {
-  try {
+import * as context from './context';
+import * as stateHelper from './state-helper';
+
+actionsToolkit.run(
+  // main
+  async () => {
    const inputs: context.Inputs = await context.getInputs();
-    const dockerConfigHome: string = process.env.DOCKER_CONFIG || path.join(os.homedir(), '.docker');
+    stateHelper.setCleanup(inputs.cleanup);

-    // standalone if docker cli not available
-    const standalone = !(await docker.isAvailable());
+    const toolkit = new Toolkit();
+    const standalone = await toolkit.buildx.isStandalone();
    stateHelper.setStandalone(standalone);

-    core.startGroup(`Docker info`);
-    if (standalone) {
-      core.info(`Docker info skipped in standalone mode`);
-    } else {
-      await exec.exec('docker', ['version'], {
-        failOnStdErr: false
-      });
-      await exec.exec('docker', ['info'], {
-        failOnStdErr: false
-      });
+    await core.group(`Docker info`, async () => {
+      try {
+        await Docker.printVersion();
+        await Docker.printInfo();
+      } catch (e) {
+        core.info(e.message);
      }
-    core.endGroup();
+    });

-    if (util.isValidUrl(inputs.version)) {
+    let toolPath;
+    if (Util.isValidRef(inputs.version)) {
      if (standalone) {
        throw new Error(`Cannot build from source without the Docker CLI`);
      }
-      core.startGroup(`Build and install buildx`);
-      await buildx.build(inputs.version, dockerConfigHome, standalone);
-      core.endGroup();
-    } else if (!(await buildx.isAvailable(standalone)) || inputs.version) {
-      core.startGroup(`Download and install buildx`);
-      await buildx.install(inputs.version || 'latest', standalone ? context.tmpDir() : dockerConfigHome, standalone);
-      core.endGroup();
+      await core.group(`Build buildx from source`, async () => {
+        toolPath = await toolkit.buildxInstall.build(inputs.version);
+      });
+    } else if (!(await toolkit.buildx.isAvailable()) || inputs.version) {
+      await core.group(`Download buildx from GitHub Releases`, async () => {
+        toolPath = await toolkit.buildxInstall.download(inputs.version || 'latest');
+      });
+    }
+    if (toolPath) {
+      await core.group(`Install buildx`, async () => {
+        if (standalone) {
+          await toolkit.buildxInstall.installStandalone(toolPath);
+        } else {
+          await toolkit.buildxInstall.installPlugin(toolPath);
+        }
+      });
    }

-    const buildxVersion = await buildx.getVersion(standalone);
    await core.group(`Buildx version`, async () => {
-      const versionCmd = buildx.getCommand(['version'], standalone);
-      await exec.exec(versionCmd.commandLine, versionCmd.args, {
-        failOnStdErr: false
-      });
+      await toolkit.buildx.printVersion();
    });

    core.setOutput('name', inputs.name);
    stateHelper.setBuilderName(inputs.name);
+    stateHelper.setBuilderDriver(inputs.driver);

-    const credsdir = path.join(dockerConfigHome, 'buildx', 'creds', inputs.name);
-    fs.mkdirSync(credsdir, {recursive: true});
-    stateHelper.setCredsDir(credsdir);
+    fs.mkdirSync(Buildx.certsDir, {recursive: true});
+    stateHelper.setCertsDir(Buildx.certsDir);

    if (inputs.driver !== 'docker') {
-      core.startGroup(`Creating a new builder instance`);
-      const authOpts = auth.setCredentials(credsdir, 0, inputs.driver, inputs.endpoint);
-      if (authOpts.length > 0) {
-        inputs.driverOpts = [...inputs.driverOpts, ...authOpts];
+      await core.group(`Creating a new builder instance`, async () => {
+        const certsDriverOpts = Buildx.resolveCertsDriverOpts(inputs.driver, inputs.endpoint, {
+          cacert: process.env[`${context.builderNodeEnvPrefix}_0_AUTH_TLS_CACERT`],
+          cert: process.env[`${context.builderNodeEnvPrefix}_0_AUTH_TLS_CERT`],
+          key: process.env[`${context.builderNodeEnvPrefix}_0_AUTH_TLS_KEY`]
+        });
+        if (certsDriverOpts.length > 0) {
+          inputs.driverOpts = [...inputs.driverOpts, ...certsDriverOpts];
        }
-      const createCmd = buildx.getCommand(await context.getCreateArgs(inputs, buildxVersion), standalone);
-      await exec.exec(createCmd.commandLine, createCmd.args);
-      core.endGroup();
+        const createCmd = await toolkit.buildx.getCommand(await context.getCreateArgs(inputs, toolkit));
+        await exec.exec(createCmd.command, createCmd.args);
+      });
    }

    if (inputs.append) {
-      core.startGroup(`Appending node(s) to builder`);
+      await core.group(`Appending node(s) to builder`, async () => {
        let nodeIndex = 1;
-      for (const node of nodes.Parse(inputs.append)) {
-        const authOpts = auth.setCredentials(credsdir, nodeIndex, inputs.driver, node.endpoint || '');
-        if (authOpts.length > 0) {
-          node['driver-opts'] = [...(node['driver-opts'] || []), ...authOpts];
+        const nodes = yaml.load(inputs.append) as Node[];
+        for (const node of nodes) {
+          const certsDriverOpts = Buildx.resolveCertsDriverOpts(inputs.driver, `${node.endpoint}`, {
+            cacert: process.env[`${context.builderNodeEnvPrefix}_${nodeIndex}_AUTH_TLS_CACERT`],
+            cert: process.env[`${context.builderNodeEnvPrefix}_${nodeIndex}_AUTH_TLS_CERT`],
+            key: process.env[`${context.builderNodeEnvPrefix}_${nodeIndex}_AUTH_TLS_KEY`]
+          });
+          if (certsDriverOpts.length > 0) {
+            node['driver-opts'] = [...(node['driver-opts'] || []), ...certsDriverOpts];
          }
-        const appendCmd = buildx.getCommand(await context.getAppendArgs(inputs, node, buildxVersion), standalone);
-        await exec.exec(appendCmd.commandLine, appendCmd.args);
+          const appendCmd = await toolkit.buildx.getCommand(await context.getAppendArgs(inputs, node, toolkit));
+          await exec.exec(appendCmd.command, appendCmd.args);
          nodeIndex++;
        }
-      core.endGroup();
+      });
    }

-    core.startGroup(`Booting builder`);
-    const inspectCmd = buildx.getCommand(await context.getInspectArgs(inputs, buildxVersion), standalone);
-    await exec.exec(inspectCmd.commandLine, inspectCmd.args);
-    core.endGroup();
+    await core.group(`Booting builder`, async () => {
+      const inspectCmd = await toolkit.buildx.getCommand(await context.getInspectArgs(inputs, toolkit));
+      await exec.exec(inspectCmd.command, inspectCmd.args);
+    });

    if (inputs.install) {
      if (standalone) {
        throw new Error(`Cannot set buildx as default builder without the Docker CLI`);
      }
-      core.startGroup(`Setting buildx as default builder`);
-      await exec.exec('docker', ['buildx', 'install']);
-      core.endGroup();
+      await core.group(`Setting buildx as default builder`, async () => {
+        const installCmd = await toolkit.buildx.getCommand(['install']);
+        await exec.exec(installCmd.command, installCmd.args);
+      });
    }

-    core.startGroup(`Inspect builder`);
-    const builder = await buildx.inspect(inputs.name, standalone);
-    const firstNode = builder.nodes[0];
+    const builderInspect = await toolkit.builder.inspect(inputs.name);
+    const firstNode = builderInspect.nodes[0];
+
+    await core.group(`Inspect builder`, async () => {
      const reducedPlatforms: Array<string> = [];
-    for (const node of builder.nodes) {
+      for (const node of builderInspect.nodes) {
        for (const platform of node.platforms?.split(',') || []) {
          if (reducedPlatforms.indexOf(platform) > -1) {
            continue;
@ -113,35 +129,32 @@ async function run(): Promise<void> {
          reducedPlatforms.push(platform);
        }
      }
-    core.info(JSON.stringify(builder, undefined, 2));
-    core.setOutput('driver', builder.driver);
+      core.info(JSON.stringify(builderInspect, undefined, 2));
+      core.setOutput('driver', builderInspect.driver);
      core.setOutput('platforms', reducedPlatforms.join(','));
-    core.setOutput('nodes', JSON.stringify(builder.nodes, undefined, 2));
+      core.setOutput('nodes', JSON.stringify(builderInspect.nodes, undefined, 2));
      core.setOutput('endpoint', firstNode.endpoint); // TODO: deprecated, to be removed in a later version
      core.setOutput('status', firstNode.status); // TODO: deprecated, to be removed in a later version
      core.setOutput('flags', firstNode['buildkitd-flags']); // TODO: deprecated, to be removed in a later version
-    core.endGroup();
+    });

-    if (!standalone && builder.driver == 'docker-container') {
-      stateHelper.setContainerName(`buildx_buildkit_${firstNode.name}`);
-      core.startGroup(`BuildKit version`);
-      for (const node of builder.nodes) {
-        const bkvers = await buildx.getBuildKitVersion(`buildx_buildkit_${node.name}`);
-        core.info(`${node.name}: ${bkvers}`);
+    if (!standalone && builderInspect.driver == 'docker-container') {
+      stateHelper.setContainerName(`${Buildx.containerNamePrefix}${firstNode.name}`);
+      await core.group(`BuildKit version`, async () => {
+        for (const node of builderInspect.nodes) {
+          const buildkitVersion = await toolkit.buildkit.getVersion(node);
+          core.info(`${node.name}: ${buildkitVersion}`);
        }
-      core.endGroup();
+      });
    }
    if (core.isDebug() || firstNode['buildkitd-flags']?.includes('--debug')) {
      stateHelper.setDebug('true');
    }
-  } catch (error) {
-    core.setFailed(error.message);
-  }
-}
-
-async function cleanup(): Promise<void> {
+  },
+  // post
+  async () => {
    if (stateHelper.IsDebug && stateHelper.containerName.length > 0) {
-    core.startGroup(`BuildKit container logs`);
+      await core.group(`BuildKit container logs`, async () => {
        await exec
          .getExecOutput('docker', ['logs', `${stateHelper.containerName}`], {
            ignoreReturnCode: true
@ -151,14 +164,21 @@ async function cleanup(): Promise<void> {
              core.warning(res.stderr.trim());
            }
          });
-    core.endGroup();
+      });
    }

-  if (stateHelper.builderName.length > 0) {
-    core.startGroup(`Removing builder`);
-    const rmCmd = buildx.getCommand(['rm', stateHelper.builderName], /true/i.test(stateHelper.standalone));
+    if (!stateHelper.cleanup) {
+      return;
+    }
+
+    if (stateHelper.builderDriver != 'docker' && stateHelper.builderName.length > 0) {
+      await core.group(`Removing builder`, async () => {
+        const buildx = new Buildx({standalone: stateHelper.standalone});
+        const builder = new Builder({buildx: buildx});
+        if (await builder.exists(stateHelper.builderName)) {
+          const rmCmd = await buildx.getCommand(['rm', stateHelper.builderName]);
          await exec
-      .getExecOutput(rmCmd.commandLine, rmCmd.args, {
+            .getExecOutput(rmCmd.command, rmCmd.args, {
              ignoreReturnCode: true
            })
            .then(res => {
@ -166,17 +186,16 @@ async function cleanup(): Promise<void> {
                core.warning(res.stderr.trim());
              }
            });
-    core.endGroup();
+        } else {
+          core.info(`${stateHelper.builderName} does not exist`);
+        }
+      });
    }

-  if (stateHelper.credsDir.length > 0 && fs.existsSync(stateHelper.credsDir)) {
-    core.info(`Cleaning up credentials`);
-    fs.rmSync(stateHelper.credsDir, {recursive: true});
+    if (stateHelper.certsDir.length > 0 && fs.existsSync(stateHelper.certsDir)) {
+      await core.group(`Cleaning up certificates`, async () => {
+        fs.rmSync(stateHelper.certsDir, {recursive: true});
+      });
    }
-}
-
-if (!stateHelper.IsPost) {
-  run();
-} else {
-  cleanup();
-}
+  }
+);
--- a/src/nodes.ts
+++ b/src/nodes.ts
@ -1,13 +0,0 @@
-import * as yaml from 'js-yaml';
-
-export type Node = {
-  name?: string;
-  endpoint?: string;
-  'driver-opts'?: Array<string>;
-  'buildkitd-flags'?: string;
-  platforms?: string;
-};
-
-export function Parse(data: string): Node[] {
-  return yaml.load(data) as Node[];
-}
--- a/src/state-helper.ts
+++ b/src/state-helper.ts
@ -1,11 +1,12 @@
 import * as core from '@actions/core';

-export const IsPost = !!process.env['STATE_isPost'];
 export const IsDebug = !!process.env['STATE_isDebug'];
-export const standalone = process.env['STATE_standalone'] || '';
+export const standalone = /true/i.test(process.env['STATE_standalone'] || '');
 export const builderName = process.env['STATE_builderName'] || '';
+export const builderDriver = process.env['STATE_builderDriver'] || '';
 export const containerName = process.env['STATE_containerName'] || '';
-export const credsDir = process.env['STATE_credsDir'] || '';
+export const certsDir = process.env['STATE_certsDir'] || '';
+export const cleanup = /true/i.test(process.env['STATE_cleanup'] || '');

 export function setDebug(debug: string) {
  core.saveState('isDebug', debug);
@ -19,14 +20,18 @@ export function setBuilderName(builderName: string) {
  core.saveState('builderName', builderName);
 }

+export function setBuilderDriver(builderDriver: string) {
+  core.saveState('builderDriver', builderDriver);
+}
+
 export function setContainerName(containerName: string) {
  core.saveState('containerName', containerName);
 }

-export function setCredsDir(credsDir: string) {
-  core.saveState('credsDir', credsDir);
+export function setCertsDir(certsDir: string) {
+  core.saveState('certsDir', certsDir);
 }

-if (!IsPost) {
-  core.saveState('isPost', 'true');
+export function setCleanup(cleanup: boolean) {
+  core.saveState('cleanup', cleanup);
 }
--- a/src/util.ts
+++ b/src/util.ts
@ -1,8 +0,0 @@
-export function isValidUrl(url: string): boolean {
-  try {
-    new URL(url);
-  } catch (e) {
-    return false;
-  }
-  return true;
-}
--- a/tsconfig.json
+++ b/tsconfig.json
@ -1,19 +1,21 @@
 {
  "compilerOptions": {
+    "esModuleInterop": true,
    "target": "es6",
    "module": "commonjs",
+    "strict": true,
    "newLine": "lf",
    "outDir": "./lib",
    "rootDir": "./src",
-    "esModuleInterop": true,
    "forceConsistentCasingInFileNames": true,
-    "strict": true,
    "noImplicitAny": false,
+    "resolveJsonModule": true,
    "useUnknownInCatchVariables": false,
  },
  "exclude": [
+    "./__tests__/**/*",
+    "./lib/**/*",
    "node_modules",
-    "**/*.test.ts",
    "jest.config.ts"
  ]
 }
--- a/yarn.lock
+++ b/yarn.lock