"""
A script to randomly select files from a directory, open them, and allow the user to upload them to a Discord webhook, delete them, or skip to the next file.

You can specify a file size limit, whether to descend into subdirectories, and provide blacklist/whitelist patterns for file selection.

Here's an example usage:
./rand.py -fl 5 -d /path/to/directory --descend --blacklist "*.png" "*.mp4"
"""

import argparse
import os
import pathlib
import random
import sys
import termios
import tty

from random_file_picker.core import collect_files, move_file_to_meme_folder, upload_file
from random_file_picker.openers import open_file_with_geeqie, open_path_in_dolphin, open_path_with_default_app

IMG_EXTENSIONS = [
    ".jpg",
    ".jpeg",
    ".webp",
    ".png",
    ".gif",
]

def getch() -> str:
    fd = sys.stdin.fileno()
    old_settings = termios.tcgetattr(fd)
    try:
        tty.setcbreak(sys.stdin.fileno())
        ch = sys.stdin.read(1)
    finally:
        termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
    return ch

def main():
    parser = argparse.ArgumentParser(description="Random File Sender")
    parser.add_argument('-fl', '--file-limit', type=int, default=10, help='Maximum file size in MB')
    parser.add_argument('-d', '--directory', type=str, default='.', help='Directory to search for files')
    parser.add_argument('--descend', action='store_true', help='Whether to descend into subdirectories')
    parser.add_argument('--use-geeqie', action='store_true', help='Whether to use Geeqie as the image viewer')

    parser._positionals.title = 'Positional arguments'
    parser._optionals.title = 'Optional arguments'

    # Globs for blacklisting and whitelisting, --blacklist "*.png" "*.mp4"
    parser.add_argument('--blacklist', nargs='*', help='List of glob patterns to blacklist files')
    parser.add_argument('--whitelist', nargs='*', help='List of glob patterns to whitelist files')

    args = parser.parse_args()

    file_limit: int = args.file_limit
    directory: str = args.directory
    descend: bool = args.descend
    blacklist_patterns: list[str] = args.blacklist if args.blacklist else []
    whitelist_patterns: list[str] = args.whitelist if args.whitelist else []
    use_geeqie: bool = args.use_geeqie
    print(f"File size limit set to {file_limit} MB")

    search_path = pathlib.Path(directory)
    if not search_path.exists() or not search_path.is_dir():
        print(f"Directory '{directory}' does not exist or is not a directory.")
        return

    usable_files = collect_files(
        search_path=search_path,
        search_subdirectories=descend,
        file_size_limit=file_limit,
        blacklist_patterns=blacklist_patterns,
        whitelist_patterns=whitelist_patterns,
    )
    if not usable_files:
        print("No usable files found.")
        return

    pickFile = True
    while pickFile:
        file = random.choice(usable_files)

        print(f"Random file: {file}")
        if use_geeqie and str(file).lower().endswith(tuple(IMG_EXTENSIONS)):
            open_file_with_geeqie(file)
        else:
            open_path_with_default_app(file)


        selectionTime = True
        while selectionTime:
            print("Options: [u]pload, [n]ext, [d]elete, [f]older, [q]uit")
            choice = getch().lower()

            # Skip escape sequences (arrow keys)
            if choice.encode("utf-8") == b'\x1b':
                _ = getch()  # Skip the next two characters
                _ = getch()
                continue

            #print(choice.encode("utf-8"))  # For debugging purposes

            if choice == "u":
                print("\nUploading...")
                upload_file(file)
                print(f"\nSuccessfully uploaded {file}")
                print("Upload complete.")
                #selectionTime = False
            elif choice == "d":
                print("\nDeleting...")
                os.remove(file)
                usable_files.remove(file)
                selectionTime = False
            elif choice == "n":
                print("\nNext file...")
                selectionTime = False
                usable_files.remove(file)
            elif choice == "f":
                open_path_in_dolphin(file)
            elif choice == "q":
                print("\nQuitting...")
                pickFile = False
                break
            elif choice == "m":
                print("\nMoving to meme folder...")
                move_file_to_meme_folder(file)
                usable_files.remove(file)
                selectionTime = False
            else:
                print(f"\n'{choice}' is an invalid option. Please choose again.")


if __name__ == "__main__":
    main()