1 files changed, 28 insertions, 0 deletions
diff --git a/lib/tasks/gitlab/db/migration_squash.rake b/lib/tasks/gitlab/db/migration_squash.rake
new file mode 100644
index 00000000000..7ddd6c41d12
--- /dev/null
+++ b/lib/tasks/gitlab/db/migration_squash.rake
@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+namespace :gitlab do
+  namespace :db do
+    desc "GitLab | DB | squash | squash as of a version"
+    task :squash, [:version] => :environment do |_t, args|
+      require 'git'
+      git = ::Git.open(Dir.pwd)
+
+      squasher = Gitlab::Database::Migrations::Squasher.new(
+        `git ls-tree --name-only -r #{args[:version]} -- db/migrate db/post_migrate`
+      )
+
+      new_init_structure_sql = git.show(args[:version], 'db/structure.sql')
+      # Delete relevant migrations and specs
+      squasher.files_to_delete.each do |filename|
+        git.remove filename
+        puts "\tDeleting #{filename} from repo".red
+      rescue Git::GitExecuteError
+        puts "#{filename} is not in the current branch"
+      end
+      puts "\tOverwriting init_structure.sql..."
+      File.write('db/init_structure.sql', new_init_structure_sql)
+      git.add('db/init_structure.sql')
+      puts "\tDone!".white
+    end
+  end
+end