#!/usr/bin/env python

# Copyright (C) 2016 Tobias Mueller <muelli at cryptobitch.de>
#
# This file is part of GPGME.
#
# GPGME 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 2 of the License, or
# (at your option) any later version.
#
# GPGME 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 Lesser General
# Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program; if not, see <https://www.gnu.org/licenses/>.

from __future__ import absolute_import, print_function, unicode_literals

import gpg
import support

del absolute_import, print_function, unicode_literals


def check_result(result, fpr, secret):
    assert result.considered == 1 or (secret and result.considered == 3)
    assert result.no_user_id == 0
    assert not ((secret and result.imported != 0) or
                (not secret and
                 (result.imported != 0 and result.imported != 1)))
    assert result.imported_rsa == 0
    assert not ((secret and
                 (result.unchanged != 0 and result.unchanged != 1)) or
                (not secret and
                 ((result.imported == 0 and result.unchanged != 1) or
                  (result.imported == 1 and result.unchanged != 0))))
    assert result.new_user_ids == 0
    assert result.new_sub_keys == 0
    assert not ((secret and (
        (result.secret_imported == 0 and result.new_signatures != 0) or
        (result.secret_imported == 1 and result.new_signatures > 1))) or
                (not secret and result.new_signatures != 0))
    assert result.new_revocations == 0
    assert not (
        (secret and result.secret_read != 1 and result.secret_read != 3) or
        (not secret and result.secret_read != 0))
    assert not (
        (secret and result.secret_imported != 0 and result.
            secret_imported != 1 and result.
            secret_imported != 2) or (not secret and result.
                                      secret_imported != 0))
    assert not ((secret and
                 ((result.secret_imported == 0 and result.
                   secret_unchanged != 1 and result.
                   secret_unchanged != 2) or (result.
                                              secret_imported == 1 and result.
                                              secret_unchanged != 0))) or
                (not secret and result.secret_unchanged != 0))
    assert result.not_imported == 0
    if secret:
        assert not (len(result.imports) in (0, 3))
    else:
        assert not (len(result.imports) in (0, 2))

    assert fpr == result.imports[0].fpr
    assert len(result.imports) == 1 or fpr == result.imports[1].fpr
    assert result.imports[0].result == 0


c = gpg.Context()

result = c.key_import(open(support.make_filename("pubkey-1.asc"), 'rb').read())
check_result(result, "ADAB7FCC1F4DE2616ECFA402AF82244F9CD9FD55", False)

result = c.key_import(open(support.make_filename("seckey-1.asc"), 'rb').read())
check_result(result, "ADAB7FCC1F4DE2616ECFA402AF82244F9CD9FD55", True)

try:
    result = c.key_import(b"thisisnotakey")
except ValueError:
    pass
assert result == "IMPORT_PROBLEM"