Redmine

gem 'zip_kit', '~> 6.4.1'

  # for reading ZIP files generated by zip_kit

  gem 'rubyzip'

  # Streams a ZIP file built in the given block, which receives a

  # Redmine::Export::ZipArchive

  def send_zip_stream(file_name)

    user = User.current

    # Let reverse proxies such as nginx buffer the response so that

    # the application process is released without waiting for slow clients

    response.headers['x-accel-buffering'] = 'yes'

    zip_kit_stream(:filename => filename_for_content_disposition(file_name)) do |streamer|

      yield Redmine::Export::ZipArchive.new(streamer, user)

    end

  end

    if @attachments.blank?

      return

    end

    send_zip_stream("#{@container.class.to_s.downcase}-#{@container.id}-attachments.zip") do |zip|

      @attachments.each do |attachment|

        zip.add_file(attachment.filename, attachment.created_on) do |sink|

          File.open(attachment.diskfile, 'rb') {|file| IO.copy_stream(file, sink)}

        end

      end

        load_pages_for_export

        load_pages_for_export

        send_zip_stream("#{@project.identifier}-wiki.zip") do |zip|

          @wiki.pages.with_updated_on.each do |page|

            text = WikiContent.where(:page_id => page.id).pick(:text).to_s

            zip.add_file("#{archived_wiki_page_title(page)}.txt", page.updated_on) do |sink|

              IO.copy_stream(StringIO.new(text), sink)

            end

          end

        end

  def load_pages_for_export

    @pages = @wiki.pages.

                includes([:content, {:attachments => :author}]).

                to_a

  def archived_wiki_page_title(page)

    page.title.tr('\\', '_').gsub(/[\/?%*:|"'<>\n\r\x00]+/, '_')

# frozen_string_literal: true

# Redmine - project management software

# Copyright (C) 2006-  Jean-Philippe Lang

#

# 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.

module Redmine

  module Export

    class ZipArchive

      def initialize(streamer, user)

        @streamer = streamer

        @user = user

        @entry_names = Set.new

      end

      def add_file(name, modification_time, &)

        entry_name = unique_entry_name(name)

        @entry_names << entry_name

        @streamer.write_file(entry_name, :modification_time => zip_time(modification_time), &)

      end

      private

      def unique_entry_name(name)

        entry_name = name

        extname = File.extname(name)

        basename = File.basename(name, extname)

        dup_count = 0

        while @entry_names.include?(entry_name)

          dup_count += 1

          entry_name = "#{basename}(#{dup_count})#{extname}"

        end

        entry_name

      end

      # DateTime is used because zip_kit calls #utc on the given time,

      # which converts a Time object in place.

      def zip_time(time)

        @user.convert_time_to_user_timezone(time).to_datetime

      end

    end

  end

end

    assert_equal 'yes', response.headers['X-Accel-Buffering']

    zip_data = +""

    response.body.each {|chunk| zip_data << chunk}

    Zip::File.open_buffer(zip_data) do |zip|

      assert_equal ['source.rb'], zip.entries.map(&:name)

      assert_equal File.binread(Attachment.find(4).diskfile), zip.read('source.rb')

    end

    zip_data = +''

    @response.body.each {|chunk| zip_data << chunk}

    Zip::File.open_buffer(zip_data) do |zip|

      zip.each do |entry|

          :content => entry.get_input_stream.read,

# frozen_string_literal: true

# Redmine - project management software

# Copyright (C) 2006-  Jean-Philippe Lang

#

# 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_relative '../../../../test_helper'

class ZipArchiveTest < ActiveSupport::TestCase

  # Uses a server time zone other than UTC so that a local time converted

  # to UTC can be detected

  def setup

    @original_tz = ENV['TZ']

    ENV['TZ'] = 'HST10'

  end

  def teardown

    ENV['TZ'] = @original_tz

  end

  def test_add_file_should_rename_duplicate_names

    entries = write_zip(User.find(2)) do |zip|

      3.times { zip.add_file('foo.txt', Time.now) {|sink| sink << 'foo'} }

    end

    assert_equal ['foo.txt', 'foo(1).txt', 'foo(2).txt'], entries.map(&:name)

  end

  def test_add_file_should_store_modification_time_in_user_time_zone

    user = User.find(2)

    user.pref.update!(:time_zone => 'Tokyo')

    entries = write_zip(user) do |zip|

      zip.add_file('foo.txt', Time.utc(2026, 1, 1, 18, 4, 6)) {|sink| sink << 'foo'}

    end

    assert_equal [2026, 1, 2, 3, 4, 6], dos_time_components(entries.first)

    assert_equal Time.utc(2026, 1, 1, 18, 4, 6), entries.first.extra[:universaltime].mtime.utc

  end

  def test_add_file_should_store_modification_time_in_server_local_time_without_user_time_zone

    user = User.find(2)

    user.pref.update!(:time_zone => '')

    entries = write_zip(user) do |zip|

      zip.add_file('foo.txt', Time.utc(2026, 1, 1, 18, 4, 6)) {|sink| sink << 'foo'}

    end

    # 18:04:06 UTC in the server time zone (HST, UTC-10)

    assert_equal [2026, 1, 1, 8, 4, 6], dos_time_components(entries.first)

    assert_equal Time.utc(2026, 1, 1, 18, 4, 6), entries.first.extra[:universaltime].mtime.utc

  end

  private

  def write_zip(user)

    buffer = StringIO.new(+''.b)

    ZipKit::Streamer.open(buffer) do |streamer|

      yield Redmine::Export::ZipArchive.new(streamer, user)

    end

    Zip::File.open_buffer(buffer.string).entries

  end

  def dos_time_components(entry)

    dos_time = entry.instance_variable_get(:@time)

    [dos_time.year, dos_time.month, dos_time.day, dos_time.hour, dos_time.min, dos_time.sec]

  end

end