From 1370d21460255914e03754fb0f20052e6b3002d7 Mon Sep 17 00:00:00 2001 From: Jesse Morgan Date: Sat, 28 Mar 2015 23:14:33 -0700 Subject: Crude first pass at downloading files. --- app/controllers/application_controller.rb | 4 ++++ app/controllers/downloads_controller.rb | 22 ++++++++++++++++++++++ 2 files changed, 26 insertions(+) (limited to 'app/controllers') diff --git a/app/controllers/application_controller.rb b/app/controllers/application_controller.rb index d83690e..a994170 100644 --- a/app/controllers/application_controller.rb +++ b/app/controllers/application_controller.rb @@ -2,4 +2,8 @@ class ApplicationController < ActionController::Base # Prevent CSRF attacks by raising an exception. # For APIs, you may want to use :null_session instead. protect_from_forgery with: :exception + + def not_found + raise ActionController::RoutingError.new('Not Found') + end end diff --git a/app/controllers/downloads_controller.rb b/app/controllers/downloads_controller.rb index 300d0b1..2681b0a 100644 --- a/app/controllers/downloads_controller.rb +++ b/app/controllers/downloads_controller.rb @@ -1,3 +1,6 @@ +require 'aws-sdk-core/s3' +require 'fileutils' + class DownloadsController < ApplicationController before_action :set_download, only: [:show, :edit, :update, :destroy] @@ -7,6 +10,25 @@ class DownloadsController < ApplicationController @downloads = Download.all end + # GET /downloads/:filename + def download + download = Download.find_by(filename: params[:filename]) or not_found + + s3client = Aws::S3::Client.new() + s3obj = Aws::S3::Object.new(ENV['S3_DOWNLOADS_BUCKET'], download.filename, client: s3client) + resp = s3obj.get + + FileUtils.mkdir_p(Rails.root.join('tmp', 'downloads')) + + cached_filename = Rails.root.join('tmp', 'downloads', download.filename) + + File.open(cached_filename, 'wb') do |file| + file.write resp.body.read + end + + send_file cached_filename + end + # GET /downloads/1 # GET /downloads/1.json def show -- cgit v1.2.3