lib/controllers/reports_controller.rb.template spec/reports_controller_spec.rb test.sqlite3 test.xls @@ -1 +1,2 @@ +Mime::Type.register "application/vnd.ms-excel", :xls init.rb @@ -1,8 +1,7 @@ require 'fileutils' dir = File.dirname(__FILE__) -FileUtils.cp Dir[File.join(dir, 'lib', 'controllers', '*.rb')], File.join(dir, '../../../app/controllers') +FileUtils.cp File.join(dir, 'lib', 'controllers', 'reports_controller.rb.template'), File.join(dir, '../../../app/controllers/reports_controller.rb') FileUtils.cp_r Dir[File.join(dir, 'views', '*')], File.join(dir, '../../../app/views') - puts IO.read(File.join(File.dirname(__FILE__), 'INSTALL')) \ No newline at end of file install.rb @@ -1,4 +1,8 @@ +require "spreadsheet/excel" +require 'iconv' + class ReportsController < ApplicationController + include Spreadsheet def index @reports = Report.find :all @@ -6,27 +10,52 @@ class ReportsController < ApplicationController def show @report = Report.find(params[:id]) - @results = @report.run + data = @report.run respond_to do |format| format.html do - if @results.is_a? Array - @results = @results.inject('<table>') do |table, row| - table << '<tr>' - table << row.inject('') {|_row, cell| _row << "<td>#{cell.to_s}</td>"} - table << '</tr>' - end << '</table>' - end + @table = create_html_table data end format.csv do - if @results.is_a? Array - @results = @results.inject('') do |table, row| - table << row.map {|cell| cell.to_s.gsub(/\s+/, ' ')}.join(';') - table << "\n" - end - end - render :text => @results + render :text => create_csv_table(data) + headers['Content-Disposition'] = "attachment; filename=\"#{@report.name.downcase.gsub(/\s+/, '_')}_#{Date.today.to_s}.csv\"" + end + format.xls do + render :text => create_excel_table(data) + headers['Content-Disposition'] = "attachment; filename=\"#{@report.name.downcase.gsub(/\s+/, '_')}_#{Date.today.to_s}.xls\"" + headers['Cache-Control'] = '' + end + end + end + + def create_html_table(_table) + _table.inject('<table>') do |table, row| + table << '<tr>' + table << row.inject('') {|_row, cell| _row << "<td>#{cell.to_s}</td>"} + table << '</tr>' + end << '</table>' + end + + def create_csv_table(_table) + _table.inject('') do |table, row| + table << row.map {|cell| cell.to_s.gsub(/\s+/, ' ')}.join(';') + table << "\n" + end + end + + def create_excel_table(_table) + out = StringIO.new + workbook = Excel.new(out) + format = Format.new + + worksheet = workbook.add_worksheet + _table.each_with_index do |row, i| + row.each_with_index do |col, j| + worksheet.write i, j, Iconv.conv('ISO-8859-1', 'UTF-8', col.to_s), format # excel doesn't seem to like utf-8 end end + + workbook.close + out.string end end lib/controllers/reports_controller.rb @@ -10,4 +10,49 @@ ActiveRecord::Base.establish_connection :adapter => 'sqlite3', :database => File # migrate require File.dirname(__FILE__) + '/../generators/dead_simple_reports/templates/migration' -CreateDeadSimpleReports.up \ No newline at end of file +CreateDeadSimpleReports.up + +# rails controller stuff + +class ApplicationController + class MimeTypeFormat + + [:html, :csv, :xls].each do |mime_type| + + eval <<-CODE + def #{mime_type}(&block) + @#{mime_type}_block = block + end + + def #{mime_type}_view + @#{mime_type}_block.call + end + + CODE + + end + + end + + def respond_to(&formatter) + @format = MimeTypeFormat.new + formatter.call @format + @format + end + + def params + {} + end + + def headers + {} + end + + def render(options = {}) + @view = options[:text] + end + + def view + @view + end +end \ No newline at end of file spec/spec_helper.rb @@ -2,6 +2,12 @@ <ol> <%- @reports.each do |report| -%> - <li><%= link_to h(report.name), :action => 'show', :id => report.id %> (<%= link_to 'CSV', :action => 'show', :id => report.id, :format => 'csv' %>)</li> + <li> + <%= link_to h(report.name), :action => 'show', :id => report.id %> + ( + <%= link_to 'CSV', :action => 'show', :id => report.id, :format => 'csv' %> + <%= link_to 'Excel', :action => 'show', :id => report.id, :format => 'xls' %> + ) + </li> <%- end -%> </ol> \ No newline at end of file views/reports/index.rhtml @@ -1,17 +1,6 @@ <h1><%= h @report.name %></h1> -<p>Format: <%= link_to 'CSV', :format => 'csv' %></p> +<p>Format: <%= link_to 'CSV', :format => 'csv' %>, <%= link_to 'Excel', :format => 'xls' %></p> -<style> - table { - border-collapse: collapse; - } - - td { - border: 1px solid black; - padding: 3px; - } -</style> - -<%= @results %> +<%= @table %> views/reports/show.rhtml 1ce6c51ed41b92df86e66bf14ce8aa3db5831161 Alexander Lang alex@skywalker.(none) http://github.com/langalex/dead_simple_reports/commit/61fea284d033406dab26044912dec36e652a0327 61fea284d033406dab26044912dec36e652a0327 2008-04-12T12:51:43-07:00 2008-04-12T12:51:43-07:00 added excel format reports via the spreadsheet-excel gem, added specs, changed install to copy fewer files into the app so that updates are easier 90393b0b4a65eab7886ba2db6a7569a3ad619e64 Alexander Lang alex@skywalker.(none)