### $Id: get_slices.py 14387 2009-07-08 18:19:11Z faiyaza $
### $URL: https://svn.planet-lab.org/svn/sfa/trunk/sfa/methods/get_aggregates.py $
from types import StringTypes
from sfa.util.faults import *
from sfa.util.namespace import *
from sfa.util.method import Method
from sfa.util.parameter import Parameter, Mixed
from sfa.trust.auth import Auth
from sfa.server.aggregate import Aggregates

class get_geni_aggregates(Method):
    """
    Get a list of connection information for all known GENI aggregates.      

    @param cred credential string specifying the rights of the caller
    @param a Human readable name (hrn or urn), or list of hrns or None
    @return list of dictionaries with aggregate information.  
    """

    interfaces = ['registry', 'aggregate', 'slicemgr']
    
    accepts = [
        Parameter(str, "Credential string"),        
        Mixed(Parameter(str, "Human readable name (hrn or urn)"),
              Parameter(None, "hrn not specified"))
        ]
    

    returns = [Parameter(dict, "Aggregate interface information")]
    
    def call(self, cred, xrn = None):
        hrn, type = urn_to_hrn(xrn)
        self.api.auth.check(cred, 'list')
        
        geni_aggs = Aggregates(self.api, '/etc/sfa/geni_aggregates.xml').interfaces
        geni_aggs[self.api.hrn]['port'] = 12348
        geni_aggs[self.api.hrn]['urn'] = 'http://%s:12348' % geni_aggs[self.api.hrn]['addr']
        geni_aggs = geni_aggs.values()



        hrn_list = [] 
        if hrn:
            if isinstance(hrn, StringTypes):
                hrn_list = [hrn]
            elif isinstance(hrn, list):
                hrn_list = hrn
        
        if not hrn_list:
            interfaces = geni_aggs
        else:
            interfaces = [interface for interface in geni_aggs if interface['hrn'] in hrn_list]

        # Add urns
        for interface in interfaces:
            interface['urn'] = hrn_to_urn(interface['hrn'], 'authority')

        return interfaces