redmine_dmsf/app/helpers/dmsf_helper.rb

# encoding: utf-8
#
# Redmine plugin for Document Management System "Features"
#
# Copyright © 2011    Vít Jonáš <vit.jonas@gmail.com>
# Copyright © 2011-18 Karel Pičman <karel.picman@kontron.com>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

require 'tmpdir'
require 'digest'
require 'csv'

module DmsfHelper
  include Redmine::I18n

  def self.temp_dir
    if Setting.plugin_redmine_dmsf['dmsf_tmpdir'].present?
      tmpdir = Pathname.new(Setting.plugin_redmine_dmsf['dmsf_tmpdir'])
    else
      tmpdir = Pathname.new(Dir.tmpdir)
    end
    tmpdir
  end

  def self.temp_filename(filename)
    filename = sanitize_filename(filename)
    timestamp = DateTime.now.strftime("%y%m%d%H%M%S")
    while self.temp_dir.join("#{timestamp}_#{filename}").exist?
      timestamp.succ!
    end
    "#{timestamp}_#{filename}"
  end

  def self.sanitize_filename(filename)
    # get only the filename, not the whole path
    just_filename = File.basename(filename.gsub('\\\\', '/'))
    # replace all non alphanumeric, hyphens or periods with underscore
    just_filename = just_filename.gsub(/[^\w\.\-]/,'_')
    unless just_filename =~ %r{^[a-zA-Z0-9_\.\-]*$}
      # keep the extension if any
      extension = $1 if just_filename =~ %r{(\.[a-zA-Z0-9]+)$}
      just_filename = Digest::SHA256.hexdigest(just_filename)  << extension
    end
    just_filename
  end

  def self.filetype_css(filename)
    extension = File.extname(filename)
    extension = extension[1, extension.length-1]
    path = File.join(Rails.root, 'public',
      File.join(Redmine::Utils.relative_url_root, 'plugin_assets', 'redmine_dmsf', 'images', 'filetypes', "#{extension}.png"))
    if File.exist?(path)
      cls = "filetype-#{extension}";
    else
      cls = Redmine::MimeType.css_class_of(filename)
    end
    cls << ' dmsf-icon-file' if cls
    cls
  end

  def plugin_asset_path(plugin, asset_type, source)
    File.join(Redmine::Utils.relative_url_root, 'plugin_assets', plugin.to_s, asset_type, source)
  end

  def json_url
    if I18n.locale
      ret = File.join('jquery.dataTables', 'locales', "#{I18n.locale}.json")
      path = File.join(Rails.root, 'public', plugin_asset_path(:redmine_dmsf, 'javascripts', ret))
      return ret if File.exist?(path)
      Rails.logger.warn "#{path} not found"
    end
    File.join 'jquery.dataTables', 'locales', 'en.json'
  end

  def js_url
    if I18n.locale
      ret = File.join('plupload', 'js', 'i18n', "#{I18n.locale}.js")
      path = File.join(Rails.root, 'public', plugin_asset_path(:redmine_dmsf, 'javascripts', ret))
      return ret if File.exist?(path)
      Rails.logger.warn "#{path} not found"
    end
    File.join 'plupload', 'js', 'i18n', 'en.js'
  end

  def self.visible_folders(folders, project)
    allowed = Setting.plugin_redmine_dmsf['dmsf_act_as_attachable'] &&
      (project.dmsf_act_as_attachable == Project::ATTACHABLE_DMS_AND_ATTACHMENTS) &&
      User.current.allowed_to?(:display_system_folders, project)
    folders.reject{ |folder|
      if folder.system
        unless allowed
          true
        else
          issue_id = folder.title.to_i
          if issue_id > 0
            issue = Issue.find_by_id issue_id
            issue && !issue.visible?(User.current)
          else
            false
          end
        end
      else
        false
      end
    }
  end

  def self.all_children_sorted(parent, pos, ident)
    # Folders && files && links
    nodes = visible_folders(parent.dmsf_folders.visible.to_a, parent.is_a?(Project) ? parent : parent.project) + parent.dmsf_links.visible + parent.dmsf_files.visible
    # Alphabetical and type sort
    nodes.sort! do |x, y|
      if ((x.is_a?(DmsfFolder) || (x.is_a?(DmsfLink) && x.is_folder?)) &&
            (y.is_a?(DmsfFile) || (y.is_a?(DmsfLink) && y.is_file?)))
        -1
      elsif ((x.is_a?(DmsfFile) || (x.is_a?(DmsfLink) && x.is_file?)) &&
            (y.is_a?(DmsfFolder) || (y.is_a?(DmsfLink) && y.is_folder?)))
        1
      else
        x.title.downcase <=> y.title.downcase
      end
    end
    # Calculate position
    ident = 0 unless ident
    pos = (10 ** 12) unless pos
    step = (10 ** 12) / (10 ** (ident * 3))
    tree = []
    i = 0
    nodes.each do |x|
      if x.is_a?(DmsfFolder) || (x.is_a?(DmsfLink) && x.is_folder?)
        i += 1
        tree << [x, pos + (step * i)]
      else
        tree << [x, pos + (step * (i + 1))]
      end
    end
    tree
  end

  def self.dmsf_to_csv(parent, columns)
    #Redmine::Export::CSV.generate do |csv|
    CSV.generate(:force_quotes => true, :encoding => 'UTF-8') do |csv|
      # Header
      csv << columns.map { |c| c.capitalize }
      # Lines
      DmsfHelper.object_to_csv(csv, parent, columns)
    end
  end

  def self.object_to_csv(csv, parent, columns, level = -1)
    # Folder
    if level >= 0
      csv << parent.to_csv(columns, level)
    end
    # Sub-folders
    folders = parent.dmsf_folders.visible.to_a
    folders.concat(parent.folder_links.visible.to_a.map{ |l| l.folder })
    folders.compact!
    folders.sort!{ |x, y| x.title <=> y.title}
    folders.each do |f|
      DmsfHelper.object_to_csv(csv, f, columns, level + 1)
    end
    # Files
    files = parent.dmsf_files.visible.to_a
    files.concat(parent.file_links.visible.to_a.map{ |l| l.target_file })
    files.concat(parent.url_links.visible.to_a)
    files.compact!
    files.sort!{ |x, y| x.title <=> y.title}
    files.each do |file_or_link|
      csv << file_or_link.to_csv(columns, level + 1)
    end
  end

  def render_principals_for_new_email(users)
    principals_check_box_tags 'user_ids[]', users
  end

end