Add back some of the non-gitaly restore functionality behind gates

lib/backup/repository.rb

@@ -68,6 +68,9 @@ module Backup
 
     def prepare_directories
       Gitlab.config.repositories.storages.each do |name, repository_storage|
+        gitaly_migrate(:remove_repositories) do |is_enabled|
+          # TODO: Need to find a way to do this for gitaly
+          unless is_enabled
             path = repository_storage.legacy_disk_path
             next unless File.exist?(path)
 
@@ -86,22 +89,49 @@ module Backup
             end
           end
         end
+      end
+    end
 
     def restore
       prepare_directories
+      gitlab_shell = Gitlab::Shell.new
       Project.find_each(batch_size: 1000) do |project|
         path_to_project_bundle = path_to_bundle(project)
+        path_to_project_repo = path_to_repo(project)
+        project.ensure_storage_path_exists
 
+        restore_repo_status = nil
         if File.exist?(path_to_project_bundle)
           begin
-            project.repository.create_from_bundle path_to_project_bundle unless project.repository_exists?
-            progress.puts "[DONE] restoring #{project.name} repository".color(:green)
-          rescue StandardError => e
-            progress.puts "[Failed] restoring #{project.name} repository".color(:red)
+            gitlab_shell.remove_repository(project.repository_storage, project.disk_path) if project.repository_exists?
+            project.repository.create_from_bundle path_to_project_bundle
+            restore_repo_status = true
+          rescue => e
+            restore_repo_status = false
             progress.puts "Error: #{e}".color(:red)
           end
         else
-          progress.puts "[Failed] bundle file #{path_to_project_bundle} does not exist"
+          restore_repo_status = gitlab_shell.create_repository(project.repository_storage, project.disk_path)
+        end
+
+        if restore_repo_status
+          progress.puts "[DONE] restoring #{project.name} repository".color(:green)
+        else
+          progress.puts "[Failed] restoring #{project.name} repository".color(:red)
+        end
+
+        gitaly_migrate(:restore_custom_hooks) do |is_enabled|
+          # TODO: Need to find a way to do this for gitaly
+          unless is_enabled
+            in_path(path_to_tars(project)) do |dir|
+              cmd = %W(tar -xf #{path_to_tars(project, dir)} -C #{path_to_project_repo} #{dir})
+
+              output, status = Gitlab::Popen.popen(cmd)
+              unless status.zero?
+                progress_warn(project, cmd.join(' '), output)
+              end
+            end
+          end
         end
 
         wiki = ProjectWiki.new(project)
@@ -109,6 +139,7 @@ module Backup
 
         if File.exist?(path_to_wiki_bundle)
           begin
+            gitlab_shell.remove_repository(project.wiki.repository_storage, project.wiki.disk_path) if project.wiki_repository_exists?
             project.repository.create_from_bundle(path_to_wiki_bundle)
             progress.puts "[DONE] restoring #{project.name} wiki".color(:green)
           rescue StandardError => e
@@ -116,6 +147,13 @@ module Backup
             progress.puts "Error #{e}".color(:red)
           end
         end
+
+        gitaly_migrate(:create_hooks) do |is_enabled|
+          # TODO: Need to find a way to do this for gitaly
+          unless is_enabled
+            Gitlab::Git::Repository.create_hooks(path_to_project_repo, Gitlab.config.gitlab_shell.hooks_path)
+          end
+        end
       end
     end
     # rubocop:enable Metrics/AbcSize
@@ -190,5 +228,13 @@ module Backup
     def display_repo_path(project)
       project.hashed_storage?(:repository) ? "#{project.full_path} (#{project.disk_path})" : project.full_path
     end
+
+    def gitaly_migrate(method, status: Gitlab::GitalyClient::MigrationStatus::OPT_IN, &block)
+      Gitlab::GitalyClient.migrate(method, status: status, &block)
+    rescue GRPC::NotFound, GRPC::BadStatus => e
+      # Old Popen code returns [Error, output] to the caller, so we
+      # need to do the same here...
+      raise Error, e
+    end
   end
 end