from pathlib import Path
from ..color_print import color_print

def main(args):
    # Set the directory path
    script_dir = Path(__file__).resolve().parent
    dir_path = Path("")

    # Use the current script's directory or specify a path
    base_path = Path(__file__).resolve().parent 
    if args == []:
        for entry in dir_path.iterdir():
            if entry.is_dir():
                color_print.cprint(f"[FOLDER] {entry.name}", "DARKBLUE")
            elif entry.is_file():
                print(f"[FILE]   {entry.name}")
        print()
    elif args == ["--help"] or args == ["-h"]:
        print("list_files: Bro is lazy on doing this help section")
    else:
        print("list_files: Invalid arguments, try using none?")