/**
 *
 * Copyright 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;

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

import javax.xml.registry.BulkResponse;
import javax.xml.registry.BusinessLifeCycleManager;
import javax.xml.registry.JAXRException;
import javax.xml.registry.infomodel.ClassificationScheme;
import javax.xml.registry.infomodel.Concept;
import javax.xml.registry.infomodel.Key;
import javax.xml.registry.infomodel.Organization;
import javax.xml.registry.infomodel.Service;
import javax.xml.registry.infomodel.ServiceBinding;

/**
 * Remove Registry Objects
 * 
 */
public class Remover extends BaseTestCase
{
    BusinessLifeCycleManager blm = null;
    
    public Remover(BusinessLifeCycleManager blm) {
        super();
        this.blm = blm;
    }
    
    public void removeOrganization(Organization org) throws JAXRException {

    	Key key = org.getKey();
    	
    	String id = key.getId();
    	System.out.println("Deleting organization with id " + id);
    	Collection<Key> keys = new ArrayList<Key>();
    	keys.add(key);
    	BulkResponse response = blm.deleteOrganizations(keys);
    	Collection exceptions = response.getExceptions();
    	if (exceptions == null) {
    		System.out.println("Organization deleted");
    		Collection retKeys = response.getCollection();
    		Iterator keyIter = retKeys.iterator();
    		javax.xml.registry.infomodel.Key orgKey = null;
    		if (keyIter.hasNext()) {
    			orgKey = 
    				(javax.xml.registry.infomodel.Key) keyIter.next();
    			id = orgKey.getId();
    			System.out.println("Organization key was " + id);
    		}
    	}
    }
    
    public void removeClassificationScheme(ClassificationScheme cs)
    throws JAXRException
    {
        Key key = cs.getKey();
        String id = key.getId();

        System.out.println("Deleting concept with id " + id);

        Collection<Key> keys = new ArrayList<Key>();
        keys.add(key);
        BulkResponse response = blm.deleteClassificationSchemes(keys);
        
        Collection exceptions = response.getExceptions();
        if (exceptions == null) {
            Collection retKeys = response.getCollection();
            Iterator keyIter = retKeys.iterator();
            javax.xml.registry.infomodel.Key orgKey = null;
            if (keyIter.hasNext()) {
                orgKey = 
                    (javax.xml.registry.infomodel.Key) keyIter.next();
                id = orgKey.getId();
                System.out.println("Concept with ID=" + id + " was deleted");
            }
        }
    }
    
    public void removeConcept(Concept concept)
    throws JAXRException
    {
        Key key = concept.getKey();
        String id = key.getId();

        System.out.println("Deleting concept with id " + id);

        Collection<Key> keys = new ArrayList<Key>();
        keys.add(key);
        BulkResponse response = blm.deleteConcepts(keys);
        
        Collection exceptions = response.getExceptions();
        if (exceptions == null) {
            Collection retKeys = response.getCollection();
            Iterator keyIter = retKeys.iterator();
            javax.xml.registry.infomodel.Key orgKey = null;
            if (keyIter.hasNext()) {
                orgKey = 
                    (javax.xml.registry.infomodel.Key) keyIter.next();
                id = orgKey.getId();
                System.out.println("Concept with ID=" + id + " was deleted");
            }
        }
    }
    
    public void removeService(Service service)
    throws JAXRException
    {
        Key key = service.getKey();
        String id = key.getId();

        System.out.println("Deleting service with id " + id);

        Collection<Key> keys = new ArrayList<Key>();
        keys.add(key);
        BulkResponse response = blm.deleteServices(keys);
        
        Collection exceptions = response.getExceptions();
        if (exceptions == null) {
            Collection retKeys = response.getCollection();
            Iterator keyIter = retKeys.iterator();
            javax.xml.registry.infomodel.Key orgKey = null;
            if (keyIter.hasNext()) {
                orgKey = 
                    (javax.xml.registry.infomodel.Key) keyIter.next();
                id = orgKey.getId();
                System.out.println("Service with ID=" + id + " was deleted");
            }
        }
    }
    
    public void removeServiceBinding(ServiceBinding serviceBinding)
    throws JAXRException
    {
        Key key = serviceBinding.getKey();
        String id = key.getId();

        System.out.println("Deleting serviceBinding with id " + id);

        Collection<Key> keys = new ArrayList<Key>();
        keys.add(key);
        BulkResponse response = blm.deleteServiceBindings(keys);
        
        Collection exceptions = response.getExceptions();
        if (exceptions == null) {
            Collection retKeys = response.getCollection();
            Iterator keyIter = retKeys.iterator();
            javax.xml.registry.infomodel.Key orgKey = null;
            if (keyIter.hasNext()) {
                orgKey = 
                    (javax.xml.registry.infomodel.Key) keyIter.next();
                id = orgKey.getId();
                System.out.println("ServiceBinding with ID=" + id + " was deleted");
            }
        }
    }

 

}