# encoding: utf-8
#
# Redmine plugin for Document Management System "Features"
#
# Copyright (C) 2011-16 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.

desc <<-END_DESC
DMSF maintenance task
  * Create missing MD5 digest for all file revisions

Available options:
  *dry_run - test, no changes to the database

Example:
  bundle exec rake redmine:dmsf_create_digests RAILS_ENV="production"
  bundle exec rake redmine:dmsf_create_digests dry_run=1 RAILS_ENV="production"
END_DESC

namespace :redmine do
  task :dmsf_create_digests => :environment do
    m = DmsfDigest.new
    m.create_digests
  end
end

class DmsfDigest

  def initialize
    @dry_run = ENV['dry_run']
  end

  def create_digests
    revisions = DmsfFileRevision.where("digest IS NULL OR digest = ''").all
    count = revisions.count
    n = 0
    revisions.each_with_index do |rev, i|
      if rev.create_digest
        rev.save unless @dry_run
        n += 1
      end
      # Progress bar
      print "\r#{i * 100 / count}%"
    end
    print "\r100%\n"
    # Result
    puts "#{n}/#{DmsfFileRevision.count} revisions updated."
  end

end
