#!/usr/bin/python3
# -*- encoding: utf-8 -*-
# Copyright © 2016 the Snipe Contributors
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following
# disclaimer in the documentation and/or other materials provided
# with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
# CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
# TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
# TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
# THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.

'''
Unit tests for the doc munching stuff in the text module
'''


import docutils.core
import sys
import unittest

sys.path.append('..')
sys.path.append('../lib')

from snipe.chunks import Chunk  # noqa: E402
import snipe.help as help       # noqa: E402
import snipe.keymap as keymap   # noqa: E402
import snipe.text as text       # noqa: E402
import snipe.util as util       # noqa: E402


class TestRendering(unittest.TestCase):
    def test_RSTRenderer(self):
        pub = parse_rest(TEXT)

        renderer = text.RSTRenderer()
        renderer.process(pub.writer.document)

        import logging
        logging.debug('%s', repr(renderer.output))
        logging.debug('\n%s', renderer.flat())

        self.assertEqual(TEXT_rendered, renderer.output)

    def test_markdown_to_chunk(self):
        self.assertEqual(
            [(set(), 'some text\n')], text.markdown_to_chunk('some text'))

    def test_wordboundaries1(self):
        self.assertEqual(
            rest_chunks('A line of text.')[0],
            (0, [(set(), 'A line of text.\n')]))

    def test_wordboundaries2(self):
        self.assertEqual(
            rest_chunks('A *line* of text.')[0],
            (0, [(set(), 'A '), ({'bold'}, 'line'), (set(), ' of text.\n')]))

    def test_wordboundaries3(self):
        self.assertEqual(
            rest_chunks('A line of *text.*')[0],
            (0, [(set(), 'A line of '), ({'bold'}, 'text.'), (set(), '\n')]))

    def test_wordboundaries4(self):
        self.assertEqual(
            rest_chunks('A line of *text*.')[0],
            (0, [(set(), 'A line of '), ({'bold'}, 'text'), (set(), '.\n')]))

    def test_wordboundaries5(self):
        self.assertEqual(
            rest_chunks('A line?\nof text.')[0],
            (0, [(set(), 'A line? of text.\n')]))
        self.assertEqual(
            rest_chunks('A line? of\n*text.*')[0],
            (0, [(set(), 'A line? of '), ({'bold'}, 'text.'), (set(), '\n')]))
        self.assertEqual(
            rest_chunks('A line? of\n``text.``')[0],
            (0, [(set(), 'A line? of '), ({'bold'}, 'text.'), (set(), '\n')]))

    def test_wordboundaries6(self):
        self.assertEqual(
            rest_chunks('A line of ``text.``')[0],
            (0, [(set(), 'A line of '), ({'bold'}, 'text.'), (set(), '\n')]))


class TestInterrogate(unittest.TestCase):
    def test(self):
        help.HelpBrowser.base_module = sys.modules[__name__]
        self.assertEqual(
            rest_flat('.. interrogate_keymap:: InterrogateMe'),
            '! exclam\nThis thing. Does a thing.\n\n')
        self.assertEqual(
            rest_flat(
                '.. interrogate_config:: ' + __name__ + '.InterrogateMe'),
            'thing\nA config key for InterrogateConfig to find\n\n')


class TestToc(unittest.TestCase):
    def test(self):
        self.assertEqual(
            rest_flat('.. toc::b'),
            '')


class TestXHTML(unittest.TestCase):
    def test_xhtml_to_chunk(self):
        self.assertEqual(text.xhtml_to_chunk(
            '<blarf>foo</blarf>'), Chunk([
                (('bold',), '<blarf>'),
                ((), 'foo'),
                (('bold',), '</blarf>'),
                ((), '\n')]))
        self.assertEqual(
            text.xhtml_to_chunk('foo'), [(set(), 'foo\n')])
        self.assertEqual(
            text.xhtml_to_chunk('one\ntwo\n<pre>three\nfour\n</pre>'),
            [(set(), 'one two \nthree\nfour\n')])
        self.assertEqual(
            text.xhtml_to_chunk(
                '<blockquote><b><code>foo</code></b></blockquote>'),
            [({'bold', 'bg:#3d3d3d'}, '  foo'), (set(), '\n')])
        self.assertEqual(
            text.xhtml_to_chunk('<a>foo</a>'),
            [({'fg:#6666ff', 'underline'}, 'foo'), (set(), '\n')])
        self.assertEqual(
            list(text.xhtml_to_chunk('<b>\none\ntwo\n</b>')),
            [({'bold'}, ' one two '), (set(), '\n')])
        self.assertEqual(
            list(text.xhtml_to_chunk('<b>one<br/>two</b>')),
            [({'bold'}, 'one\ntwo'), (set(), '\n')])
        self.assertEqual(
            list(text.xhtml_to_chunk('<code>' + 'aaa ' * 20 + '</code>')), [
                ({'bg:#3d3d3d'}, 'aaa' + ' aaa' * 17),
                (set(), '\n'),
                ({'bg:#3d3d3d'}, 'aaa aaa '),
                (set(), '\n')])


