class Gitlab::QA::Scenario::Test::Integration::Geo

def fast_ssh_key_lookup_commands
  @fast_ssh_key_lookup_commands ||= [
    %(echo -e "\n#{fast_ssh_key_lookup_content.chomp}" >> /assets/sshd_config),
    'gitlab-ctl restart sshd'
  ]
end

def fast_ssh_key_lookup_content
  @fast_ssh_key_lookup_content ||= <<~CONTENT
  # Enable fast SSH key lookup - https://docs.gitlab.com/ee/administration/operations/fast_ssh_key_lookup.html
  AuthorizedKeysCommand /opt/gitlab/embedded/service/gitlab-shell/bin/gitlab-shell-authorized-keys-check git %u %k
  AuthorizedKeysCommandUser git
  CONTENT
end

def git_lfs_install_commands
  @git_lfs_install_commands ||= [
    "cd /tmp ; curl -qsL https://github.com/git-lfs/git-lfs/releases/download/v#{GIT_LFS_VERSION}/git-lfs-linux-amd64-v#{GIT_LFS_VERSION}.tar.gz | tar xzvf -",
    'cp /tmp/git-lfs /usr/local/bin'
  ]
end

def perform(release)
  release = Release.new(release)
  raise ArgumentError, 'Geo is EE only!' unless release.ee?
  Runtime::Env.require_license!
  Component::Gitlab.perform do |primary|
    primary.release = release
    primary.name = 'gitlab-primary'
    primary.network = 'geo'
    primary.omnibus_config = <<~OMNIBUS
      geo_primary_role['enable'] = true;
      postgresql['listen_address'] = '0.0.0.0';
      postgresql['trust_auth_cidr_addresses'] = ['0.0.0.0/0','0.0.0.0/0'];
      postgresql['md5_auth_cidr_addresses'] = ['0.0.0.0/0'];
      postgresql['max_replication_slots'] = 1;
      gitlab_rails['db_key_base'] = '4dd58204865eb41bca93bd38131d51cc';
      gitlab_rails['monitoring_whitelist'] = ['0.0.0.0/0'];
      sidekiq['concurrency'] = 2;
      unicorn['worker_processes'] = 2;
    OMNIBUS
    primary.exec_commands = fast_ssh_key_lookup_commands + git_lfs_install_commands
    primary.instance do
      Component::Gitlab.perform do |secondary|
        secondary.release = release
        secondary.name = 'gitlab-secondary'
        secondary.network = 'geo'
        secondary.omnibus_config = <<~OMNIBUS
          geo_secondary_role['enable'] = true;
          gitlab_rails['db_key_base'] = '4dd58204865eb41bca93bd38131d51cc';
          sidekiq['concurrency'] = 2;
          unicorn['worker_processes'] = 2;
          gitlab_rails['monitoring_whitelist'] = ['0.0.0.0/0'];
        OMNIBUS
        secondary.exec_commands = fast_ssh_key_lookup_commands + git_lfs_install_commands
        secondary.act do
          # TODO, we do not wait for secondary to start because of
          # https://gitlab.com/gitlab-org/gitlab-ee/issues/3999
          #
          # rubocop:disable Style/Semicolon
          prepare; start; reconfigure; process_exec_commands
          # shellout to instance specs
          puts 'Running Geo primary / secondary specs!'
          Component::Specs.perform do |specs|
            specs.suite = 'QA::EE::Scenario::Test::Geo'
            specs.release = release
            specs.network = 'geo'
            specs.args = [
              '--primary-address', primary.address,
              '--primary-name', primary.name,
              '--secondary-address', secondary.address,
              '--secondary-name', secondary.name
            ]
          end
          teardown
        end
      end
    end
  end
end