Chris@1295: # Redmine - project management software
Chris@1295: # Copyright (C) 2006-2012  Jean-Philippe Lang
Chris@1295: #
Chris@1295: # This program is free software; you can redistribute it and/or
Chris@1295: # modify it under the terms of the GNU General Public License
Chris@1295: # as published by the Free Software Foundation; either version 2
Chris@1295: # of the License, or (at your option) any later version.
Chris@1295: #
Chris@1295: # This program is distributed in the hope that it will be useful,
Chris@1295: # but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@1295: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Chris@1295: # GNU General Public License for more details.
Chris@1295: #
Chris@1295: # You should have received a copy of the GNU General Public License
Chris@1295: # along with this program; if not, write to the Free Software
Chris@1295: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
Chris@1295: 
Chris@1295: require 'zlib'
Chris@1295: 
Chris@1295: class WikiContent < ActiveRecord::Base
Chris@1295:   self.locking_column = 'version'
Chris@1295:   belongs_to :page, :class_name => 'WikiPage', :foreign_key => 'page_id'
Chris@1295:   belongs_to :author, :class_name => 'User', :foreign_key => 'author_id'
Chris@1295:   validates_presence_of :text
Chris@1295:   validates_length_of :comments, :maximum => 255, :allow_nil => true
Chris@1295: 
Chris@1295:   acts_as_versioned
Chris@1295: 
Chris@1295:   def visible?(user=User.current)
Chris@1295:     page.visible?(user)
Chris@1295:   end
Chris@1295: 
Chris@1295:   def project
Chris@1295:     page.project
Chris@1295:   end
Chris@1295: 
Chris@1295:   def attachments
Chris@1295:     page.nil? ? [] : page.attachments
Chris@1295:   end
Chris@1295: 
Chris@1295:   # Returns the mail adresses of users that should be notified
Chris@1295:   def recipients
Chris@1295:     notified = project.notified_users
Chris@1295:     notified.reject! {|user| !visible?(user)}
Chris@1295:     notified.collect(&:mail)
Chris@1295:   end
Chris@1295: 
Chris@1295:   # Return true if the content is the current page content
Chris@1295:   def current_version?
Chris@1295:     true
Chris@1295:   end
Chris@1295: 
Chris@1295:   class Version
Chris@1295:     belongs_to :page, :class_name => '::WikiPage', :foreign_key => 'page_id'
Chris@1295:     belongs_to :author, :class_name => '::User', :foreign_key => 'author_id'
Chris@1295:     attr_protected :data
Chris@1295: 
Chris@1295:     acts_as_event :title => Proc.new {|o| "#{l(:label_wiki_edit)}: #{o.page.title} (##{o.version})"},
Chris@1295:                   :description => :comments,
Chris@1295:                   :datetime => :updated_on,
Chris@1295:                   :type => 'wiki-page',
Chris@1295:                   :url => Proc.new {|o| {:controller => 'wiki', :action => 'show', :project_id => o.page.wiki.project, :id => o.page.title, :version => o.version}}
Chris@1295: 
Chris@1295:     acts_as_activity_provider :type => 'wiki_edits',
Chris@1295:                               :timestamp => "#{WikiContent.versioned_table_name}.updated_on",
Chris@1295:                               :author_key => "#{WikiContent.versioned_table_name}.author_id",
Chris@1295:                               :permission => :view_wiki_edits,
Chris@1295:                               :find_options => {:select => "#{WikiContent.versioned_table_name}.updated_on, #{WikiContent.versioned_table_name}.comments, " +
Chris@1295:                                                            "#{WikiContent.versioned_table_name}.#{WikiContent.version_column}, #{WikiPage.table_name}.title, " +
Chris@1295:                                                            "#{WikiContent.versioned_table_name}.page_id, #{WikiContent.versioned_table_name}.author_id, " +
Chris@1295:                                                            "#{WikiContent.versioned_table_name}.id",
Chris@1295:                                                 :joins => "LEFT JOIN #{WikiPage.table_name} ON #{WikiPage.table_name}.id = #{WikiContent.versioned_table_name}.page_id " +
Chris@1295:                                                           "LEFT JOIN #{Wiki.table_name} ON #{Wiki.table_name}.id = #{WikiPage.table_name}.wiki_id " +
Chris@1295:                                                           "LEFT JOIN #{Project.table_name} ON #{Project.table_name}.id = #{Wiki.table_name}.project_id"}
Chris@1295: 
Chris@1295:     after_destroy :page_update_after_destroy
Chris@1295: 
Chris@1295:     def text=(plain)
Chris@1295:       case Setting.wiki_compression
Chris@1295:       when 'gzip'
Chris@1295:       begin
Chris@1295:         self.data = Zlib::Deflate.deflate(plain, Zlib::BEST_COMPRESSION)
Chris@1295:         self.compression = 'gzip'
Chris@1295:       rescue
Chris@1295:         self.data = plain
Chris@1295:         self.compression = ''
Chris@1295:       end
Chris@1295:       else
Chris@1295:         self.data = plain
Chris@1295:         self.compression = ''
Chris@1295:       end
Chris@1295:       plain
Chris@1295:     end
Chris@1295: 
Chris@1295:     def text
Chris@1295:       @text ||= begin
Chris@1295:         str = case compression
Chris@1295:               when 'gzip'
Chris@1295:                 Zlib::Inflate.inflate(data)
Chris@1295:               else
Chris@1295:                 # uncompressed data
Chris@1295:                 data
Chris@1295:               end
Chris@1295:         str.force_encoding("UTF-8") if str.respond_to?(:force_encoding)
Chris@1295:         str
Chris@1295:       end
Chris@1295:     end
Chris@1295: 
Chris@1295:     def project
Chris@1295:       page.project
Chris@1295:     end
Chris@1295: 
Chris@1295:     # Return true if the content is the current page content
Chris@1295:     def current_version?
Chris@1295:       page.content.version == self.version
Chris@1295:     end
Chris@1295: 
Chris@1295:     # Returns the previous version or nil
Chris@1295:     def previous
Chris@1295:       @previous ||= WikiContent::Version.
Chris@1295:         reorder('version DESC').
Chris@1295:         includes(:author).
Chris@1295:         where("wiki_content_id = ? AND version < ?", wiki_content_id, version).first
Chris@1295:     end
Chris@1295: 
Chris@1295:     # Returns the next version or nil
Chris@1295:     def next
Chris@1295:       @next ||= WikiContent::Version.
Chris@1295:         reorder('version ASC').
Chris@1295:         includes(:author).
Chris@1295:         where("wiki_content_id = ? AND version > ?", wiki_content_id, version).first
Chris@1295:     end
Chris@1295: 
Chris@1295:     private
Chris@1295: 
Chris@1295:     # Updates page's content if the latest version is removed
Chris@1295:     # or destroys the page if it was the only version
Chris@1295:     def page_update_after_destroy
Chris@1295:       latest = page.content.versions.reorder("#{self.class.table_name}.version DESC").first
Chris@1295:       if latest && page.content.version != latest.version
Chris@1295:         raise ActiveRecord::Rollback unless page.content.revert_to!(latest)
Chris@1295:       elsif latest.nil?
Chris@1295:         raise ActiveRecord::Rollback unless page.destroy
Chris@1295:       end
Chris@1295:     end
Chris@1295:   end
Chris@1295: end