topic2k/pygcgen

# -*- coding: utf-8 -*-

from __future__ import print_function

import codecs
import os
import re
import sys

from .generator import Generator
from .options_parser import OptionsParser
from .pygcgen_exceptions import ChangelogGeneratorError

if sys.version_info.major == 3:
    # noinspection PyCompatibility
    from builtins import object


def checkname(filename):
    if not os.path.exists(filename):
        return filename
    mtch = re.match(r'(?P<filename>.*?)(?P<nr>\d+)(?P<rest>($|.md)?)',
                    filename)
    if mtch:
        # filename exists, add a number
        nr = int(mtch.group("nr")) + 1
        filename = "{0}{1}{2}".format(mtch.group("filename"), nr,
                                      mtch.group("rest"))
    else:
        # filename exists, but doesn't end with a number -> add one.
        filename = filename.rsplit(".", 1)
        filename = filename[0] + "_1" + (("." + filename[1]) if
                                         len(filename) > 1 else "")
    return checkname(filename)


class ChangelogGenerator(object):
    """ Class responsible for whole change log generation cycle. """

    def __init__(self, options=None):
        """
        :type options: list
        :param options: command line arguments
        """

        self.options = OptionsParser(options).options
        self.generator = Generator(self.options)

    def run(self):
        """
        The entry point of this script to generate change log
        'ChangelogGeneratorError' Is thrown when one
        of the specified tags was not found in list of tags.
        """
        if not self.options.project or not self.options.user:
            print("Project and/or user missing. "
                  "For help run:\n  pygcgen --help")
            return

        if not self.options.quiet:
            print("Generating changelog...")

        log = None
        try:
            log = self.generator.compound_changelog()
        except ChangelogGeneratorError as err:
            print("\n\033[91m\033[1m{}\x1b[0m".format(err.args[0]))
            exit(1)
        if not log:
            if not self.options.quiet:
                print("Empty changelog generated. {} not written.".format(
                    self.options.output)
                )
            return

        if self.options.no_overwrite:
            out = checkname(self.options.output)
        else:
            out = self.options.output

        with codecs.open(out, "w", "utf-8") as fh:
            fh.write(log)

        if not self.options.quiet:
            print("Done!")
            print("Generated changelog written to {}".format(out))


def run():
    ChangelogGenerator(sys.argv[1:]).run()


# def run_gui():
#     import wx
#     from .gui import GeneratorApp
#
#
#     app = GeneratorApp()
#     wx.MessageBox("Not yet implemented.", os.path.basename(sys.argv[0]))
#     # win = MainFrame(None).Show()
#     # app.MainLoop()
#


if __name__ == "__main__":
    run()