"""example_data."""
import glob
import os

from flask import current_app
from spiff_workflow_webapp.models.process_model import ProcessModelInfo
from spiff_workflow_webapp.services.spec_file_service import SpecFileService
from spiff_workflow_webapp.services.process_model_service import ProcessModelService


class ExampleDataLoader:
    """ExampleDataLoader."""

    def create_spec(self, id, display_name="", description="", filepath=None, master_spec=False,
                    process_group_id='', display_order=0, from_tests=False, standalone=False, library=False):
        """Assumes that a directory exists in static/bpmn with the same name as the given id.

        further assumes that the [id].bpmn is the primary file for the process model.
        returns an array of data models to be added to the database.
        """
        global file
        spec = ProcessModelInfo(id=id,
                                display_name=display_name,
                                description=description,
                                process_group_id=process_group_id,
                                display_order=display_order,
                                is_master_spec=master_spec,
                                standalone=standalone,
                                library=library,
                                primary_file_name="",
                                primary_process_id="",
                                is_review=False,
                                libraries=[])
        workflow_spec_service = ProcessModelService()
        workflow_spec_service.add_spec(spec)

        if not filepath and not from_tests:
            filepath = os.path.join(current_app.root_path, 'static', 'bpmn', id, "*.*")
        if not filepath and from_tests:
            filepath = os.path.join(current_app.root_path, '..', 'tests', 'data', id, "*.*")

        files = glob.glob(filepath)
        for file_path in files:
            if os.path.isdir(file_path):
                continue  # Don't try to process sub directories

            noise, file_extension = os.path.splitext(file_path)
            filename = os.path.basename(file_path)
            is_primary = filename.lower() == id + '.bpmn'
            file = None
            try:
                file = open(file_path, 'rb')
                data = file.read()
                SpecFileService.add_file(workflow_spec=spec, file_name=filename, binary_data=data)
                if is_primary:
                    SpecFileService.set_primary_bpmn(spec, filename, data)
                    workflow_spec_service = ProcessModelService()
                    workflow_spec_service.update_spec(spec)
            except IsADirectoryError:
                # Ignore sub directories
                pass
            finally:
                if file:
                    file.close()
        return spec