import unittest

try:
    from fontTools.pens.cocoaPen import CocoaPen
    from AppKit import NSBezierPathElementMoveTo, NSBezierPathElementLineTo
    from AppKit import NSBezierPathElementCurveTo, NSBezierPathElementClosePath

    PATH_ELEMENTS = {
        # NSBezierPathElement key      desc
        NSBezierPathElementMoveTo:    'moveto',
        NSBezierPathElementLineTo:    'lineto',
        NSBezierPathElementCurveTo:   'curveto',
        NSBezierPathElementClosePath: 'close',
    }

    PYOBJC_AVAILABLE = True
except ImportError:
    PYOBJC_AVAILABLE = False


def draw(pen):
    pen.moveTo((50, 0))
    pen.lineTo((50, 500))
    pen.lineTo((200, 500))
    pen.curveTo((350, 500), (450, 400), (450, 250))
    pen.curveTo((450, 100), (350, 0), (200, 0))
    pen.closePath()


def cocoaPathToString(path):
    num_elements = path.elementCount()
    output = []
    for i in range(num_elements - 1):
        elem_type, elem_points = path.elementAtIndex_associatedPoints_(i)
        elem_type = PATH_ELEMENTS[elem_type]
        path_points = " ".join([f"{p.x} {p.y}" for p in elem_points])
        output.append(f"{elem_type} {path_points}")
    return " ".join(output)


@unittest.skipUnless(PYOBJC_AVAILABLE, "pyobjc not installed")
class CocoaPenTest(unittest.TestCase):
    def test_draw(self):
        pen = CocoaPen(None)
        draw(pen)
        self.assertEqual(
            "moveto 50.0 0.0 lineto 50.0 500.0 lineto 200.0 500.0 curveto 350.0 500.0 450.0 400.0 450.0 250.0 curveto 450.0 100.0 350.0 0.0 200.0 0.0 close ",
            cocoaPathToString(pen.path)
        )

    def test_empty(self):
        pen = CocoaPen(None)
        self.assertEqual("", cocoaPathToString(pen.path))


if __name__ == '__main__':
    import sys
    sys.exit(unittest.main())
