package com.c2kernel.entity.transfer;

import java.io.File;
import java.util.ArrayList;

import com.c2kernel.lookup.EntityPath;
import com.c2kernel.lookup.ldap.NextKeyManager;
import com.c2kernel.process.Gateway;
import com.c2kernel.utils.FileStringUtility;
import com.c2kernel.utils.Logger;

/**************************************************************************
 *
 * $Revision: 1.5 $
 * $Date: 2005/04/26 06:48:13 $
 *
 * Copyright (C) 2003 CERN - European Organization for Nuclear Research
 * All rights reserved.
 **************************************************************************/

public class TransferSet {

    public ArrayList<TransferItem> items;

    public TransferSet() {
    }

    public TransferSet(int[] sysKeys) {
        items = new ArrayList<TransferItem>();
        for (int sysKey : sysKeys) {
            try {
                items.add(new TransferItem(sysKey));
            } catch (Exception ex) {
                Logger.error("Could not add item "+sysKey);
                Logger.error(ex);
            }
        }
    }

    public void exportPackage(File dir) throws Exception {
        if (items==null || items.size() == 0)
            throw new Exception("Nothing to dump");
        FileStringUtility.createNewDir(dir.getAbsolutePath());
        for (TransferItem element : items) {
            try {
                element.exportItem(new File(dir, String.valueOf(element.sysKey)), "/");
            } catch (Exception ex) {
                Logger.error("Error dumping item "+element.sysKey);
                Logger.error(ex);
            }
        }

        try {
            String self = Gateway.getMarshaller().marshall(this);
            FileStringUtility.string2File(new File(dir, "transferSet.xml"), self);
        } catch (Exception ex) {
            Logger.error("Error writing header file");
            Logger.error(ex);
        }
    }

    public void importPackage(File rootDir) {
        for (TransferItem element : items) {
            Logger.msg(5, "Importing "+element.sysKey);
            try {
                element.importItem(new File(rootDir, String.valueOf(element.sysKey)));
            } catch (Exception ex) {
                Logger.error("Import of item "+element.sysKey+" failed. Rolling back");
                Logger.error(ex);
                Gateway.getStorage().abort(element);
            }
        }
        checkLastKey();
    }

    private void checkLastKey()
    {
        // find highest key in out import set
        int packageLastKey = 0;
        for (TransferItem element : items) {
            if (element.sysKey > packageLastKey)
                packageLastKey = element.sysKey;
        }

        try
        {   // find the current last key
            NextKeyManager nextKeyMan = Gateway.getLookup().getNextKeyManager();
            EntityPath lastKey = nextKeyMan.getLastEntityPath();
            Logger.msg(1, "Last key imported was "+packageLastKey+". LDAP lastkey was "+lastKey.getSysKey());


            if (packageLastKey > lastKey.getSysKey()) { // set new last
                Logger.msg(1, "Updating lastKey to "+packageLastKey);
                nextKeyMan.writeLastEntityKey(packageLastKey);
            }
        }
        catch (Exception ex)
        {
            Logger.error("Exception::LoadKeys::processFile() " + ex);
        }
    }
}