package eu.dnetlib.datasource.publisher;

import java.util.List;

import com.google.common.collect.Lists;
import eu.dnetlib.common.rmi.DNetRestDocumentation;
import eu.dnetlib.datasource.publisher.clients.ClientResponse;
import eu.dnetlib.datasource.publisher.clients.DatasourceInfoRetriever;
import eu.dnetlib.datasource.publisher.model.DatasourceResponse;

import eu.dnetlib.datasource.publisher.model.IdentifiersResponse;
import eu.dnetlib.datasource.publisher.model.Response;
import io.swagger.annotations.*;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PathVariable;

@Controller
@DNetRestDocumentation
public class DatasourcesApiController implements DatasourcesApi {

	private static final Log log = LogFactory.getLog(DatasourcesApiController.class);

	@Autowired
	private DatasourceInfoRetriever dsInfoRetriever;

	@Override
    public ResponseEntity<IdentifiersResponse> listIds() {
	    try {
		    return rsp(dsInfoRetriever.listIds(), HttpStatus.OK);
	    } catch (ApiException e) {
		    return rsp(new IdentifiersResponse(), HttpStatus.INTERNAL_SERVER_ERROR);
	    }
    }

    @Override
    public ResponseEntity<DatasourceResponse> getDs(@ApiParam(value = "ID of datasource to fetch", required=true ) @PathVariable("id") String id) {

	    if (log.isDebugEnabled()) {
		    log.debug(String.format("getDatasourceInfo(dsId = %s)", id));
	    }
	    final long start = System.nanoTime();
	    final ResponseEntity<DatasourceResponse> rsp = rsp(new DatasourceResponse(), HttpStatus.OK);

	    final ClientResponse clientResponse = dsInfoRetriever.getInfo(id);
	    BeanUtils.copyProperties(clientResponse.getDatasourceResponse(), rsp.getBody());

	    if (!clientResponse.getErrors().isEmpty()) {
	    	return rsp(clientResponse.getDatasourceResponse(), HttpStatus.INTERNAL_SERVER_ERROR);
	    }

	    final long time = (System.nanoTime() - start) / 1000000;

	    rsp.getBody().getResponseHeader().setDatasourceId(id).setQueryTime(time);
	    log.debug(String.format("%s ms", time));

	    return rsp;
    }

	private <T extends Response> ResponseEntity<T> rsp(final T info, final HttpStatus status) {
		info.getResponseHeader().setStatusCode(status.value());
		return new ResponseEntity<>(info, status);
	}

}