import os
import sys
import time
import string
from ctypes import windll
from random import randint
from configparser import ConfigParser


def get_drives():
    drives = []
    bitmask = windll.kernel32.GetLogicalDrives()
    for letter in string.ascii_uppercase:
        if bitmask & 1:
            drives.append(f"{letter}:\\")
        bitmask >>= 1
    return drives


def read_config(drive):
    if "PWDI.ini" in os.listdir(drive):
        config = ConfigParser()
        config.read(f"{drive}PWDI.ini")
        PWDI_CONFIG = config["PWDI_CONFIG"]
        return {
            "timeout": int(PWDI_CONFIG.get("timeout", 10)),
            "filename": PWDI_CONFIG.get("filename", "PWDI.temp"),
            "filesize": int(PWDI_CONFIG.get("filesize", 1024 * 1024)),
            "delfile": int(PWDI_CONFIG.get("delfile", 1))
        }


drives = get_drives()
idle_drives = []
for drive in drives:
    if "PWDI.ini" in os.listdir(drive):
        idle_drives.append(drive)

config_idle_drives = {}
time_idle_drives = {}
for idle_drive in idle_drives:
    config = ConfigParser()
    config.read(f"{idle_drive}PWDI.ini")
    PWDI_CONFIG = config["PWDI_CONFIG"]
    config_idle_drives[idle_drive] = {
        "timeout": int(PWDI_CONFIG.get("timeout", 10)),
        "filename": PWDI_CONFIG.get("filename", "PWDI.temp"),
        "filesize": int(PWDI_CONFIG.get("filesize", 1024 * 1024)),
        "delfile": int(PWDI_CONFIG.get("delfile", 1))
    }
    time_idle_drives[idle_drive] = time.time()

while True:
    time.sleep(1)
    now = time.time()
    for idle_drive in idle_drives:
        if not ((now - time_idle_drives[idle_drive]) > config_idle_drives[idle_drive]["timeout"]):
            continue
        # print(f"Idling {idle_drive}")
        with open(f'{idle_drive}{config_idle_drives[idle_drive]["filename"]}', "wb") as f:
            bstr = b"".join((randint(0, 255)).to_bytes(length=1, byteorder=sys.byteorder) for i in range(config_idle_drives[idle_drive]["filesize"]))
            f.write(bstr)
        if config_idle_drives[idle_drive]["delfile"] and config_idle_drives[idle_drive]["filename"] in os.listdir(
                idle_drive):
            os.remove(f'{idle_drive}{config_idle_drives[idle_drive]["filename"]}')
        time_idle_drives[idle_drive] = time.time()

[PWDI_CONFIG]
timeout=30
filesize=131072
filename=PWDI.temp
delfile=1