from os import walk, remove
from os.path import sep, join, exists, normpath
from re import findall, sub
from distutils.core import setup
from distutils.command.install import install
from pprint import pprint
from fileinput import FileInput
from re import sub, match

from Configuration import *

class Setup:

    config = Configuration()
    manifest_path = "MANIFEST"

    def __init__(self):
        pass

    def remove_old_mainfest(self):
        path = self.manifest_path
        if exists(path):
            remove(path)

    def build_package_list(self):
        packages = []
        config = self.config.get_section("setup")
        locations = [config["package-root"]] + config["additional-packages"]
        for location in locations:
            if exists(location):
                for root, dirs, files in walk(location, followlinks=True):
                    packages.append(root.replace(sep, "."))
        return packages

    def build_data_map(self):
        include = []
        config = self.config.get_section("setup")
        exclude = map(normpath, config["data-exclude"])
        for root, dirs, files in walk("."):
            dirs = self.remove_excluded(dirs, root, exclude)
            files = [join(root, f) for f in self.remove_excluded(files, root,
                                                                 exclude)]
            if files:
                include.append((normpath(join(config["installation-path"],
                                              root)), files))
        return include

    def remove_excluded(self, paths, root, exclude):
        removal = []
        for path in paths:
            if normpath(join(root, path)) in exclude:
                removal.append(path)
        for path in removal:
            paths.remove(path)
        return paths

    def translate_title(self):
        config = self.config.get_section("setup")
        title = config["title"].replace(" ", config["whitespace-placeholder"])
        return sub("[^\w-]", config["special-char-placeholder"], title)

    def build_description(self):
        description = ""
        path = self.config.get("setup", "description-file")
        if exists(path):
            description = "\n%s\n%s\n%s" % (file(path).read(),
                                            "Changelog\n=========",
                                            self.translate_changelog())
        return description

    def translate_changelog(self):
        translation = ""
        path = self.config.get("setup", "changelog")
        if exists(path):
            lines = file(path).readlines()
            package_name = lines[0].split()[0]
            for line in lines:
                line = line.strip()
                if line.startswith(package_name):
                    version = findall("\((.*)\)", line)[0]
                    translation += "\n%s\n%s\n" % (version, "-" * len(version))
                elif line and not line.startswith("--"):
                    if line.startswith("*"):
                        translation += line + "\n"
                    else:
                        translation += "  " + line + "\n"
        return translation

    def setup(self, windows=[], options={}):
	print "running setup..."
        self.remove_old_mainfest()
        config = self.config.get_section("setup")
	scripts = []
	if config["init-script"]:
	    scripts.append(config["init-script"])
        setup(cmdclass={"install": insert_resource_path},
              name=self.translate_title(),
              packages=self.build_package_list(),
              scripts=scripts,
              data_files=self.build_data_map(),
              requires=config["requirements"],
              version=config["version"],
              description=config["summary"],
              classifiers=config["classifiers"],
              long_description=self.build_description(),
              license=config["license"],
              platforms=config["platforms"],
              author=config["contact-name"],
              author_email=config["contact-email"],
              url=config["url"],
	      windows=windows,
	      options=options)


class insert_resource_path(install):

    def run(self):
        install.run(self)
        self.edit_game_object_file()

    def edit_game_object_file(self):
        config = Configuration().get_section("setup")
        for path in self.get_outputs():
            if path.endswith(config["main-object"]):
                for line in FileInput(path, inplace=True):
                    pattern = "^ *{0} *=.*".\
                              format(config["resource-path-identifier"])
                    if match(pattern, line):
                        line = sub("=.*$", "= \"{0}\"".\
                                   format(config["installation-path"]), line)
                    print line.strip("\n")

from os import sep, getcwd
from os.path import join, exists, basename, dirname, expanduser
from sys import argv
from re import match
from pprint import pformat

from ConfigParser import RawConfigParser

