# This file has parts from Buildbot and is modifyed by Gentoo Authors. 
# Buildbot 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, version 2.
#
# This program 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
# this program; if not, write to the Free Software Foundation, Inc., 51
# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Copyright Buildbot Team Members
# Origins: buildbot.db.*
# Modifyed by Gentoo Authors.
# Copyright 2021 Gentoo Authors

import uuid
import sqlalchemy as sa

from twisted.internet import defer

from buildbot.db import base
class VersionsConnectorComponent(base.DBConnectorComponent):

    @defer.inlineCallbacks
    def getVersionByName(self, name, p_uuid):
        def thd(conn):
            tbl = self.db.model.versions
            q = tbl.select()
            q = q.where(tbl.c.name == name)
            q = q.where(tbl.c.package_uuid == p_uuid)
            res = conn.execute(q)
            row = res.fetchone()
            if not row:
                return None
            return self._row2dict(conn, row)
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def getVersionByUuid(self, uuid):
        def thd(conn):
            tbl = self.db.model.versions
            q = tbl.select()
            q = q.where(tbl.c.uuid == uuid)
            res = conn.execute(q)
            row = res.fetchone()
            if not row:
                return None
            return self._row2dict(conn, row)
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def addVersion(self, name, package_uuid, file_hash, commit_id, change_id):
        def thd(conn, no_recurse=False):
            try:
                tbl = self.db.model.versions
                q = tbl.insert()
                r = conn.execute(q, dict(name=name,
                                         package_uuid=package_uuid,
                                         file_hash=file_hash,
                                         commit_id=commit_id,
                                         change_id=change_id))
            except (sa.exc.IntegrityError, sa.exc.ProgrammingError):
                uuid = None
            else:
                uuid = r.inserted_primary_key[0]
            return uuid
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def delVersion(self, uuid):
        deleted_at = int(self.master.reactor.seconds())
        def thd(conn, no_recurse=False):
        
                tbl = self.db.model.versions
                q = tbl.update()
                q = q.where(tbl.c.uuid == uuid)
                conn.execute(q, deleted=True,
                                deleted_at=deleted_at)
        yield self.db.pool.do(thd)

    @defer.inlineCallbacks
    def addKeyword(self, version_uuid, keyword_id, status):
        def thd(conn, no_recurse=False):
            try:
                tbl = self.db.model.versions_keywords
                q = tbl.insert()
                r = conn.execute(q, dict(version_uuid=version_uuid,
                                         keyword_id=keyword_id,
                                         status=status))
            except (sa.exc.IntegrityError, sa.exc.ProgrammingError):
                uuid = None
            else:
                uuid = r.inserted_primary_key[0]
            return uuid
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def addMetadata(self, version_uuid, metadata, value):
        def thd(conn, no_recurse=False):
            try:
                tbl = self.db.model.versions_metadata
                q = tbl.insert()
                r = conn.execute(q, dict(version_uuid=version_uuid,
                                         metadata=metadata,
                                         value=value))
            except (sa.exc.IntegrityError, sa.exc.ProgrammingError):
                id = None
            else:
                id = r.inserted_primary_key[0]
            return uuid
        res = yield self.db.pool.do(thd)
        return res

    #FIXME: return sorted by id
    @defer.inlineCallbacks
    def getMetadataByUuidAndMatadata(self, uuid, metadata):
        def thd(conn):
            tbl = self.db.model.versions_metadata
            q = tbl.select()
            q = q.where(tbl.c.version_uuid == uuid)
            q = q.where(tbl.c.metadata == metadata)
            return [self._row2dict_version_metadata(conn, row)
                for row in conn.execute(q).fetchall()]
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def getEbuildsByPackage(self, p_uuid, deleted=False):
        def thd(conn):
            tbl = self.db.model.versions
            q = tbl.select()
            q = q.where(tbl.c.deleted == deleted)
            q = q.where(tbl.c.package_uuid == p_uuid)
            return [self._row2dict(conn, row)
                for row in conn.execute(q).fetchall()]
        res = yield self.db.pool.do(thd)
        return res

    @defer.inlineCallbacks
    def removeVersion(self, uuid):
        def thd(conn, no_recurse=False):
                tbl = self.db.model.versions
                q = tbl.delete()
                q = q.where(tbl.c.uuid == uuid)
                conn.execute(q)
        yield self.db.pool.do(thd)

    @defer.inlineCallbacks
    def removeVersionMetadata(self, version_uuid):
        def thd(conn, no_recurse=False):
                tbl = self.db.model.versions_metadata
                q = tbl.delete()
                q = q.where(tbl.c.version_uuid == version_uuid)
                conn.execute(q)
        yield self.db.pool.do(thd)

    @defer.inlineCallbacks
    def removeVersionKeyword(self, version_uuid):
        def thd(conn, no_recurse=False):
                tbl = self.db.model.versions_keywords
                q = tbl.delete()
                q = q.where(tbl.c.version_uuid == version_uuid)
                conn.execute(q)
        yield self.db.pool.do(thd)


    def _row2dict(self, conn, row):
        return dict(
            uuid=row.uuid,
            name=row.name,
            package_uuid=row.package_uuid,
            file_hash=row.file_hash,
            commit_id=row.commit_id,
            change_id = row.change_id,
            deleted=row.deleted,
            deleted_at=row.deleted_at
            )

    def _row2dict_version_metadata(self, conn, row):
        return dict(
            id=row.id,
            version_uuid=row.version_uuid,
            metadata=row.metadata,
            value=row.value
            )