# SPDX-FileCopyrightText: 2022 James R. Barlow
# SPDX-License-Identifier: CC0-1.0

from __future__ import annotations

import os
import re
from datetime import datetime, timedelta, timezone
from pathlib import Path
from xml.etree import ElementTree as ET

import pytest
from conftest import skip_if_ci, skip_if_pypy
from hypothesis import assume, example, given, settings
from hypothesis import strategies as st
from hypothesis.strategies import integers
from lxml.etree import XMLSyntaxError

import pikepdf
from pikepdf import Dictionary, Name, PasswordError, Pdf, Stream
from pikepdf.models.metadata import (
    XMP_NS_DC,
    XMP_NS_PDF,
    XMP_NS_XMP,
    DateConverter,
    PdfMetadata,
    XmpDocument,
    decode_pdf_date,
)

# pylint: disable=redefined-outer-name,pointless-statement


@pytest.fixture
def libxmp_meta():
    try:
        libxmp = pytest.importorskip('libxmp')
    except Exception as e:  # pylint: disable=broad-except
        # libxmp raises a nonstandard exception sometimes so the standard
        # pytest.importorskip doesn't trap it *irritated sigh*.
        pytest.skip(str(e))
    return libxmp.XMPMeta


@pytest.fixture
def vera(resources):
    # Has XMP but no docinfo
    with Pdf.open(resources / 'veraPDF test suite 6-2-10-t02-pass-a.pdf') as pdf:
        yield pdf


@pytest.fixture
def graph(resources):
    # Has XMP and docinfo, all standard format XMP
    with Pdf.open(resources / 'graph.pdf') as pdf:
        yield pdf


@pytest.fixture
def sandwich(resources):
    # Has XMP, docinfo, <?adobe-xap-filters esc="CRLF"?>, shorthand attribute XMP
    with Pdf.open(resources / 'sandwich.pdf') as pdf:
        yield pdf


@pytest.fixture
def trivial(resources):
    # Has no XMP or docinfo
    with Pdf.open(resources / 'pal-1bit-trivial.pdf') as pdf:
        yield pdf


@pytest.fixture
def invalid_creationdate(resources):
    # Has nuls in docinfo, old PDF
    with Pdf.open(resources / 'invalid_creationdate.pdf') as pdf:
        yield pdf


def test_lowlevel(sandwich):
    meta = sandwich.open_metadata()
    assert meta._qname('pdf:Producer') == '{http://ns.adobe.com/pdf/1.3/}Producer'
    assert (
        meta._prefix_from_uri('{http://ns.adobe.com/pdf/1.3/}Producer')
        == 'pdf:Producer'
    )
    assert 'pdf:Producer' in meta
    assert '{http://ns.adobe.com/pdf/1.3/}Producer' in meta
    assert 'xmp:CreateDate' in meta
    assert meta['xmp:ModifyDate'].startswith('2017')
    assert len(meta) > 0
    assert meta['dc:title'] == 'Untitled'

    assert 'pdf:invalid' not in meta
    assert '{http://ns.adobe.com/pdf/1.3/}invalid' not in meta
    with pytest.raises(TypeError):
        assert ['hi'] in meta

    with pytest.raises(KeyError):
        meta['dc:invalid']
    with pytest.raises(KeyError):
        meta['{http://ns.adobe.com/pdf/1.3/}invalid']
    with pytest.raises(KeyError):
        meta['{http://invalid.com/ns/}doublyinvalid']


def test_no_info(vera, outdir):
    assert vera.trailer.get('/Info') is None, 'need a test file with no /Info'

    assert len(vera.docinfo) == 0
    creator = 'pikepdf test suite'
    vera.docinfo['/Creator'] = creator
    assert vera.docinfo.is_indirect, "/Info must be an indirect object"
    vera.save(outdir / 'out.pdf')

    with Pdf.open(outdir / 'out.pdf') as new:
        assert new.docinfo['/Creator'] == creator


def test_update_info(graph, outdir):
    new_title = '我敢打赌，你只是想看看这意味着什么'
    graph.docinfo['/Title'] = new_title
    graph.save(outdir / 'out.pdf')

    with Pdf.open(outdir / 'out.pdf') as new:
        assert new.docinfo['/Title'] == new_title
        assert graph.docinfo['/Author'] == new.docinfo['/Author']

        with pytest.raises(ValueError):
            new.docinfo = Dictionary({'/Keywords': 'bob'})

        new.docinfo = new.make_indirect(Dictionary({'/Keywords': 'bob'}))
        assert new.docinfo.is_indirect, "/Info must be an indirect object"


def test_copy_info(vera, graph, outdir):
    vera.docinfo = vera.copy_foreign(graph.docinfo)  # vera has no docinfo
    assert vera.docinfo.is_indirect, "/Info must be an indirect object"
    vera.save(outdir / 'out.pdf')


def test_del_info(graph, outpdf):
    assert Name.Info in graph.trailer
    del graph.docinfo
    assert Name.Info not in graph.trailer
    graph.save(outpdf)
    with Pdf.open(outpdf) as new:
        assert Name.Info not in new.trailer


def test_add_new_xmp_and_mark(trivial):
    with trivial.open_metadata(
        set_pikepdf_as_editor=False, update_docinfo=False
    ) as xmp_view:
        assert not xmp_view

    with trivial.open_metadata(update_docinfo=False) as xmp:
        assert not xmp  # No changes at this point
    del xmp

    print(trivial.Root.Metadata.read_bytes())

    with trivial.open_metadata(update_docinfo=False) as xmp:
        assert xmp['pdf:Producer'] == 'pikepdf ' + pikepdf.__version__
        assert 'xmp:MetadataDate' in xmp


