D-Net

package eu.dnetlib.openaire.usermanagement;

import com.google.gson.Gson;

import com.google.gson.GsonBuilder;

import eu.dnetlib.openaire.user.pojos.RegisteredService;

import eu.dnetlib.openaire.usermanagement.utils.RegisteredServicesUtils;

import eu.dnetlib.openaire.usermanagement.utils.TokenUtils;

import org.apache.commons.validator.routines.UrlValidator;

import org.apache.log4j.Logger;

import org.mitre.openid.connect.model.OIDCAuthenticationToken;

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

import org.springframework.security.core.context.SecurityContextHolder;

import org.springframework.web.context.support.SpringBeanAutowiringSupport;

import javax.servlet.ServletConfig;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.io.IOException;

import java.net.URL;

import java.sql.SQLException;

public class RegisterServiceServlet extends HttpServlet {

    private Logger logger = Logger.getLogger(RegisterServiceServlet.class);

    public void init(ServletConfig config) throws ServletException {

        super.init(config);

        SpringBeanAutowiringSupport.processInjectionBasedOnServletContext(this,

                config.getServletContext());

    }

    @Autowired

    private RegisteredServicesUtils registeredServicesUtils;

    public void doGet(HttpServletRequest request, HttpServletResponse response)

            throws ServletException, IOException {

        response.setContentType("text/html");

        request.getRequestDispatcher("./registerService.jsp").include(request, response);

    }

    public void doPost(HttpServletRequest request, HttpServletResponse response)

            throws ServletException, IOException {

        //TODO check user's limit

        response.setContentType("text/html");

        String name = request.getParameter("first_name").trim();

        String description = request.getParameter("description").trim();

        String keyType = request.getParameter("key_radio").trim();

        System.out.println("key type " + keyType);

/*

        if (keyType.equals("uri")) {

            String jwksUri = request.getParameter("uri");

            System.out.println("JWKS URI " + jwksUri);

            UrlValidator urlValidator = new UrlValidator();

            if (!urlValidator.isValid(jwksUri)){

                request.getSession().setAttribute("msg_key_uri_error_display", "display:block");

            }

        } else {

            String jwksString = request.getParameter("value");

            System.out.println("JWKS String " + jwksString);

            Gson gson = new GsonBuilder().registerTypeAdapter(Jwks.class, new ServiceResponseDeserializer()).create();

            Jwks jwks = gson.fromJson(jwksString, Jwks.class);

            System.out.println(jwks.keys);

            System.out.println(jwks.keys.length);

        }

*/

        OIDCAuthenticationToken authentication = (OIDCAuthenticationToken) SecurityContextHolder.getContext().getAuthentication();

        String userid = authentication.getSub();

        String email = authentication.getUserInfo().getEmail();

        String accessToken = authentication.getAccessTokenValue();

        ServiceResponse serviceResponse = null;

        if ( name!= null && !name.isEmpty() && userid != null && !userid.isEmpty() &&

                    email != null && !email.isEmpty()) {

            String serverMessage = TokenUtils.registerService(userid, name, description, email, accessToken);

            System.out.println("Server message" + serverMessage);

            if (serverMessage == null) {

                request.getSession().setAttribute("message", "There was an error registering your service. Please try again later.");

                response.sendRedirect("./registeredServices");

            }

            serviceResponse = new Gson().fromJson(serverMessage, ServiceResponse.class);

            System.out.println("SERVICE RESPONSE " + serviceResponse);

        }

        String aai_id = serviceResponse.getId();

        String owner = userid;

        RegisteredService registeredService = new RegisteredService(aai_id, owner, name);

        try {

            registeredServicesUtils.addRegistedService(registeredService);

        } catch (SQLException sqle) {

            System.out.println("SQLE SKATA " +  sqle.getMessage());

            logger.error("SQLE SKATA", sqle);

            request.getSession().setAttribute("message", "There was an error registering your service. Please try again later.");

            response.sendRedirect("./registeredServices");

        }

        request.getSession().setAttribute("accessToken", authentication.getAccessTokenValue());

        request.getSession().setAttribute("refreshToken", authentication.getRefreshTokenValue());

        response.sendRedirect("./registeredServices");

    }

}