# A persistent cache of pre-compiled formulas

import os
import stat
import pickle
import hashlib


class TimeStampedObject:
    "An object and the last time it was updated"

    def __init__(self, obj, time, file=None):
        self.time = time
        self.obj = obj
        self.cache_file = file


class T:
    def __init__(self, dir):
        self.dir = os.path.expanduser(dir)
        self.files = {}

    def init(self):
        os.makedirs(self.dir, mode=0o770, exist_ok=True)
        self._loadHash()

    def _indexName(self):
        return os.path.join(self.dir, "index_v00.pkl")

    def _loadHash(self):
        if os.path.exists(self._indexName()):
            self.files = self.loadPickledFile(self._indexName())

    def save(self):
        self.createPickledFile(self._indexName(), self.files)

    def clear(self):
        if not os.path.exists(self.dir):
            return
        for f in os.listdir(self.dir):
            try:
                os.remove(os.path.join(self.dir, f))
            except BaseException:
                pass

    def makefilename(self, name, ext):
        return os.path.join(self.dir, f"fract4d_{name}{ext}")

    def hashcode(self, s, *extras):
        hash = hashlib.md5(s.encode("utf-8"))
        for x in extras:
            hash.update(x.encode("utf-8"))

        return hash.hexdigest()

    def makePickleName(self, s, *extras):
        name = self.hashcode(s, *extras) + ".pkl"
        fullname = os.path.join(self.dir, name)
        return fullname

    def createPickledFile(self, file, contents):
        with open(file, "wb") as f:
            pickle.dump(contents, f, True)
            # print "created %s" % file

    def loadPickledFile(self, file):
        with open(file, "rb") as f:
            contents = pickle.load(f)
        return contents

    def getcontents(self, file, parser):
        mtime = os.stat(file)[stat.ST_MTIME]

        tso = self.files.get(file, None)
        if tso:
            if tso.time == mtime:
                return tso.obj

        with open(file) as f:
            val = parser(f)

        hashname = self.makePickleName(file)
        # self.createPickledFile(hashname,val)
        self.files[file] = TimeStampedObject(val, mtime, hashname)

        return val