def test_update_docinfo(vera):
    with vera.open_metadata(set_pikepdf_as_editor=False, update_docinfo=True) as xmp:
        pass
    assert xmp['pdf:Producer'] == vera.docinfo[Name.Producer]
    assert xmp['xmp:CreatorTool'] == vera.docinfo[Name.Creator]
    assert xmp['dc:creator'][0] == vera.docinfo[Name.Author]

    # Test delete propagation
    with vera.open_metadata(set_pikepdf_as_editor=False, update_docinfo=True) as xmp:
        del xmp['dc:creator']
        with pytest.raises(KeyError, match='dc:nonexistent'):
            del xmp['dc:nonexistent']
    assert 'dc:creator' not in xmp
    assert Name.Author not in vera.docinfo


@pytest.mark.parametrize(
    'filename', list((Path(__file__).parent / 'resources').glob('*.pdf'))
)
def test_roundtrip(filename):
    try:
        with Pdf.open(filename) as pdf:
            with pdf.open_metadata() as xmp:
                for k in xmp.keys():
                    if 'Date' not in k:
                        xmp[k] = 'A'
            assert '<?xpacket' not in str(xmp)
    except PasswordError:
        return


def test_build_metadata(trivial, graph, outdir):
    with trivial.open_metadata(set_pikepdf_as_editor=False) as xmp:
        xmp.load_from_docinfo(graph.docinfo)
    trivial.save(outdir / 'tmp.pdf')

    with pikepdf.open(outdir / 'tmp.pdf') as pdf:
        assert pdf.Root.Metadata.Type == Name.Metadata
        assert pdf.Root.Metadata.Subtype == Name.XML
        with pdf.open_metadata(set_pikepdf_as_editor=False) as xmp:
            assert 'pdf:Producer' not in xmp
            xmp_date = xmp['xmp:CreateDate']
            docinfo_date = decode_pdf_date(trivial.docinfo[Name.CreationDate])
            assert xmp_date == docinfo_date.isoformat()


def test_python_xmp_validate_add(trivial, libxmp_meta):
    with trivial.open_metadata() as xmp:
        xmp['dc:creator'] = ['Bob', 'Doug']
        xmp['dc:title'] = 'Title'
        xmp['dc:publisher'] = {'Mackenzie'}

    xmp_str = str(xmp).replace('\n', '')
    assert '<rdf:Seq><rdf:li>Bob</rdf:li><rdf:li>Doug</rdf:li>' in xmp_str
    assert '<rdf:Bag><rdf:li>Mackenzie</rdf:li>' in xmp_str

    xmpmeta = libxmp_meta(xmp_str=str(xmp))
    DC = XMP_NS_DC
    assert xmpmeta.does_array_item_exist(DC, 'creator', 'Bob')
    assert xmpmeta.does_array_item_exist(DC, 'creator', 'Doug')
    assert xmpmeta.get_localized_text(DC, 'title', None, 'x-default') == 'Title'
    assert xmpmeta.does_array_item_exist(DC, 'publisher', 'Mackenzie')


def test_python_xmp_validate_change_list(graph, libxmp_meta):
    with graph.open_metadata() as xmp:
        assert 'dc:creator' in xmp
        xmp['dc:creator'] = ['Dobby', 'Kreacher']
    assert str(xmp)
    xmpmeta = libxmp_meta(xmp_str=str(xmp))
    DC = XMP_NS_DC
    assert xmpmeta.does_array_item_exist(DC, 'creator', 'Dobby')
    assert xmpmeta.does_array_item_exist(DC, 'creator', 'Kreacher')


def test_python_xmp_validate_change(sandwich, libxmp_meta):
    with sandwich.open_metadata() as xmp:
        assert 'xmp:CreatorTool' in xmp
        xmp['xmp:CreatorTool'] = 'Creator'  # Exists as a xml tag text
        xmp['pdf:Producer'] = 'Producer'  # Exists as a tag node
    assert str(xmp)
    xmpmeta = libxmp_meta(xmp_str=str(xmp))
    assert xmpmeta.does_property_exist(XMP_NS_XMP, 'CreatorTool')
    assert xmpmeta.does_property_exist(XMP_NS_PDF, 'Producer')


def test_decode_pdf_date():
    VALS = [
        ('20160220040559', datetime(2016, 2, 20, 4, 5, 59)),
        ("20180101010101Z00'00'", datetime(2018, 1, 1, 1, 1, 1, tzinfo=timezone.utc)),
        ("20180101010101Z00'00", datetime(2018, 1, 1, 1, 1, 1, tzinfo=timezone.utc)),
        ("20180101010101Z", datetime(2018, 1, 1, 1, 1, 1, tzinfo=timezone.utc)),
        ("20180101010101+0000", datetime(2018, 1, 1, 1, 1, 1, tzinfo=timezone.utc)),
        (
            "20180101010101+0100",
            datetime(2018, 1, 1, 1, 1, 1, tzinfo=timezone(timedelta(hours=1))),
        ),
    ]
    for s, d in VALS:
        assert decode_pdf_date(s) == d


def test_date_docinfo_from_xmp():
    VALS = [
        ('2018-12-04T03:02:01', "D:20181204030201"),
        ('2018-12-15T07:36:43Z', "D:20181215073643+00'00"),
        ('2018-12-04T03:02:01-01:00', "D:20181204030201-01'00"),
    ]
    for xmp_val, docinfo_val in VALS:
        assert DateConverter.docinfo_from_xmp(xmp_val) == docinfo_val


def test_reduced_precision_date_docinfo_from_xmp():
    VALS = [
        ('2023', 'D:2023'),
        ('2023-11', 'D:202311'),
        ('1999', 'D:1999'),
        ('2000-01', 'D:200001'),
    ]
    for xmp_val, docinfo_val in VALS:
        assert DateConverter.docinfo_from_xmp(xmp_val) == docinfo_val


