from conf import hook

""" Pre-Test Hook: ServerFiles
This hook is used to define a set of files on the server's virtual file system.
server_files is expected to be dictionary that maps filenames to their
contents. In the future, this can be used to add additional metadata to the
files using the WgetFile class too.

This hook also does some additional processing on the contents of the file. Any
text between {{and}} is replaced by the contents of a class variable of the
same name. This is useful in creating files that contain an absolute link to
another file on the same server. """


@hook()
class ServerFiles:
    def __init__(self, server_files):
        self.server_files = server_files

    def __call__(self, test_obj):
        for server, files in zip(test_obj.servers, self.server_files):
            files_content = {f.name: test_obj._replace_substring(f.content)
                             for f in files}
            files_rules = {f.name: test_obj.get_server_rules(f)
                           for f in files}
            server.server_conf(files_content, files_rules)
