from baserow.contrib.builder.workflow_actions.models import (
    LocalBaserowCreateRowWorkflowAction,
    LocalBaserowUpdateRowWorkflowAction,
    NotificationWorkflowAction,
)


class WorkflowActionFixture:
    def create_notification_workflow_action(self, **kwargs):
        return self.create_workflow_action(NotificationWorkflowAction, **kwargs)

    def create_upsert_row_workflow_action(self, model_class, **kwargs):
        if "service" not in kwargs:
            user = kwargs.pop("user", self.create_user())
            integration = self.create_local_baserow_integration(
                application=kwargs["page"].builder, user=user
            )
            kwargs["service"] = self.create_local_baserow_upsert_row_service(
                integration=integration,
            )
        return self.create_workflow_action(model_class, **kwargs)

    def create_local_baserow_create_row_workflow_action(self, **kwargs):
        return self.create_upsert_row_workflow_action(
            LocalBaserowCreateRowWorkflowAction, **kwargs
        )

    def create_local_baserow_update_row_workflow_action(self, **kwargs):
        return self.create_upsert_row_workflow_action(
            LocalBaserowUpdateRowWorkflowAction, **kwargs
        )

    def create_workflow_action(self, model_class, **kwargs):
        if "order" not in kwargs:
            kwargs["order"] = 0

        if "page" not in kwargs and "element" in kwargs:
            kwargs["page"] = kwargs["element"].page

        return model_class.objects.create(**kwargs)