create_service.rb

# frozen_string_literal: true

module DataExports
  # Service used to Create Data Export
  class CreateService < BaseService
    DataExportCreateError = Class.new(StandardError)
    def initialize(user = nil, params = {})
      super
    end

    def execute
      @data_export = DataExport.new(params)
      assign_initial_export_attributes

      if @data_export.valid?
        @data_export.export_type == 'analysis' ? validate_analysis_ids : validate_sample_ids
        @data_export.save
        DataExports::CreateJob.perform_later(@data_export)
      end

      @data_export
    rescue DataExports::CreateService::DataExportCreateError => e
      @data_export.errors.add(:base, e.message)
      @data_export
    end

    private

    # sample and linelist exports pass the namespace the user is exporting from and authorize the selected samples
    # based on the namespace
    def validate_sample_ids
      namespace = Namespace.find(params['export_parameters']['namespace_id'])

      authorize! namespace, to: :export_data?

      samples = authorized_export_samples(namespace, params['export_parameters']['ids'])

      return unless samples.count != params['export_parameters']['ids'].count

      raise DataExportCreateError,
            I18n.t('services.data_exports.create.invalid_export_samples')
    end

    def validate_analysis_ids
      workflow_executions = if params['export_parameters']['analysis_type'] == 'project'
                              authorized_export_project_workflows
                            else
                              authorized_export_user_workflows
                            end
      if workflow_executions.count != params['export_parameters']['ids'].count

        raise DataExportCreateError,
              I18n.t('services.data_exports.create.invalid_export_workflow_executions')
      end

      validate_workflow_executions_state(workflow_executions)
    end

    def assign_initial_export_attributes
      @data_export.user = current_user
      @data_export.status = 'processing'
      @data_export.name = nil if params.key?('name') && params['name'].empty?
    end

    def authorized_export_samples(namespace, sample_ids)
      authorized_scope(Sample, type: :relation, as: :namespace_samples,
                               scope_options: { namespace:, minimum_access_level: Member::AccessLevel::ANALYST })
        .where(id: sample_ids)
    end

    def authorized_export_project_workflows
      project_namespace = Namespace.find(params['export_parameters']['namespace_id'])
      authorize! project_namespace, to: :export_data?
      authorized_scope(WorkflowExecution, type: :relation, as: :automated,
                                          scope_options: { project: project_namespace.project })
        .where(id: params['export_parameters']['ids'])
    end

    def authorized_export_user_workflows
      authorized_scope(WorkflowExecution, type: :relation, as: :user, scope_options: { user: current_user })
        .where(id: params['export_parameters']['ids'])
    end

    def validate_workflow_executions_state(workflow_executions)
      completed_workflow_executions = workflow_executions.where(state: 'completed')

      return if completed_workflow_executions.count == workflow_executions.count

      raise DataExportCreateError,
            I18n.t('services.data_exports.create.non_completed_workflow_executions')
    end
  end
end