# -*- coding: UTF-8 -*-
# Copyright 2020 by Kurt Rathjen. All Rights Reserved.
#
# This library is free software: you can redistribute it and/or modify it
# under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version. This library is distributed in the
# hope that it will be useful, but WITHOUT ANY WARRANTY; without even the
# implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
# See the GNU Lesser General Public License for more details.
# You should have received a copy of the GNU Lesser General Public
# License along with this library. If not, see <http://www.gnu.org/licenses/>.

from functools import partial

from studiovendor.Qt import QtGui
from studiovendor.Qt import QtWidgets

from .separatoraction import SeparatorAction


class GroupByMenu(QtWidgets.QMenu):

    def __init__(self, name, parent, dataset):
        super(GroupByMenu, self).__init__(name, parent)

        self._dataset = dataset
        self.aboutToShow.connect(self.populateMenu)

    def setDataset(self, dataset):
        """
        Set the dataset model for the menu:
        
        :type dataset: studiolibrary.Dataset
        """
        self._dataset = dataset

    def dataset(self):
        """
        Get the dataset model for the menu.
        
        :rtype: studiolibrary.Dataset 
        """
        return self._dataset

    def setGroupBy(self, groupName, groupOrder):
        """
        Set the group by value for the dataset.
        
        :type groupName: str 
        :type groupOrder: str
        """
        if groupName:
            value = [groupName + ":" + groupOrder]
        else:
            value = None

        self.dataset().setGroupBy(value)
        self.dataset().search()

    def populateMenu(self):
        """
        Show the menu options.
        """
        self.clear()

        groupBy = self.dataset().groupBy()
        if groupBy:
            currentField = groupBy[0].split(":")[0]
            currentOrder = "dsc" if "dsc" in groupBy[0] else "asc"
        else:
            currentField = ""
            currentOrder = ""

        action = SeparatorAction(u"分组", self)
        self.addAction(action)

        action = self.addAction("None")
        action.setCheckable(True)

        if not currentField:
            action.setChecked(True)

        callback = partial(self.setGroupBy, None, None)
        action.triggered.connect(callback)

        fields = self.dataset().fields()

        for field in fields:

            if not field.get("groupable"):
                continue

            name = field.get("name")

            action = self.addAction(name.title())
            action.setCheckable(True)

            if currentField == name:
                action.setChecked(True)
            else:
                action.setChecked(False)

            callback = partial(self.setGroupBy, name, currentOrder)
            action.triggered.connect(callback)

        action = SeparatorAction(u"分组顺序", self)
        self.addAction(action)

        action = self.addAction("Ascending")
        action.setCheckable(True)
        action.setChecked(currentOrder == "asc")

        callback = partial(self.setGroupBy, currentField, "asc")
        action.triggered.connect(callback)

        action = self.addAction("Descending")
        action.setCheckable(True)
        action.setChecked(currentOrder == "dsc")

        callback = partial(self.setGroupBy, currentField, "dsc")
        action.triggered.connect(callback)
