# =============================================================================
# Copyright (C) 2014 Ryan Holmes
#
# This file is part of pyfa.
#
# pyfa is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pyfa 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
# =============================================================================

import copy

from eos import db
from eos.saveddata.targetResists import TargetResists as es_TargetResists


class ImportError(Exception):
    pass


class TargetResists(object):
    instance = None

    @classmethod
    def getInstance(cls):
        if cls.instance is None:
            cls.instance = TargetResists()

        return cls.instance

    def getTargetResistsList(self):
        return db.getTargetResistsList()

    def getTargetResists(self, name):
        return db.getTargetResists(name)

    def newPattern(self, name):
        p = es_TargetResists(0.0, 0.0, 0.0, 0.0)
        p.name = name
        db.save(p)
        return p

    def renamePattern(self, p, newName):
        p.name = newName
        db.save(p)

    def deletePattern(self, p):
        db.remove(p)

    def copyPattern(self, p):
        newP = copy.deepcopy(p)
        db.save(newP)
        return newP

    def saveChanges(self, p):
        db.save(p)

    def importPatterns(self, text):
        lookup = {}
        current = self.getTargetResistsList()
        for pattern in current:
            lookup[pattern.name] = pattern

        imports, num = es_TargetResists.importPatterns(text)
        for pattern in imports:
            if pattern.name in lookup:
                match = lookup[pattern.name]
                match.__dict__.update(pattern.__dict__)
            else:
                db.save(pattern)
        db.commit()

        lenImports = len(imports)
        if lenImports == 0:
            raise ImportError("No patterns found for import")
        if lenImports != num:
            raise ImportError("%d patterns imported from clipboard; %d had errors" % (num, num - lenImports))

    def exportPatterns(self):
        patterns = self.getTargetResistsList()
        patterns.sort(key=lambda p: p.name)
        return es_TargetResists.exportPatterns(*patterns)