mirror of https://github.com/leafspark/AutoGGUF
113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
from PySide6.QtCore import QPoint, Qt
|
|
from PySide6.QtWidgets import QHBoxLayout, QLabel, QMenuBar, QPushButton, QWidget
|
|
|
|
|
|
class CustomTitleBar(QWidget):
|
|
def __init__(self, parent=None) -> None:
|
|
super().__init__(parent)
|
|
self.parent = parent
|
|
layout = QHBoxLayout(self)
|
|
layout.setContentsMargins(10, 5, 10, 5)
|
|
|
|
# Add the favicon
|
|
# TODO: uncomment this
|
|
# self.icon_label = QLabel()
|
|
# self.icon_label.setPixmap(QPixmap(resource_path("assets/favicon.ico")))
|
|
# layout.addWidget(self.icon_label)
|
|
|
|
# Add app title (bolded)
|
|
self.title = QLabel("<b>AutoGGUF</b>") # Use HTML tags for bolding
|
|
layout.addWidget(self.title)
|
|
|
|
# Add menubar here
|
|
self.menubar = QMenuBar()
|
|
layout.addWidget(self.menubar) # Add menubar to the layout
|
|
|
|
layout.addStretch(1) # This pushes the buttons to the right
|
|
|
|
# Add minimize and close buttons
|
|
self.minimize_button = QPushButton("—")
|
|
self.close_button = QPushButton("✕")
|
|
|
|
for button in (self.minimize_button, self.close_button):
|
|
button.setFixedSize(30, 30)
|
|
button.setStyleSheet(
|
|
"""
|
|
QPushButton {
|
|
border: none;
|
|
background-color: transparent;
|
|
}
|
|
QPushButton:hover {
|
|
background-color: rgba(255, 255, 255, 0.1);
|
|
}
|
|
"""
|
|
)
|
|
|
|
# Enable mouse tracking for smoother movement
|
|
self.setMouseTracking(True)
|
|
|
|
# Add maximize button
|
|
self.maximize_button = QPushButton("□")
|
|
self.maximize_button.setFixedSize(30, 30)
|
|
self.maximize_button.setStyleSheet(
|
|
"""
|
|
QPushButton {
|
|
border: none;
|
|
background-color: transparent;
|
|
padding: 2px;
|
|
font-size: 15px;
|
|
}
|
|
QPushButton:hover {
|
|
background-color: rgba(255, 255, 255, 0.1);
|
|
}
|
|
"""
|
|
)
|
|
self.maximize_button.clicked.connect(self.toggle_maximize)
|
|
|
|
layout.addWidget(self.minimize_button)
|
|
layout.addWidget(self.maximize_button)
|
|
layout.addWidget(self.close_button)
|
|
|
|
self.minimize_button.clicked.connect(self.parent.showMinimized)
|
|
self.close_button.clicked.connect(self.parent.close)
|
|
|
|
self.start = QPoint(0, 0)
|
|
self.pressing = False
|
|
self.isMaximized = False # Flag to track maximization state
|
|
self.normal_size = None # Store the normal window size
|
|
|
|
def mousePressEvent(self, event) -> None:
|
|
if event.button() == Qt.LeftButton:
|
|
self.start = event.globalPos() - self.parent.frameGeometry().topLeft()
|
|
self.pressing = True
|
|
|
|
def mouseMoveEvent(self, event) -> None:
|
|
if self.pressing:
|
|
new_pos = event.globalPos() - self.start
|
|
screen = self.parent.screen()
|
|
screen_geo = screen.availableGeometry()
|
|
|
|
# Check if the new position would put the titlebar below the taskbar
|
|
if (
|
|
new_pos.y() + self.parent.height() > screen_geo.bottom()
|
|
): # Use screen_geo.bottom()
|
|
new_pos.setY(screen_geo.bottom() - self.parent.height())
|
|
|
|
self.parent.move(new_pos)
|
|
|
|
def mouseReleaseEvent(self, event) -> None:
|
|
self.pressing = False
|
|
|
|
def toggle_maximize(self):
|
|
if self.isMaximized:
|
|
self.parent.showNormal()
|
|
if self.normal_size:
|
|
self.parent.resize(self.normal_size)
|
|
self.maximize_button.setText("□") # Change back to maximize symbol
|
|
self.isMaximized = False
|
|
else:
|
|
self.normal_size = self.parent.size() # Store the current size
|
|
self.parent.showMaximized()
|
|
self.maximize_button.setText("❐") # Change to restore symbol
|
|
self.isMaximized = True
|