Datamosher.py

from tkinter import Tk
from tkinter.filedialog import askopenfilename
import string, random, os, shutil
import time

def isCorrupted(file_path):
    """Check if the file is corrupted or otherwise fucked."""
    # ToDo: implement a more reliable media file checker using ffmpeg 
    try:
        with open(file_path, 'rb') as file:
            file.read()  # Try to read the file
        return False  # File is readable
    except (IOError, OSError) as e:
        print(f"File corruption detected: {e}")
        return True  # File is fucked
    except Exception as e:
        print(f"Unexpected error: {e}")
        return True  # CORRUPTION!
    

def getFileInfo(fileSource, state='checkIfExists'):
    """Check if file already exists and figure what to do about it."""
    fileName, fileExtension = os.path.splitext(fileSource)
    if state == 'replace':
        newFileName = fileName
    else:
        newFileName = fileName + ' Mosh'
        for i in range(1, 99):
            if os.path.exists(newFileName + fileExtension):
                newFileName = f"{fileName} Mosh {i}"
            else:
                break
    return fileName, fileExtension, newFileName

def handleFile(fileSource, newFileName, operation):
    """FIle operations"""
    tmp_txt = newFileName + '.txt'
    shutil.copyfile(fileSource, tmp_txt)

    with open(tmp_txt, 'rb') as txtFile:
        fileContent = txtFile.read()

    fileContent = operation(fileContent)

    with open(tmp_txt, 'wb') as txtFile:
        txtFile.write(fileContent)

    final_path = newFileName + os.path.splitext(fileSource)[1]
    os.rename(tmp_txt, final_path)
    return final_path

def handle_choice(fileSource, choice):
    """Handles user choice."""
    operations = {
        '1': replaceMosh,
        '2': insertMosh,
        '3': copyPartMosh
    }
    if choice in operations:
        return operations[choice](fileSource)
    else:
        print("Bad choice. Please select 1, 2, or 3.")
        return None

def replaceMosh(fileSource):
    """Replace random bytes in the file with other random bytes."""
    replacable = [''.join(random.sample(string.ascii_lowercase + string.digits, 2)) for _ in range(random.randint(1, 20))]
    replacee = [''.join(random.sample(string.ascii_lowercase + string.digits, 2)) for _ in range(len(replacable))]

    fileName, fileExtension, newFileName = getFileInfo(fileSource)

    def operation(fileContent):
        for old, new in zip(replacable, replacee):
            fileContent = fileContent.replace(str.encode(old), str.encode(new))
        return fileContent

    return handleFile(fileSource, newFileName, operation)

def insertMosh(fileSource):
    """Add random characters to the file at random positions."""
    fileName, fileExtension, newFileName = getFileInfo(fileSource)

    def operation(fileContent):
        length = len(fileContent)
        for i in range(random.randint(10, 200)):
            if length == 0:
                position = 0
            else:
                position = random.randint(0, length - 1)
            randomChar = ''.join(random.sample(string.ascii_lowercase + string.digits, 2)) * random.randint(200, 40000)
            fileContent = fileContent[:position] + str.encode(randomChar) + fileContent[position:]
            length = len(fileContent)
        return fileContent

    return handleFile(fileSource, newFileName, operation)

def copyPartMosh(fileSource):
    """Repeat random chunks of the file at random positions."""
    fileName, fileExtension, newFileName = getFileInfo(fileSource)

    def operation(fileContent):
        length = len(fileContent)
        for _ in range(random.randint(10, 30)):
            if length <= 200:
                position1 = 0
            else:
                position1 = random.randint(0, length - 200)
            position2 = min(position1 + random.randint(200, 4000), length)
            chunk = fileContent[position1:position2]
            repeated_chunk = chunk * random.randint(10, 200)
            insert_position = random.randint(0, length)
            fileContent = fileContent[:insert_position] + repeated_chunk + fileContent[insert_position:]
            length = len(fileContent)
        return fileContent

    return handleFile(fileSource, newFileName, operation)

def main():
    root = Tk()
    root.withdraw()
    fileSource = askopenfilename()
    root.destroy()
    if not fileSource:
        print("No file selected. Exiting.")
        return

    choice = input('1 for ReplaceMosh\n2 for InsertMosh\n3 for Random Chunk copying\n:')
    newFilePath = handle_choice(fileSource, choice)
    if not newFilePath:
        return

    while True:
        if isCorrupted(newFilePath):
            print("File is corrupted. Retrying...")
            try:
                os.remove(newFilePath)  # Delete the corrupted mosh file
            except Exception as e:
                print(f"Error deleting file: {e}")
                return  # Fail gracefully if must

            # retry: create a new mosh version from original
            newFilePath = handle_choice(fileSource, choice)
            if not newFilePath:
                return
            time.sleep(0.5)
        else:
            print("File is not corrupted. We did it bois.")
            break  # Exit the program if the file is not corrupted

if __name__ == "__main__":
    main()

'''
if isCorrupted(newFileName + '.jpg') == False:
    while isCorrupted(newFileName + '.jpg') == False:
        mosh(fileSource)
        time.sleep(3)
else:
    print('yay')
'''