from django.core.management.base import BaseCommand, CommandError
from django.template.loader import render_to_string
from django.conf import settings
from navbar.templatetags.navbar import get_base_context
import os


class Command(BaseCommand):
    help = 'Generates the CSS files used by Pygments'

    def export_static(self, curr_lang):
        self.stdout.write('Plain HTML export for language %s…' % curr_lang)
        output_dir = os.path.join(self.export_dir, 'html')
        output_file = 'khanav_%s.html' % curr_lang
        os.makedirs(output_dir, mode=0o755, exist_ok=True)
        with open(os.path.join(output_dir, output_file), 'w') as f:
            ctx = get_base_context(curr_lang, '/')
            html = render_to_string('navbar/navbar.html', ctx)
            f.write(html)

    def export_docuwiki(self, langs):
        self.stdout.write(
            'Docuwiki export for languages %s…' % ', '.join(langs)
        )
        output_langs = []
        for lang in langs:
            ctx = get_base_context(lang, '/')
            output_langs.append({
                'name': lang,
                'html': render_to_string('navbar/navbar.html', ctx)
            })
        output_dir = os.path.join(self.export_dir, 'docuwiki')
        os.makedirs(output_dir, mode=0o755, exist_ok=True)
        with open(os.path.join(output_dir, 'khanav.php'), 'w') as f:
            f.write(render_to_string('navbar/export_docuwiki.html', {
                'output_langs': output_langs
            }))

    def add_arguments(self, parser):
        parser.add_argument(
            '--dir', '-d',
            help='Export directory.',
            default=os.path.join(settings.BASE_DIR, 'build/navbar'),
            dest='export_dir'
        )
        parser.add_argument(
            '--lang', '-l',
            help='Language to export.',
            action='append',
            dest='langs'
        )
        parser.add_argument(
            '--plain', '--html',
            help='Export for plain HTML.',
            action='store_true',
            dest='static'
        )
        parser.add_argument(
            '--docuwiki',
            help='Export for docuwiki.',
            action='store_true',
            dest='docuwiki'
        )

    def handle(self, *args, **options):
        if options['langs'] is None:
            raise CommandError('No language specified.')
        if 'all' in options['langs']:
            options['langs'] = [name for name, _ in settings.LANGUAGES]
        self.export_dir = options['export_dir']

        if options['static']:
            for lang in options['langs']:
                self.export_static(lang)
        if options['docuwiki']:
            self.export_docuwiki(options['langs'])
        self.stdout.write(self.style.SUCCESS('Ok.'))