navbar.py

from PyQt5.QtCore import Qt, QUrl
from PyQt5.QtWidgets import QMainWindow, QLineEdit, QShortcut, QLabel
from PyQt5.QtWebEngineWidgets import QWebEngineView
from methods.database_methods import DBMethods

from methods.connect_methods import BrowserConnect
from methods.tabs_methods import TabsMethods
from methods.sidebar_methods import SideBarMethods

from methods.json_methods import ConfigMethods

from icecream import ic

import sys, re, datetime


class NavBar(QMainWindow):
    def __init__(self):
        super().__init__()
        self.init_tabs()

    def init_tabs(self):
        self.tabs = TabsMethods().create_tabs()
        
        self.tabs.tabBarDoubleClicked.connect(self.tab_open_doubleclick)
        self.tabs.currentChanged.connect(self.current_tab_changed)
        self.tabs.tabCloseRequested.connect(self.close_current_tab)
        self.setCentralWidget(self.tabs)
       
        self.nav_toolbar = TabsMethods().create_navigation_toolbar(self)
        self.addToolBar(self.nav_toolbar)
        
        self.urlBar = QLineEdit()
        self.urlBar.returnPressed.connect(self.goToUrl)
        self.nav_toolbar.addWidget(self.urlBar)
        
        
        self.label = QLabel()
        self.nav_toolbar.addWidget(self.label)
    

        self.addShortcut = QShortcut('Ctrl+T', self)
        self.addShortcut.activated.connect(self.addNewTab)
        
        self.closeShortcut = QShortcut('Ctrl+W', self)
        self.closeShortcut.activated.connect(self.close_current_tab)
        
        #self.setWindowFlags(Qt.WindowType.FramelessWindowHint) -> retira os botoes de fechar, minimizar e maximizar
        recentlyUrls = DBMethods().getRecentlyUrls()
        
        titlesUrls = [(item[1], item[2]) for item in recentlyUrls]
        ic(titlesUrls)
        
        for title, url in titlesUrls:
            self.addNewTab(title, QUrl(url))
        
        self.showMaximized()
        self.setWindowTitle('ACS Browser')
        
        self.addToolBar(Qt.ToolBarArea.LeftToolBarArea, SideBarMethods().create_sidebar(self))
    
    
    def addNewTab(self, title = 'No Title Find',  url = None, label="Blank",):
        if url is None:
            url = QUrl(ConfigMethods().loadJson()['newTabURL'])

        browser = QWebEngineView()
        browser.load(url)
              
        currentTabIndex = self.tabs.addTab(browser, label)
        self.tabs.setCurrentIndex(currentTabIndex)
        self.tabs.setTabText(currentTabIndex, title)
        
        
        browser.urlChanged.connect(lambda url, browser=browser:
                                     BrowserConnect().update_urlBar(self, url, browser))
        

        browser.titleChanged.connect(lambda _, i=currentTabIndex, browser=browser:
                                     self.tabs.setTabText(i, self.tabs.currentWidget().page().title()))
        
        
        browser.iconChanged.connect(lambda _, browser=browser:
                                     BrowserConnect().updateIcon(self, browser))
        
        browser.page().loadFinished.connect(self.loadDBMethods)
        


    def goToUrl(self):
        url = QUrl(self.urlBar.text())
        urlBarTxt = self.urlBar.text()
        
        self.searchEngineDefault = ''
        
        match ConfigMethods().loadJson()['searchEngine']:
            case 'Google':
                self.searchEngineDefault = 'https://www.google.com/search?q='
            case 'Yahoo':
                self.searchEngineDefault = 'https://search.yahoo.com/search?q='
            
            case 'Bing':
                self.searchEngineDefault = 'https://www.bing.com/search?q='
            
            case 'DuckDuckGo':
                self.searchEngineDefault = 'https://duckduckgo.com/?q='
            
            case _:
                self.searchEngineDefault = 'https://www.google.com/search?q='
            
        
        valid_url = re.compile(
    r"^(http|https)?:?(\/\/)?(www\.)?[-a-zA-Z0-9@:%._\+~#=]{1,256}\.[a-zA-Z0-9()]{1,6}\b([-a-zA-Z0-9()@:%_\+.~#?&//=]*)"
        )
        
        if valid_url.search(urlBarTxt) and not any(
            i in urlBarTxt for i in ("http://", "https://", "file:///")
        ):
            url = f'https://{urlBarTxt}'
            
            
        elif '/' not in urlBarTxt:
            url = f'{self.searchEngineDefault}{self.urlBar.text()}'
            
        self.tabs.currentWidget().load(QUrl(url))
      
      

    def tab_open_doubleclick(self, currentTabIndex):
        if currentTabIndex == -1:
            self.addNewTab()
                        
                        
            
    def current_tab_changed(self):
        url = self.tabs.currentWidget().url()
        BrowserConnect().update_urlBar(self, url, self.tabs.currentWidget())
        
         
    def close_current_tab(self):
        if self.tabs.count() < 2:
            sys.exit()
        
        self.tabs.currentWidget().close()
        #o .close() termina o processo do widget atual, impedindo que uma aba ja fechada venha ocupar espaço na memoria
        #assim, o browser pelo menos n fica ocupando gigas na memoria qnd se tem apenas 1 aba aberta
        self.tabs.removeTab(self.tabs.currentIndex())
        
    
    def loadDBMethods(self):
        title = self.tabs.currentWidget().page().title()
        url = DBMethods().convertUrlToStr(self.tabs.currentWidget().page().url())
        date = DBMethods().convertDateToBR(datetime.datetime.now())
        
        
        # history
        DBMethods().replaceOldData('history', 'url', url, 2)


        DBMethods().insert(
            "INSERT INTO history (title,url,date) VALUES (:title,:url,:date)",
            {"title": title, "url": url, "date": date}
                      )
        
        # zoom
        zoomInDB = DBMethods().getCurrentZoomPage(self.tabs.currentWidget().page().url())
        self.tabs.currentWidget().setZoomFactor(float(zoomInDB))
        self.label.setText(f"Zoom atual: {self.tabs.currentWidget().zoomFactor():.1f}")
        
        
        # save recently page
        DBMethods().replaceOldData('tabs', 'url', url, 2)
        DBMethods().saveRecentlyPage(title, url)