import subprocess
from datetime import datetime

_file_date_format = "%Y:%m:%d %H:%M:%S%z"
_original_date_format = "%Y:%m:%d %H:%M:%S.%f%z"

_date_fields = {
    'file_modification_date/time': _file_date_format,
    'file_access_date/time': _file_date_format,
    'file_inode_change_date/time': _file_date_format,
    'date/time_original': _original_date_format,
    'create_date': _original_date_format,
    'modify_date': _original_date_format,
}

_integer_fields = [
    'image_width',
    'image_height',
    'iso',
    'exposure_time',
    'shutter_count',
    'shutter_speed',
    'jpg_from_raw_start',
    'jpg_from_raw_length',
    'thumbnail_offset',
    'thumbnail_length',
    'sr2_sub_ifd_offset',
    'sr2_sub_ifd_length',
    'exif_image_width',
    'exif_image_height',
    'shutter_count_2',
    'stops_above_base_iso',
    'sony_iso',
    'iso_setting',
    'iso_auto_min',
    'iso_auto_max',
    'bits_per_sample',
    'strip_byte_counts',
    'rows_per_strip',
    'strip_offsets',
    'x_resolution',
    'y_resolution',
    'samples_per_pixel',
    'sequence_file_number',
    'digital_zoom_ratio',
    'sequence_image_number',
    'focus_position_2',
]

_decimal_fields = [
    'aperture',
    'megapixels',
    'light_value',
    'blue_balance',
    'sony_f_number',
    'sony_max_aperture_value',
    'sony_f_number_2',
    'f_number',
    'max_aperture_value',
    'brightness_value',
    'sony_exposure_time',
    'sony_exposure_time_2',
]


def execute_exiftool(img_file: str) -> dict:
    res = subprocess.run(
        ['exiftool', img_file],
        capture_output=True,
        text=True,
    )

    if res.returncode != 0:
        raise Exception(res.stderr)

    exif_metadata = {}

    for line in res.stdout.splitlines():
        parts = line.split(':', 1)
        exif_metadata[parts[0].strip().lower().replace(' ', '_')] = parts[1].strip()

    for (date_field, date_format) in _date_fields.items():
        if date_field in exif_metadata:
            exif_metadata[date_field] = datetime.strptime(exif_metadata[date_field], date_format)

    for field in _integer_fields:
        if field in exif_metadata:
            try:
                exif_metadata[field] = int(exif_metadata[field])
            except ValueError as e:
                print(f'Failed to convert {field} ({exif_metadata[field]}) to an integer.')
                raise e

    for field in _decimal_fields:
        if field in exif_metadata:
            try:
                exif_metadata[field] = float(exif_metadata[field])
            except ValueError as e:
                print(f'Failed to convert {field} ({exif_metadata[field]}) to a float.')
                raise e

    return exif_metadata