from grow.pods.storage import base_storage
import errno
import jinja2
import os
import shutil


class FileStorage(base_storage.BaseStorage):

    @staticmethod
    def open(filename, mode=None):
        if mode is None:
            mode = 'r'
        return open(filename, mode=mode)

    @staticmethod
    def read(filename):
        fp = open(filename)
        content = fp.read()
        fp.close()
        return content

    @staticmethod
    def modified(filename):
        return os.stat(filename).st_mtime

    @staticmethod
    def size(filename):
        return os.path.getsize(filename)

    @staticmethod
    def stat(filename):
        return os.stat(filename)

    @staticmethod
    def listdir(dirpath):
        paths = []
        for root, dirs, files in os.walk(dirpath, followlinks=True):
            for filename in files:
                path = os.path.join(root, filename)[len(dirpath):]
                paths.append(path)
        return paths

    @staticmethod
    def JinjaLoader(path):
        return jinja2.FileSystemLoader(path)

    @classmethod
    def write(cls, path, content):
        dirname = os.path.dirname(path)
        try:
            os.makedirs(dirname)
        except OSError as e:
            if e.errno == errno.EEXIST and os.path.isdir(dirname):
                pass
            else:
                raise
        fp = cls.open(path, mode='w')
        fp.write(content)
        fp.close()
        return fp

    @staticmethod
    def exists(filename):
        return os.path.exists(filename)

    @staticmethod
    def delete(filename):
        return os.remove(filename)

    @staticmethod
    def delete_dir(dirpath):
        shutil.rmtree(dirpath)

    @staticmethod
    def copy_to(paths, target_paths):
        # TODO(jeremydw): Rename to bulk_copy_to.
        for i, path in enumerate(paths):
            target_path = target_paths[i]
            shutil.copyfile(path, target_path)
            shutil.copystat(path, target_path)

    @staticmethod
    def move_to(path, target_path):
        os.rename(path, target_path)