def test_reduced_precision_date_xmp_from_docinfo():
    VALS = [
        ('D:2023', '2023'),
        ('D:202311', '2023-11'),
        ('D:1999', '1999'),
        ('D:200001', '2000-01'),
    ]
    for docinfo_val, xmp_val in VALS:
        assert DateConverter.xmp_from_docinfo(docinfo_val) == xmp_val


def test_reduced_precision_date_roundtrip():
    VALS = ['2023', '2023-11', '1999', '2000-01']
    for xmp_date in VALS:
        pdf_date = DateConverter.docinfo_from_xmp(xmp_date)
        roundtrip = DateConverter.xmp_from_docinfo(pdf_date)
        assert roundtrip == xmp_date


def test_reduced_precision_date_in_metadata(trivial, outpdf):
    with trivial.open_metadata() as meta:
        meta['xmp:CreateDate'] = '2023'
    trivial.save(outpdf)

    with pikepdf.open(outpdf) as pdf:
        with pdf.open_metadata() as meta:
            assert meta['xmp:CreateDate'] == '2023'
        assert str(pdf.docinfo.get(Name.CreationDate, '')).startswith('D:2023')


def test_reduced_precision_date_year_month_in_metadata(trivial, outpdf):
    with trivial.open_metadata() as meta:
        meta['xmp:CreateDate'] = '2023-11'
    trivial.save(outpdf)

    with pikepdf.open(outpdf) as pdf:
        with pdf.open_metadata() as meta:
            assert meta['xmp:CreateDate'] == '2023-11'
        assert str(pdf.docinfo.get(Name.CreationDate, '')).startswith('D:202311')


@given(
    integers(-9999, 9999),
    integers(0, 99),
    integers(0, 99),
    integers(0, 99),
    integers(0, 99),
    integers(0, 99),
)
@example(1, 1, 1, 0, 0, 0)
def test_random_dates(year, month, day, hour, mins, sec):
    date_args = year, month, day, hour, mins, sec
    xmp = '{:04d}-{:02d}-{:02d}T{:02d}:{:02d}:{:02d}'.format(*date_args)
    docinfo = 'D:{:04d}{:02d}{:02d}{:02d}{:02d}{:02d}'.format(*date_args)

    try:
        converted = DateConverter.docinfo_from_xmp(xmp)
    except ValueError:
        pass
    else:
        assert converted == docinfo

    try:
        converted = DateConverter.xmp_from_docinfo(docinfo)
    except ValueError:
        pass
    else:
        assert converted == xmp


def test_bad_char_rejection(trivial):
    with trivial.open_metadata() as xmp:
        xmp['dc:description'] = 'Bad characters \x00 \x01 \x02'
        xmp['dc:creator'] = ['\ue001bad', '\ufff0bad']
    ET.fromstring(str(xmp))


def test_xpacket_generation(sandwich):
    xmpstr1 = sandwich.Root.Metadata.read_bytes()
    xpacket_begin = b'<?xpacket begin='
    xpacket_end = b'<?xpacket end='
    assert xmpstr1.startswith(xpacket_begin)

    with sandwich.open_metadata() as xmp:
        xmp['dc:creator'] = ['Foo']

    xmpstr2 = sandwich.Root.Metadata.read_bytes()
    assert xmpstr2.startswith(xpacket_begin)

    def only_one_substring(s, subs):
        return s.find(subs) == s.rfind(subs)

    assert only_one_substring(xmpstr2, xpacket_begin)
    assert only_one_substring(xmpstr2, xpacket_end)


def test_no_rdf_subtags(graph):
    xmp = graph.open_metadata()
    assert '{http://www.w3.org/1999/02/22-rdf-syntax-ns#}Alt' not in xmp.keys()
    assert '{http://www.w3.org/1999/02/22-rdf-syntax-ns#}Bag' not in xmp.keys()
    assert '{http://www.w3.org/1999/02/22-rdf-syntax-ns#}li' not in xmp.keys()


def test_remove_attribute_metadata(sandwich):
    with sandwich.open_metadata() as xmp:
        del xmp['pdfaid:part']
    assert 'pdfaid:part' not in xmp
    assert 'pdfaid:conformance' in xmp

    with sandwich.open_metadata() as xmp:
        del xmp['pdfaid:conformance']

    # Ensure the whole node was deleted
    assert not re.search(r'rdf:Description xmlns:[^\s]+ rdf:about=""/', str(xmp))


def test_docinfo_problems(sandwich, invalid_creationdate):
    sandwich.Root.Metadata = Stream(
        sandwich,
        b"""
        <?xpacket begin='\xc3\xaf\xc2\xbb\xc2\xbf' id='W5M0MpCehiHzreSzNTczkc9d'?>
        <?adobe-xap-filters esc="CRLF"?>
        <x:xmpmeta xmlns:x='adobe:ns:meta/' x:xmptk='XMP toolkit 2.9.1-13, framework 1.6'>
        <rdf:RDF xmlns:rdf='http://www.w3.org/1999/02/22-rdf-syntax-ns#' xmlns:iX='http://ns.adobe.com/iX/1.0/'>
        <rdf:Description rdf:about='uuid:873a76ba-4819-11f4-0000-5c5716666531' xmlns:pdf='http://ns.adobe.com/pdf/1.3/' pdf:Producer='GPL Ghostscript 9.26'/>
        <rdf:Description rdf:about='uuid:873a76ba-4819-11f4-0000-5c5716666531' xmlns:xmp='http://ns.adobe.com/xap/1.0/'><xmp:ModifyDate>2019-01-04T00:44:42-08:00</xmp:ModifyDate>
        <xmp:CreateDate>2019-01-04T00:44:42-08:00</xmp:CreateDate>
        <xmp:CreatorTool>Acrobat 4.0 Scan Plug-in for Windows&#0;</xmp:CreatorTool></rdf:Description>
        <rdf:Description rdf:about='uuid:873a76ba-4819-11f4-0000-5c5716666531' xmlns:xapMM='http://ns.adobe.com/xap/1.0/mm/' xapMM:DocumentID='uuid:873a76ba-4819-11f4-0000-5c5716666531'/>
        <rdf:Description rdf:about='uuid:873a76ba-4819-11f4-0000-5c5716666531' xmlns:dc='http://purl.org/dc/elements/1.1/' dc:format='application/pdf'><dc:title><rdf:Alt><rdf:li xml:lang='x-default'>Untitled</rdf:li></rdf:Alt></dc:title></rdf:Description>
        </rdf:RDF>
        </x:xmpmeta>
        """,
    )
    meta = sandwich.open_metadata()
    meta._load()  # File has invalid XML sequence &#0;
    with meta:
        with pytest.warns(UserWarning) as warned:
            meta.load_from_docinfo(invalid_creationdate.docinfo)
        assert 'could not be copied' in warned[0].message.args[0]
        with pytest.raises(ValueError):
            meta.load_from_docinfo(invalid_creationdate.docinfo, raise_failure=True)

    with pytest.warns(UserWarning) as warned:
        with meta as xmp:
            xmp['xmp:CreateDate'] = 'invalid date'
        assert 'could not be updated' in warned[0].message.args[0]


