package com.blackboard.custom;
import com.iplanet.sso.*;
import java.net.*;
import javax.servlet.http.*;
import java.net.*;
import java.util.*;
import blackboard.platform.BbServiceManager;
import blackboard.platform.log.LogService;
import blackboard.platform.security.authentication.BaseAuthenticationModule;
import blackboard.platform.security.authentication.BbAuthenticationFailedException;
import blackboard.platform.security.authentication.BbCredentialsNotFoundException;
import blackboard.platform.security.authentication.BbSecurityException;

public class SJESIdentity6 extends BaseAuthenticationModule {
    LogService _logger;
    SSOToken token = null;
    /**
     * @param request
     * @param response
     * @throws BbSecurityException
     * @throws BbAuthenticationFailedException
     * @throws BbCredentialsNotFoundException
     * @return
     *
     * Retrieve from the header "userid" username and return
     * "userid" is the string configured to be returned in
     * AMAgent.properties file from the uid attribute in the directory server schema
     * See also the headerAttributes comments in the AMAgent.properties file.
     *
     **/
    public String doAuthenticate(HttpServletRequest request, HttpServletResponse response)
    throws BbSecurityException, BbAuthenticationFailedException, BbCredentialsNotFoundException {
        // Get the default authentication techniques result
        String username = null;
        _logger.logInfo("SJES doAuthenticate, username = " + username);
        Enumeration enum = request.getHeaderNames();
        _logger.logInfo("SJES After Enum");
        if (enum.hasMoreElements()==false) {
            _logger.logInfo("SJES enum = empty");
        }
        while (enum.hasMoreElements()){
            String name = (String) enum.nextElement();
            _logger.logInfo("SJES name = " + name);
            _logger.logInfo("SJES Value = " + request.getHeader(name));
        }
        String uid = request.getHeader("userid");
        _logger.logInfo("SJES doAuthenticate, uid = " + uid);
        username = uid;
        _logger.logInfo("SJES returning username = " + username);
        return username;
    }
    /**
     * @param request
     * @param response
     * @throws BbSecurityException
     *
     * Logout the user from the identity server using SSO API and Blackboard using super.
     * Also note that we have to destroy the cookie manually to update the agent cache and
     * hence avoid any incosistency
     *
     */
    public void doLogout(HttpServletRequest request, HttpServletResponse response) throws BbSecurityException {
        try {
            _logger.logInfo("SJES doLogout" );
            Cookie ssocookie = null;
            Cookie[] cookies = request.getCookies() ;
            String cName =  (new com.iplanet.am.util.SystemProperties()).get("com.iplanet.am.cookie.name");
            _logger.logInfo("SJES cookieName=" + cName);
            for (int k=0; k<cookies.length; k++) {
                if( cName.equalsIgnoreCase(cookies[k].getName())) {
                    ssocookie = cookies[k];
                }
            }
            
            String tokenString = ssocookie.getValue();
            _logger.logInfo("SJES tokenString =" + tokenString);
            
            String decodedtokenString = URLDecoder.decode(tokenString);
            _logger.logInfo("SJES decodetokenString =" + decodedtokenString);
            SSOTokenManager mgr = SSOTokenManager.getInstance();
            _logger.logInfo("SJES mgr for token=" + mgr);
            SSOToken token = mgr.createSSOToken(decodedtokenString);
            if(mgr.isValidToken(token)){
                
                String tokenHost = token.getHostName();
                _logger.logInfo("SJES hostname =" + tokenHost);
                java.security.Principal principal = token.getPrincipal();
                _logger.logInfo("SJES Principal =" + principal.getName());
                String authType = token.getAuthType();
                _logger.logInfo("SJES authType =" + authType);
                int level = token.getAuthLevel();
                _logger.logInfo("SJES level =" + level);
                InetAddress ipAddress = token.getIPAddress();
                _logger.logInfo("SJES ipAddress =" + ipAddress.getHostAddress());
                long maxTime = token.getMaxSessionTime();
                _logger.logInfo("SJES maxTime of =" + maxTime);
                long idleTime = token.getIdleTime();
                _logger.logInfo("SJES idleTime =" + idleTime);
                long maxIdleTime = token.getMaxIdleTime();
                _logger.logInfo("SJES maxIdleTime =" + maxIdleTime);
                long timeleft = token.getTimeLeft();
                _logger.logInfo("SJES timeleft =" + timeleft);
                mgr.destroyToken(token);
                _logger.logInfo("Destroy of token : " + token + " succeeded");
                ssocookie.setMaxAge(0);
                ssocookie.setPath("/");
                ssocookie.setDomain(YOUR_DOMAIN_STRING_HERE);// Specify the correct domain.
                ssocookie.setValue("null");
                _logger.logInfo("SJES cookie set to Null");
		response.addCookie(ssocookie);
                _logger.logInfo("Destroy of cookie completed");
            }
            else {
                _logger.logInfo("Token is Invalid");
            }
            super.doLogout(request,response);
        }
        catch (Exception e){
            _logger.logInfo("Exception of doLogout ");
            _logger.logInfo("Exception is " + e);
        }
        
    }
    /**
     * @return
     * Refer to the Blackboard Integration Guide for further details
     * Just overiding an class from the Base Class
     */
    public String getAuthType() {
        return "custom";
    }
    private static final String[] PROP_KEYS = { "impl"};
    /**
     *
     * Refer to the Blackboard Integration Guide for further details
     * Just overiding an class from the Base Class
     *
     * @return  */
    public String[] getPropKeys() {
        return  PROP_KEYS;
    }
    public SJESIdentity6() {
        try {
            // Set _logger
            _logger = BbServiceManager.getLogService();
        }
        catch (Exception e) {
            e.printStackTrace();
        }
        _logger.logInfo("SJES Auth Module: instantiated");
    }
}