Contents

require 'sinatra'
require 'sinatra/base'
require 'json'
require_relative './directory_not_found_error'
require_relative './configuration'

module Ferver
  class Controller < Sinatra::Base
    before do
      @ferver_list = FileList.new(current_ferver_path)
    end

    error Ferver::DirectoryNotFoundError do
      halt 500, "Ferver: Directory '#{current_ferver_path}' not found."
    end

    # redirect to file list
    get '/' do
      redirect to('/files')
    end

    # list files
    get '/files' do
      if request.preferred_type.to_s == 'application/json'
        content_type :json

        ferver_list.all.to_json
      else
        erb :index, locals: { file_list: ferver_list.all,
                              ferver_path: File.expand_path(current_ferver_path),
                              file_count: ferver_list.size }
      end
    end

    # download file
    get '/files/:id' do
      halt(400, 'Bad request') unless valid_file_request?

      if ferver_list.file_id_valid?(file_id_request.value)
        file_name = ferver_list.file_by_id(file_id_request.value)
        file = FileList.path_for_file(current_ferver_path, file_name)

        send_file(file, disposition: 'attachment', filename: File.basename(file))
      else
        status 404
      end
    end

    private

    attr_reader :ferver_list

    def file_id_request
      @file_id_request ||= FileIdRequest.new(params[:id])
    end

    def valid_file_request?
      file_id_request.valid?
    end

    def current_ferver_path
      Ferver.configuration.directory_path
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
ferver-1.2.1	lib/ferver/controller.rb