redmine_dmsf/app/models/dmsf_link.rb

# encode: utf-8
#
# Redmine plugin for Document Management System "Features"
#
# 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.

class DmsfLink < ActiveRecord::Base

  include ActiveModel::Validations

  belongs_to :project
  belongs_to :dmsf_folder
  belongs_to :deleted_by_user, :class_name => 'User', :foreign_key => 'deleted_by_user_id'
  belongs_to :user

  validates :name, :presence => true
  validates_length_of :name, :maximum => 255
  validates_length_of :external_url, :maximum => 255
  validate :validate_url

  def validate_url
    if self.target_type == 'DmsfUrl'
      begin
        if self.external_url.present?
          URI.parse self.external_url
        else
          errors.add :external_url, :invalid
        end
      rescue URI::InvalidURIError
        errors.add :external_url, :invalid
      end
    end
  end

  STATUS_DELETED = 1.freeze
  STATUS_ACTIVE = 0.freeze

  scope :visible, -> { where(:deleted => STATUS_ACTIVE) }
  scope :deleted, -> { where(:deleted => STATUS_DELETED) }

  def target_folder_id
    if self.target_type == DmsfFolder.model_name.to_s
      self.target_id
    else
      f = DmsfFile.find_by_id self.target_id
      f.dmsf_folder_id if f
    end
  end

  def target_folder
    unless @target_folder
      if self.target_folder_id
        @target_folder = DmsfFolder.find_by_id self.target_folder_id
      end
    end
    @target_folder
  end

  def target_file_id
    self.target_id if self.target_type == DmsfFile.model_name.to_s
  end

  def target_file
    unless @target_file
      if self.target_file_id
        @target_file = DmsfFile.find_by_id self.target_file_id
      end
    end
    @target_file
  end

  def target_project
    unless @target_project
      @target_project = Project.find_by_id self.target_project_id
    end
    @target_project
  end

  def folder
    unless @folder
      if self.dmsf_folder_id
        @folder = DmsfFolder.find_by_id self.dmsf_folder_id
      end
    end
    @folder
  end

  def title
    self.name
  end

  def self.find_link_by_file_name(project, folder, filename)
    links = DmsfLink.where(
      :project_id => project.id,
      :dmsf_folder_id => folder ? folder.id : nil,
      :target_type => DmsfFile.model_name.to_s).visible.all
    links.each do |link|
      return link if link.target_file && (link.target_file.name == filename)
    end
    nil
  end

  def path
    if self.target_type == DmsfFile.model_name.to_s
      path = self.target_file.dmsf_path.map { |element| element.is_a?(DmsfFile) ? element.display_name : element.title }.join('/') if self.target_file
    else
      path = self.target_folder ? self.target_folder.dmsf_path_str : ''
    end
    path.insert(0, "#{self.target_project.name}:") if self.project_id != self.target_project_id
    if path && path.length > 50
      return "#{path[0, 25]}...#{path[-25, 25]}"
    end
    path
  end

  def copy_to(project, folder)
    link = DmsfLink.new
    link.target_project_id = self.target_project_id
    link.target_id = self.target_id
    link.target_type = self.target_type
    link.name = self.name
    link.external_url = self.external_url
    link.project_id = project.id
    link.dmsf_folder_id = folder ? folder.id : nil
    link.save
    link
  end

  def container
    if self.folder && self.folder.system
      Issue.where(:id => self.folder.title.to_i).first
    end
  end

  def delete(commit = false)
    if commit
      if self.container.is_a?(Issue)
        self.container.dmsf_file_removed(self.target_file)
      end
      self.destroy
    else
      self.deleted = STATUS_DELETED
      self.deleted_by_user = User.current
      save(:validate => false)
    end
  end

  def restore
    if self.dmsf_folder_id && (self.dmsf_folder.nil? || self.dmsf_folder.deleted?)
      errors[:base] << l(:error_parent_folder)
      return false
    end
    self.deleted = STATUS_ACTIVE
    self.deleted_by_user = nil
    save(:validate => false)
  end

  def is_folder?
    self.target_type == 'DmsfFolder'
  end

  def is_file?
    !is_folder?
  end

  def to_csv(columns, level)
    csv = []
    if self.target_type == 'DmsfUrl'
      # Project
      csv << self.project.name if columns.include?(l(:field_project))
      # Id
      csv << self.id if columns.include?('id')
      # Title
      csv << self.title.insert(0, ' ' * level) if columns.include?('title')
      # Extension
      csv << '' if columns.include?('extension')
      # Size
      csv << '' if columns.include?('size')
      # Modified
      csv << format_time(self.updated_at) if columns.include?('modified')
      # Version
      csv << '' if columns.include?('version')
      # Workflow
      csv << '' if columns.include?('workflow')
      # Author
      csv << self.user.name if columns.include?('author')
      # Last approver
      csv << '' if columns.include?(l(:label_last_approver))
      # Url
      csv << self.external_url if columns.include?(l(:label_document_url))
      # Revision
      csv << '' if columns.include?(l(:label_last_revision_id))
      # Custom fields
      cfs = CustomField.where(:type => 'DmsfFileRevisionCustomField').order(:position)
      cfs.each do |c|
        csv << '' if columns.include?(c.name)
      end
    end
    csv
  end

end