#!/usr/bin/env python
# vim:fileencoding=utf-8
# License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>

import codecs

from . import BaseTest, set_text_in_line

from kitty.data_types import Line, Cursor
from kitty.utils import is_simple_string, wcwidth


class TestDataTypes(BaseTest):

    def test_line_ops(self):
        t = 'Testing with simple text'
        l = Line(len(t))
        set_text_in_line(l, t)
        self.ae(l, l)
        self.ae(str(l), t)
        self.ae(str(l.copy()), t)
        l.continued = False
        l2 = l.copy()
        self.assertFalse(l2.continued)
        self.ae(l, l2)
        l2.char[1] = 23
        self.assertNotEqual(l, l2)

        c = Cursor(3, 5)
        c.hidden = True
        c.bold = c.italic = c.reverse = c.strikethrough = True
        c.fg = c.bg = c.decoration_fg = 0x0101
        self.ae(c, c)
        c2, c3 = c.copy(), c.copy()
        self.ae(c, c.copy())
        c2.bold = c2.hidden = False
        self.assertNotEqual(c, c2)
        l.apply_cursor(c2, 3)
        self.ae(c2, l.cursor_from(3, ypos=c2.y))
        l.apply_cursor(c2, 0, len(l))
        for i in range(len(l)):
            c2.x = i
            self.ae(c2, l.cursor_from(i, ypos=c2.y))
        l = Line(5)
        l.apply_cursor(c3, 0)
        l.copy_char(0, l, 1)
        c3.x, c3.hidden = 1, False
        self.ae(l.cursor_from(1, ypos=c3.y), c3)

        t = '0123456789'
        lo = Line(len(t))
        set_text_in_line(lo, t)
        l = lo.copy()
        l.right_shift(4, 2)
        self.ae(str(l), '0123454567')
        l = lo.copy()
        l.right_shift(0, 0)
        self.ae(l, lo)
        l.right_shift(0, 1)
        self.ae(str(l), '0' + t[:-1])
        l = lo.copy()
        l.left_shift(0, 2)
        self.ae(str(l), t[2:] + '89')
        l = lo.copy()
        l.left_shift(7, 3)
        self.ae(str(l), t)

        l = Line(1)
        l.set_decoration(0, 2)
        q = Cursor()
        for x in 'bold italic reverse strikethrough'.split():
            getattr(l, 'set_' + x)(0, True)
            setattr(q, x, True)
        q.decoration = 2
        l.set_decoration(0, q.decoration)
        c = l.cursor_from(0)
        self.ae(c, q)

    def test_utils(self):
        d = codecs.getincrementaldecoder('utf-8')('strict').decode
        self.ae(tuple(map(wcwidth, 'a1\0コ')), (1, 1, 0, 2))
        for s in ('abd38453*(+\n\t\f\r !\0~[]{}()"\':;<>/?ASD`',):
            self.assertTrue(is_simple_string(s))
            self.assertTrue(is_simple_string(d(s.encode('utf-8'))))
        self.assertFalse(is_simple_string('a1コ'))