#!/usr/bin/env python3

import errno
import os
import stat
import sys
import tempfile

from posix_parity import cleanup_dir
from posix_parity import cleanup_paths
from posix_parity import compare_calls
from posix_parity import fail
from posix_parity import join
from posix_parity import should_compare_inode
from posix_parity import stat_cmp_basic
from posix_parity import temp_dir
from posix_parity import touch


def stat_cmp_basic_with_inode(lhs, rhs):
    return stat_cmp_basic(lhs, rhs) and lhs.st_ino == rhs.st_ino


def main():
    if len(sys.argv) != 2:
        print("usage: TEST_posix_chmod <mountpoint>", file=sys.stderr)
        return 1

    mount = sys.argv[1]
    stcmp = stat_cmp_basic_with_inode if should_compare_inode(mount) else stat_cmp_basic

    with tempfile.TemporaryDirectory() as native:
        merge_base = temp_dir(mount)
        try:
            native_base = join(native, os.path.basename(merge_base))
            os.makedirs(native_base, exist_ok=True)

            merge_file = join(merge_base, "file")
            native_file = join(native_base, "file")
            merge_missing = join(merge_base, "missing")
            native_missing = join(native_base, "missing")
            merge_notdir = join(merge_base, "notdir")
            native_notdir = join(native_base, "notdir")

            cleanup_paths([merge_file, merge_notdir])

            touch(merge_file, b"x", 0o644)
            touch(native_file, b"x", 0o644)
            touch(merge_notdir, b"x", 0o644)
            touch(native_notdir, b"x", 0o644)

            err = compare_calls(
                "chmod success",
                lambda: os.chmod(merge_file, 0o600),
                lambda: os.chmod(native_file, 0o600),
            )
            if err:
                return fail(err)

            err = compare_calls(
                "chmod stat parity",
                lambda: os.lstat(merge_file),
                lambda: os.lstat(native_file),
                stcmp,
            )
            if err:
                return fail(err)

            err = compare_calls(
                "chmod ENOENT",
                lambda: os.chmod(merge_missing, 0o600),
                lambda: os.chmod(native_missing, 0o600),
            )
            if err:
                return fail(err)

            err = compare_calls(
                "chmod ENOTDIR",
                lambda: os.chmod(join(merge_notdir, "child"), 0o600),
                lambda: os.chmod(join(native_notdir, "child"), 0o600),
            )
            if err:
                return fail(err)

            if os.geteuid() != 0:
                err = compare_calls(
                    "chmod setuid parity",
                    lambda: os.chmod(merge_file, stat.S_ISUID | 0o644),
                    lambda: os.chmod(native_file, stat.S_ISUID | 0o644),
                )
                if err:
                    return fail(err)
            else:
                _ = errno.EPERM

            return 0
        finally:
            cleanup_dir(merge_base)


if __name__ == "__main__":
    raise SystemExit(main())
