Mini Shell

"""
Common functions for managing mounts
"""

import logging
import os

import salt.utils.files
import salt.utils.stringutils
import salt.utils.versions
import salt.utils.yaml

log = logging.getLogger(__name__)


def _read_file(path):
    """
    Reads and returns the contents of a text file
    """
    try:
        with salt.utils.files.fopen(path, "rb") as contents:
            return salt.utils.yaml.safe_load(contents)
    except OSError:
        return {}


def get_cache(opts):
    """
    Return the mount cache file location.
    """
    return os.path.join(opts["cachedir"], "mounts")


def read_cache(opts):
    """
    Write the mount cache file.
    """
    cache_file = get_cache(opts)
    return _read_file(cache_file)


def write_cache(cache, opts):
    """
    Write the mount cache file.
    """
    cache_file = get_cache(opts)

    try:
        _cache = salt.utils.stringutils.to_bytes(salt.utils.yaml.safe_dump(cache))
        with salt.utils.files.fopen(cache_file, "wb+") as fp_:
            fp_.write(_cache)
        return True
    except OSError:
        log.error("Failed to cache mounts", exc_info_on_loglevel=logging.DEBUG)
        return False