report.rb in rails_db_admin-3.2.0

- old
+ new
@@ -1,31 +1,244 @@
 class Report < ActiveRecord::Base
   attr_protected :created_at, :updated_at
   attr_accessible :name, :internal_identifier
-  
+
   validates :name, :internal_identifier, :uniqueness => true
 
   before_create :set_default_template
 
+  after_create :create_report_files!
+
+  before_destroy :delete_report_files!
+
+  has_file_assets
+
+  has_party_roles
+
+  is_json :meta_data
+
+  REPORT_STRUCTURE = ['stylesheets', 'javascripts', 'images', 'templates']
+
+  class << self
+    def iid(internal_identifier)
+      find_by_internal_identifier(internal_identifier)
+    end
+
+    def import(file)
+      ActiveRecord::Base.transaction do
+        name_and_iid = file.original_filename.to_s.gsub(/(^.*(\\|\/))|(\.zip$)/, '')
+        report_name = name_and_iid.split('[').first
+        report_iid = name_and_iid.split('[').last.gsub(']', '')
+        return false unless valid_report?(file)
+        Report.skip_callback(:create, :after, :create_report_files!)
+        report = Report.create(:name => report_name, :internal_identifier => report_iid)
+        report.import(file)
+        Report.set_callback(:create, :after, :create_report_files!)
+      end
+    end
+
+    def make_tmp_dir
+      Pathname.new(Rails.root.to_s + "/tmp/reports/tmp_#{Time.now.to_i.to_s}/").tap do |dir|
+        FileUtils.mkdir_p(dir) unless dir.exist?
+      end
+    end
+
+    def valid_report?(file)
+      valid = false
+      Zip::ZipFile.open(file.path) do |zip|
+        zip.sort.each do |entry|
+          entry.name.split('/').each do |file|
+            valid = true if REPORT_STRUCTURE.include?(file)
+          end
+        end
+      end
+      valid
+    end
+  end
+
+  def base_dir
+    "#{Rails.root}/public/compass_ae_reports/#{self.internal_identifier}"
+  end
+
+  def url
+    "/public/compass_ae_reports/#{self.internal_identifier}"
+  end
+
+  def import(file)
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => Rails.application.config.erp_tech_svcs.file_storage)
+    file = ActionController::UploadedTempfile.new("uploaded-report").tap do |f|
+      f.puts file.read
+      f.original_filename = file.original_filename
+      f.read # no idea why we need this here, otherwise the zip can't be opened
+    end unless file.path
+
+    Zip::ZipFile.open(file.path) do |zip|
+      zip.each do |entry|
+        if entry.name.split('/').last == 'base.html.erb'
+          name = entry.name.sub(/__MACOSX\//, '')
+          data = ''
+          entry.get_input_stream { |io| data = io.read }
+          self.template = data
+          self.save
+          self.add_file(self.template, File.join(self.url, 'templates', "base.html.erb"))
+        elsif entry.name.split('/').last == 'meta_data.yml'
+          data = ''
+          entry.get_input_stream { |io| data = io.read }
+          data = StringIO.new(data) if data.present?
+          self.meta_data = YAML.load(data)
+          self.save
+        elsif entry.name.split('/').last == 'query.sql'
+          data = ''
+          entry.get_input_stream { |io| data = io.read }
+          self.query = data
+          self.save
+        else
+          if entry.file?
+            name = entry.name.sub(/__MACOSX\//, '')
+            data = ''
+            entry.get_input_stream { |io| data = io.read }
+            data = StringIO.new(data) if data.present?
+            report_file = self.files.where("name = ? and directory = ?", File.basename(name), File.join(self.url, File.dirname(name))).first
+            unless report_file.nil?
+              report_file.data = data
+              report_file.save
+            else
+              self.add_file(data, File.join(file_support.root, self.url, name))
+            end
+          end
+        end
+      end
+    end
+
+  end
+
+  def export
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => Rails.application.config.erp_tech_svcs.file_storage)
+    tmp_dir = Report.make_tmp_dir
+    (tmp_dir + "#{name}[#{internal_identifier}].zip").tap do |file_name|
+      file_name.unlink if file_name.exist?
+      Zip::ZipFile.open(file_name.to_s, Zip::ZipFile::CREATE) do |zip|
+        files.each do |file|
+          contents = file_support.get_contents(File.join(file_support.root, file.directory, file.name))
+          relative_path = file.directory.sub("#{url}", '')
+          path = FileUtils.mkdir_p(File.join(tmp_dir, relative_path))
+          full_path = File.join(path, file.name)
+          File.open(full_path, 'wb+') { |f| f.puts(contents) }
+          zip.add(File.join(relative_path[1..relative_path.length], file.name), full_path) if ::File.exists?(full_path)
+        end
+        ::File.open(tmp_dir + 'query.sql', 'wb+') { |f| f.puts(query) }
+        zip.add('query.sql', tmp_dir + 'query.sql')
+        ::File.open(tmp_dir + 'meta_data.yml', 'wb+') { |f| f.puts(meta_data.to_yaml) }
+        zip.add('meta_data.yml', tmp_dir + 'meta_data.yml')
+
+      end
+    end
+  end
+
+  def stylesheet_path(source)
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => ErpTechSvcs::Config.file_storage)
+    File.join(file_support.root, self.url, 'stylesheets', source)
+  end
+
+  def image_path(source)
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => ErpTechSvcs::Config.file_storage)
+    File.join(file_support.root, self.url, 'images', source)
+  end
+
+  def template_path(source)
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => ErpTechSvcs::Config.file_storage)
+    File.join(file_support.root, self.url, 'templates', source)
+  end
+
+  def javascript_path(source)
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => ErpTechSvcs::Config.file_storage)
+    File.join(file_support.root, self.url, 'javascripts', source)
+  end
+
+  def create_report_files!
+    self.add_file(self.template, File.join(self.url, 'templates', "base.html.erb"))
+    self.add_file(self.default_stylesheet,File.join(self.url, 'stylesheets', "#{self.internal_identifier}.css") )
+  end
+
+  def delete_report_files!
+    file_support = ErpTechSvcs::FileSupport::Base.new(:storage => ErpTechSvcs::Config.file_storage)
+    file_support.delete_file(File.join(file_support.root, self.url), :force => true)
+  end
+
   def set_default_template
     self.template =
