#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os, sys, subprocess, glob as file_glob

def copy_to_fat(image_path, source_dir):
    if not os.path.isfile(image_path):
        print(f"Error: FAT32 image {image_path} does not exist.")
        sys.exit(1)

    if not os.path.isdir(source_dir):
        print(f"Error: {source_dir} is not a valid directory.")
        sys.exit(1)

    try:
        files_to_copy = file_glob.glob(os.path.join(source_dir, "*"))

        if not files_to_copy:
            print(f"Warning: No files found in {source_dir}, nothing to copy.")
            sys.exit(1)
        
        command = ["mcopy", "-spm", "-i", image_path] + files_to_copy + ["::"]
        subprocess.run(command, check=True)
    except Exception as e:
        print(f"Error: failed: {e}")
        sys.exit(1)

if __name__ == "__main__":
    if len(sys.argv) != 3:
        print("HELP: kimg.py <fat32_image> <source_directory>")
        sys.exit(1)

    image_path = sys.argv[1]
    source_dir = sys.argv[2]

    copy_to_fat(image_path, source_dir)

    print("INFO: Image created successfully.")