# Copyright (c) 2015 Ultimaker B.V.
# Uranium is released under the terms of the AGPLv3 or higher.

class Version(object):
    def __init__(self, version):
        super().__init__()
        if isinstance(version, str):
            version_list = version.replace("-", ".").replace("_", ".").split(".")
        else:
            version_list = version
        self._major = 0
        self._minor = 0
        self._revision = 0
        try:
            self._major = int(version_list[0])
            self._minor = int(version_list[1])
            self._revision = int(version_list[2])
        except IndexError:
            pass

    def getMajor(self):
        return self._major

    def getMinor(self):
        return self._minor

    def getRevision(self):
        return self._revision

    def __gt__ (self, other):
        if type(other) is Version:
            return other.__lt__(self)
        elif type(other) is str:
            return Version(other).__lt__(self)
        else:
            return False

    def __lt__(self, other):
        if type(other) is Version:
            if self._major < other.getMajor():
                return True
            if self._minor < other.getMinor() and self._major == other.getMajor():
                return True
            if self._revision < other.getRevision() and self._major == other.getMajor() and self._minor == other.getMinor():
                return True
            return False
        elif type(other) is str:
            return self < Version(other)
        else:
            return False

    def __eq__(self, other):
        if type(other) is Version:
            return self._major == other.getMajor() and self._minor == other.getMinor() and self._revision == other.getRevision()
        elif type(other) is str:
            return self == Version(other)
        else:
            return False

    def __str__(self):
        return "%s.%s.%s" %(self._major, self._minor, self._revision)

    def __hash__(self):
        return hash(self.__str__())