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 perform(release, *rspec_args)
  release = QA::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.seed_admin_token = false
    primary.omnibus_configuration << <<~OMNIBUS
      gitlab_rails['db_key_base'] = '4dd58204865eb41bca93bd38131d51cc';
      geo_primary_role['enable'] = true;
      gitlab_rails['db_password'] = 'mypass';
      gitlab_rails['db_pool'] = 5;
      gitlab_rails['geo_node_name'] = '#{primary.name}';
      gitlab_rails['monitoring_whitelist'] = ['0.0.0.0/0'];
      gitlab_rails['packages_enabled'] = true;
      postgresql['listen_address'] = '0.0.0.0';
      postgresql['max_replication_slots'] = 1;
      postgresql['md5_auth_cidr_addresses'] = ['0.0.0.0/0'];
      postgresql['sql_user_password'] = 'e1d1469ec5f533651918b4567a3ed1ae';
      postgresql['trust_auth_cidr_addresses'] = ['0.0.0.0/0','0.0.0.0/0'];
      sidekiq['concurrency'] = 2;
      puma['worker_processes'] = 2;
    OMNIBUS
    primary.exec_commands = fast_ssh_key_lookup_commands + QA::Scenario::CLICommands.git_lfs_install_commands
    primary.instance do
      Component::Gitlab.perform do |secondary|
        secondary.release = release
        secondary.name = 'gitlab-secondary'
        secondary.network = 'geo'
        secondary.omnibus_configuration << <<~OMNIBUS
          geo_secondary['db_fdw'] = true;
          geo_secondary_role['enable'] = true;
          gitlab_rails['db_key_base'] = '4dd58204865eb41bca93bd38131d51cc';
          gitlab_rails['db_password'] = 'mypass';
          gitlab_rails['db_pool'] = 5;
          gitlab_rails['geo_node_name'] = '#{secondary.name}';
          gitlab_rails['monitoring_whitelist'] = ['0.0.0.0/0'];
          gitlab_rails['packages_enabled'] = true;
          postgresql['listen_address'] = '0.0.0.0';
          postgresql['md5_auth_cidr_addresses'] = ['0.0.0.0/0'];
          postgresql['sql_user_password'] = 'e1d1469ec5f533651918b4567a3ed1ae';
          sidekiq['concurrency'] = 2;
          puma['worker_processes'] = 2;
        OMNIBUS
        secondary.exec_commands += fast_ssh_key_lookup_commands + QA::Scenario::CLICommands.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
          Runtime::Logger.info('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,
              *rspec_args
            ]
          end
          teardown
        end
      end
    end
  end
end