from os import makedirs
from os.path import exists, join
from sys import exc_info
from time import strftime

from pygame import image

from GameChild import *
from Input import *

class ScreenGrabber(GameChild):

    def __init__(self, game):
        GameChild.__init__(self, game)
        self.subscribe_to(self.get_custom_event_id(), self.save_display)

    def save_display(self, event):
        if self.is_command(event, "capture-screen"):
            directory = self.get_configuration().get("screen-captures", "path")
            try:
                if not exists(directory):
                    makedirs(directory)
                name = self.build_name()
                path = join(directory, name)
                capture = image.save(self.get_screen(), path)
                self.print_debug("Saved screen capture to {0}".format(path))
            except:
                self.print_debug("Couldn't save screen capture to {0}, {1}".\
                                 format(directory, exc_info()[1]))

    def build_name(self):
        config = self.get_configuration().get_section("screen-captures")
        prefix = config["file-name-format"]
        extension = config["file-extension"]
        return "{0}.{1}".format(strftime(prefix), extension)

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{0}\n{1}\n{2}".format(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_resources_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_resources_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["resources-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 pprint import pformat

from ConfigParser import RawConfigParser

class Configuration(RawConfigParser):

    default_project_file_rel_path = "config"
    default_resources_paths = [".", "resources"]

    def __init__(self, project_file_rel_path=None, resources_path=None,
                 type_declarations=None):
        RawConfigParser.__init__(self)
        self.project_file_rel_path = project_file_rel_path
        self.resources_path = resources_path
        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()))
        set_option(section, "additional-packages", "")
        set_option(section, "title", "") 
        set_option(section, "classifiers", "")
        set_option(section, "resources-search-path", "./, resources/")
        set_option(section, "installation-dir", "/usr/local/share/games/")
        set_option(section, "changelog", "changelog")
        set_option(section, "description-file", "")
        set_option(section, "init-script", "") 
        set_option(section, "version", "")
        set_option(section, "summary", "")
        set_option(section, "license", "")
        set_option(section, "platforms", "")
        set_option(section, "contact-name", "")
        set_option(section, "contact-email", "")
        set_option(section, "url", "")
        set_option(section, "requirements", "")
        set_option(section, "main-object", "pgfw/Game.py")
        set_option(section, "resources-path-identifier", "resources_path")
        set_option(section, "special-char-placeholder", "_")
        set_option(section, "whitespace-placeholder", "-")
        set_option(section, "icon-path", "")
        set_option(section, "windows-dist-path", "dist/win/")
        section = "display"
        add_section(section)
        set_option(section, "dimensions", "480, 320")
        set_option(section, "frame-duration", "33")
        set_option(section, "wait-duration", "2")
        set_option(section, "caption", "") 
        set_option(section, "centered", "yes")
        section = "screen-captures"
        add_section(section)
        set_option(section, "rel-path", "caps")
        set_option(section, "file-name-format", "%Y-%m-%d_%H:%M:%S")
        set_option(section, "file-extension", "png")
        section = "mouse"
        add_section(section)
        set_option(section, "double-click-time-limit", ".5")
        section = "keys"
        add_section(section)
        set_option(section, "up", "K_UP, K_w")
        set_option(section, "right", "K_RIGHT, K_d")
        set_option(section, "down", "K_DOWN, K_s")
        set_option(section, "left", "K_LEFT, K_a")
        set_option(section, "capture-screen", "K_F9")
        set_option(section, "toggle-fullscreen", "K_F11")
        set_option(section, "reset-game", "K_F8")
        section = "joy"
        add_section(section)
        set_option(section, "advance", "7")
        set_option(section, "pause", "7")
        section = "event"
        add_section(section)
        set_option(section, "custom-event-id", "USEREVENT")
        set_option(section, "command-event-name", "command")
        section = "audio"
        add_section(section)
        set_option(section, "sfx-path", "aud/fx/")

    def read_project_config_file(self):
        path = self.locate_project_config_file()
        if path:
            self.read(path)
        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.resources_path:
            installed_path = join(self.resources_path, rel_path)
            if exists(installed_path):
                return installed_path

    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_resources_search_path()
        self.set_screen_captures_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")))

    def set_resources_search_path(self):
        section, option = "setup", "resources-search-path"
        search_path = self.get(section, option)
        if self.resources_path:
            search_path.append(self.resources_path)
        else:
            search_path.append(self.get("setup", "installation-path"))
        self.set(section, option, search_path)

    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"]:
                return True if value == "yes" else 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))
        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")))

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

    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)

    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)

    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
    

class TypeDeclarations(dict):

    list_member_sep = ','
    defaults = {"display": {"int": ["frame-duration", "wait-duration"],
                            "bool": "centered",
                            "int-list": "dimensions"},
                "screen-captures": {"path": "path"},
                "setup": {"list": ["classifiers", "resources-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"},
                "keys": {"list": ["up", "right", "down", "left"]},
                "joy": {"int": ["advance", "pause"]},
                "audio": {"path": "sfx-path"}}
    additional_defaults = {}

    def __init__(self):
        dict.__init__(self, {"bool": [], "int": [], "float": [], "path": [],
                             "list": [], "int-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)