path: root/cgitize/main.py

# Copyright (c) 2018 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "cgitize" project.
# For details, see https://github.com/egor-tensin/cgitize.
# Distributed under the MIT License.

from argparse import ArgumentParser
import logging
import sys

from cgitize.cgit import CGitRepositories, CGitServer
from cgitize.config import Config
from cgitize.utils import setup_logging
from cgitize.version import __version__


def parse_args(argv=None):
    if argv is None:
        argv = sys.argv[1:]
    parser = ArgumentParser()
    parser.add_argument('--config', '-c', metavar='PATH',
                        default=Config.DEFAULT_PATH,
                        help='config file path')
    parser.add_argument('--repo', metavar='REPO_ID',
                        nargs='*', dest='repos',
                        help='repos to pull')
    parser.add_argument('--force', '-f', action='store_true',
                        help='overwrite existing repositories')
    parser.add_argument('--verbose', '-v', action='store_true',
                        help='verbose log output')
    parser.add_argument('--version', '-V', action='version',
                        version=f'%(prog)s {__version__}')
    return parser.parse_args(argv)


def setup_error_header_file(error_header_path, success, error=None):
    with open(error_header_path, 'w') as file:
        contents = ''
        if not success:
            contents = '''<p style="text-align: center; color: red; font-weight: bold;">'''
            contents += '''Some repositories couldn't be updated, please check application logs for details.'''
            if error is not None:
                contents += f'''<br>{type(error).__name__}: {error}'''
            contents += '''</p>\n'''
        file.write(contents)


def main(argv=None):
    args = parse_args(argv)
    with setup_logging(args.verbose):
        config = Config.read(args.config)
        success = True
        error = None

        try:
            cgit_server = CGitServer(config.main.clone_url)
            output = CGitRepositories(config.main.output_dir, cgit_server, force=args.force)
            for repo in config.parse_repositories():
                if args.repos is None or repo.name in args.repos:
                    success = success and output.update(repo)
        except Exception as e:
            success = False
            error = e

        if success:
            logging.info('All repositories were updated successfully')
        else:
            logging.warning("Some repositories couldn't be updated!")

        setup_error_header_file(config.main.error_header_path, success, error)
        return int(not success)


if __name__ == '__main__':
    sys.exit(main())