# -*- coding: utf-8 -*-
#
# Copyright © 2012 - 2019 Michal Čihař <michal@cihar.com>
#
# This file is part of Weblate <https://weblate.org/>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
#

import argparse
import json

from django.core.management.base import BaseCommand

from weblate.auth.models import User
from weblate.lang.models import Language
from weblate.trans.models import Project


class Command(BaseCommand):
    help = 'imports userdata from JSON dump of database'

    def add_arguments(self, parser):
        parser.add_argument(
            'json-file',
            type=argparse.FileType('r'),
            help='JSON file containing user data to import',
        )

    @staticmethod
    def import_watched(profile, userprofile):
        """Import user subscriptions."""
        # Add subscriptions
        for subscription in userprofile['watched']:
            try:
                profile.watched.add(Project.objects.get(slug=subscription))
            except Project.DoesNotExist:
                continue

    @staticmethod
    def update_languages(profile, userprofile):
        """Update user language preferences."""
        profile.language = userprofile['language']
        for lang in userprofile['secondary_languages']:
            profile.secondary_languages.add(Language.objects.auto_get_or_create(lang))
        for lang in userprofile['languages']:
            profile.languages.add(Language.objects.auto_get_or_create(lang))

    def handle_compat(self, data):
        """Compatibility with pre 3.6 dumps."""
        if 'basic' in data:
            return
        data['basic'] = {'username': data['username']}
        data['profile'] = {
            'translated': data['translated'],
            'suggested': data['suggested'],
            'language': data['language'],
            'uploaded': data.get('uploaded', 0),
            'secondary_languages': data['secondary_languages'],
            'languages': data['languages'],
            'watched': data['subscriptions'],
        }

    def handle(self, **options):
        """Create default set of groups.

        Also ptionally updates them and moves users around to default group.
        """
        userdata = json.load(options['json-file'])
        options['json-file'].close()

        for userprofile in userdata:
            self.handle_compat(userprofile)
            username = userprofile['basic']['username']
            try:
                user = User.objects.get(username=username)
                update = False
                profile = user.profile
                if not profile.language:
                    update = True

                # Merge stats
                profile.translated += userprofile['profile']['translated']
                profile.suggested += userprofile['profile']['suggested']
                profile.uploaded += userprofile['profile']['uploaded']

                # Update fields if we should
                if update:
                    self.update_languages(profile, userprofile['profile'])

                # Add subscriptions
                self.import_watched(profile, userprofile['profile'])

                profile.save()
            except User.DoesNotExist:
                self.stderr.write('User not found: {0}\n'.format(username))
