package com.c2kernel.entity.transfer;

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

import com.c2kernel.common.ObjectNotFoundException;
import com.c2kernel.entity.C2KLocalObject;
import com.c2kernel.entity.TraceableEntity;
import com.c2kernel.lifecycle.instance.Workflow;
import com.c2kernel.lookup.DomainPath;
import com.c2kernel.lookup.EntityPath;
import com.c2kernel.lookup.Path;
import com.c2kernel.persistency.ClusterStorage;
import com.c2kernel.persistency.outcome.Outcome;
import com.c2kernel.process.Gateway;
import com.c2kernel.property.Property;
import com.c2kernel.property.PropertyArrayList;
import com.c2kernel.utils.CastorXMLUtility;
import com.c2kernel.utils.FileStringUtility;
import com.c2kernel.utils.Logger;

public class TransferItem {
    public ArrayList<String> domainPaths;
    public int sysKey;
    static int importAgentId;

    public TransferItem() throws Exception {
        try {
            importAgentId = Gateway.getLDAPLookup().getRoleManager().getAgentPath("system").getSysKey();
        } catch (ObjectNotFoundException e) {
            Logger.error("TransferItem - System user not found!");
            throw e;
        }
    }

    public TransferItem(int sysKey) throws Exception {
        this.sysKey = sysKey;
        domainPaths = new ArrayList<String>();
        Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null);
        Enumeration<Path> paths = Gateway.getLDAPLookup().search(new DomainPath(), name.getValue());
        while (paths.hasMoreElements()) {
            DomainPath thisPath = (DomainPath)paths.nextElement();
            domainPaths.add(thisPath.toString());
        }
    }

    public void exportItem(File dir, String path) throws Exception {
        Logger.msg("Path " + path + " in " + sysKey);
        String[] contents = Gateway.getStorage().getClusterContents(sysKey, path);
        if (contents.length > 0) {
            FileStringUtility.createNewDir(dir.getCanonicalPath());
            for (String content : contents) {
                exportItem(new File(dir, content), path + "/" + content);
            }
        } else { //no children, try to dump object
            try {
                C2KLocalObject obj = Gateway.getStorage().get(sysKey, path, null);
                Logger.msg("Dumping object " + path + " in " + sysKey);
                File dumpPath = new File(dir.getCanonicalPath() + ".xml");
                FileStringUtility.string2File(dumpPath, CastorXMLUtility.marshall(obj));
                return;
            } catch (ObjectNotFoundException ex) {
            } // not an object
        }
    }

    public void importItem(File dir) throws Exception {
        // check if already exists
        try {
            Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null);
            throw new Exception("Syskey " + sysKey + " already in use as " + name.getValue());
        } catch (Exception ex) {
        }

        // retrieve objects
        ArrayList<String> objectFiles = FileStringUtility.listDir(dir.getCanonicalPath(), false, true);
        ArrayList<C2KLocalObject> objects = new ArrayList<C2KLocalObject>();
        for (String element : objectFiles) {
            String xmlFile = FileStringUtility.file2String(element);
            C2KLocalObject newObj;
            String choppedPath = element.substring(dir.getCanonicalPath().length()+1, element.length()-4);
            Logger.msg(choppedPath);
            if (choppedPath.startsWith(ClusterStorage.OUTCOME))
                newObj = new Outcome(choppedPath, xmlFile);
            else
                newObj = (C2KLocalObject)CastorXMLUtility.unmarshall(xmlFile);

            objects.add(newObj);
        }

        // create item
        EntityPath entityPath = new EntityPath(sysKey);
        TraceableEntity newItem = (TraceableEntity)Gateway.getCorbaServer().createEntity(entityPath);
        Gateway.getLDAPLookup().add(entityPath);

        PropertyArrayList props = new PropertyArrayList();
        Workflow wf = null;
        // put objects
        for (C2KLocalObject obj : objects) {
            if (obj instanceof Property)
                props.list.add((Property)obj);
            else if (obj instanceof Workflow)
                wf = (Workflow)obj;
        }

        if (wf == null)
            throw new Exception("No workflow found in import for "+sysKey);

        // init item
        newItem.initialise(importAgentId, CastorXMLUtility.marshall(props), CastorXMLUtility.marshall(wf.search("workflow/domain")));

        // store objects
        importByType(ClusterStorage.COLLECTION, objects);
        importByType(ClusterStorage.HISTORY, objects);
        importByType(ClusterStorage.OUTCOME, objects);
        importByType(ClusterStorage.VIEWPOINT, objects);
        Gateway.getStorage().commit(this);
        // add domPaths
        for (String element : domainPaths) {
            DomainPath newPath = new DomainPath(element, entityPath);
            Gateway.getLDAPLookup().add(newPath);
        }
    }

    private void importByType(String type, ArrayList<C2KLocalObject> objects) throws Exception {
        for (C2KLocalObject element : objects) {
            if (element.getClusterType().equals(type))
                Gateway.getStorage().put(sysKey, element, this);
        }

    }
}