class Gitlab::QA::Release

def api_project_name
  project_name.gsub('ce', 'foss').gsub('-ee', '')
end

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
  return if valid?
  raise InvalidImageNameError, "The release image name '#{@release}' does not have the expected format."
end

def login_params
  return if Runtime::Env.skip_pull?
  if dev_gitlab_org?
    Runtime::Env.require_qa_dev_access_token!
    {
      username: Runtime::Env.gitlab_dev_username,
      password: Runtime::Env.dev_access_token_variable,
      registry: DEV_REGISTRY
    }
  elsif omnibus_mirror?
    username, password = if Runtime::Env.ci_job_token && Runtime::Env.ci_pipeline_source.include?('pipeline')
                           ['gitlab-ci-token', Runtime::Env.ci_job_token]
                         elsif Runtime::Env.qa_container_registry_access_token
                           [Runtime::Env.gitlab_username, Runtime::Env.qa_container_registry_access_token]
                         else
                           Runtime::Env.require_qa_access_token!
                           [Runtime::Env.gitlab_username, Runtime::Env.qa_access_token]
                         end
    {
      username: username,
      password: password,
      registry: COM_REGISTRY
    }
  end
end

def omnibus_mirror?
  image.start_with?("#{COM_REGISTRY}/gitlab-org/build/omnibus-gitlab-mirror/")
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
  @qa_image ||= if omnibus_mirror?
                  omnibus_project = image.match(CUSTOM_GITLAB_IMAGE_REGEX)[:project]
                  gitlab_project = "/gitlab-org/gitlab/"
                  "#{image.gsub(omnibus_project, gitlab_project)}-qa"
                else
                  "#{image}-qa"
                end
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