from typing import Optional

from cli.cli_command import NeofsCliCommand
from shell import CommandResult

from .blockchain_network_type import NetworkType


class NeoGoDb(NeofsCliCommand):
    def dump(
            self,
            config_path: str,
            out: str,
            network: NetworkType = NetworkType.PRIVATE,
            count: int = 0,
            start: int = 0,
    ) -> CommandResult:
        """ dump blocks (starting with block #1) to the file

        Args:
            config_path (str):     path to config
            network (NetworkType): Select network type (default: private)
            count (int):           number of blocks to be processed (default or 0: all chain)
                                   (default: 0)
            start (int):           block number to start from (default: 0) (default: 0)
            out (srt):             Output file (stdout if not given)

        Returns:
            str: Command string

        """
        return self._execute(
            "db dump",
            **{network.value: True},
            **{
                param: param_value
                for param, param_value in locals().items()
                if param not in ["self"]
            }
        )

    def restore(
            self,
            config_path: str,
            input_file: str,
            network: NetworkType = NetworkType.PRIVATE,
            count: int = 0,
            dump: Optional[str] = None,
            incremental: bool = False,
    ) -> CommandResult:
        """ dump blocks (starting with block #1) to the file

        Args:
            config_path (str):     path to config
            network (NetworkType): Select network type (default: private)
            count (int):           number of blocks to be processed (default or 0: all chain)
                                   (default: 0)
            input_file (str):      Input file (stdin if not given)
            dump (str):            directory for storing JSON dumps
            incremental (bool):    use if dump is incremental

        Returns:
            str: Command string

        """
        return self._execute(
            "db restore",
            **{network.value: True},
            **{
                param: param_value
                for param, param_value in locals().items()
                if param not in ["self"]
            }
        )