# frozen_string_literal: true class WikiPage class Meta < ApplicationRecord include Gitlab::Utils::StrongMemoize CanonicalSlugConflictError = Class.new(ActiveRecord::RecordInvalid) WikiPageInvalid = Class.new(ArgumentError) self.table_name = 'wiki_page_meta' belongs_to :project has_many :slugs, class_name: 'WikiPage::Slug', foreign_key: 'wiki_page_meta_id', inverse_of: :wiki_page_meta has_many :events, as: :target, dependent: :delete_all # rubocop:disable Cop/ActiveRecordDependent validates :title, presence: true validates :project_id, presence: true validate :no_two_metarecords_in_same_project_can_have_same_canonical_slug scope :with_canonical_slug, ->(slug) do joins(:slugs).where(wiki_page_slugs: { canonical: true, slug: slug }) end alias_method :resource_parent, :project class << self # Return the (updated) WikiPage::Meta record for a given wiki page # # If none is found, then a new record is created, and its fields are set # to reflect the wiki_page passed. # # @param [String] last_known_slug # @param [WikiPage] wiki_page # # This method raises errors on validation issues. def find_or_create(last_known_slug, wiki_page) raise WikiPageInvalid unless wiki_page.valid? project = wiki_page.wiki.project known_slugs = [last_known_slug, wiki_page.slug].compact.uniq raise 'No slugs found! This should not be possible.' if known_slugs.empty? transaction do updates = wiki_page_updates(wiki_page) found = find_by_canonical_slug(known_slugs, project) meta = found || create!(updates.merge(project_id: project.id)) meta.update_state(found.nil?, known_slugs, wiki_page, updates) # We don't need to run validations here, since find_by_canonical_slug # guarantees that there is no conflict in canonical_slug, and DB # constraints on title and project_id enforce our other invariants # This saves us a query. meta end end def find_by_canonical_slug(canonical_slug, project) meta, conflict = with_canonical_slug(canonical_slug) .where(project_id: project.id) .limit(2) if conflict.present? meta.errors.add(:canonical_slug, 'Duplicate value found') raise CanonicalSlugConflictError.new(meta) end meta end private def wiki_page_updates(wiki_page) last_commit_date = wiki_page.version_commit_timestamp || Time.now.utc { title: wiki_page.title, created_at: last_commit_date, updated_at: last_commit_date } end end def canonical_slug strong_memoize(:canonical_slug) { slugs.canonical.first&.slug } end def canonical_slug=(slug) return if @canonical_slug == slug if persisted? transaction do slugs.canonical.update_all(canonical: false) page_slug = slugs.create_with(canonical: true).find_or_create_by(slug: slug) page_slug.update_columns(canonical: true) unless page_slug.canonical? end else slugs.new(slug: slug, canonical: true) end @canonical_slug = slug end def update_state(created, known_slugs, wiki_page, updates) update_wiki_page_attributes(updates) insert_slugs(known_slugs, created, wiki_page.slug) self.canonical_slug = wiki_page.slug end private def update_wiki_page_attributes(updates) # Remove all unnecessary updates: updates.delete(:updated_at) if updated_at == updates[:updated_at] updates.delete(:created_at) if created_at <= updates[:created_at] updates.delete(:title) if title == updates[:title] update_columns(updates) unless updates.empty? end def insert_slugs(strings, is_new, canonical_slug) creation = Time.current.utc slug_attrs = strings.map do |slug| { wiki_page_meta_id: id, slug: slug, canonical: (is_new && slug == canonical_slug), created_at: creation, updated_at: creation } end slugs.insert_all(slug_attrs) unless !is_new && slug_attrs.size == 1 @canonical_slug = canonical_slug if is_new || strings.size == 1 end def no_two_metarecords_in_same_project_can_have_same_canonical_slug return unless project_id.present? && canonical_slug.present? offending = self.class.with_canonical_slug(canonical_slug).where(project_id: project_id) offending = offending.where.not(id: id) if persisted? if offending.exists? errors.add(:canonical_slug, 'each page in a wiki must have a distinct canonical slug') end end end end