-"<h3><%= title %></h3>
-
+    "<%= bootstrap_load %>
+  <%= report_stylesheet_link_tag '#{self.internal_identifier}','#{self.internal_identifier}.css' %>
+<h3> <%= title %> </h3>
 <table>
-  <tr>
-  <% columns.each do |column| %>
-    <th><%= column %></th>
-  <% end %>
-  </tr>
-  <% rows.each do |row| %>
+  <thead>
     <tr>
-    <% row.values.each do |value| %>
-	 <td><%= value %></td>
+    <% columns.each do |column| %>
+      <th> <%= column %> </th>
     <% end %>
     </tr>
+  </thead>
+  <% rows.each do |row| %>
+    <tr>
+      <% row.values.each do |value| %>
+            <td> <%= value %> </td>
+      <% end %>
+    </tr>
   <% end %>
 </table>
+<%= jquery_load %>"
+  end
 
-<%= report_download_link(unique_name, :csv, 'Download CSV') %> |
-<%= report_download_link(unique_name, :pdf, 'Download PDF') %>"
+  def default_stylesheet
+    "body{
+  padding: 5px; 
+}
+
+table{
+  width: 100%;
+}
+
+table tr{
+  page-break-inside: avoid !important;
+}
+
+table th{
+  border: 1px solid black;
+  color: white;
+  border-collapse:collapse;
+  font-size: 16px;
+  vertical-align:center;
+  text-align: center;
+  background-color: 537697;
+  padding-top: 2px;
+  padding-bottom: 2px;
+  padding-left: 15px;
+  padding-right: 15px;
+
+}
+
+table td{
+  border: 1px solid black;
+  border-collapse:collapse;
+  font-size: 14px;
+  vertical-align:top;
+  padding-top: 2px;
+  padding-left: 15px;
+  padding-right: 15px;
+}
+
+td.no-border {
+    border: 0px;
+    height: 20px;
+}
+
+.bold{
+    font-weight: bold;
+    height: 20px;
+}
+
+.left-padded{
+    padding-left: 20px;
+}"
   end
-end
\ No newline at end of file
+
+end