/*

  * 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");

     }

 }