#
# Copyright(c) 2019-2021 Intel Corporation
# SPDX-License-Identifier: BSD-3-Clause
#

import test_tools.common.linux_command as linux_comm
import type_def.size as size
from core.test_run import TestRun


class Ddrescue(linux_comm.LinuxCommand):
    def __init__(self):
        linux_comm.LinuxCommand.__init__(self, TestRun.executor, 'ddrescue')
        self.source_path = None
        self.destination_path = None
        self.param_name_prefix = "--"

    def source(self, value):
        self.source_path = value
        return self

    def destination(self, value):
        self.destination_path = value
        return self

    def reverse(self):
        return self.set_flags("reverse")

    def synchronous(self):
        return self.set_flags("synchronous")

    def direct(self):
        return self.set_flags("direct")

    def force(self):
        return self.set_flags("force")

    def block_size(self, value: size.Size):
        return self.set_param('sector-size', int(value.get_value()))

    def size(self, value: size.Size):
        return self.set_param('size', int(value.get_value()))

    def __str__(self):
        command = linux_comm.LinuxCommand.__str__(self)
        command += f" {self.source_path} {self.destination_path}"
        return command