class Configuration(RawConfigParser):

    default_project_file_rel_path = "config"
    default_resource_paths = [".", "resource"]

    def __init__(self, project_file_rel_path=None, resource_path=None,
                 type_declarations=None):
        RawConfigParser.__init__(self)
        self.project_file_rel_path = project_file_rel_path
        self.resource_path = resource_path
        self.modifiable = {}
        self.order = []
        self.set_type_declarations(type_declarations)
        self.set_defaults()
        self.read_project_config_file()
        self.modify_defaults()
        self.print_debug(self)

    def set_type_declarations(self, type_declarations):
        if type_declarations is None:
            type_declarations = TypeDeclarations()
        self.type_declarations = type_declarations

    def translate_path(self, path):
        new = ""
        if path and path[0] == sep:
            new += sep
        return expanduser("{0}{1}".format(new, join(*path.split(sep))))

    def set_defaults(self):
        add_section = self.add_section
        set_option = self.set
        section = "setup"
        add_section(section)
        set_option(section, "package-root", basename(getcwd()), False)
        set_option(section, "additional-packages", "", False)
        set_option(section, "title", "", False)
        set_option(section, "classifiers", "", False)
        set_option(section, "resource-search-path", "./, resource/", False)
        set_option(section, "installation-dir", "/usr/local/share/games/",
                   False)
        set_option(section, "changelog", "changelog", False)
        set_option(section, "description-file", "", False)
        set_option(section, "init-script", "", False)
        set_option(section, "version", "", False)
        set_option(section, "summary", "", False)
        set_option(section, "license", "", False)
        set_option(section, "platforms", "", False)
        set_option(section, "contact-name", "", False)
        set_option(section, "contact-email", "", False)
        set_option(section, "url", "", False)
        set_option(section, "requirements", "", False)
        set_option(section, "main-object", "pgfw/Game.py", False)
        set_option(section, "resource-path-identifier", "resource_path", False)
        set_option(section, "special-char-placeholder", "_", False)
        set_option(section, "whitespace-placeholder", "-", False)
        set_option(section, "windows-dist-path", "dist/win/", False)
        set_option(section, "windows-icon-path", "", False)
        set_option(section, "lowercase-boolean-true", "yes", False)
        section = "display"
        add_section(section)
        set_option(section, "dimensions", "480, 360", False)
        set_option(section, "frame-duration", "40", False)
        set_option(section, "wait-duration", "2", False)
        set_option(section, "caption", "", False)
        set_option(section, "centered", "yes", False)
        set_option(section, "icon-path", "", False)
        set_option(section, "skip-frames", "no", False)
        set_option(section, "fullscreen", "no", False)
        set_option(section, "windowed-flag", "wi", False)
        set_option(section, "show-framerate", "no", False)
        set_option(section, "framerate-display-flag", "fr", False)
        set_option(section, "framerate-text-size", "16", False)
        set_option(section, "framerate-text-color", "0, 0, 0", False)
        set_option(section, "framerate-text-background", "255, 255, 255", False)
        section = "input"
        add_section(section)
        set_option(section, "release-suffix", "-release", False)
        section = "sprite"
        add_section(section)
        set_option(section, "transparent-color", "magenta", False)
        section = "screen-captures"
        add_section(section)
        set_option(section, "rel-path", "caps", False)
        set_option(section, "file-name-format", "%Y%m%d%H%M%S", False)
        set_option(section, "file-extension", "png", False)
        section = "video-recordings"
        add_section(section)
        set_option(section, "enable", "no", False)
        set_option(section, "rel-path", "vids", False)
        set_option(section, "directory-name-format", "%Y%m%d%H%M%S", False)
        set_option(section, "file-extension", "png", False)
        set_option(section, "frame-format", "RGB", False)
        set_option(section, "framerate", "100", False)
        section = "mouse"
        add_section(section)
        set_option(section, "visible", "yes", False)
        set_option(section, "double-click-time-limit", ".5", False)
        section = "keys"
        add_section(section)
        set_option(section, "up", "K_UP, K_w", False)
        set_option(section, "right", "K_RIGHT, K_d", False)
        set_option(section, "down", "K_DOWN, K_s", False)
        set_option(section, "left", "K_LEFT, K_a", False)
        set_option(section, "capture-screen", "K_F9", False)
        set_option(section, "toggle-fullscreen", "K_F11", False)
        set_option(section, "reset-game", "K_F8", False)
        set_option(section, "record-video", "K_F10", False)
        set_option(section, "mute", "K_F12", False)
        set_option(section, "toggle-interpolator", "K_F7", False)
        section = "joy"
        add_section(section)
        set_option(section, "advance", "7", False)
        set_option(section, "pause", "7", False)
        set_option(section, "select", "6", False)
        section = "event"
        add_section(section)
        set_option(section, "user-event-id", "USEREVENT", False)
        set_option(section, "command-id-offset", "1", False)
        set_option(section, "command-key", "command", False)
        set_option(section, "cancel-flag-key", "cancel", False)
        section = "audio"
        add_section(section)
        set_option(section, "sfx-path", "aud/fx/", False)
        section = "interpolator-gui"
        add_section(section)
        set_option(section, "margin", "80", False)
        set_option(section, "marker-color", "255, 0, 0", False)
        set_option(section, "marker-size", "11", False)
        set_option(section, "curve-color", "0, 255, 0", False)
        set_option(section, "label-size", "16", False)
        set_option(section, "label-precision", "2", False)
        set_option(section, "axis-label-count", "8", False)
        set_option(section, "prompt-size", "380, 60", False)
        set_option(section, "prompt-border-color", "255, 0, 0", False)
        set_option(section, "prompt-border-width", "3", False)
        set_option(section, "prompt-character-limit", "21", False)
        set_option(section, "prompt-text-size", "42", False)
        set_option(section, "template-nodeset", "L 0 0, 1000 1", False)
        set_option(section, "template-nodeset-name", "template", False)
        set_option(section, "flat-y-range", "1", False)

    def add_section(self, name):
        if name not in self.order:
            self.order.append(name)
        RawConfigParser.add_section(self, name)

    def set(self, section, option, value, modifiable=True):
        if modifiable:
            if section not in self.order:
                self.order.append(section)
            if section not in self.modifiable:
                self.modifiable[section] = []
            if option not in self.modifiable[section]:
                self.modifiable[section].append(option)
        RawConfigParser.set(self, section, option, value)

    def read_project_config_file(self):
        path = self.locate_project_config_file()
        if path:
            fp = open(path)
            self.set_modifiable(fp)
            fp.seek(0)
            self.readfp(fp)
            fp.seek(0)
            self.set_order(fp)
            fp.close()
        else:
            self.print_debug("No configuration file found")

    def locate_project_config_file(self):
        rel_path = self.project_file_rel_path
        if not rel_path:
            rel_path = self.default_project_file_rel_path
        if exists(rel_path) and not self.is_shared_mode():
            return rel_path
        if self.resource_path:
            installed_path = join(self.resource_path, rel_path)
            if exists(installed_path):
                return installed_path

    def set_order(self, fp):
        self.order = order = []
        for line in file(self.locate_project_config_file()):
            result = match("^\s*\[(.*)\]\s*$", line)
            if result:
                order.append(result.group(1))

    def set_modifiable(self, fp):
        config = RawConfigParser()
        config.readfp(fp)
        modifiable = self.modifiable
        for section in config._sections:
            if section not in modifiable:
                modifiable[section] = []
            for option in config._sections[section]:
                if option != "__name__" and option not in modifiable[section]:
                    modifiable[section].append(option)

    def is_shared_mode(self):
        return "-s" in argv

    def print_debug(self, statement):
        if self.is_debug_mode():
            print statement

    def is_debug_mode(self):
        return "-d" in argv

    def modify_defaults(self):
        self.set_installation_path()
        self.set_resource_search_path()
        self.set_screen_captures_path()
        self.set_video_recordings_path()
        self.set_data_exclusion_list()
        self.set_requirements()

    def set_installation_path(self):
        self.set("setup", "installation-path",
                 join(self.get("setup", "installation-dir"),
                      self.get("setup", "package-root")), False)

    def set_resource_search_path(self):
        section, option = "setup", "resource-search-path"
        search_path = self.get(section, option)
        if self.resource_path:
            search_path.append(self.resource_path)
        else:
            search_path.append(self.get("setup", "installation-path"))
        self.set(section, option, search_path, False)

    def get(self, section, option):
        value = RawConfigParser.get(self, section, option)
        if value is None:
            value = self.get_substitute(section, option)
        return self.cast_value(section, option, value)

    def get_substitute(self, section, option):
        if section == "display":
            if option == "caption":
                return self.get("setup", "title")

    def cast_value(self, section, option, value):
        pair = section, option
        types = self.type_declarations
        if type(value) == str:
            if pair in types["bool"]:
                if value.lower() == self.get("setup", "lowercase-boolean-true"):
                    return True
                return False
            elif pair in types["int"]:
                return int(value)
            elif pair in types["float"]:
                return float(value)
            elif pair in types["path"]:
                return self.translate_path(value)
            elif pair in types["list"]:
                if value == "":
                    return []
                else:
                    return map(str.strip, value.split(types.list_member_sep))
            elif pair in types["int-list"]:
                return map(int, value.split(types.list_member_sep))
            elif pair in types["float-list"]:
                return map(float, value.split(types.list_member_sep))
        return value

    def set_screen_captures_path(self):
        section, option = "screen-captures", "path"
        if not self.has_option(section, option):
            self.set(section, option, join(self.build_home_path(),
                                           self.get(section, "rel-path")),
                     False)

    def build_home_path(self):
        return join("~", "." + self.get("setup", "package-root"))

    def set_video_recordings_path(self):
        section, option = "video-recordings", "path"
        if not self.has_option(section, option):
            self.set(section, option, join(self.build_home_path(),
                                           self.get(section, "rel-path")),
                     False)

    def set_data_exclusion_list(self):
        section, option = "setup", "data-exclude"
        exclude = []
        if self.has_option(section, option):
            exclude = self.get(section, option)
        exclude += [".git", ".gitignore", "README", "build/", "dist/",
                    "setup.py", "MANIFEST", "PKG-INFO",
                    self.get("setup", "changelog"),
                    self.get("setup", "package-root")]
        for location in self.get("setup", "additional-packages"):
            exclude.append(location)
        self.set(section, option, exclude, False)

    def set_requirements(self):
        section, option = "setup", "requirements"
        requirements = []
        if self.has_option(section, option):
            requirements = self.get(section, option)
        if "pygame" not in requirements:
            requirements.append("pygame")
        self.set(section, option, requirements, False)

    def get_section(self, section):
        assignments = {}
        for option in self.options(section):
            assignments[option] = self.get(section, option)
        return assignments

    def __repr__(self):
        config = {}
        for section in self.sections():
            config[section] = self.get_section(section)
        return pformat(config, 2, 1)

    def items(self, section):
        items = []
        for option in self.options(section):
            items.append((option, self.get(section, option)))
        return items

    def write(self, fp=None):
        modifiable = self.modifiable
        use_main = fp is None
        if use_main:
            path = self.locate_project_config_file()
            if not path:
                path = join(self.resource_path or "",
                            self.default_project_file_rel_path)
            fp = open(path, "w")
        break_line = False
        for section in self.order:
            if section in modifiable:
                break_line and fp.write("\n")
                fp.write("[%s]\n" % section)
                for option in modifiable[section]:
                    if self.has_option(section, option):
                        value = self.get(section, option)
                        fp.write("%s = %s\n" % (option,
                                                self.get_raw_value(value)))
                break_line = True
        if use_main:
            fp.close()

    def get_raw_value(self, value):
        if isinstance(value, list):
            raw = ""
            for ii, value in enumerate(value):
                if ii:
                    raw += ", "
                raw += str(value)
        else:
            raw = str(value)
        return raw

    def clear_section(self, section):
        if self.has_section(section):
            for option in self.options(section):
                self.remove_option(section, option)


