#!/usr/bin/env python3

import argparse
import os
import re
from pathlib import Path

from avdb.db import DBOps, DBConnection

def process_single_file(db_ops, file_path):
    """
    Process a single image file.

    Args:
        db_ops (DBOps): An instance of DBOps for database operations.
        file_path (str): Path to the image file.
    """
    print(f"Processing file: {file_path}")
    db_ops.process_image_analysis(file_path)

def process_directory(db_ops, directory_path):
    """
    Process all image files in a directory.

    Args:
        db_ops (DBOps): An instance of DBOps for database operations.
        directory_path (str): Path to the directory containing image files.
    """
    print(f"Processing directory: {directory_path}")
    for root, _, files in os.walk(directory_path):
        # print(files)
        # print(root)
        for file in files:
            m  = re.match(r'img_\d{4}\.jpeg', file.lower())
            if m is not None:
                file_path = os.path.join(root, file)
                process_single_file(db_ops, file_path)

def main():
    # Parse command-line arguments
    parser = argparse.ArgumentParser(description="Run image analysis for a single file or all files in a directory.")
    parser.add_argument("path", help="Path to a single image file or a directory containing image files.")
    parser.add_argument("--dbname", default="testdb", help="Name of the database to use (default: testdb).")
    args = parser.parse_args()

    # Initialize database connection and operations
    with DBConnection(args.dbname).get_connection() as db_conn:
        db_ops = DBOps(db_conn)

        # Determine if the path is a file or directory
        if os.path.isfile(args.path):
            process_single_file(db_ops, args.path)
        elif os.path.isdir(args.path):
            process_directory(db_ops, args.path)
        else:
            print(f"Error: Path '{args.path}' does not exist.")
            sys.exit(1)

if __name__ == '__main__':
    main()