def test_docinfo_delete_missing(sandwich):
    with sandwich.open_metadata() as m:
        d = Dictionary(Creator="test creator")
        assert 'xmp:CreateDate' in m
        assert m['xmp:CreatorTool'] != 'test creator'
        m.load_from_docinfo(d, delete_missing=True)
        assert m['xmp:CreatorTool'] == 'test creator'
        assert 'xmp:CreateDate' not in m


def test_present_bug_empty_tags(trivial):
    trivial.Root.Metadata = Stream(
        trivial,
        b"""
        <?xpacket begin='\xc3\xaf\xc2\xbb\xc2\xbf' id='W5M0MpCehiHzreSzNTczkc9d'?>
        <?adobe-xap-filters esc="CRLF"?>
        <x:xmpmeta xmlns:x='adobe:ns:meta/' x:xmptk='XMP toolkit 2.9.1-13, framework 1.6'>
        <rdf:RDF xmlns:rdf='http://www.w3.org/1999/02/22-rdf-syntax-ns#' xmlns:iX='http://ns.adobe.com/iX/1.0/'>
        <rdf:Description rdf:about=""><dc:creator xmlns:dc="http://purl.org/dc/elements/1.1/"><rdf:Seq><rdf:li/></rdf:Seq></dc:creator></rdf:Description>
        </rdf:RDF>
        </x:xmpmeta>
        """,
    )
    with trivial.open_metadata(update_docinfo=True) as meta:
        assert len(meta) > 0
    assert Name.Author not in trivial.docinfo


def test_wrong_xml(sandwich):
    sandwich.Root.Metadata = Stream(
        sandwich,
        b"""
        <test><xml>This is valid xml but not valid XMP</xml></test>
    """.strip(),
    )
    # With lazy loading removed, ValueError is raised immediately in open_metadata()
    with pytest.raises(ValueError, match='not XMP'):
        sandwich.open_metadata(strict=True)


def test_no_x_xmpmeta(trivial):
    trivial.Root.Metadata = Stream(
        trivial,
        b"""
        <?xpacket begin="\xef\xbb\xbf" id="W5M0MpCehiHzreSzNTczkc9d"?>
        <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
                xmlns:xmp="http://ns.adobe.com/xap/1.0/">
        <rdf:Description rdf:about=""
                        xmlns:pdfaid="http://www.aiim.org/pdfa/ns/id/"
                        xmlns:xmp="http://ns.adobe.com/xap/1.0/">
            <pdfaid:part>1</pdfaid:part>
            <pdfaid:conformance>A</pdfaid:conformance>
            <xmp:CreatorTool>Simple Scan 3.30.2</xmp:CreatorTool>
            <xmp:CreateDate>2019-02-05T07:08:46+01:00</xmp:CreateDate>
            <xmp:ModifyDate>2019-02-05T07:08:46+01:00</xmp:ModifyDate>
            <xmp:MetadataDate>2019-02-05T07:08:46+01:00</xmp:MetadataDate>
        </rdf:Description>
        </rdf:RDF>
        <?xpacket end="w"?>
    """.strip(),
    )

    with trivial.open_metadata() as xmp:
        assert xmp._get_rdf_root() is not None
        xmp['pdfaid:part'] = '2'
    assert xmp['pdfaid:part'] == '2'


@pytest.mark.parametrize(
    'xml',
    [
        (b"      \n   "),
        (b" <"),
        (
            b'<?xpacket begin="\xef\xbb\xbf" id="W5M0MpCehiHzreSzNTczkc9d"?>\n'
            b'<?xpacket end=""?>\n'
        ),
        (
            b'<?xpacket begin="" id=""?>\n'
            b'<x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="">\n'
            b'</x:xmpmeta>\n'
            b'<?xpacket end=""?>\n'
        ),
    ],
)
def test_degenerate_xml_recoverable(trivial, xml):
    trivial.Root.Metadata = trivial.make_stream(xml)
    with trivial.open_metadata(strict=False) as xmp:
        xmp['pdfaid:part'] = '2'
    assert xmp['pdfaid:part'] == '2'

    with trivial.open_metadata(strict=True) as xmp:
        xmp['pdfaid:part'] = '5'


