D-Net

package eu.dnetlib.openaire.user.api.services;

import com.google.gson.Gson;

import com.unboundid.ldap.sdk.LDAPException;

import eu.dnetlib.openaire.user.MigrationUser;

import eu.dnetlib.openaire.user.dao.SQLMigrationUserDAO;

import eu.dnetlib.openaire.user.ldap.MUserActionsLDAP;

import javax.servlet.http.HttpServletRequest;

import javax.ws.rs.*;

import javax.ws.rs.core.Context;

import javax.ws.rs.core.MediaType;

import javax.ws.rs.core.Response;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.List;

/**

 * Created by sofia on 17/11/2016.

 */

@Path("/users")

public class Test2Service {

    @GET

    @Path("/{userId}")

    @Produces(MediaType.APPLICATION_JSON)

    public Response getUserById(@PathParam("userId") int userId)

    {

        SQLMigrationUserDAO mUserDAO = new SQLMigrationUserDAO();

        try {

            MigrationUser mUser = mUserDAO.fetchById(userId);

            // Invalide user ID

            if (mUser == null) {

                String errorMessageJson = compose404Message("Cannot find user with id " + userId + ".");

                return Response

                        .status(Response.Status.NOT_FOUND)

                        .entity(errorMessageJson)

                        .type(MediaType.APPLICATION_JSON)

                        .build();

            }

            return Response.status(200).entity(composeDataResponse(mUser)).build();

        }

        catch (SQLException e) {

            return Response

                    .status(Response.Status.INTERNAL_SERVER_ERROR)

                    .entity(compose500Message("Fail to fetch users.", e))

                    .type(MediaType.APPLICATION_JSON)

                    .build();

        }

    }

    @GET

    @Path("/authenticate")

    @Produces(MediaType.APPLICATION_JSON)

    public Response authenticateUser(@QueryParam("username") String username, @QueryParam("password") String password)

    {

        SQLMigrationUserDAO mUserDAO = new SQLMigrationUserDAO();

        try {

            MigrationUser mUser = mUserDAO.fetchByUid(username);

            // if user was not found

            if (mUser == null) {

                String errorMessageJson = compose404Message("Cannot find user with username " + username + ".");

                return Response

                        .status(Response.Status.NOT_FOUND)

                        .entity(errorMessageJson)

                        .type(MediaType.APPLICATION_JSON)

                        .build();

            }

            boolean authenticated = MUserActionsLDAP.authenticateUser(mUser.getEmail(), password);

            if (!authenticated) {

                return Response

                        .status(Response.Status.UNAUTHORIZED)

                        .entity(compose401Message("User " + username + " could not be authenticated."))

                        .type(MediaType.APPLICATION_JSON)

                        .build();

            }

            return Response.status(200).entity(composeDataResponse(mUser)).build();

        }

        catch (LDAPException exc) {

            return Response

                    .status(Response.Status.INTERNAL_SERVER_ERROR)

                    .entity(compose500Message("LDAP error.", exc))

                    .type(MediaType.APPLICATION_JSON)

                    .build();

        }

        catch (SQLException exc) {

            return Response

                    .status(Response.Status.INTERNAL_SERVER_ERROR)

                    .entity(compose500Message("Fail to fetch users.", exc))

                    .type(MediaType.APPLICATION_JSON)

                    .build();

        }

    }

    /* JSON Utility Methods */

    private String compose401Message(String message) {

        return  "{ \"status\" : \"error\", \"code\" : \"401\", \"message\" : \"  " + message +" \" }";

    }

    private String compose404Message(String message) {

        return  "{ \"status\" : \"error\", \"code\" : \"404\", \"message\" : \"  " + message +" \" }";

    }

    private String compose500Message(String message, Exception exception) {

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

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

    }

    private String composeDataResponse(MigrationUser user) {

        return " { \"status\" : \"success\", \"code\": \"200\", " + "\"data\" : " + new Gson().toJson(user) + " }";

    }

}