#!/usr/bin/env python

# Copyright (C) 2016 g10 Code GmbH
#
# 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 os
import gpg
import support
_ = support  # to appease pyflakes.

del absolute_import, print_function, unicode_literals

expected_notations = {
    "laughing@me": ("Just Squeeze Me",
                    gpg.constants.sig.notation.HUMAN_READABLE),
    "preferred-email-encoding@pgp.com":
    ("pgpmime", gpg.constants.sig.notation.HUMAN_READABLE |
     gpg.constants.sig.notation.CRITICAL),
    None: ("https://www.gnu.org/policy/", 0),
}

# GnuPG prior to 2.1.13 did not report the critical flag correctly.
with gpg.Context() as c:
    version = c.engine_info.version
    have_correct_sig_data = not (
        version.startswith("1.") or version.startswith("2.0.") or
        (version.startswith("2.1.") and int(version[4:]) < 13))


def check_result(result):
    assert len(result.signatures) == 1, "Unexpected number of signatures"
    sig = result.signatures[0]
    assert len(sig.notations) == len(expected_notations)

    for r in sig.notations:
        assert 'name_len' not in dir(r)
        assert 'value_len' not in dir(r)
        assert r.name in expected_notations
        value, flags = expected_notations.pop(r.name)

        assert r.value == value, \
            "Expected {!r}, got {!r}".format(value, r.value)
        assert r.human_readable \
            == bool(flags & gpg.constants.sig.notation.HUMAN_READABLE)
        assert r.critical \
            == (bool(flags & gpg.constants.sig.notation.CRITICAL)
                if have_correct_sig_data else False)

    assert len(expected_notations) == 0


source = gpg.Data("Hallo Leute\n")
signed = gpg.Data()

c = gpg.Context()
for name, (value, flags) in expected_notations.items():
    c.sig_notation_add(name, value, flags)

c.op_sign(source, signed, gpg.constants.sig.mode.NORMAL)

signed.seek(0, os.SEEK_SET)
sink = gpg.Data()
c.op_verify(signed, None, sink)
result = c.op_verify_result()
check_result(result)