150 lines
4.7 KiB
Ruby
150 lines
4.7 KiB
Ruby
# 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
|