# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

from ci.task.generic import BaseTask

# ======================================================================================================================


class BaseLinuxTask(BaseTask):
    def __init__(self, host: str, cmd: str):
        ssh_cmd: str = f"ssh -C {host} \"bash -l -c \'{cmd}\'\""
        super().__init__(ssh_cmd)


class CheckoutOnLinux(BaseLinuxTask):
    def __init__(self, host: str, repository: str, branch: str):
        cmd: str = f"cd {repository} && git pull origin && git checkout {branch}"
        super().__init__(host, cmd)


class CompileOnLinux(BaseLinuxTask):
    def __init__(self, host: str, repository: str, target: str, is_debug: bool):
        debug_flag: str = "DEBUG=yes" if is_debug else "DEBUG=no"
        profiler_flag: str = "PROFILER=yes" if not is_debug else "PROFILER=no"
        cmd: str = f"cd {repository} && make {profiler_flag} {debug_flag} {target}"
        super().__init__(host, cmd)


class RunOnLinux(BaseLinuxTask):
    def __init__(self, host: str, repository: str, target: str, is_debug: bool, is_sudo: bool, config_path: str):
        debug_flag: str = "DEBUG=yes" if is_debug else "DEBUG=no"
        sudo_cmd: str = "sudo -E" if is_sudo else ""
        profiler_flag: str = "PROFILER=yes" if not is_debug else "PROFILER=no"
        cmd: str = f"cd {repository} && {sudo_cmd} make -j 1 CONFIG_PATH={config_path} {profiler_flag} {debug_flag} {target} 2> out.stderr && cat out.stderr >&2 || ( cat out.stderr >&2 ; exit 1 )"
        super().__init__(host, cmd)


class CleanupOnLinux(BaseLinuxTask):
    def __init__(self, host: str, repository: str, is_sudo: bool, branch: str):
        sudo_cmd: str = "sudo -E" if is_sudo else ""
        cmd: str = f"cd {repository} && {sudo_cmd} make clean && git checkout {branch} && git clean -fdx ; sudo -E rm -rf /dev/shm/demikernel* ; sudo pkill -f demikernel*"
        super().__init__(host, cmd)


class CloneOnLinux(BaseLinuxTask):
    def __init__(self, host: str, path: str, repository: str, branch: str):
        cmd: str = f"cd {path} && git clone {repository} --branch {branch}"
        super().__init__(host, cmd)


class MakeRedisOnLinux(BaseLinuxTask):
    def __init__(self, host: str, path: str):
        cmd: str = f"cd {path}/redis  && make MALLOC=libc all"
        super().__init__(host, cmd)


class RunredisServerOnLinux(BaseLinuxTask):
    def __init__(self, host: str, redis_path: str, is_sudo: str, env: str, params: str):
        sudo_cmd: str = "sudo -E" if is_sudo else ""
        redis_cmd: str = f"{sudo_cmd} {env} ./src/redis-server {params} --daemonize yes"
        cmd: str = f"cd {redis_path} && {redis_cmd}"
        super().__init__(host, cmd)


class RunRedisBenchmarkOnLinux(BaseLinuxTask):
    def __init__(self, host: str, redis_path: str, params: str, timeout: int = 120):
        cmd: str = f"cd {redis_path} && timeout {timeout} ./src/redis-benchmark {params}"
        super().__init__(host, cmd)


class StopRedisServerOnLinux(BaseLinuxTask):
    def __init__(self, host: str, redis_path: str, params: str, timeout: int = 10):
        cmd: str = f"cd {redis_path} && timeout {timeout} ./src/redis-cli {params} shutdown"
        super().__init__(host, cmd)


class CleanupRedisOnLinux(BaseLinuxTask):
    def __init__(self, host: str, process_name: str, redis_path: str):
        cmd: str = f"sudo pkill -e {process_name} ; rm -rf {redis_path}"
        super().__init__(host, cmd)