/*
 * JBoss, Home of Professional Open Source
 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags.
 * See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 * This copyrighted material is made available to anyone wishing to use,
 * modify, copy, or redistribute it subject to the terms and conditions
 * of the GNU Lesser General Public License, v. 2.1.
 * This program is distributed in the hope that it will be useful, but WITHOUT A
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License,
 * v.2.1 along with this distribution; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA  02110-1301, USA.
 *
 * (C) 2008,
 * @author JBoss Inc.
 */
package org.jboss.test.jbossts.taskdefs;

import com.arjuna.ats.arjuna.objectstore.ObjectStore;
import com.arjuna.ats.arjuna.common.Uid;
import com.arjuna.ats.arjuna.gandiva.ClassName;
import com.arjuna.ats.arjuna.exceptions.ObjectStoreException;
import com.arjuna.ats.arjuna.state.InputObjectState;
import com.arjuna.ats.arjuna.AtomicAction;

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

/**
 * Utility class for interaction with a transaction log
 */
public class TransactionLog
{
   /**
    * Default object type for store operations
    */
   public static final String DEFAULT_OBJECT_TYPE = "StateManager/BasicAction/TwoPhaseCoordinator/AtomicAction";

   private ObjectStore store;

   public TransactionLog(String storeDir, String impleType)
   {
      init(storeDir, impleType);
   }

   private void init(String storeDir, String impleType)
   {
      System.setProperty("com.arjuna.ats.arjuna.objectstore.objectStoreDir", storeDir);

      if (impleType != null)
         store = new ObjectStore(new ClassName(impleType), (String) null);
      else
         store = new ObjectStore();
   }

   /**
    * Remove any committed objects from the storer
    * @param objectType the type of objects that should be removed
    * @return the number of objects that were purged
    * @throws ObjectStoreException the store implementation was unable to remove a committed object
    */
   public int clearXids(String objectType) throws ObjectStoreException
   {
      Collection<Uid> uids = getIds(objectType);

      for (Uid uid : uids)
         store.remove_committed(uid, objectType);

      return uids.size();
   }

   public Collection<Uid> getIds(String objectType) throws ObjectStoreException
   {
      return getIds(null, objectType);
   }

   /**
    * Get a list object ids for a given object type
    *
    * @param ids holder for the returned uids
    * @param objectType The type of object to search in the store for
    * @return all objects of the given type
    * @throws ObjectStoreException the store implementation was unable retrieve all types of objects
    */
   public Collection<Uid> getIds(Collection<Uid> ids, String objectType) throws ObjectStoreException
   {
      if (ids == null)
         ids = new ArrayList<Uid> ();


      InputObjectState types = new InputObjectState();

      if (store.allTypes(types))
      {
         String theName;

         try
         {
            boolean endOfList = false;

            while (!endOfList)
            {
               theName = types.unpackString();

               if (theName.compareTo("") == 0)
                  endOfList = true;
               else
               {
                  if (objectType != null && !theName.equals(objectType))
                     continue;

                  InputObjectState uids = new InputObjectState();

                  if (store.allObjUids(theName, uids))
                  {
                     try
                     {
                        boolean endOfUids = false;

                        while (!endOfUids)
                        {
                           Uid theUid = new Uid(Uid.nullUid());
                           theUid.unpack(uids);

                           if (theUid.equals(Uid.nullUid()))
                              endOfUids = true;
                           else
                              ids.add(theUid);
                        }
                     }
                     catch (Exception e)
                     {
                        // end of uids!
                     }
                  }

                  System.out.println();
               }
            }
         }
         catch (Exception e)
         {
            System.err.println(e);

            // end of list!
         }
      }

      return ids;
   }

   public int getStatus(Uid uid)
   {
      AtomicAction action = new AtomicAction(uid);

      action.activate();

      return action.status();
   }
}