class TeamsController < ApplicationController before_action :load_vars, only: %i(parse_sheet import_samples export_samples export_projects export_projects_modal) before_action :check_create_samples_permissions, only: %i(parse_sheet import_samples) before_action :check_view_samples_permission, only: [:export_samples] before_action :check_export_projects_permissions, only: %i(export_projects_modal export_projects) def parse_sheet session[:return_to] ||= request.referer unless import_params[:file] return parse_sheet_error(t('teams.parse_sheet.errors.no_file_selected')) end if import_params[:file].size > Rails.configuration.x.file_max_size_mb.megabytes error = t('general.file.size_exceeded', file_size: Rails.configuration.x.file_max_size_mb) return parse_sheet_error(error) end begin sheet = SpreadsheetParser.open_spreadsheet(import_params[:file]) @header, @columns = SpreadsheetParser.first_two_rows(sheet) if @header.empty? || @columns.empty? return parse_sheet_error(t('teams.parse_sheet.errors.empty_file')) end # Fill in fields for dropdown @available_fields = @team.get_available_sample_fields # Truncate long fields @available_fields.update(@available_fields) do |_k, v| v.truncate(Constants::NAME_TRUNCATION_LENGTH_DROPDOWN) end # Save file for next step (importing) @temp_file = TempFile.new( session_id: session.id, file: import_params[:file] ) if @temp_file.save TempFile.destroy_obsolete(@temp_file.id) respond_to do |format| format.json do render json: { html: render_to_string( partial: 'samples/parse_samples_modal.html.erb' ) } end end else return parse_sheet_error( t('teams.parse_sheet.errors.temp_file_failure') ) end rescue ArgumentError, CSV::MalformedCSVError return parse_sheet_error(t('teams.parse_sheet.errors.invalid_file', encoding: ''.encoding)) rescue TypeError return parse_sheet_error(t('teams.parse_sheet.errors.invalid_extension')) end end def import_samples session[:return_to] ||= request.referer respond_to do |format| if import_params[:file_id] @temp_file = TempFile.find_by_id(import_params[:file_id]) if @temp_file # Check if session_id is equal to prevent file stealing if @temp_file.session_id == session.id # Check if mappings exists or else we don't have anything to parse if import_params[:mappings] @sheet = SpreadsheetParser.open_spreadsheet(@temp_file.file) # Check for duplicated values h1 = import_params[:mappings].clone.delete_if { |_, v| v.empty? } if h1.length == h1.invert.length # Check if there exist mapping for sample name (it's mandatory) if import_params[:mappings].value?('-1') result = @team.import_samples(@sheet, import_params[:mappings], current_user) nr_of_added = result[:nr_of_added] total_nr = result[:total_nr] if result[:status] == :ok # If no errors are present, redirect back # to samples table flash[:success] = t( "teams.import_samples.success_flash", nr: nr_of_added, samples: t( "teams.import_samples.sample", count: total_nr ) ) @temp_file.destroy format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:success) render json: { status: :ok } } else # Otherwise, also redirect back, # but display different message flash[:alert] = t( "teams.import_samples.partial_success_flash", nr: nr_of_added, samples: t( "teams.import_samples.sample", count: total_nr ) ) @temp_file.destroy format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:alert) render json: { status: :unprocessable_entity } } end else # This is currently the only AJAX error response flash_alert = t( "teams.import_samples.errors.no_sample_name") format.html { flash[:alert] = flash_alert redirect_to session.delete(:return_to) } format.json { render json: { html: render_to_string({ partial: "parse_error.html.erb", locals: { error: flash_alert } }) }, status: :unprocessable_entity } end else # This code should never execute unless user tampers with # JS (selects same column in more than one dropdown) flash_alert = t( "teams.import_samples.errors.duplicated_values") format.html { flash[:alert] = flash_alert redirect_to session.delete(:return_to) } format.json { render json: { html: render_to_string({ partial: "parse_error.html.erb", locals: { error: flash_alert } }) }, status: :unprocessable_entity } end else @temp_file.destroy flash[:alert] = t( "teams.import_samples.errors.no_data_to_parse") format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:alert) render json: { status: :unprocessable_entity } } end else @temp_file.destroy flash[:alert] = t( "teams.import_samples.errors.session_expired") format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:alert) render json: { status: :unprocessable_entity } } end else # No temp file to begin with, so no need to destroy it flash[:alert] = t( "teams.import_samples.errors.temp_file_not_found") format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:alert) render json: { status: :unprocessable_entity } } end else flash[:alert] = t( "teams.import_samples.errors.temp_file_not_found") format.html { redirect_to session.delete(:return_to) } format.json { flash.keep(:alert) render json: { status: :unprocessable_entity } } end end end def export_samples if export_params[:sample_ids] && export_params[:header_ids] generate_samples_zip else flash[:alert] = t('zip_export.export_error') end redirect_back(fallback_location: root_path) end def export_projects unless export_proj_requests_exceeded? current_user.export_vars['num_of_export_all_last_24_hours'] += 1 current_user.save generate_export_projects_zip render json: { flash: t('projects.export_projects.success_flash') }, status: :ok end end def export_projects_modal if @exp_projects.present? limit = (ENV['EXPORT_ALL_LIMIT_24_HOURS'] || 3).to_i curr_num = current_user.export_vars['num_of_export_all_last_24_hours'] if export_proj_requests_exceeded? render json: { html: render_to_string( partial: 'projects/export/error.html.erb', locals: { limit: limit } ), title: t('projects.export_projects.error_title'), status: 'error' } else render json: { html: render_to_string( partial: 'projects/export/modal.html.erb', locals: { num_projects: @exp_projects.size, limit: limit, num_of_requests_left: limit - curr_num - 1 } ), title: t('projects.export_projects.modal_title') } end end end def routing_error(error = 'Routing error', status = :not_found, exception=nil) redirect_to root_path end private def parse_sheet_error(error) respond_to do |format| format.html do flash[:alert] = error session[:return_to] ||= request.referer redirect_to session.delete(:return_to) end format.json do render json: { message: error }, status: :unprocessable_entity end end end def load_vars @team = Team.find_by_id(params[:id]) unless @team render_404 end end def import_params params.permit(:id, :file, :file_id, mappings: {}).to_h end def export_params params.permit(sample_ids: [], header_ids: []).to_h end def export_projects_params params.permit(:id, project_ids: []).to_h end def check_create_samples_permissions render_403 unless can_create_samples?(@team) end def check_view_samples_permission unless can_read_team?(@team) render_403 end end def check_export_projects_permissions render_403 unless can_read_team?(@team) if export_projects_params[:project_ids] @exp_projects = Project.where(id: export_projects_params[:project_ids]) @exp_projects.each do |project| render_403 unless can_export_project?(current_user, project) end end end def export_proj_requests_exceeded? # Check if user has enough requests for the day limit = (ENV['EXPORT_ALL_LIMIT_24_HOURS'] || 3).to_i !limit.zero? \ && current_user.export_vars['num_of_export_all_last_24_hours'] >= limit end def generate_samples_zip zip = ZipExport.create(user: current_user) zip.generate_exportable_zip( current_user, @team.to_csv( Sample.where(id: export_params[:sample_ids]), export_params[:header_ids] ), :samples ) end def generate_export_projects_zip ids = @exp_projects.where(team_id: @team).index_by(&:id) options = { team: @team } zip = TeamZipExport.create(user: current_user) zip.generate_exportable_zip( current_user, ids, :teams, options ) ids end end