src/main/java/org/apache/ws/scout/registry/infomodel/OrganizationImpl.java

/*
 * Copyright 2001-2004 The Apache Software Foundation.
 *
 * 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 org.apache.ws.scout.registry.infomodel;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import javax.xml.registry.JAXRException;
import javax.xml.registry.LifeCycleManager;
import javax.xml.registry.UnsupportedCapabilityException;
import javax.xml.registry.infomodel.Organization;
import javax.xml.registry.infomodel.PostalAddress;
import javax.xml.registry.infomodel.Service;
import javax.xml.registry.infomodel.TelephoneNumber;
import javax.xml.registry.infomodel.User;

/**
 * Organization Interface
 * * Implements JAXR Interface.
 * For futher details, look into the JAXR API Javadoc.
 *
 * @author <a href="mailto:anil@apache.org">Anil Saldhana</a>
 * @author <a href="mailto:geirm@apache.org">Geir Magnusson Jr.</a>
 */
public class OrganizationImpl extends RegistryObjectImpl implements Organization
{
    private User primaryContact;
    private Set<User> users = new HashSet<User>();
    private Set<TelephoneNumber> telephoneNumbers = new HashSet<TelephoneNumber>();
    private Collection<Service> services = new ArrayList<Service>();

    public OrganizationImpl(LifeCycleManager lifeCycleManager)
    {
        super(lifeCycleManager);
    }

    public User getPrimaryContact() throws JAXRException
    {
        //TODO: How do we fix this? Run JAXRQueryTest and you will hit this problem.
        //
        //  gmj : I think I fixed w/ primary contact hack...
        //
        //if (primaryContact == null) {
        //    throw new IllegalStateException("primaryContact is null and the spec says we cannot return a null value");
        //}
        return primaryContact;
    }

    public void setPrimaryContact(User user) throws JAXRException
    {
        if (user == null)
        {
            throw new IllegalArgumentException("primaryContact must not be null");
        }

        /*
         * first check to see if user already exists in user set
         */

        primaryContact = user;

        if (!users.contains(user)) {
            addUser(user);
        }
    }

    public void addUser(User user) throws JAXRException
    {
        doPrimaryContactHack(user);

        users.add(user);
        ((UserImpl) user).setOrganization(this);
    }

    /**
     *
     *  to solve the getPrimaryContactProblem(), if we have no defined
     *  primaryContact, we'll designate the first user as such
     *
     * @param user
     */
    private void doPrimaryContactHack(User user) {

        if (primaryContact == null && users.size() == 0) {
            primaryContact = user;
        }
    }

    public void addUsers(Collection collection) throws JAXRException
    {
        // do this by hand to ensure all members are actually instances of User
        for (Iterator iterator = collection.iterator(); iterator.hasNext();)
        {
            User user = (User) iterator.next();
            addUser(user);
        }
    }

    public Collection<User> getUsers() throws JAXRException
    {
        return users;
    }

    public void removeUser(User user) throws JAXRException
    {
        if (user != null) {
            users.remove(user);
        }

        /*
         * more primaryContact hakiness - nothing says that you can't
         * remove the user that is the PC...
         */

        if (!users.contains(primaryContact)) {
            primaryContact = null;
        }
    }

    public void removeUsers(Collection collection) throws JAXRException
    {
        if (collection != null) {
            users.removeAll(collection);
        }

        /*
         * more primaryContact hakiness - nothing says that you can't
         * remove the user that is the PC...
         */

        if (!users.contains(primaryContact)) {
            primaryContact = null;
        }

    }

    public Collection<TelephoneNumber> getTelephoneNumbers(String phoneType) throws JAXRException
    {
        Set<TelephoneNumber> filteredNumbers;
        if (phoneType == null)
        {
            filteredNumbers = telephoneNumbers;
        } else
        {
            filteredNumbers = new HashSet<TelephoneNumber>(telephoneNumbers.size());
            for (Iterator i = telephoneNumbers.iterator(); i.hasNext();)
            {
                TelephoneNumber number = (TelephoneNumber) i.next();
                if (phoneType.equals(number.getType()))
                {
                    filteredNumbers.add(number);
                }
            }
        }
        return filteredNumbers;
    }

    public void setTelephoneNumbers(Collection collection) throws JAXRException
    {
        // do this by hand to ensure all members are actually instances of TelephoneNumber
        Set<TelephoneNumber> numbers = new HashSet<TelephoneNumber>(collection.size());
        for (Object number : collection) {
            numbers.add((TelephoneNumber) number);
        }
        this.telephoneNumbers = numbers;
    }

    public void addService(Service service) throws JAXRException
    {
        services.add(service);

        /*
         * we need to tell the service who it's organization is so
         * we can set the UDDI Business Key...
         */

        service.setProvidingOrganization(this);
    }

    public void addServices(Collection collection) throws JAXRException
    {
        // do this by hand to ensure all members are actually instances of Service
        for (Iterator iterator = collection.iterator(); iterator.hasNext();)
        {
            Service service = (Service) iterator.next();

            addService(service);
        }
    }

    public Collection<Service> getServices() throws JAXRException
    {
        return services;
    }

    public void removeService(Service service) throws JAXRException
    {
    	services.remove(service);
    }

    public void removeServices(Collection collection) throws JAXRException
    {
        services.removeAll(collection);
    }

    ///////////////////////////////////////////////////////////////////////////
    // Level 1 features must throw exceptions
    ///////////////////////////////////////////////////////////////////////////

    public Organization getParentOrganization() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public Collection<Organization> getDescendantOrganizations() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public Organization getRootOrganization() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public void addChildOrganization(Organization organization) throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public void addChildOrganizations(Collection collection) throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public int getChildOrganizationCount() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public Collection<Organization> getChildOrganizations() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public void removeChildOrganization(Organization organization) throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public void removeChildOrganizations(Collection collection) throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public PostalAddress getPostalAddress() throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }

    public void setPostalAddress(PostalAddress postalAddress) throws JAXRException
    {
        throw new UnsupportedCapabilityException("Level 1 feature");
    }
}