import tempfile

from weevely.core.module import Module
from weevely.core.vectors import ModuleExec


class Read(Module):
    """Read remote file from the remote filesystem."""

    aliases = ["cat"]

    def init(self):
        self.register_info({"author": ["Emilio Pinna"], "license": "GPLv3"})

        self.register_arguments(
            [
                {"name": "rpath", "help": "Remote file path"},
                {"name": "-vector", "choices": ("file", "fread", "file_get_contents", "base64")},
            ]
        )

    def run(self, **kwargs):
        # Get a temporary file name
        temp_file = tempfile.NamedTemporaryFile()
        self.args["lpath"] = temp_file.name

        arg_vector = ["-vector", self.args.get("vector")] if self.args.get("vector") else []

        # Run file_download
        result = ModuleExec(
            "file_download", [self.args.get("rpath"), "${lpath}"] + arg_vector, name="file_download"
        ).run(self.args)

        # Delete temp file
        temp_file.close()

        return result