class TypeDeclarations(dict):

    list_member_sep = ','

    defaults = {

        "display": {"int": ["frame-duration", "wait-duration",
                            "framerate-text-size"],

                    "bool": ["centered", "skip-frames", "fullscreen",
                             "show-framerate"],

                    "int-list": ["dimensions", "framerate-text-color",
                                 "framerate-text-background"]},

        "screen-captures": {"path": ["rel-path", "path"]},

        "video-recordings": {"path": ["rel-path", "path"],

                             "int": "framerate",

                             "bool": "enable"},

        "setup": {"list": ["classifiers", "resource-search-path",
                           "requirements", "data-exclude",
                           "additional-packages"],

                  "path": ["installation-dir", "changelog", "description-file",
                           "main-object", "icon-path", "windows-dist-path",
                           "package-root"]},

        "mouse": {"float": "double-click-time-limit",

                  "bool": "visible"},

        "keys": {"list": ["up", "right", "down", "left"]},

        "joy": {"int": ["advance", "pause", "select"]},

        "audio": {"path": "sfx-path"},

        "event": {"int": "command-id-offset"},

        "interpolator-gui": {"int": ["margin", "marker-size", "label-size",
                                     "axis-label-count", "label-precision",
                                     "prompt-border-width",
                                     "prompt-character-limit",
                                     "prompt-text-size", "flat-y-range"],

                             "int-list": ["marker-color", "curve-color",
                                          "prompt-size",
                                          "prompt-border-color"]},

        }

    additional_defaults = {}

    def __init__(self):
        dict.__init__(self, {"bool": [], "int": [], "float": [], "path": [],
                             "list": [], "int-list": [], "float-list": []})
        self.add_chart(self.defaults)
        self.add_chart(self.additional_defaults)

    def add(self, cast, section, option):
        self[cast].append((section, option))

    def add_chart(self, chart):
        for section, declarations in chart.iteritems():
            for cast, options in declarations.iteritems():
                if type(options) != list:
                    options = [options]
                for option in options:
                    self.add(cast, section, option)