@settings(deadline=None)
@given(st.integers(min_value=1, max_value=1350))
@example(531)
@example(548)
@example(1154)
@example(1155)
@example(1195)
@example(1303)
@pytest.mark.filterwarnings('ignore:The DocumentInfo field')
def test_truncated_xml(resources, idx):
    with Pdf.open(resources / 'sandwich.pdf') as sandwich:
        data = sandwich.Root.Metadata.read_bytes()
        assume(idx < len(data))

        sandwich.Root.Metadata = sandwich.make_stream(data[0:idx])
        try:
            with sandwich.open_metadata(strict=True) as xmp:
                xmp['pdfaid:part'] = '5'
        except (XMLSyntaxError, AssertionError):
            pass

        with sandwich.open_metadata(strict=False) as xmp:
            xmp['pdfaid:part'] = '7'


def test_pdf_version_update(graph, outdir, libxmp_meta):
    def get_xmp_version(filename):
        with pikepdf.open(filename) as pdf:
            meta = pdf.open_metadata()
            xmp = libxmp_meta(xmp_str=str(meta))
            try:
                return xmp.get_property('http://ns.adobe.com/pdf/1.3/', 'PDFVersion')
            except Exception:
                return ''

    # We don't update PDFVersion unless it is present, even if we change the PDF version
    graph.save(
        outdir / 'empty_xmp_pdfversion.pdf',
        force_version='1.7',
        fix_metadata_version=True,
    )
    assert get_xmp_version(outdir / 'empty_xmp_pdfversion.pdf') == ''

    # Add PDFVersion field for remaining tests
    with graph.open_metadata() as m:
        m['pdf:PDFVersion'] = graph.pdf_version

    # Confirm we don't update the field when the flag is false
    graph.save(
        outdir / 'inconsistent_version.pdf',
        force_version='1.6',
        fix_metadata_version=False,
    )
    assert get_xmp_version(outdir / 'inconsistent_version.pdf') == '1.3'

    # Confirm we update if present
    graph.save(outdir / 'consistent_version.pdf', force_version='1.5')
    assert get_xmp_version(outdir / 'consistent_version.pdf') == '1.5'


def test_extension_level(trivial, outpdf):
    trivial.save(outpdf, min_version=('1.6', 314159))
    with pikepdf.open(outpdf) as pdf:
        assert pdf.pdf_version >= '1.6' and pdf.extension_level == 314159

    trivial.save(outpdf, force_version=('1.7', 42))
    with pikepdf.open(outpdf) as pdf:
        assert pdf.pdf_version == '1.7' and pdf.extension_level == 42

    with pytest.raises(TypeError):
        trivial.save(outpdf, force_version=('1.7', 'invalid extension level'))


@settings(deadline=60000)
@given(
    st.dictionaries(
        keys=st.sampled_from(
            [
                "/Author",
                "/Subject",
                "/Title",
                "/Keywords",
                "/Producer",
                "/CreationDate",
                "/Creator",
                "/ModDate",
                "/Dummy",
            ]
        ),
        values=st.recursive(
            st.none() | st.binary(max_size=16) | st.booleans(),
            lambda children: st.lists(children, min_size=0, max_size=4),
            max_leaves=2,
        ),
    )
)
@skip_if_pypy
def test_random_docinfo(docinfo):
    p = pikepdf.new()
    with p.open_metadata() as m:
        pdf_docinfo = pikepdf.Dictionary(docinfo)

        try:
            m.load_from_docinfo(pdf_docinfo, raise_failure=True)
        except ValueError as e:
            assert 'could not be copied to XMP' in str(e) or '/Dummy' in str(e)
        else:
            ET.fromstring(str(m))  # ensure we can parse it


@given(
    st.dictionaries(
        keys=st.sampled_from(
            [
                "/Author",
                "/Subject",
                "/Title",
                "/Keywords",
                "/Producer",
                "/Creator",
            ]
        ),
        values=st.none() | st.characters(blacklist_categories=('Cs',)),
    )
)
@skip_if_pypy
@skip_if_ci
def test_random_valid_docinfo(docinfo):
    p = pikepdf.new()
    with p.open_metadata() as m:
        pdf_docinfo = pikepdf.Dictionary(docinfo)

        m.load_from_docinfo(pdf_docinfo, raise_failure=True)
        ET.fromstring(str(m))  # ensure we can parse it


@pytest.mark.parametrize('author', ['Queen, C.', 'King, S.'])
def test_issue_162(trivial, author):
    trivial.Root.Metadata = Stream(
        trivial,
        b"""
        <?xpacket begin="\xef\xbb\xbf" id="W5M0MpCehiHzreSzNTczkc9d"?>
        <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#"
                xmlns:xmp="http://ns.adobe.com/xap/1.0/"
                xmlns:dc="http://purl.org/dc/elements/1.1/">
        <rdf:Description rdf:about="" xmlns:dc="http://purl.org/dc/elements/1.1/" dc:creator="Foo"></rdf:Description>
        </rdf:RDF>
        <?xpacket end="w"?>""",
    )
    with trivial.open_metadata() as m:
        docinfo = pikepdf.Dictionary(Author=author)
        with pytest.warns(UserWarning, match=r'Merging elements'):
            m.load_from_docinfo(docinfo, raise_failure=True)
        assert m['dc:creator'] == [author]


def test_set_empty_string(graph):
    with graph.open_metadata() as m:
        m['dc:title'] = 'a'

    generated_xmp = graph.Root.Metadata.read_bytes()
    print(generated_xmp)
    assert generated_xmp.count(b'<dc:title>') == 1


@pytest.mark.parametrize('fix_metadata', [True, False])
def test_dont_create_empty_xmp(trivial, outpdf, fix_metadata):
    trivial.save(outpdf, fix_metadata_version=fix_metadata)

    with pikepdf.open(outpdf) as p:
        assert Name.Metadata not in p.Root


@pytest.mark.parametrize('fix_metadata', [True, False])
def test_dont_create_empty_docinfo(trivial, outpdf, fix_metadata):
    del trivial.trailer.Info
    trivial.save(outpdf, fix_metadata_version=fix_metadata)

    with pikepdf.open(outpdf) as p:
        assert Name.Info not in p.trailer


