Source code for pyocutil.main

"""
main file
"""

import os
import os.path
import subprocess
import sys

import pylogconf.core
from pytconf import config_arg_parse_and_launch, get_free_args, register_endpoint, register_main

from pyocutil.configs import ConfigSymlinkInstall, ConfigVerbose
from pyocutil.static import APP_NAME, DESCRIPTION, VERSION_STR
from pyocutil.utils import debug, do_install, file_gen, no_err_run, touch_mkdir_many




@register_endpoint(
    description="Install symlinks to things in a folder",
    configs=[
        ConfigSymlinkInstall,
        ConfigVerbose,
    ],
)
def symlink_install() -> None:
    cwd = os.getcwd()
    # first unlink all paths in target leading back to here
    if os.path.isdir(ConfigSymlinkInstall.target_folder):
        for file in os.listdir(ConfigSymlinkInstall.target_folder):
            full = os.path.join(ConfigSymlinkInstall.target_folder, file)
            if os.path.islink(full):
                link_target = os.path.realpath(full)
                if link_target.startswith(cwd) and ConfigSymlinkInstall.doit:
                    debug(f"unlinking [{full}]")
                    os.unlink(full)
    else:
        os.mkdir(ConfigSymlinkInstall.target_folder)
    # now create the new links
    for root, directories, files in file_gen(ConfigSymlinkInstall.source_folder, ConfigSymlinkInstall.recurse):
        for file in files:
            source = os.path.abspath(os.path.join(root, file))
            target = os.path.join(ConfigSymlinkInstall.target_folder, file)
            do_install(source, target, ConfigSymlinkInstall.force, ConfigSymlinkInstall.doit)
        for directory in directories:
            source = os.path.abspath(os.path.join(root, directory))
            target = os.path.join(ConfigSymlinkInstall.target_folder, directory)
            do_install(source, target, ConfigSymlinkInstall.force, ConfigSymlinkInstall.doit)





@register_endpoint(
    description="Remove one folder from each file name",
    allow_free_args=True,
)
def remove_folders() -> None:
    result = []
    for filename in get_free_args():
        r = os.sep.join(filename.split(os.sep)[1:])
        result.append(r)
    print(" ".join(result), end="")





@register_endpoint(
    description="Touch a file, possibly creating folders for it",
    allow_free_args=True,
)
def touch_mkdir() -> None:
    touch_mkdir_many(get_free_args())





@register_endpoint(
    description="Return with no error even if underlying process returns error",
    allow_free_args=True,
)
def no_err() -> None:
    no_err_run(get_free_args())





@register_endpoint(
    description="Collect stdout and stderr and print them only in error",
    allow_free_args=True,
    min_free_args=1,
    configs=[
        ConfigVerbose,
    ],
)
def only_print_on_error() -> None:
    if ConfigVerbose.verbose:
        print(" ".join(get_free_args()))
    with subprocess.Popen(get_free_args(), stdout=subprocess.PIPE, stderr=subprocess.PIPE) as pr:
        (out_out, out_err) = pr.communicate()
        status = pr.returncode
        if status:
            print(out_out.decode(), end="")
            print(out_err.decode(), end="")
            sys.exit(status)





@register_endpoint(
    description="disregard exit code of command and issue error if there is output",
    allow_free_args=True,
    min_free_args=1,
    configs=[
        ConfigVerbose,
    ],
)
def error_on_output() -> None:
    if ConfigVerbose.verbose:
        print(" ".join(get_free_args()))
    with subprocess.Popen(get_free_args(), stdout=subprocess.PIPE, stderr=subprocess.PIPE) as pr:
        (out_out, out_err) = pr.communicate()
        _ = pr.returncode
    if len(out_out) > 0 or len(out_err) > 0:
        print(out_out.decode(), end="")
        print(out_err.decode(), end="")
        sys.exit(1)
    else:
        sys.exit(0)





@register_endpoint(
    description="reverse the exitcode of a command",
    allow_free_args=True,
    min_free_args=1,
    configs=[
        ConfigVerbose,
    ],
)
def reverse_exitcode() -> None:
    if ConfigVerbose.verbose:
        print(" ".join(get_free_args()))
    exitcode = subprocess.call(get_free_args())
    sys.exit(int(not bool(exitcode)))





@register_endpoint(
    description="wrap css validation command line",
    allow_free_args=True,
    min_free_args=1,
)
def wrapper_css_validator() -> None:
    args = get_free_args()
    out = subprocess.check_output(args).decode()
    do_print = False
    error = False
    for line in out.split("\n"):
        if line.startswith("Sorry"):
            do_print = True
            error = True
        if line.startswith("Valid"):
            do_print = False
        if do_print:
            print(line)
    if error:
        sys.exit(1)





@register_main(
    main_description=DESCRIPTION,
    app_name=APP_NAME,
    version=VERSION_STR,
)
def main():
    pylogconf.core.setup()
    config_arg_parse_and_launch()



if __name__ == "__main__":
    main()