import os
import ConfigParser, StringIO

def read_config(rootDir = None, defaults = None, nick="octofuss"):
    """
    Read octofuss configuration, returning a ConfigParser object
    """
    if rootDir == None:
        rootDir = os.environ.get(nick.upper() + "_CONFDIR", "/etc/" + nick)
    files = []
    def trytouse(path):
        if os.path.exists(path):
            files.append(path)

    # Start with the main config file
    trytouse(os.path.join(rootDir, nick + ".conf"))

    # Add snippets found in rc.d style directory
    subdir = os.path.join(rootDir, nick + ".conf.d")
    if os.path.isdir(subdir):
        for file in sorted(os.listdir(subdir)):
            if file.startswith('#'): continue
            if file.startswith('.'): continue
            if file.endswith('~'): continue
            if file.endswith('.bak'): continue
            trytouse(os.path.join(subdir, file))

    config = ConfigParser.ConfigParser()
    if defaults != None:
        infile = StringIO.StringIO(defaults)
        config.readfp(infile, "defaults")
    config.read(files)
    return config