import os
from amitools.vamos.path import VolumeManager, resolve_sys_path
from amitools.vamos.cfgcore import ConfigDict


def path_volume_resolve_sys_path_test(tmpdir):
    rsp = resolve_sys_path
    p = str(tmpdir)
    assert rsp(p) == p
    # user home
    assert rsp("~") == os.path.expanduser("~")
    # env var
    os.environ["TEST_PATH"] = p
    assert rsp("${TEST_PATH}") == p
    assert rsp("${TEST_PATH}/bla") == os.path.join(p, "bla")


def path_volume_add_del_test(tmpdir):
    v = VolumeManager()
    assert v.setup()
    assert v.get_all_names() == []
    my_path = str(tmpdir.mkdir("bla"))
    no_path = str(tmpdir.join("hugo"))
    # ok
    vol = v.add_volume("My:" + my_path)
    assert vol
    assert v.get_all_names() == ["My"]
    assert v.is_volume("MY")
    assert vol.is_setup
    assert vol.get_path() == my_path
    assert v.add_volume("foo:" + my_path)
    assert v.get_all_names() == ["My", "foo"]
    # duplicate path mapping
    assert not v.add_volume("foo:" + my_path)
    # duplicate path name
    assert not v.add_volume("my:" + no_path)
    # invalid path
    assert not v.add_volume("foo:" + no_path)
    # ok
    assert v.del_volume("my")
    assert not vol.is_setup
    assert v.get_all_names() == ["foo"]
    # invalid name
    assert not v.del_volume("baz")
    # shutdown
    v.shutdown()


def path_volume_add_local_test(tmpdir):
    vols_dir = str(tmpdir.join("volumes"))
    v = VolumeManager(vols_base_dir=vols_dir)
    v.setup()
    # without create
    assert not v.add_volume("My")
    # with create
    vol = v.add_volume("My?create")
    assert vol
    # check for vol dir
    vol_path = os.path.join(vols_dir, "My")
    assert os.path.isdir(vol_path)
    assert vol.get_path() == vol_path
    # create multiple
    vols = v.add_volumes(["foo?create", "bar?create"])
    assert vols
    for vol in vols:
        vol_path = os.path.join(vols_dir, vol.get_name())
        assert os.path.isdir(vol_path)
        assert vol.get_path() == vol_path
    # shutdown
    v.shutdown()


def path_volume_create_rel_sys_path_test(tmpdir):
    v = VolumeManager()
    org = tmpdir.mkdir("bla")
    my_path = str(org)
    # ok
    vol = v.add_volume("My:" + my_path)
    assert vol
    # single path
    path = vol.create_rel_sys_path("bla")
    assert path == str(org.join("bla"))
    assert os.path.isdir(path)
    # multi path
    path = vol.create_rel_sys_path(["foo", "bar"])
    assert path == str(org.join("foo").join("bar"))
    assert os.path.isdir(path)


def path_volume_sys_to_ami_test(tmpdir):
    v = VolumeManager()
    mp = tmpdir.mkdir("bla")
    my_path = str(mp)
    no_path = str(tmpdir.join("hugo"))
    mp2 = mp.mkdir("blub")
    my_path2 = str(mp2)
    assert v.add_volume("My:" + my_path)
    assert v.add_volume("nested:" + my_path2)
    # exisitng path
    s2a = v.sys_to_ami_path
    assert s2a(my_path) == "My:"
    assert s2a(str(mp.join("foo"))) == "My:foo"
    # expect nested path
    assert s2a(my_path2) == "nested:"
    assert s2a(str(mp2.join("bla/blub"))) == "nested:bla/blub"
    # non existing
    assert s2a(str(tmpdir)) is None
    # not abosulte
    assert s2a("bla") is None


def path_volume_ami_to_sys_test(tmpdir):
    v = VolumeManager()
    mp = tmpdir.mkdir("bla")
    my_path = str(mp)
    mp2 = mp.mkdir("Foo").mkdir("BAR").mkdir("baZ")
    # case insensitive file system?
    ci_fs = os.path.exists(os.path.join(my_path, "foo"))
    sub_path = str(mp2)
    assert v.add_volume("My:" + my_path)
    # base path
    a2s = v.ami_to_sys_path
    assert a2s("my:") == my_path
    assert a2s("my:unkown/PATH") == os.path.join(my_path, "unkown", "PATH")
    # follow along case of path in sys fs
    assert a2s("my:foo/bar/baz") == sub_path
    # fast mode on case insensitive fs does not adjust ami path
    if ci_fs:
        assert a2s("my:foo", True) == os.path.join(my_path, "foo")
    else:
        assert a2s("my:foo", True) == os.path.join(my_path, "Foo")


def path_volume_cfg_test(tmpdir):
    my_path = str(tmpdir.mkdir("bla"))
    v = VolumeManager()
    cfg = ConfigDict({"volumes": ["my:" + my_path]})
    assert v.parse_config(cfg)
    assert v.get_all_names() == ["my"]
    assert v.is_volume("MY")


def path_volume_create_test(tmpdir):
    v = VolumeManager(str(tmpdir))
    assert v.setup()
    spec = "my:" + str(tmpdir) + "/bla"
    # dir does not exist -> can't create
    assert not v.add_volume(spec)
    # create
    assert v.add_volume(spec + "?create")
    # check
    assert tmpdir.join("bla").check(dir=1)
    # shutdown
    v.shutdown()


def path_volume_temp_test(tmpdir):
    v = VolumeManager(str(tmpdir))
    assert v.setup()
    spec = "my:" + str(tmpdir)
    # create temp
    spec += "/bla"
    assert v.add_volume(spec + "?temp")
    # check that temp dir exists
    assert tmpdir.join("bla").check(dir=1)
    # shutdown
    v.shutdown()
    # now temp is gone
    assert not tmpdir.join("bla").check()
