import pytest
import py.path

from dotfiles.dotfile import Dotfile
from dotfiles.exceptions import IsSymlink, NotASymlink
from dotfiles.exceptions import TargetExists, TargetMissing, Exists


def _dotfile(repo, name, target=None):
    return Dotfile(repo.homedir.join(name),
                   repo.path.join(target if target is not None else name))


def test_str(repo):
    dotfile = _dotfile(repo, '.a', '.b')
    assert str(dotfile) == repo.homedir.join('.a')


def test_short_name(repo):
    dotfile = _dotfile(repo, '.foo')
    assert dotfile.name == repo.homedir.join('.foo')
    assert dotfile.short_name(repo.homedir) == '.foo'


def test_state_error(repo):
    dotfile = _dotfile(repo, '.vimrc')
    assert dotfile.state == 'error'


def test_state_missing(repo):
    dotfile = _dotfile(repo, '.vimrc')
    dotfile.target.ensure()
    assert dotfile.state == 'missing'


def test_state_conflict(repo):
    dotfile = _dotfile(repo, '.vimrc')
    dotfile.target.ensure()
    dotfile.name.ensure()
    assert dotfile.state == 'conflict'


def test_state_ok(repo):
    dotfile = _dotfile(repo, '.vimrc', 'vimrc')
    dotfile.target.ensure()
    dotfile.name.mksymlinkto(dotfile.target)
    assert dotfile.state == 'ok'
    dotfile.name.remove()
    assert dotfile.state == 'missing'


@pytest.mark.parametrize('path', ['.foo', '.foo/bar/baz'])
def test_add(repo, path):
    dotfile = _dotfile(repo, path)
    dotfile.target.ensure()
    dotfile.name.ensure()

    with pytest.raises(TargetExists):
        dotfile.add()

    dotfile.target.remove()
    dotfile.add()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(file=1, link=1)
    assert dotfile.name.samefile(dotfile.target)

    with pytest.raises(IsSymlink):
        dotfile.add()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(file=1, link=1)
    assert dotfile.name.samefile(dotfile.target)


@pytest.mark.parametrize('path', ['.foo', '.foo/bar/baz'])
def test_remove(repo, path):
    dotfile = _dotfile(repo, path)
    py.path.local(dotfile.name.dirname).ensure_dir()
    dotfile.name.mksymlinkto(dotfile.target)

    with pytest.raises(TargetMissing):
        dotfile.remove()

    dotfile.target.ensure()
    dotfile.remove()

    assert dotfile.target.check(exists=0)
    assert dotfile.name.check(file=1, link=0)

    with pytest.raises(NotASymlink):
        dotfile.remove()

    assert dotfile.target.check(exists=0)
    assert dotfile.name.check(file=1, link=0)


@pytest.mark.parametrize('path', ['.foo', '.foo/bar/baz'])
def test_link(repo, path):
    dotfile = _dotfile(repo, path)

    with pytest.raises(TargetMissing):
        dotfile.link()

    dotfile.target.ensure()
    dotfile.link()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(file=1, link=1)
    assert dotfile.name.samefile(dotfile.target)

    with pytest.raises(Exists):
        dotfile.link()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(file=1, link=1)
    assert dotfile.name.samefile(dotfile.target)


@pytest.mark.parametrize('path', ['.foo', '.foo/bar/baz'])
def test_unlink(repo, path):
    dotfile = _dotfile(repo, path)

    with pytest.raises(NotASymlink):
        dotfile.unlink()

    py.path.local(dotfile.name.dirname).ensure_dir()
    dotfile.name.mksymlinkto(dotfile.target)

    with pytest.raises(TargetMissing):
        dotfile.unlink()

    dotfile.target.ensure()
    dotfile.unlink()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(exists=0)

    with pytest.raises(NotASymlink):
        dotfile.unlink()

    assert dotfile.target.check(file=1, link=0)
    assert dotfile.name.check(exists=0)