# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from __future__ import annotations

from typing import TYPE_CHECKING

from crossbench.action_runner.action.action import Action
from crossbench.action_runner.action.action_type import ActionType

if TYPE_CHECKING:
  from crossbench.action_runner.base import ActionRunner
  from crossbench.runner.run import Run


class ScreenshotAction(Action):
  TYPE: ActionType = ActionType.SCREENSHOT

  def run_with(self, run: Run, action_runner: ActionRunner) -> None:
    action_runner.screenshot(run, self)