import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QToolBar, QAction, QLineEdit
from PyQt5.QtCore import QUrl
from PyQt5.QtGui import QIcon
from PyQt5.QtWebEngineWidgets import QWebEngineView


class MOSAAA(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("MOSAAA Browser")
        self.setWindowIcon(QIcon("mosaaa.png"))  # ✅ Set custom icon
        self.setGeometry(100, 100, 1200, 800)

        self.browser = QWebEngineView()

        # ✅ Set custom homepage to your local HTML file
        self.browser.setUrl(QUrl.fromLocalFile(
            "C:/Users/robyn/OneDrive/Desktop/alienfolder/index.html"
        ))

        self.setCentralWidget(self.browser)
        self.create_toolbar()

    def create_toolbar(self):
        navbar = QToolBar()
        self.addToolBar(navbar)

        # Back Button
        back_btn = QAction("Back", self)
        back_btn.triggered.connect(self.browser.back)
        navbar.addAction(back_btn)

        # Forward Button
        forward_btn = QAction("Forward", self)
        forward_btn.triggered.connect(self.browser.forward)
        navbar.addAction(forward_btn)

        # Reload Button
        reload_btn = QAction("Reload", self)
        reload_btn.triggered.connect(self.browser.reload)
        navbar.addAction(reload_btn)

        # Home Button
        home_btn = QAction("Home", self)
        home_btn.triggered.connect(self.navigate_home)
        navbar.addAction(home_btn)

        # URL Bar
        self.url_bar = QLineEdit()
        self.url_bar.returnPressed.connect(self.navigate_to_url)
        navbar.addWidget(self.url_bar)

        self.browser.urlChanged.connect(self.update_url)

    def navigate_home(self):
        self.browser.setUrl(QUrl.fromLocalFile(
            "C:/Users/robyn/OneDrive/Desktop/alienfolder/index.html"
        ))

    def navigate_to_url(self):
        url = self.url_bar.text()
        if not url.startswith("http") and not url.startswith("file://"):
            url = "http://" + url
        self.browser.setUrl(QUrl(url))

    def update_url(self, q):
        self.url_bar.setText(q.toString())


if __name__ == '__main__':
    app = QApplication(sys.argv)
    app.setApplicationName("MOSAAA Browser")
    window = MOSAAA()
    window.show()
    sys.exit(app.exec_())