// Copyright (C) 2017 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.gerrit.elasticsearch;

import static java.util.stream.Collectors.toList;

import com.google.gson.JsonParser;
import com.google.inject.Inject;
import com.google.inject.Singleton;
import java.io.IOException;
import java.util.Collections;
import java.util.List;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.HttpGet;
import org.elasticsearch.client.Response;
import org.elasticsearch.client.RestClient;

@Singleton
class ElasticIndexVersionDiscovery {
  private final RestClient client;

  @Inject
  ElasticIndexVersionDiscovery(ElasticRestClientBuilder clientBuilder) {
    this.client = clientBuilder.build();
  }

  List<String> discover(String prefix, String indexName) throws IOException {
    String name = prefix + indexName + "_";
    Response response = client.performRequest(HttpGet.METHOD_NAME, name + "*/_aliases");

    if (response.getStatusLine().getStatusCode() == HttpStatus.SC_OK) {
      return new JsonParser()
          .parse(AbstractElasticIndex.getContent(response))
          .getAsJsonObject()
          .entrySet()
          .stream()
          .map(e -> e.getKey().replace(name, ""))
          .collect(toList());
    }
    return Collections.emptyList();
  }
}