from ..zips import open_zip


_style_map_path = "mammoth/style-map"


def write_style_map(fileobj, style_map):
    with open_zip(fileobj, "a") as zip_file:
        zip_file.write_str(_style_map_path, style_map)


def read_style_map(fileobj):
    with open_zip(fileobj, "r") as zip_file:
        if zip_file.exists(_style_map_path):
            return zip_file.read_str(_style_map_path)