unepwcmc/SAPI

class CitesTrade::ExportsController < CitesTradeController
  respond_to :json

  def download
    respond_to do |format|
      format.html {
        search = Trade::ShipmentsExportFactory.new(search_params.merge({
          :per_page => Trade::ShipmentsExport::PUBLIC_CSV_LIMIT
        }))
        result = search.export
        if result.is_a?(Array)
          # this was added in order to prevent download managers from
          # failing when chunked_transfer_encoding is set in nginx (1.8.1)
          file_path = Pathname.new(result[0]).realpath
          response.headers['Content-Length'] = File.size(file_path).to_s
          send_file file_path, result[1]
          Trade::TradeDataDownloadLogger.log_download request, search_params, search.total_cnt
        else
          redirect_to cites_trade_root_url
        end
      }
      format.json {
        search = Trade::ShipmentsExportFactory.new(search_params.merge({
          :report_type => :raw # get the raw count
        }))
        render :json => {
          :total => search.total_cnt,
          :csv_limit => Trade::ShipmentsExport::PUBLIC_CSV_LIMIT,
          :web_limit => Trade::ShipmentsExport::PUBLIC_WEB_LIMIT
        }
      }
    end
  end
end