gitlab-org--gitlab-foss/danger/database/Dangerfile

# frozen_string_literal: true

gitlab_danger = GitlabDanger.new(helper.gitlab_helper)

SCHEMA_NOT_UPDATED_MESSAGE_SHORT = <<~MSG
New %<migrations>s added but %<schema>s wasn't updated.
MSG

SCHEMA_NOT_UPDATED_MESSAGE_FULL = <<~MSG
**#{SCHEMA_NOT_UPDATED_MESSAGE_SHORT}**

Usually, when adding new %<migrations>s, %<schema>s should be
updated too (unless the migration isn't changing the DB schema
and isn't the most recent one).
MSG

DB_MESSAGE = <<~MSG
This merge request requires a database review. To make sure these
changes are reviewed, take the following steps:

1. Ensure the merge request has ~database and ~"database::review pending" labels.
   If the merge request modifies database files, Danger will do this for you.
1. Prepare your MR for database review according to the
   [docs](https://docs.gitlab.com/ee/development/database_review.html#how-to-prepare-the-merge-request-for-a-database-review).
1. Assign and mention the database reviewer suggested by Reviewer Roulette.
MSG

DB_FILES_MESSAGE = <<~MSG
The following files require a review from the Database team:
MSG

DATABASE_APPROVED_LABEL = 'database::approved'

non_geo_db_schema_updated = !git.modified_files.grep(%r{\Adb/schema\.rb}).empty?
geo_db_schema_updated = !git.modified_files.grep(%r{\Aee/db/geo/schema\.rb}).empty?

non_geo_migration_created = !git.added_files.grep(%r{\A(db/(post_)?migrate)/}).empty?
geo_migration_created = !git.added_files.grep(%r{\Aee/db/geo/(post_)?migrate/}).empty?

format_str = gitlab_danger.ci? ? SCHEMA_NOT_UPDATED_MESSAGE_FULL : SCHEMA_NOT_UPDATED_MESSAGE_SHORT

if non_geo_migration_created && !non_geo_db_schema_updated
  warn format(format_str, migrations: 'migrations', schema: gitlab_danger.html_link("db/schema.rb"))
end

if geo_migration_created && !geo_db_schema_updated
  warn format(format_str, migrations: 'Geo migrations', schema: gitlab_danger.html_link("ee/db/geo/schema.rb"))
end

return unless gitlab_danger.ci?
return if gitlab.mr_labels.include?(DATABASE_APPROVED_LABEL)

db_paths_to_review = helper.changes_by_category[:database]

if gitlab.mr_labels.include?('database') || db_paths_to_review.any?
  message 'This merge request adds or changes files that require a ' \
    'review from the [Database team](https://gitlab.com/groups/gl-database/-/group_members).'

  markdown(DB_MESSAGE)
  markdown(DB_FILES_MESSAGE + helper.markdown_list(db_paths_to_review)) if db_paths_to_review.any?

  database_labels = helper.missing_database_labels(gitlab.mr_labels)

  if database_labels.any?
    gitlab.api.update_merge_request(gitlab.mr_json['project_id'],
                                        gitlab.mr_json['iid'],
                                        labels: (gitlab.mr_labels + database_labels).join(','))
  end
end