# -*- coding: utf-8 -*-

# Trac imports.
from trac.core import *
from trac.db import *

# Trac interfaces.
from trac.env import IEnvironmentSetupParticipant

# Last discussion database shcema version.
last_db_version = 5

class DiscussionInit(Component):
    """
        Initialise database and environment for discussion component.
    """
    implements(IEnvironmentSetupParticipant)

    # IEnvironmentSetupParticipant.
    def environment_created(self):
        pass

    def environment_needs_upgrade(self, db):
        cursor = db.cursor()

        # Is database up to date?
        return self._get_db_version(cursor) != last_db_version

    def upgrade_environment(self, db):
        cursor = db.cursor()

        # Get current database schema version.
        db_version = self._get_db_version(cursor)

        # Perform incremental upgrades.
        for I in range(db_version + 1, last_db_version + 1):
            script_name  = 'db%i' % (I)
            module = __import__('tracdiscussion.db.%s' % (script_name),
            globals(), locals(), ['do_upgrade'])
            module.do_upgrade(self.env, cursor)

        # Finish database changes.
        db.commit()

    def _get_db_version(self, cursor):
        try:
            sql = ("SELECT value "
                   "FROM system "
                   "WHERE name='discussion_version'")
            self.log.debug(sql)
            cursor.execute(sql)
            for row in cursor:
                return int(row[0])
            return 0
        except:
            return 0