from flask import current_app
from flask_migrate import upgrade
import os
from app.models import (
    CorpusFollowerRole,
    Role,
    SpaCyNLPPipelineModel,
    TesseractOCRPipelineModel,
    User
)


def init_app(app):
    @app.cli.command('deploy')
    def deploy():
        ''' Run deployment tasks. '''
        # Make default directories
        print('Make default directories')
        base_dir = current_app.config['NOPAQUE_DATA_DIR']
        default_dirs = [
            os.path.join(base_dir, 'tmp'),
            os.path.join(base_dir, 'users')
        ]
        for dir in default_dirs:
            if os.path.exists(dir):
                if not os.path.isdir(dir):
                    raise NotADirectoryError(f'{dir} is not a directory')
            else:
                os.mkdir(dir)

        # migrate database to latest revision
        print('Migrate database to latest revision')
        upgrade()

        # Insert/Update default database values
        print('Insert/Update default Roles')
        Role.insert_defaults()
        print('Insert/Update default Users')
        User.insert_defaults()
        print('Insert/Update default CorpusFollowerRoles')
        CorpusFollowerRole.insert_defaults()
        print('Insert/Update default SpaCyNLPPipelineModels')
        SpaCyNLPPipelineModel.insert_defaults()
        print('Insert/Update default TesseractOCRPipelineModels')
        TesseractOCRPipelineModel.insert_defaults()