/*
 *  Copyright (c) 2020, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 *
 *  WSO2 Inc. licenses this file to you 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 org.wso2.synapse.samples.jaxrs.foodsample;

import org.apache.commons.codec.binary.Base64;
import org.wso2.synapse.samples.jaxrs.foodsample.bean.Token;

import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response;

@javax.inject.Singleton
@Path("/foodservice/")
public class FoodService {

    private int unauthorizedReqCount = 0;
    private int tokenReqCount = 0;
    private int appleServiceRequests = 0;

    @POST
    @Path("/token")
    @Consumes(MediaType.APPLICATION_FORM_URLENCODED)
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAccessToken(@Context HttpHeaders httpHeaders,
                                   MultivaluedMap<String, String> tokenRequestParams) {

        tokenReqCount++;

        String basicHeader = httpHeaders.getHeaderString("Authorization");

        if (validateBasicAuthHeader(basicHeader)) {
            return Response.status(Response.Status.OK).entity(new Token(Constants.accessToken, Constants.expiresIn,
                    Constants.tokenType)).build();
        }
        return Response.status(Response.Status.UNAUTHORIZED).entity("Invalid Credentials").build();
    }

    @POST
    @Path("/token1")
    @Consumes(MediaType.APPLICATION_FORM_URLENCODED)
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAccessToken1(@Context HttpHeaders httpHeaders,
                                    MultivaluedMap<String, String> tokenRequestParams) {

        String basicHeader = httpHeaders.getHeaderString("Authorization");

        if (validateBasicAuthHeader(basicHeader)) {
            return Response.status(Response.Status.OK).entity(new Token(Constants.accessToken, Constants.expiresIn,
                    Constants.tokenType)).build();
        }
        return Response.status(Response.Status.UNAUTHORIZED).entity("Invalid Credentials").build();
    }

    @POST
    @Path("/token2")
    @Consumes(MediaType.APPLICATION_FORM_URLENCODED)
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAccessToken2(@Context HttpHeaders httpHeaders,
                                   MultivaluedMap<String, String> tokenRequestParams) {

        String basicHeader = httpHeaders.getHeaderString("Authorization");

        if (validateBasicAuthHeader(basicHeader)) {
            return Response.status(Response.Status.OK).entity(new Token(Constants.accessToken2, Constants.expiresIn,
                    Constants.tokenType)).build();
        }
        return Response.status(Response.Status.UNAUTHORIZED).entity("Invalid Credentials").build();
    }

    @POST
    @Path("/custom-token")
    @Consumes(MediaType.APPLICATION_FORM_URLENCODED)
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAccessTokenWithCustomParams(@Context HttpHeaders httpHeaders,
                                                   MultivaluedMap<String, String> tokenRequestParams) {

        String basicHeader = httpHeaders.getHeaderString("Authorization");

        if (validateBasicAuthHeader(basicHeader) && validateCustomHeader(httpHeaders) && validateCustomParams(tokenRequestParams)) {
            return Response.status(Response.Status.OK).entity(new Token(Constants.accessToken, Constants.expiresIn,
                    Constants.tokenType)).build();
        }
        return Response.status(Response.Status.UNAUTHORIZED).entity("Invalid Credentials").build();
    }

    @POST
    @Path("/password-token")
    @Consumes(MediaType.APPLICATION_FORM_URLENCODED)
    @Produces(MediaType.APPLICATION_JSON)
    public Response getAccessTokenWithPasswordGrant(@Context HttpHeaders httpHeaders,
                                                    MultivaluedMap<String, String> tokenRequestParams) {

        String basicHeader = httpHeaders.getHeaderString("Authorization");

        if (validateBasicAuthHeader(basicHeader) && validateCustomParams(tokenRequestParams) &&
                validatePasswordCredentials(tokenRequestParams)) {
            return Response.status(Response.Status.OK).entity(new Token(Constants.accessToken, Constants.expiresIn,
                    Constants.tokenType)).build();
        }
        return Response.status(Response.Status.UNAUTHORIZED).entity("Invalid Credentials").build();
    }

    @GET
    @Path("/food")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getFoodItem(@Context HttpHeaders httpHeaders) {

        String authorizationHeader = httpHeaders.getHeaderString("Authorization");
        if (authorizationHeader != null) {
            String token = authorizationHeader.split(" ")[1];
            if (token.equals(Constants.accessToken)) {
                return Response.status(Response.Status.OK).entity(tokenReqCount).build();
            }
        }
        return Response.status(Response.Status.UNAUTHORIZED).build();
    }

    @GET
    @Path("/apple")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getFoodItem2(@Context HttpHeaders httpHeaders) {

        appleServiceRequests++;
        String authorizationHeader = httpHeaders.getHeaderString("Authorization");
        if (authorizationHeader != null) {
            String token = authorizationHeader.split(" ")[1];
            if (token.equals(Constants.accessToken2)) {
                return Response.status(Response.Status.OK).entity(appleServiceRequests).build();
            }
        }
        return Response.status(Response.Status.UNAUTHORIZED).build();
    }

    @GET
    @Path("/unauthorized")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getUnauthorizedItem(@Context HttpHeaders httpHeaders) {

        unauthorizedReqCount++;
        return Response.status(Response.Status.UNAUTHORIZED).entity(unauthorizedReqCount).build();
    }

    private boolean validateCredentials(MultivaluedMap<String, String> tokenRequestParams) {

        String refreshToken = tokenRequestParams.getFirst("refresh_token");
        String clientIdInBody = tokenRequestParams.getFirst("client_id");
        String clientSecretInBody = tokenRequestParams.getFirst("client_secret");

        if (refreshToken != null && !refreshToken.equals(Constants.refreshToken)) {
            return false;
        }

        return clientIdInBody.equals(Constants.clientId) && clientSecretInBody.equals(Constants.clientSecret);
    }

    private boolean validatePasswordCredentials(MultivaluedMap<String, String> tokenRequestParams) {

        String username = tokenRequestParams.getFirst("username");
        String password = tokenRequestParams.getFirst("password");
        String grantType = tokenRequestParams.getFirst("grant_type");

        return username.equals(Constants.username) && password.equals(Constants.password) &&
                grantType.equals(Constants.passwordGrant);
    }

    private boolean validateBasicAuthHeader(String basicHeader) {

        String credentials = basicHeader.substring(6).trim();

        String decodedCredentials = new String(new Base64().decode(credentials.getBytes()));
        String clientId = decodedCredentials.split(":")[0];
        String clientSecret = decodedCredentials.split(":")[1];

        return clientId.equals(Constants.clientId) && clientSecret.equals(Constants.clientSecret);
    }

    private boolean validateCustomParams(MultivaluedMap<String, String> tokenRequestParams) {

        String accountId = tokenRequestParams.getFirst("account_id");
        String userRole = tokenRequestParams.getFirst("user_role");

        return accountId.equals("1234") && userRole.equals("tester");
    }

    private boolean validateCustomHeader(HttpHeaders httpHeaders) {
        String userTokenHeader = httpHeaders.getHeaderString("user_token");
        String accountTokenHeader = httpHeaders.getHeaderString("account_token");

        return userTokenHeader.equals("abc#123") && accountTokenHeader.equals("zxy@123");
    }
}
