class Gitlab::QA::Release

def canonical?
  release =~ CANONICAL_REGEX
end

def canonical_image
  @canonical_image ||= "gitlab/gitlab-#{edition}"
end

def dev_gitlab_org?
  image.start_with?(DEV_REGISTRY)
end

def edition
  @edition ||=
    if canonical?
      release.match(CANONICAL_REGEX)[:edition].to_sym
    else
      release.match(CUSTOM_GITLAB_IMAGE_REGEX)[:edition].to_sym
    end
end

def ee?
  edition == :ee
end

def image
  @image ||=
    if canonical?
      "gitlab/gitlab-#{edition}"
    else
      release.match(CUSTOM_GITLAB_IMAGE_REGEX)[:image_without_tag]
    end
end

def initialize(release)
  @release = release.to_s.downcase
  raise InvalidImageNameError, "The release image name '#{@release}' does not have the expected format." unless valid?
end

def previous_stable
  # The previous stable is always gitlab/gitlab-ce:latest or
  # gitlab/gitlab-ee:latest
  self.class.new("#{canonical_image}:latest")
end

def project_name
  @project_name ||=
    if canonical?
      "gitlab-#{edition}"
    else
      "gitlab-#{release.match(CUSTOM_GITLAB_IMAGE_REGEX)[:edition]}"
    end
end

def qa_image
  "#{image}-qa"
end

def qa_tag
  if dev_gitlab_org? && (match_data = tag.match(DEV_TAG_REGEX))
    "#{match_data[:version]}-#{match_data[:gitlab_ref]}"
  else
    tag.sub(/[-\.]([ce]e)(\.(\d+))?\z/, '-\1')
  end
end

def tag
  @tag ||=
    if canonical?
      release.match(CANONICAL_REGEX)[:tag] || DEFAULT_CANONICAL_TAG
    else
      release.match(CUSTOM_GITLAB_IMAGE_REGEX)&.[](:tag) || DEFAULT_TAG
    end
end

def to_ee
  return self if ee?
  self.class.new(to_s.sub('ce:', 'ee:'))
end

def to_s
  "#{image}:#{tag}"
end

def valid?
  canonical? || release.match?(CUSTOM_GITLAB_IMAGE_REGEX)
end