SoftwareManager/core/ui/gui.py

from PySide6 import QtWidgets
from PySide6.QtCore import Qt
from PySide6.QtWidgets import QLineEdit, QPushButton, QWidget, QVBoxLayout, QListWidget, QToolBar, QDialog, QVBoxLayout, QLabel, QHBoxLayout,QComboBox
from PySide6.QtGui import QIcon, QAction
import darkdetect
import requests
from core.scraping.uztracker_scraper import scrape_uztracker
from core.scraping.rutracker_scraper import scrape_rutracker
from core.scraping.utils import get_magnet_link
from core.downloading.download import start_client
from core.downloading.download import add_magnet

def state_debug(setting):
    global debug
    if setting is True:
        debug = True
    else:
        debug = False

global tracker
tracker = "rutracker" # default tracker


class MainWindow(QtWidgets.QMainWindow, QWidget):
    def __init__(self):
        global settings_action
        super().__init__()
        searchresults = []

        self.setWindowTitle("Software Manager")
        self.setGeometry(100, 100, 800, 600)

        self.controls = QWidget()
        self.controlsLayout = QVBoxLayout()

        # Widgets
        self.searchbar = QLineEdit()
        self.searchbar.setPlaceholderText("Search for software...")
        self.searchbar.setClearButtonEnabled(True)
        self.searchbar.setMinimumHeight(30)
        self.searchbar.returnPressed.connect(self.return_pressed) # Triggers scraping functions on enter
        self.button = QtWidgets.QPushButton("Download")
        self.softwareList = QListWidget()

        container = QWidget()
        containerLayout = QVBoxLayout()
        containerLayout.addWidget(self.searchbar)
        
        containerLayout.addWidget(self.softwareList)
        self.softwareList.addItems(searchresults)

        containerLayout.addWidget(self.button)
        # download button triggers
        self.button.clicked.connect(lambda: self.download_selected())
        self.button.clicked.connect(lambda: self.get_selected_item())

        container.setLayout(containerLayout)
        self.setCentralWidget(container)

        containerLayout.setAlignment(Qt.AlignmentFlag.AlignBottom)

        toolbar = QToolBar("Main Toolbar")
        self.addToolBar(toolbar)
        toolbar.setLayoutDirection(Qt.RightToLeft)
        
        self.tracker_list = QComboBox()
        self.tracker_list.addItems(["rutracker", "uztracker"])
        self.tracker_list.activated.connect(self.set_tracker)


        if darkdetect.isDark():
            settings_action = QAction(QIcon("core/assets/settings_dark.png"), "Settings", self)
        else:
            settings_action = QAction(QIcon("core/assets/settings.png"), "Settings", self)

        settings_action.triggered.connect(self.settings_dialog)
        toolbar.addAction(settings_action)
        toolbar.addWidget(self.tracker_list)

        toolbar.setMovable(False)

    def set_tracker(self, _):
        global tracker
        tracker = self.tracker_list.currentText()

    def settings_dialog(self):
        if debug:
            print("Settings dialog opened")
        dialog = QDialog(self)
        dialog.setWindowTitle("Settings")
        dialog.setFixedSize(400, 300)


        dialog.setLayout(QVBoxLayout())
        label = QLabel("This is a settings dialog.")
        QtWidgets.QPushButton("Close", clicked=dialog.close)
        # close button
        dialog.layout().addWidget(label)
        dialog.layout().addWidget(QtWidgets.QPushButton("Close", clicked=dialog.close))

        layout = QHBoxLayout()

        save_btn = QPushButton("Save")
        cancel_btn = QPushButton("Cancel")
        save_btn.clicked.connect(self.save_settings)
        cancel_btn.clicked.connect(dialog.reject)
        layout.addWidget(save_btn)
        layout.addWidget(cancel_btn)

        self.setLayout(QVBoxLayout())
        dialog.layout().addLayout(layout)

        
        dialog.exec()

    def save_settings(self):
        pass


    def download_selected(self):
        item = self.softwareList.currentItem()
        if item is not None:
            if debug:
                print(f"Downloading {self.softwareList.currentItem().text()}")
            self.get_item_index(self.softwareList.currentItem().text(), self.postnames, self.postlinks, debug)
        else:
            if debug:
                print("No item selected for download.")
            # add gui notification for no item selected


    def get_selected_item(self): 
        item = self.softwareList.currentItem()
        if item is not None:
            return item.text()
        return ""


    def return_pressed(self):
        search_text = self.searchbar.text()
        if debug:
            print("User searched for:", search_text)
        if tracker == "uztracker":
            response = scrape_uztracker(search_text, debug)
        if tracker == "rutracker":
            response = scrape_rutracker(search_text)
        if response:
            self.postnames, self.postlinks = response
            self.softwareList.clear()
            if self.postnames:
                self.softwareList.addItems(self.postnames)
        elif debug:
            print(f"No Results found for \"{search_text}\"")

    def get_item_index(self, item, list, listlinks, debug):
        position = list.index(item)
        if 0 <= position < len(listlinks): # note for myself: py starts counting at 0; check if number is not negative, check if number is not more than list length.
            if tracker == "uztracker":
                selected = "https://uztracker.net/" + listlinks[position].lstrip("./")
            if tracker == "rutracker":                
                selected = listlinks[position]

            if debug:
                print("Selected URL: ", selected)
            selected_magnet = get_magnet_link(selected, debug)
            start_client()
            add_magnet(selected_magnet)