From 254ee6f47eebfc00462c10756a92066e82cc1a96 Mon Sep 17 00:00:00 2001
From: Andrew Branson <andrew@andrewbranson.net>
Date: Tue, 21 Jun 2011 15:46:02 +0200
Subject: Initial commit

---
 .../com/c2kernel/entity/transfer/TransferItem.java | 131 +++++++++++++++++++++
 1 file changed, 131 insertions(+)
 create mode 100755 source/com/c2kernel/entity/transfer/TransferItem.java

(limited to 'source/com/c2kernel/entity/transfer/TransferItem.java')

diff --git a/source/com/c2kernel/entity/transfer/TransferItem.java b/source/com/c2kernel/entity/transfer/TransferItem.java
new file mode 100755
index 0000000..a869741
--- /dev/null
+++ b/source/com/c2kernel/entity/transfer/TransferItem.java
@@ -0,0 +1,131 @@
+package com.c2kernel.entity.transfer;
+
+import java.io.File;
+import java.util.*;
+
+import com.c2kernel.common.ObjectNotFoundException;
+import com.c2kernel.entity.*;
+import com.c2kernel.lifecycle.instance.Workflow;
+import com.c2kernel.lookup.*;
+import com.c2kernel.persistency.ClusterStorage;
+import com.c2kernel.persistency.outcome.Outcome;
+import com.c2kernel.process.Gateway;
+import com.c2kernel.property.*;
+import com.c2kernel.utils.*;
+
+public class TransferItem {
+    public ArrayList 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();
+        Property name = (Property)Gateway.getStorage().get(sysKey, ClusterStorage.PROPERTY + "/Name", null);
+        Enumeration 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 (int i = 0; i < contents.length; i++) {
+                exportItem(new File(dir, contents[i]), path + "/" + contents[i]);
+            }
+        } 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) {
+        }
+
+        ArrayList events, outcomes, viewpoints = new ArrayList();
+        // retrieve objects
+        ArrayList objectFiles = FileStringUtility.listDir(dir.getCanonicalPath(), false, true);
+        ArrayList objects = new ArrayList();
+        for (Iterator iter = objectFiles.iterator(); iter.hasNext();) {
+            String element = (String)iter.next();
+            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 (Iterator iter = objects.iterator(); iter.hasNext();) {
+            C2KLocalObject obj = (C2KLocalObject)iter.next();
+            if (obj instanceof Property)
+                props.list.add(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 (Iterator iter = domainPaths.iterator(); iter.hasNext();) {
+            String element = (String)iter.next();
+            DomainPath newPath = new DomainPath(element, entityPath);
+            Gateway.getLDAPLookup().add(newPath);
+        }
+    }
+    
+    private void importByType(String type, ArrayList objects) throws Exception {
+        for (Iterator iter = objects.iterator(); iter.hasNext();) {
+            C2KLocalObject element = (C2KLocalObject)iter.next();
+            if (element.getClusterType().equals(type))
+                Gateway.getStorage().put(sysKey, element, this);
+        }
+        
+    }
+}
\ No newline at end of file
-- 
cgit v1.2.3