D-Net

package eu.dnetlib.openaire.rest;

import com.google.gson.JsonObject;

import eu.dnetlib.data.claims.entity.Metrics;

import eu.dnetlib.data.claims.handler.MetricsHandler;

import eu.dnetlib.data.claims.sql.SQLStoreException;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import org.springframework.web.bind.annotation.CrossOrigin;

import javax.ws.rs.CookieParam;

import javax.ws.rs.GET;

import javax.ws.rs.Path;

import javax.ws.rs.Produces;

import javax.ws.rs.core.MediaType;

import java.sql.SQLException;

import java.util.HashMap;

import java.util.Map;

@Component

@CrossOrigin(origins = "*")

@Path("/report")

public class MetricsService {

    private static final Logger logger = LogManager.getLogger(MetricsService.class);

    @Autowired

    public Authorization authorization = null;

    @Autowired

    private MetricsHandler metricsHandler = null;

    public static void main(String[] args) {}

//    @GET

//    @Path("/euprojects")

//    @Produces(MediaType.APPLICATION_JSON)

//    public Response getNumOfEUProjects() {

//        try {

//            int count = metricsHandler.countEUProjects();

//            return Response.status(200).entity(composeDataResponse(count)).build();

//        } catch (SQLStoreException |Exception e) {

//            logger.error("Could not fetch number of EU projects", e);

//            return Response.status(Response.Status.INTERNAL_SERVER_ERROR).entity(

//                    compose500Message("Fail to fetch number of EU projects.", e)).

//                    type(MediaType.APPLICATION_JSON).build();

//        }

//    }

//

//

//    @GET

//    @Path("/countries")

//    @Produces(MediaType.APPLICATION_JSON)

//    public Response getNumOfCountriesInClaims() {

//        try {

//            int count = metricsHandler.countCountriesFromClaimedBy();

//            return Response.status(200).entity(composeDataResponse(count)).build();

//        } catch (SQLStoreException |Exception e) {

//            logger.error("Could not fetch number of countries for people who claimed", e);

//            return Response.status(Response.Status.INTERNAL_SERVER_ERROR).entity(

//                    compose500Message("Fail to fetch number of countries for people who claimed.", e)).

//                    type(MediaType.APPLICATION_JSON).build();

//        }

//    }

//

//    @GET

//    @Path("/research-results")

//    @Produces(MediaType.APPLICATION_JSON)

//    public Response getNumOfUniqueResearchResults() {

//        try {

//            int count = metricsHandler.countUniqueResearchResults();

//            return Response.status(200).entity(composeDataResponse(count)).build();

//        } catch (SQLStoreException |Exception e) {

//            logger.error("Could not fetch number of unique research results claimed", e);

//            return Response.status(Response.Status.INTERNAL_SERVER_ERROR).entity(

//                    compose500Message("Fail to fetch number of unique research results claimed.", e)).

//                    type(MediaType.APPLICATION_JSON).build();

//        }

//    }

    @GET

    @Path("/calc")

    @Produces(MediaType.TEXT_PLAIN)

    public String calculateMetrics(

//            @HeaderParam("X-XSRF-TOKEN") String token,

            @CookieParam("AccessToken") String  cookie

    ) throws SQLStoreException, Exception {

        if(cookie == null || cookie.isEmpty()){

            return "Forbidden: You don't have permission to access. Maybe you are not registered.";

        }

        if(authorization.isPortalAdministrator(cookie)) {

            metricsHandler.calculateMetrics();

            return "metrics calculated!";

        }

        return "Forbidden: You don't have permission to access.";

    }

    @GET

    @Path("/metrics")

    @Produces(MediaType.TEXT_PLAIN)

    public String getMetrics() throws SQLStoreException, SQLException {

        Metrics metrics = metricsHandler.getMetrics();

        String response = "";

        if(metrics != null) {

//            response += "# TYPE claims gauge\n";

//            response += "claims " + metrics.getTotal_claims() + "\n";

//

//            response += "# TYPE claims_users gauge\n";

//            response += "claims_users " + metrics.getTotal_users() + "\n";

//

//            response += "# TYPE claims_projects gauge\n";

//            response += "claims_projects " + metrics.getProjects() + "\n";

//

//            response += "# TYPE claims_eu_projects gauge\n";

//            response += "claims_eu_projects " + metrics.getEu_projects() + "\n";

//

//            response += "# TYPE claims_countries gauge\n";

//            response += "claims_countries " + metrics.getCountries() + "\n";

//

//            response += "# TYPE claims_unique_research_results gauge\n";

//            response += "claims_unique_research_results " + metrics.getResearch_results() + "\n";

            Map<String, Integer> worksPerDashboard = new HashMap<>();

            // 1st approach

//            response += "# TYPE orcid_works gauge\n";

            response += "# TYPE claims_recorded gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

                if(dashboardMetrics.get("claims") != null) {

                    response += "claims_recorded{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("claims") + "\n";

                }

            }

            response += "# TYPE claims_users gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

                if(dashboardMetrics.get("users") != null) {

                    response += "claims_users{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("users") + "\n";

                }

            }

            response += "# TYPE claims_projects gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

                if(dashboardMetrics.get("projects") != null) {

                    response += "claims_projects{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("projects") + "\n";

                }

            }

            response += "# TYPE claims_eu_projects gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

                if(dashboardMetrics.get("eu_projects") != null) {

                    response += "claims_eu_projects{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("eu_projects") + "\n";

                }

            }

            response += "# TYPE claims_countries gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

                if(dashboardMetrics.get("countries") != null) {

                    response += "claims_countries{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("countries") + "\n";

                }

            }

            response += "# TYPE claims_unique_research_results gauge\n";

            for(JsonObject dashboardMetrics : metrics.getMetrics_per_dashboard()) {

//            for(Object dashboardMetricsObj : (List) metrics.getMetrics_per_dashboard()) {

//                Map<String, Object> dashboardMetrics = (HashMap<String, Object>) dashboardMetricsObj;

                if(dashboardMetrics.get("research_results") != null) {

                    response += "claims_unique_research_results{environment=" + dashboardMetrics.get("environment") + " portal=" + dashboardMetrics.get("dashboard") + "}" + " " + dashboardMetrics.get("research_results") + "\n";

                }

            }

            if(metrics.getDate() != null) {

                response += "claims_last_metrics_updater_run_timestamp_seconds " + (metrics.getDate().getTime() / 1000) + "\n";

            }

        }

        return response;

    }

    private String composeResults(int total, String label) {

        return "\""+label+"\": \""+total+"\"";

    }

    private String composeTotalResults(int total) {

        return "\"total\": \""+total+"\"";

    }

    private String composeDataResponse(int count) {

        return " { \"status\" : \"success\", \"code\": \"200\",  "+composeTotalResults(count) + " }";

    }

//    private String composeDataResponse(Metrics metrics) {

//        return " { \"status\" : \"success\", \"code\": \"200\",  "

//                +composeResults(metrics.getCountries(), "countries")

//                +composeResults(metrics.getEu_projects(), "eu_projects")

//                +composeResults(metrics.getResearch_results(), "research_results")+ " }";

//    }

    private String compose500Message(String message, Throwable throwable) {

        return  "{ \"status\" : \"fail\", \"code\" : \"500\", \"message\" : \"  " + message + "\", " +

                "\"description\" : \""+  throwable.getMessage() +"\" }";

    }

}