Module materialize.zippy.mz_actions

# Copyright Materialize, Inc. and contributors. All rights reserved.
#
# Use of this software is governed by the Business Source License
# included in the LICENSE file at the root of this repository.
#
# As of the Change Date specified in that file, in accordance with
# the Business Source License, use of this software will be governed
# by the Apache License, Version 2.0.

from typing import List, Set, Type

from materialize.mzcompose import Composition
from materialize.zippy.framework import Action, Capability
from materialize.zippy.mz_capabilities import MzIsRunning


class MzStart(Action):
    def run(self, c: Composition) -> None:
        c.up("materialized")
        c.wait_for_materialized()

    def provides(self) -> List[Capability]:
        return [MzIsRunning()]


class MzStop(Action):
    @classmethod
    def requires(self) -> Set[Type[Capability]]:
        return {MzIsRunning}

    def run(self, c: Composition) -> None:
        c.kill("materialized")

    def removes(self) -> Set[Type[Capability]]:
        return {MzIsRunning}