S3CFG.java

/**
 *  Copyright 2007-2008 University Of Southern California
 *
 *  Licensed 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 edu.isi.pegasus.common.credential.impl;

import edu.isi.pegasus.common.credential.CredentialHandler;
import edu.isi.pegasus.planner.catalog.classes.Profiles;

import java.io.File;
import java.util.Map;

import edu.isi.pegasus.planner.catalog.site.classes.SiteCatalogEntry;
import edu.isi.pegasus.planner.classes.PegasusBag;
import edu.isi.pegasus.planner.namespace.Namespace;




/**
 * A convenience class that allows us to determine the path to the user s3cfg file.
 *
 * @author Mats Rynge
 * @author Karan Vahi
 *
 * @version $Revision$
 */
public class S3CFG  extends Abstract implements CredentialHandler {

    /**
     * The name of the environment variable that specifies the path to the
     * s3cfg file.
     */
    public static final String S3CFG_FILE_VARIABLE = "S3CFG";
    
    
    private static final String S3CFG_PEGASUS_PROFILE_KEY = S3CFG.S3CFG_FILE_VARIABLE.toLowerCase() ;//has to be lowercased

    /**
     * The description
     */
    private static final String DESCRIPTION = "S3 Conf File Credential Handler";

    /**
     * The local path to the credential
     */
    private String mLocalCredentialPath;

    /**
     * The default constructor.
     */
    public S3CFG(){
        super();
    }
    
    /**
     * Initializes the credential implementation. Implementations require
     * access to the logger, properties and the SiteCatalog Store.
     *
     * @param bag  the bag of Pegasus objects.
     */
    public void initialize( PegasusBag bag ){
        super.initialize( bag );
        mLocalCredentialPath = this.getLocalPath();
    }
    
    /**
     * Returns the path to S3CFG . The order of preference is as follows
     *
     * - If a S3CFG is specified as a Pegasus Profile in the site catalog
     * - Else the path on the local site
     *
     * @param site   the  site handle
     *
     * @return  the path to S3CFG.S3CFG_FILE_VARIABLE for the site.
     */
    public  String getPath( String site ){

        SiteCatalogEntry siteEntry = mSiteStore.lookup( site );
        //check if one is specified in site catalog entry
        String path = ( siteEntry == null )? null :
                       (String)siteEntry.getProfiles().get( Profiles.NAMESPACES.pegasus).get( S3CFG.S3CFG_FILE_VARIABLE.toLowerCase() );

        return( path == null ) ?
                //PM-731 return the path on the local site
                mLocalCredentialPath:
                path;
       
    }

    /**
     * Returns the path to user cred on the local site. 
     * The order of preference is as follows
     *
     * - If a S3CFG.S3CFG_FILE_VARIABLE is specified in the site catalog entry as a Pegasus Profile that is used, else the corresponding env profile for backward support
     * - Else S3CFG.S3CFG_FILE_VARIABLE Pegasus Profile specified in the properties, else the corresponding env profile for backward support
     * - Else the one pointed to by the environment variable S3CFG.S3CFG_FILE_VARIABLE
     * - Else the default path of ~/.pegasus/s3cfg
     * - Else the legacy default path of ~/.s3cfg
     * 
     * @param site   the  site catalog entry object.
     *
     * @return  the path to user cred.
     */
    public String getLocalPath(){
        SiteCatalogEntry siteEntry = mSiteStore.lookup( "local" );

        //check if corresponding Pegasus Profile is specified in site catalog entry
        String cred = ( siteEntry == null )? null :
                        (String)siteEntry.getProfiles().get( Profiles.NAMESPACES.pegasus).get( S3CFG.S3CFG_PEGASUS_PROFILE_KEY );
        if( cred == null && siteEntry != null ){
            //try to check for an env profile in the site entry 
            cred = (String)siteEntry.getProfiles().get( Profiles.NAMESPACES.env).get( S3CFG.S3CFG_FILE_VARIABLE );
        }
        
        //try from properites file
        if( cred == null ){
            //load the pegasus profile from property file 
            Namespace profiles = mProps.getProfiles( Profiles.NAMESPACES.pegasus );
            cred = (String)profiles.get( S3CFG.S3CFG_PEGASUS_PROFILE_KEY  );
        }
        if( cred == null ) {
            //load the env profile from the  property file
            Namespace env = mProps.getProfiles(Profiles.NAMESPACES.env);
            cred = (String)env.get( S3CFG.S3CFG_FILE_VARIABLE  );
        }
        
        if( cred == null){
            //check if S3CFG is specified in the environment
            Map<String,String> envs = System.getenv();
            if( envs.containsKey( S3CFG.S3CFG_FILE_VARIABLE ) ){
                cred = envs.get( S3CFG.S3CFG_FILE_VARIABLE  );
            }
            
            if (cred == null) {
                // New default
                cred = envs.get("HOME") + "/.pegasus/s3cfg";

                File cfg = new File(cred);
                if (!cfg.isFile()) {
                    // Old default
                    cred = envs.get("HOME") + "/.s3cfg";
                }
            }
        }
        
        return cred;
    }

   

    
    /**
     * returns the basename of the path to the local credential
     * 
     * @param site  the site handle
     */
    public String getBaseName( String site ) {
        File path = new File(this.getPath( site ) );
        return path.getName();
    }

    /**
     * Returns the env or pegasus profile key that needs to be associated
     * for the credential.
     * 
     * @return the name of the environment variable.
     */
    public String getProfileKey( ){
        return S3CFG.S3CFG_FILE_VARIABLE;
    }

    /**
     * Returns the name of the environment variable that needs to be set
     * for the job associated with the credential.
     *
     * @return the name of the environment variable.
     */
    public String getEnvironmentVariable( String site ){
        return S3CFG.S3CFG_FILE_VARIABLE + "_" + this.getSiteNameForEnvironmentKey(site);
    }

    /**
     * Returns the description for the implementing handler
     *
     * @return  description
     */
    public String getDescription(){
        return S3CFG.DESCRIPTION;
    }
}