import marshal
import types
from Carbon import Folder
from Carbon import Folders

class PrefObject:

    def __init__(self, dict = None):
        if dict == None:
            self._prefsdict = {}
        else:
            self._prefsdict = dict

    def __len__(self):
        return len(self._prefsdict)

    def __delattr__(self, attr):
        if self._prefsdict.has_key(attr):
            del self._prefsdict[attr]
        else:
            raise AttributeError, 'delete non-existing instance attribute'

    def __getattr__(self, attr):
        if attr == '__members__':
            keys = self._prefsdict.keys()
            keys.sort()
            return keys
        try:
            return self._prefsdict[attr]
        except KeyError:
            raise AttributeError, attr

    def __setattr__(self, attr, value):
        if attr[0] <> '_':
            self._prefsdict[attr] = value
        else:
            self.__dict__[attr] = value

    def getprefsdict(self):
        return self._prefsdict


class PrefFile(PrefObject):

    def __init__(self, path, creator = 'Pyth'):
        # Find the preferences folder and our prefs file, create if needed.
        self.__path = path
        self.__creator = creator
        self._prefsdict = {}
        try:
            prefdict = marshal.load(open(self.__path, 'rb'))
        except (IOError, ValueError):
            # file not found, or currupt marshal data
            pass
        else:
            for key, value in prefdict.items():
                if type(value) == types.DictType:
                    self._prefsdict[key] = PrefObject(value)
                else:
                    self._prefsdict[key] = value

    def save(self):
        prefdict = {}
        for key, value in self._prefsdict.items():
            if type(value) == types.InstanceType:
                prefdict[key] = value.getprefsdict()
                if not prefdict[key]:
                    del prefdict[key]
            else:
                prefdict[key] = value
        marshal.dump(prefdict, open(self.__path, 'wb'))
        try:
            MacOS.SetCreatorAndType(self.__path, self.__creator, 'pref')
        except:
            pass

    def __getattr__(self, attr):
        if attr == '__members__':
            keys = self._prefsdict.keys()
            keys.sort()
            return keys
        try:
            return self._prefsdict[attr]
        except KeyError:
            if attr[0] <> '_':
                self._prefsdict[attr] = PrefObject()
                return self._prefsdict[attr]
            else:
                raise AttributeError, attr


_prefscache = {}

def GetPrefs(prefname, creator = 'Pyth'):
    import macostools, os
    if _prefscache.has_key(prefname):
        return _prefscache[prefname]
    # Find the preferences folder and our prefs file, create if needed.
    fsr = Folder.FSFindFolder(Folders.kOnSystemDisk, 'pref', 1)
    prefsfolder = fsr.as_pathname()
    path = os.path.join(prefsfolder, prefname)
    head, tail = os.path.split(path)
    # make sure the folder(s) exist
    macostools.mkdirs(head)

    preffile = PrefFile(path, creator)
    _prefscache[prefname] = preffile
    return preffile