import sys
import os
import PySide6
from PySide6.QtWidgets import QApplication, QMainWindow
from PySide6.QtWebChannel import QWebChannel
from PySide6.QtCore import Slot
from mainwindow import Ui_MainWindow
import folium
from folium import template
import io
import pandas as pd

dirname = os.path.dirname(PySide6.__file__)
plugin_path = os.path.join(dirname, 'plugins', 'platforms')
os.environ['QT_QPA_PLATFORM_PLUGIN_PATH'] = plugin_path


class BridgeFolium(folium.MacroElement):
    _template = template.Template(
        """
        {% macro script(this, kwargs) %}
        var mainWindow;
        
        // Ждём загрузки страницы
        document.addEventListener('DOMContentLoaded', function() {
            // Создаём WebChannel
            new QWebChannel(qt.webChannelTransport, function(channel) {
                mainWindow = channel.objects.mainWindow;
            });
            
            // Ждём загрузки карты Leaflet
            setTimeout(function() {
                // Добавляем обработчик клика
                {{this._parent.get_name()}}.on('click', function(e) {
                    var lat = e.latlng.lat;
                    var lng = e.latlng.lng;
                    
                    if (mainWindow) {
                        mainWindow.receive_coordinates(lat, lng);
                    }
                });
            }, 1000);
        });
               
        {% endmacro %}
            """
    )


class ClickableMarker(folium.Marker):
    _template = template.Template(
        """
        {% macro script(this, kwargs) %}
            var {{ this.get_name() }} = L.marker(
                {{ this.location|tojson }},
                {{ this.options|tojavascript }}
            ).addTo({{ this._parent.get_name() }}).on('click', (e) => {
                var lat = e.latlng.lat;
                var lng = e.latlng.lng;
                
                if (mainWindow) {
                    mainWindow.marker_clicked(lat, lng, {{ this.options|tojavascript }}.id);
                }
            });
        {% endmacro %}
        """
    )


class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        self.df = pd.read_csv("./data.csv")

        self.ui.cmbPopulation.addItem('50000-100000', (50000, 100000))
        self.ui.cmbPopulation.addItem('100000-500000', (100001, 500000))
        self.ui.cmbPopulation.addItem('500000-1млн.', (500001, 1000000))
        self.ui.cmbPopulation.addItem('от 1млн.', (1000001, 100000000))
        self.ui.cmbPopulation.currentIndexChanged.connect(self.show_map)

        self.show_map()

    @Slot(float, float)
    def receive_coordinates(self, lat, lng):
        print(lat, lng)

    @Slot(float, float, int)
    def marker_clicked(self, lat, lng, _id):
        city = self.df[self.df.id == _id].iloc[0]
        self.ui.edtCity.setText(city.settlement)
        self.ui.edtInfo.setText(f"""
Население: {city.population}<br>
Муниципалитет: {city.municipality}<br>
Регион: {city.region}
        """.strip())
        

    def show_map(self):
        # создание объекты карты
        m = folium.Map(
            location=[52.286387, 104.280660],
            zoom_start=11,
            tiles="CartoDB Voyager",
            attributionControl=0
        )
        m.add_child(BridgeFolium())

        # Добавляем JavaScript в HTML-код карты
        m.get_root().html.add_child(folium.Element("""<script src="qrc:///qtwebchannel/qwebchannel.js"></script>"""))
        self.channel = QWebChannel()
        self.channel.registerObject("mainWindow", self)
        self.ui.web.page().setWebChannel(self.channel)

        ###
        left, right = self.ui.cmbPopulation.currentData()
        cities = self.df.loc[(left <= self.df.population) & (self.df.population <=right)]

        sw = cities[['latitude_dd', 'longitude_dd']].min().values.tolist()
        ne = cities[['latitude_dd', 'longitude_dd']].max().values.tolist()
        for city in cities.itertuples():
            ClickableMarker(
                location=[city.latitude_dd, city.longitude_dd],
                id=city.id,
                icon=folium.Icon(icon="cat", prefix='fa')
            ).add_to(m)

        m.fit_bounds([sw, ne])

        ###

        # подключение карты к интерфейсу
        m_data = io.BytesIO()
        m.save(m_data, close_file=False)
        self.ui.web.setHtml(m_data.getvalue().decode())



if __name__ == "__main__":
    app = QApplication(sys.argv)

    window = MainWindow()
    window.show()

    sys.exit(app.exec())