"""Add the default hook to all projects

Revision ID: 26af5c3602a0
Revises: 644ef887bb6f
Create Date: 2016-10-08 12:14:31.155018

"""

from __future__ import print_function, absolute_import

# revision identifiers, used by Alembic.
revision = '26af5c3602a0'
down_revision = '644ef887bb6f'

from alembic import op
import sqlalchemy as sa


try:
    import pagure.lib.plugins
    from pagure.lib import model
except ImportError:
    import sys
    sys.path.insert(0, '.')
    import pagure.lib.plugins
    from pagure.lib import model



def upgrade():
    ''' Add the default hook to all existing projects.
    '''

    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()

    # Update all the existing projects
    for project in session.query(model.Project).all():
        print('Installing %s' % project.fullname)
        # Install the default hook
        plugin = pagure.lib.plugins.get_plugin('default')
        dbobj = plugin.db_object()
        dbobj.active = True
        dbobj.project_id = project.id
        session.add(dbobj)
        session.flush()
        plugin.set_up(project)
        plugin.install(project, dbobj)
        # Save the change
        session.commit()


def downgrade():
    engine = op.get_bind()
    Session = sa.orm.scoped_session(sa.orm.sessionmaker())
    Session.configure(bind=engine)
    session = Session()

    # Update all the existing projects
    for project in session.query(model.Project).all():
        # Install the default hook
        plugin = pagure.lib.plugins.get_plugin('default')
        dbobj = plugin.db_object()
        dbobj.active = False
        dbobj.project_id = project.id
        session.add(dbobj)
        session.flush()
        plugin.remove(project, dbobj)

        # Save the change
        session.commit()