BigW Consortium Gitlab

move_personal_snippet_files.rb 2.95 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
module Gitlab
  module BackgroundMigration
    class MovePersonalSnippetFiles
      delegate :select_all, :execute, :quote_string, to: :connection

      def perform(relative_source, relative_destination)
        @source_relative_location = relative_source
        @destination_relative_location = relative_destination

        move_personal_snippet_files
      end

      def move_personal_snippet_files
        query = "SELECT uploads.path, uploads.model_id FROM uploads "\
                "INNER JOIN snippets ON snippets.id = uploads.model_id WHERE uploader = 'PersonalFileUploader'"
        select_all(query).each do |upload|
          secret = upload['path'].split('/')[0]
          file_name = upload['path'].split('/')[1]

          move_file(upload['model_id'], secret, file_name)
          update_markdown(upload['model_id'], secret, file_name)
        end
      end

      def move_file(snippet_id, secret, file_name)
        source_dir = File.join(base_directory, @source_relative_location, snippet_id.to_s, secret)
        destination_dir = File.join(base_directory, @destination_relative_location, snippet_id.to_s, secret)

        source_file_path = File.join(source_dir, file_name)
        destination_file_path = File.join(destination_dir, file_name)

        unless File.exist?(source_file_path)
          say "Source file `#{source_file_path}` doesn't exist. Skipping."
          return
        end

        say "Moving file #{source_file_path} -> #{destination_file_path}"

        FileUtils.mkdir_p(destination_dir)
        FileUtils.move(source_file_path, destination_file_path)
      end

      def update_markdown(snippet_id, secret, file_name)
        source_markdown_path = File.join(@source_relative_location, snippet_id.to_s, secret, file_name)
        destination_markdown_path = File.join(@destination_relative_location, snippet_id.to_s, secret, file_name)

        source_markdown = "](#{source_markdown_path})"
        destination_markdown = "](#{destination_markdown_path})"
        quoted_source = quote_string(source_markdown)
        quoted_destination = quote_string(destination_markdown)

        execute("UPDATE snippets "\
                "SET description = replace(snippets.description, '#{quoted_source}', '#{quoted_destination}'), description_html = NULL "\
                "WHERE id = #{snippet_id}")

        query = "SELECT id, note FROM notes WHERE noteable_id = #{snippet_id} "\
                "AND noteable_type = 'Snippet' AND note IS NOT NULL"
        select_all(query).each do |note|
          text = note['note'].gsub(source_markdown, destination_markdown)
          quoted_text = quote_string(text)

          execute("UPDATE notes SET note = '#{quoted_text}', note_html = NULL WHERE id = #{note['id']}")
        end
      end

      def base_directory
        File.join(Rails.root, 'public')
      end

      def connection
        ActiveRecord::Base.connection
      end

      def say(message)
        Rails.logger.debug(message)
      end
    end
  end
end