def test_issue_100(trivial):
    with (
        trivial.open_metadata() as m,
        pytest.warns(UserWarning, match="no XMP equivalent"),
    ):
        m.load_from_docinfo({'/AAPL:Example': pikepdf.Array([42])})


def test_issue_135_title_rdf_bag(trivial):
    with (
        trivial.open_metadata(update_docinfo=True) as xmp,
        pytest.warns(UserWarning, match="Merging elements"),
    ):
        xmp['dc:title'] = {'Title 1', 'Title 2'}
    with trivial.open_metadata(update_docinfo=False) as xmp:
        assert b'Title 1; Title 2</rdf:li></rdf:Alt></dc:title>' in xmp._get_xml_bytes()


def test_xmp_metadatadate_timezone(sandwich, outpdf):
    with sandwich.open_metadata(set_pikepdf_as_editor=True):
        pass
    sandwich.save(outpdf)
    with Pdf.open(outpdf) as pdf:
        with pdf.open_metadata() as m:
            dt = datetime.fromisoformat(m['xmp:MetadataDate'])
            assert dt.tzinfo is not None
            assert dt.tzinfo == timezone.utc


def test_modify_not_opened(graph):
    m = graph.open_metadata()
    with pytest.raises(RuntimeError, match='not opened for editing'):
        m['pdf:Producer'] = 'pytest'


def test_exception_undoes_edits(graph):
    try:
        with graph.open_metadata() as m:
            m['dc:format'] = 'application/pdf-demo'
            raise ValueError('test_exception_aborts_edits')
    except ValueError as e:
        if 'test_exception_aborts_edits' not in str(e):
            raise
        m = graph.open_metadata()
        assert m['dc:format'] != 'application/pdf-demo'


def test_xxe(trivial, outdir):
    secret = outdir / 'secret.txt'
    secret.write_text("This is a secret")
    trivial.Root.Metadata = Stream(
        trivial,
        b"""\
<?xpacket begin='\xef\xbb\xbf' id='W5M0MpCehiHzreSzNTczkc9d'?>
<!DOCTYPE rdf:RDF [<!ENTITY xxe SYSTEM "file://%s">]>
<x:xmpmeta xmlns:x='adobe:ns:meta/' x:xmptk='Image'>
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
<note>
<to>&xxe;</to>
<from>xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx</from>
</note>
</rdf:RDF>
</x:xmpmeta>
<?xpacket end='w'?>
    """
        % os.fsencode(secret),
    )
    with trivial.open_metadata() as m:
        assert 'This is a secret' not in str(m)


def test_qname_no_namespace():
    assert PdfMetadata._qname('abc') == 'abc'


def test_register_xmlns():
    PdfMetadata.register_xml_namespace('http://github.com/pikepdf/pikepdf/', 'pikepdf')
    assert (
        PdfMetadata._qname('pikepdf:foo') == '{http://github.com/pikepdf/pikepdf/}foo'
    )


def test_undocumented_pdfx_identifier(trivial):
    trivial.Root.Metadata = Stream(
        trivial,
        """\
<x:xmpmeta xmlns:x="adobe:ns:meta/" x:xmptk="Adobe XMP Core 4.0-c316 44.253921, Sun Oct 01 2006 17:14:39">
   <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
      <rdf:Description xmlns:dc="http://purl.org/dc/elements/1.1/" rdf:about="">
         <dc:format>application/pdf</dc:format>
         <dc:title>
            <rdf:Alt>
               <rdf:li xml:lang="x-default">A</rdf:li>
            </rdf:Alt>
         </dc:title>
         <dc:creator>
            <rdf:Seq>
               <rdf:li>B</rdf:li>
               <rdf:li>C</rdf:li>
            </rdf:Seq>
         </dc:creator>
         <dc:description>
            <rdf:Alt>
               <rdf:li xml:lang="x-default">D</rdf:li>
            </rdf:Alt>
         </dc:description>
         <dc:publisher>
            <rdf:Bag>
               <rdf:li>E</rdf:li>
            </rdf:Bag>
         </dc:publisher>
      </rdf:Description>
      <rdf:Description xmlns:prism="http://prismstandard.org/namespaces/basic/3.0/" rdf:about="">
         <prism:aggregationType>journal</prism:aggregationType>
         <prism:publicationName>Journal</prism:publicationName>
         <prism:copyright>© 2023</prism:copyright>
         <prism:issn>1234-5678</prism:issn>
         <prism:volume>1</prism:volume>
         <prism:number>1</prism:number>
         <prism:coverDisplayDate>March-April 2023</prism:coverDisplayDate>
         <prism:pageRange>1-2</prism:pageRange>
         <prism:startingPage>1</prism:startingPage>
         <prism:endingPage>2</prism:endingPage>
      </rdf:Description>
      <rdf:Description xmlns:pdfx="http://ns.adobe.com/pdfx/1.3/" rdf:about="">
         <TGji2z9aLyPf_nweGmtj-nLNNnt2Nlwj7yteGngiQzd2Sy937zt-Jo9eQmdmKn9yJmMmTma/>
         <pdfx:robots>noindex</pdfx:robots>
      </rdf:Description>
      <rdf:Description xmlns:xap="http://ns.adobe.com/xap/1.0/" rdf:about="">
         <xap:CreatorTool>NA</xap:CreatorTool>
         <xap:CreateDate>2023-03-07T23:47:54+05:30</xap:CreateDate>
         <xap:ModifyDate>2023-03-07T23:48:24+05:30</xap:ModifyDate>
         <xap:MetadataDate>2023-03-07T23:48:24+05:30</xap:MetadataDate>
      </rdf:Description>
      <rdf:Description xmlns:xapRights="http://ns.adobe.com/xap/1.0/rights/" rdf:about="">
         <xapRights:Marked>True</xapRights:Marked>
      </rdf:Description>
      <rdf:Description xmlns:pdf="http://ns.adobe.com/pdf/1.3/" rdf:about="">
         <pdf:Producer>Acrobat Distiller 8.1.0 (Windows)</pdf:Producer>
      </rdf:Description>
      <rdf:Description xmlns:xapMM="http://ns.adobe.com/xap/1.0/mm/" rdf:about="">
         <xapMM:DocumentID>uuid:81b6d753-a810-4613-b207-111111111111</xapMM:DocumentID>
         <xapMM:InstanceID>uuid:0d50ecbf-b7d1-4b76-b4a7-222222222222</xapMM:InstanceID>
      </rdf:Description>
   </rdf:RDF>
</x:xmpmeta>
""".encode(),
    )
    with trivial.open_metadata() as m:
        list(m.items())


