import sys
from contextlib import contextmanager
from io import StringIO

from cmake_language_server.formatter import Formatter, main
from cmake_language_server.parser import ListParser


def make_formatter_test(liststr: str, expect: str):
    def test():
        tokens, remain = ListParser().parse(liststr)
        actual = Formatter().format(tokens)
        assert actual == expect

    return test


test_command = make_formatter_test('a()', 'a()\n')
test_command_tolower = make_formatter_test('A()', 'a()\n')
test_remove_space = make_formatter_test('''
  #a
  b ( c )  # d
''', '''\
#a
b(c)  # d
''')
test_indent_if = make_formatter_test(
    '''
if()
a()  # a
 else()
# b
b()
endif()
''', '''\
if()
  a()  # a
else()
  # b
  b()
endif()
''')
test_indent_if_nested = make_formatter_test(
    '''
if()
if()
a()
b()
endif()
endif()
''', '''\
if()
  if()
    a()
    b()
  endif()
endif()
''')
test_argument = make_formatter_test('a( b c  d)', 'a(b c d)\n')
test_argument_multiline = make_formatter_test(
    '''
if()
a(b c
d  # e
f
# g
)  # h
endif()
''', '''\
if()
  a(
    b c
    d  # e
    f
    # g
  )  # h
endif()
''')


@contextmanager
def mock_stdin(buf: str):
    stdin = sys.stdin
    sys.stdin = StringIO(buf)
    yield
    sys.stdin = stdin


def test_main_stdin(capsys):
    with mock_stdin(' a()'):
        main([])
    captured = capsys.readouterr()
    assert captured.out == 'a()\n'
    assert captured.err == ''


def test_main_stdin_diff(capsys):
    with mock_stdin(' a()'):
        main(['-d'])
    captured = capsys.readouterr()
    assert '- a()' in captured.out
    assert '+a()' in captured.out
    assert captured.err == ''


def test_main_file_1(capsys, tmp_path):
    testfile1 = tmp_path / 'list1.cmake'
    with testfile1.open('w') as fp:
        fp.write(' a()')

    main([str(testfile1)])
    captured = capsys.readouterr()
    assert captured.out == 'a()\n'
    assert captured.err == ''


def test_main_file_2(capsys, tmp_path):
    testfile1 = tmp_path / 'list1.cmake'
    with testfile1.open('w') as fp:
        fp.write(' a()')
    testfile2 = tmp_path / 'list2.cmake'
    with testfile2.open('w') as fp:
        fp.write(' b()')

    main([str(testfile1), str(testfile2)])
    captured = capsys.readouterr()
    assert captured.out == 'a()\nb()\n'
    assert captured.err == ''


def test_main_inplace(capsys, tmp_path):
    testfile1 = tmp_path / 'list1.cmake'
    with testfile1.open('w') as fp:
        fp.write(' a()')

    main(['-i', str(testfile1)])
    captured = capsys.readouterr()
    assert captured.out == ''
    assert captured.err == ''

    with testfile1.open() as fp:
        content = fp.read()
    assert content == 'a()\n'


def test_main_diff(capsys, tmp_path):
    testfile1 = tmp_path / 'list1.cmake'
    with testfile1.open('w') as fp:
        fp.write(' a()')

    main(['-d', str(testfile1)])
    captured = capsys.readouterr()
    assert str(testfile1) in captured.out
    assert '- a()' in captured.out
    assert '+a()' in captured.out
    assert captured.err == ''