/*
 * Copyright (c) 2017, 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.sample.identity.oauth2;

import org.apache.commons.lang.StringUtils;

import java.util.Properties;
import java.util.logging.Logger;

/**
 * Sample OAuth2 application config that is loaded from the config file.
 */
public class ApplicationConfig {

    private static Logger LOGGER = Logger.getLogger(ApplicationConfig.class.getName());

    public static final String CONSUMER_KEY = "ConsumerKey";
    public static final String CONSUMER_SECRET = "ConsumerSecret";
    public static final String SCOPE = "Scope";
    public static final String CALLBACK_URL = "CallbackURL";

    public static final String IS_HOST_NAME = "IdentityServerHostName";
    public static final String IS_PORT = "IdentityServerPort";
    public static final String AUTHORIZE_ENDPOINT_CONTEXT = "AuthorizeEndpointContext";
    public static final String ACCESS_TOKEN_ENDPOINT_CONTEXT = "AccessTokenEndpointContext";
    public static final String LOGOUT_ENDPOINT_CONTEXT = "LogoutEndpointContext";
    public static final String USER_INFO_ENDPOINT_CONTEXT = "UserInfoEndpointContext";
    public static final String SESSION_IFRAME_ENDPOINT_CONTEXT = "SessionIFrameEndpointContext";

    private static Properties properties = new Properties();

    public static Properties getProperties() {
        return properties;
    }

    public static void setProperties(Properties properties) {
        ApplicationConfig.properties = properties;
    }

    public static String getConsumerKey() {
        String value = "";
        if (StringUtils.isNotEmpty(getProperties().getProperty(CONSUMER_KEY))) {
            value = getProperties().getProperty(CONSUMER_KEY);
        }
        return value;
    }

    public static String getConsumerSecret() {
        String value = "";
        if (StringUtils.isNotEmpty(getProperties().getProperty(CONSUMER_SECRET))) {
            value = getProperties().getProperty(CONSUMER_SECRET);
        }
        return value;
    }

    public static String getScope() {
        String value = "";
        if (StringUtils.isNotEmpty(getProperties().getProperty(SCOPE))) {
            value = getProperties().getProperty(SCOPE);
        }
        return value;
    }

    public static String getCallbackUrl() {
        String value = "";
        if (StringUtils.isNotEmpty(getProperties().getProperty(CALLBACK_URL))) {
            value = getProperties().getProperty(CALLBACK_URL);
        }
        return value;
    }

    public static String getAuthorizeEndpointContext() {
        return buildURL(getProperties().getProperty(AUTHORIZE_ENDPOINT_CONTEXT));
    }

    public static String getAccessTokenEndpointContext() {
        return buildURL(getProperties().getProperty(ACCESS_TOKEN_ENDPOINT_CONTEXT));
    }

    public static String getLogoutEndpointContext() {
        return buildURL(getProperties().getProperty(LOGOUT_ENDPOINT_CONTEXT));
    }

    public static String getUserInforEndpointContext() {
        return buildURL(getProperties().getProperty(USER_INFO_ENDPOINT_CONTEXT));
    }

    public static String getSessionIframeEndpointContext() {
        String iframeURL = buildURL(getProperties().getProperty(SESSION_IFRAME_ENDPOINT_CONTEXT));
        if (StringUtils.isNotEmpty(iframeURL) && iframeURL.endsWith("client_id=")) {
            iframeURL += getConsumerKey();
        }
        return iframeURL;
    }

    public static String getISHostName() {
        String value = "localhost";
        if (StringUtils.isNotEmpty(getProperties().getProperty(IS_HOST_NAME))) {
            value = getProperties().getProperty(IS_HOST_NAME);
        }
        return value;
    }

    public static int getISPort() {
        int value = 0;
        try {
            int tmpValue = Integer.parseInt(getProperties().getProperty(IS_PORT));
            if (tmpValue > 0) {
                value = tmpValue;
            }
        } catch (NumberFormatException e) {
            LOGGER.warning("Invalid input for Identity Server PORT.");
        }
        return value;
    }

    private static String buildURL(String context) {
        String buildURL = "";
        if (StringUtils.isNotEmpty(context)) {
            if (!context.startsWith("https")) {
                String endpointURL = "https://" + getISHostName();
                if (getISPort() != 0) {
                    endpointURL += ":" + getISPort();
                }
                if (context.startsWith("/")) {
                    buildURL = endpointURL + context;
                } else {
                    buildURL += endpointURL + "/" + context;
                }
            } else {
                buildURL = context;
            }
        }
        return buildURL;
    }
}

@@ -3862,127 +3862,484 @@
     
-if (StringUtils.isNotEmpty(iframeURL) && iframeURL.endsWith(%22client_id=%22)) %7B%0A            iframeURL += getConsumerKey();
+String consumerKey = getConsumerKey() ;%0A        if (StringUtils.isNotEmpty(iframeURL) && StringUtils.isNotEmpty(consumerKey)) %7B%0A            if(iframeURL.contains(%22?%22))%7B%0A                if(iframeURL.trim().endsWith(%22?%22)) %7B%0A                    iframeURL += %22client_id=%22 + consumerKey;%0A                %7Delse%7B%0A                    iframeURL += %22&client_id=%22 + consumerKey;%0A                %7D%0A            %7Delse%7B%0A                iframeURL += %22?client_id=%22 + consumerKey;%0A            %7D
 %0A   

Update class comment