@pytest.mark.parametrize(
    'assign,expect', [([], []), ([''], [None]), ([None], [None]), (['Foo'], ['Foo'])]
)
def test_empty_list(graph, outpdf, assign, expect):
    with graph.open_metadata() as m:
        m['dc:creator'] = assign
        assert (
            m['dc:creator'] == expect
        ), f"Before saving, expected {expect}, got {m['dc:creator']}"

    graph.save(outpdf)

    with pikepdf.open(outpdf) as pdf:
        with pdf.open_metadata() as m:
            assert (
                m['dc:creator'] == expect
            ), f"After saving, expected {expect}, got {m['dc:creator']}"


def test_docinfo_wrong_type():
    pdf = pikepdf.Pdf.new()
    pdf.trailer.Info = 42

    assert isinstance(pdf.docinfo, pikepdf.Dictionary)

    pdf.trailer.Info = Dictionary(Key=42)

    assert isinstance(pdf.docinfo, pikepdf.Dictionary)
    assert pdf.docinfo.is_indirect


# --- XmpDocument standalone tests ---


class TestXmpDocumentStandalone:
    """Tests for XmpDocument used independently of PdfMetadata."""

    def test_create_empty(self):
        """Creating XmpDocument with no data creates valid empty XMP."""
        xmp = XmpDocument()
        assert len(xmp) == 0
        xml = xmp.to_bytes()
        assert b'rdf:RDF' in xml
        assert b'xpacket' in xml

    def test_create_from_bytes(self):
        """XmpDocument can parse existing XMP bytes."""
        xmp_bytes = b"""\
<?xpacket begin="\xef\xbb\xbf" id="W5M0MpCehiHzreSzNTczkc9d"?>
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about=""
      xmlns:dc="http://purl.org/dc/elements/1.1/">
    <dc:format>application/pdf</dc:format>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
<?xpacket end="w"?>
"""
        xmp = XmpDocument(xmp_bytes)
        assert 'dc:format' in xmp
        assert xmp['dc:format'] == 'application/pdf'

    def test_get_simple_property(self):
        """Get a simple string property."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:pdf="http://ns.adobe.com/pdf/1.3/">
    <pdf:Producer>TestProducer</pdf:Producer>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['pdf:Producer'] == 'TestProducer'
        assert xmp.get('pdf:Producer') == 'TestProducer'
        assert xmp.get('pdf:Missing') is None
        assert xmp.get('pdf:Missing', 'default') == 'default'

    def test_get_language_alternative(self):
        """Get a language alternative (rdf:Alt) property."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:dc="http://purl.org/dc/elements/1.1/">
    <dc:title>
      <rdf:Alt>
        <rdf:li xml:lang="x-default">My Title</rdf:li>
      </rdf:Alt>
    </dc:title>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['dc:title'] == 'My Title'

    def test_get_ordered_array(self):
        """Get an ordered array (rdf:Seq) property."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:dc="http://purl.org/dc/elements/1.1/">
    <dc:creator>
      <rdf:Seq>
        <rdf:li>Author One</rdf:li>
        <rdf:li>Author Two</rdf:li>
      </rdf:Seq>
    </dc:creator>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['dc:creator'] == ['Author One', 'Author Two']

    def test_get_unordered_array(self):
        """Get an unordered array (rdf:Bag) property."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:dc="http://purl.org/dc/elements/1.1/">
    <dc:subject>
      <rdf:Bag>
        <rdf:li>keyword1</rdf:li>
        <rdf:li>keyword2</rdf:li>
      </rdf:Bag>
    </dc:subject>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['dc:subject'] == {'keyword1', 'keyword2'}

    def test_set_simple_property(self):
        """Set a simple string property."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'MyProducer'
        assert xmp['pdf:Producer'] == 'MyProducer'

    def test_set_language_alternative(self):
        """Setting dc:title auto-wraps in rdf:Alt."""
        xmp = XmpDocument()
        xmp['dc:title'] = 'New Title'
        assert xmp['dc:title'] == 'New Title'
        # Verify it's wrapped in Alt
        xml = xmp.to_bytes()
        assert b'rdf:Alt' in xml
        assert b'xml:lang="x-default"' in xml

    def test_set_ordered_array(self):
        """Set an ordered array (list)."""
        xmp = XmpDocument()
        xmp['dc:creator'] = ['Author A', 'Author B']
        assert xmp['dc:creator'] == ['Author A', 'Author B']
        xml = xmp.to_bytes()
        assert b'rdf:Seq' in xml

    def test_set_unordered_array(self):
        """Set an unordered array (set)."""
        xmp = XmpDocument()
        xmp['dc:subject'] = {'tag1', 'tag2'}
        assert xmp['dc:subject'] == {'tag1', 'tag2'}
        xml = xmp.to_bytes()
        assert b'rdf:Bag' in xml

    def test_update_existing_property(self):
        """Update an existing property value."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:pdf="http://ns.adobe.com/pdf/1.3/">
    <pdf:Producer>OldProducer</pdf:Producer>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['pdf:Producer'] == 'OldProducer'
        xmp['pdf:Producer'] = 'NewProducer'
        assert xmp['pdf:Producer'] == 'NewProducer'

    def test_delete_property(self):
        """Delete a property."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about="" xmlns:pdf="http://ns.adobe.com/pdf/1.3/">
    <pdf:Producer>ToDelete</pdf:Producer>
  </rdf:Description>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert 'pdf:Producer' in xmp
        del xmp['pdf:Producer']
        assert 'pdf:Producer' not in xmp

    def test_delete_returns_success(self):
        """The delete() method returns whether deletion succeeded."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'Test'
        assert xmp.delete('pdf:Producer') is True
        assert xmp.delete('pdf:Producer') is False  # Already deleted

    def test_delete_missing_raises_keyerror(self):
        """Deleting non-existent key with del raises KeyError."""
        xmp = XmpDocument()
        with pytest.raises(KeyError):
            del xmp['pdf:Missing']

    def test_contains(self):
        """Test key membership."""
        xmp = XmpDocument()
        assert 'pdf:Producer' not in xmp
        xmp['pdf:Producer'] = 'Test'
        assert 'pdf:Producer' in xmp

    def test_len(self):
        """Test length."""
        xmp = XmpDocument()
        assert len(xmp) == 0
        xmp['pdf:Producer'] = 'Test'
        assert len(xmp) == 1
        xmp['dc:title'] = 'Title'
        assert len(xmp) == 2

    def test_iter(self):
        """Test iteration over keys."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'Test'
        xmp['dc:title'] = 'Title'
        keys = list(xmp)
        assert len(keys) == 2

    def test_to_bytes_with_xpacket(self):
        """Serialization includes xpacket markers by default."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'Test'
        xml = xmp.to_bytes()
        assert xml.startswith(b'<?xpacket begin')
        assert b'<?xpacket end="w"?>' in xml

    def test_to_bytes_without_xpacket(self):
        """Serialization can exclude xpacket markers."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'Test'
        xml = xmp.to_bytes(xpacket=False)
        assert not xml.startswith(b'<?xpacket')
        assert b'<?xpacket end' not in xml

    def test_str(self):
        """String representation is XML without xpacket."""
        xmp = XmpDocument()
        xmp['pdf:Producer'] = 'Test'
        s = str(xmp)
        assert '<?xpacket' not in s
        assert 'rdf:RDF' in s

    def test_roundtrip(self):
        """Parse -> modify -> serialize -> parse roundtrip."""
        xmp = XmpDocument()
        xmp['dc:title'] = 'Original'
        xmp['dc:creator'] = ['Author']
        xmp['pdf:Producer'] = 'Producer'

        # Serialize and re-parse
        xml = xmp.to_bytes()
        xmp2 = XmpDocument(xml)

        assert xmp2['dc:title'] == 'Original'
        assert xmp2['dc:creator'] == ['Author']
        assert xmp2['pdf:Producer'] == 'Producer'

    def test_qname_conversion(self):
        """Test qualified name conversion."""
        assert XmpDocument.qname('dc:title') == '{http://purl.org/dc/elements/1.1/}title'
        assert XmpDocument.qname('pdf:Producer') == '{http://ns.adobe.com/pdf/1.3/}Producer'
        # Already qualified names pass through
        assert XmpDocument.qname('{http://example.com}foo') == '{http://example.com}foo'

    def test_prefix_from_uri(self):
        """Test URI to prefix conversion."""
        xmp = XmpDocument()
        assert xmp.prefix_from_uri('{http://purl.org/dc/elements/1.1/}title') == 'dc:title'
        assert xmp.prefix_from_uri('{http://ns.adobe.com/pdf/1.3/}Producer') == 'pdf:Producer'

    def test_register_namespace(self):
        """Test custom namespace registration."""
        XmpDocument.register_xml_namespace('http://example.com/ns/', 'example')
        xmp = XmpDocument()
        xmp['example:custom'] = 'value'
        assert xmp['example:custom'] == 'value'
        xml = xmp.to_bytes()
        assert b'http://example.com/ns/' in xml

    def test_invalid_xml_overwrite_mode(self):
        """Invalid XML is replaced with empty XMP in overwrite mode."""
        xmp = XmpDocument(b'not xml at all', overwrite_invalid_xml=True)
        assert len(xmp) == 0  # Empty XMP created

    def test_invalid_xml_strict_mode(self):
        """Invalid XML raises error in strict mode."""
        with pytest.raises(XMLSyntaxError):
            XmpDocument(b'not xml at all', overwrite_invalid_xml=False)

    def test_valid_xml_but_not_xmp_overwrite_mode(self):
        """Valid XML but not XMP is replaced in overwrite mode."""
        xmp = XmpDocument(b'<root><child/></root>', overwrite_invalid_xml=True)
        assert len(xmp) == 0  # Empty XMP created

    def test_valid_xml_but_not_xmp_strict_mode(self):
        """Valid XML but not XMP raises in strict mode."""
        with pytest.raises(ValueError, match='not XMP'):
            XmpDocument(b'<root><child/></root>', overwrite_invalid_xml=False)

    def test_attribute_style_property(self):
        """Properties can be expressed as XML attributes."""
        xmp_bytes = b"""\
<x:xmpmeta xmlns:x="adobe:ns:meta/">
<rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">
  <rdf:Description rdf:about=""
      xmlns:pdf="http://ns.adobe.com/pdf/1.3/"
      pdf:Producer="AttributeProducer"/>
</rdf:RDF>
</x:xmpmeta>
"""
        xmp = XmpDocument(xmp_bytes)
        assert xmp['pdf:Producer'] == 'AttributeProducer'
