#!/usr/bin/python3

from pathlib import Path
from PyQt6.QtWidgets import QMessageBox


class Utils:
    def __init__(self, app):
        self.app = app

        self.home_path = str(Path.home())

    def popup_init(self):
        self.usc_popup = QMessageBox()  # create a popup window that notifies the user that we have unsaved changes
        self.usc_popup.setWindowTitle("Unsaved Changes!")  # usc means unsaved changes
        self.usc_popup.setText("The file you are trying to close has unsaved changes.")
        self.usc_popup.setStandardButtons(
            QMessageBox.StandardButton.Save |
            QMessageBox.StandardButton.Discard |
            QMessageBox.StandardButton.Cancel
        )

    def unsaved_changes_popup(self):
        button = self.usc_popup.exec()

        match button:
            case QMessageBox.StandardButton.Save:
                return "save"
            case QMessageBox.StandardButton.Discard:
                return "discard"
            case QMessageBox.StandardButton.Cancel:
                return "cancel"

    def bstring_to_oz(self, data):  # convert binary data to a string of ones and zeros (oz)
        oz_bytes = []

        for byte in data:
            oz_bytes.append(format(byte, "08b"))

        oz_string = " ".join(oz_bytes)

        return oz_string

    def oz_string_to_bstring(self, oz_string):  # convert a string of zeroes and ones to a binary string
        oz_bytes = oz_string.split()

        bytes_int = []

        for byte in oz_bytes:
            bytes_int.append(int(byte, 2))

        binary_string = bytes(bytes_int)

        return binary_string