class TestMarkdownXHTMLChunk(unittest.TestCase):
    def test(self):
        input = 'foo'
        xhtml = text.markdown_to_xhtml(input)
        self.assertEqual(xhtml, '<p>foo</p>')
        chunk = text.xhtml_to_chunk(xhtml)
        self.assertEqual(chunk.tagsets(), [((), 'foo\n')])

    def test_extensions(self):
        input = 'foo\n\n    bar\n\nbaz'
        xhtml = text.markdown_to_xhtml(input)
        self.assertEqual(
            xhtml, '<p>foo</p>\n<pre><code>bar\n</code></pre>\n<p>baz</p>')
        input = 'foo\n\n~~~~~~~~\nbar\n~~~~~~~~\n\nbaz'
        xhtml = text.markdown_to_xhtml(input)
        self.assertEqual(
            xhtml, '<p>foo</p>\n<pre><code>bar\n</code></pre>\n\n<p>baz</p>')
        input = 'foo\n\n~~~~~~~~.quote\nbar\n~~~~~~~~\n\nbaz\n'
        print('zog')
        xhtml = text.markdown_to_xhtml(input)
        self.assertEqual(
            xhtml,
            '<p>foo</p>\n<blockquote>\n<p>bar</p>\n</blockquote>\n\n'
            '<p>baz</p>')


TEXT = '''
=============
a Title
=============

.. a comment

Here is some text.  Here is some more text.  Blah blah blah.  Foo.  Bar.  Flarb.
(The previous should get wrapped if everything is good.)

One.
Two.
Three.
(The above shold end up on one line.)

myslack.slack.com).  You
add ``.slack name=myname`` to the ``backends`` configuration variable (which is
``;`` separated), and get an api key from ``https://api.slack.com/web`` and put
it in ``~/.snipe/netrc`` like so: ::

 machine myslack.slack.com login myself@example.com password frob-9782504613-8396512704-9784365210-7960cf


(You need to have already signed up for the relevant slack instance by other
means.)

filler
------

Lorem ipsum dolor sit amet, consecteturadipiscingelit,seddoeiusmodtemporincididuntutlaboreetdoloremagnaaliqua.Utenimadminimveniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
'''  # noqa: E501

TEXT_rendered = [
    (0, Chunk([((), ''), (('bold',), 'a Title'), ((), '\n')])),
    (8, Chunk([((), '\n')])),
    (9, Chunk([
        ((), 'Here is some text.  Here is some more text.  Blah blah blah.'
             '  Foo.\n')])),
    (76, Chunk([
        ((), 'Bar.  Flarb. (The previous should get wrapped if everything'
             ' is good.)\n')])),
    (146, Chunk([((), '\n')])),
    (147, Chunk([
        ((), 'One. Two. Three. (The above shold end up on one'
             ' line.)\n')])),
    (202, Chunk([((), '\n')])),
    (203, Chunk([
        ((), 'myslack.slack.com).  You add '),
        (('bold',), '.slack name=myname'),
        ((), ' to the '),
        (('bold',), 'backends'),
        ((), '\n')])),
    (267, Chunk([
        ((), 'configuration variable (which is '),
        (('bold',), ';'),
        ((), ' separated), and get an api key from '),
        (('bold',), '\n')])),
    (339, Chunk([
        (('bold',), 'https://api.slack.com/web'),
        ((), ' and put it in '),
        (('bold',), '~/.snipe/netrc'),
        ((), ' like so:\n')])),
    (403, Chunk([((), '\n')])),
    (404, Chunk([
        (('bold',),  'machine myslack.slack.com login myself@example.com'
            ' password frob-9782504613-8396512704-9784365210-7960cf'),
        ((), '\n')])),
    (509, Chunk([((), '\n')])),
    (510, Chunk([
        ((), '(You need to have already signed up for the relevant slack '
             'instance by\n')])),
    (581, Chunk([((), 'other means.)\n')])),
    (595, Chunk([((), '\n')])),
    (596, Chunk([(set(), '* '), ({'bold'}, 'filler'), (set(), '\n')])),
    (605, Chunk([((), '\n')])),
    (606, Chunk([((), 'Lorem ipsum dolor sit amet,\n')])),
    (634, Chunk([
        ((), 'consecteturadipiscingelit,seddoeiusmodtemporincididuntut'
             'laboreetdoloremagnaaliqua.Utenimadminimveniam,\n')])),
    (737, Chunk([
        ((), 'quis nostrud exercitation ullamco laboris nisi ut aliquip ex'
             ' ea commodo\n')])),
    (809, Chunk([((), 'consequat.\n')])),
    (820, Chunk([((), '\n')])),
    ]


class InterrogateMe:
    thing = util.Configurable(
        'thing', 1,
        'A config key for InterrogateConfig to find',
        coerce=int)

    @keymap.bind('!')
    def exclam(self):
        """This thing.
        Does a thing."""
        pass


def parse_rest(text):
    _, pub = docutils.core.publish_programmatically(
        docutils.io.StringInput, text, None, docutils.io.NullOutput,
        None, None, None, 'standalone', None, 'restructuredtext', None,
        'null', None, None, {}, None, None)
    return pub


def rest_flat(input):
    renderer = text.RSTRenderer()
    renderer.process(parse_rest(input).writer.document)

    return renderer.flat()


def rest_chunks(input):
    renderer = text.RSTRenderer()
    renderer.process(parse_rest(input).writer.document)

    return renderer.output


# So I can cut and paste it into tests:
# Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do
# eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim
# ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut
# aliquip ex ea commodo consequat. Duis aute irure dolor in
# reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla
# pariatur. Excepteur sint occaecat cupidatat non proident, sunt in
# culpa qui officia deserunt mollit anim id est laborum.


if __name__ == '__main__':
    unittest.main()