1 files changed, 78 insertions, 0 deletions
diff --git a/src/main/java/com/c2kernel/entity/AgentImplementation.java b/src/main/java/com/c2kernel/entity/AgentImplementation.java
new file mode 100644
index 0000000..8010114
--- /dev/null
+++ b/src/main/java/com/c2kernel/entity/AgentImplementation.java
@@ -0,0 +1,78 @@
+package com.c2kernel.entity;
+
+import com.c2kernel.common.CannotManageException;
+import com.c2kernel.common.ObjectCannotBeUpdated;
+import com.c2kernel.common.ObjectNotFoundException;
+import com.c2kernel.entity.agent.Job;
+import com.c2kernel.entity.agent.JobArrayList;
+import com.c2kernel.entity.agent.JobList;
+import com.c2kernel.lookup.AgentPath;
+import com.c2kernel.lookup.InvalidItemPathException;
+import com.c2kernel.lookup.RolePath;
+import com.c2kernel.process.Gateway;
+import com.c2kernel.utils.Logger;
+
+public class AgentImplementation extends ItemImplementation implements
+		AgentOperations {
+
+	private JobList currentJobs;
+	
+	public AgentImplementation(int systemKey) {
+		super(systemKey);
+	}
+
+	   /**
+     * Called by an activity when it reckons we need to update our joblist for it
+     */
+
+    @Override
+	public synchronized void refreshJobList(int sysKey, String stepPath, String newJobs) {
+        try {
+            JobArrayList newJobList = (JobArrayList)Gateway.getMarshaller().unmarshall(newJobs);
+
+            // get our joblist
+            if (currentJobs == null)
+                currentJobs = new JobList( mSystemKey, null);
+
+            // remove old jobs for this item
+            currentJobs.removeJobsForStep( sysKey, stepPath );
+
+            // merge new jobs in
+            for (Object name : newJobList.list) {
+                Job newJob = (Job)name;
+                Logger.msg(6, "Adding job for "+newJob.getItemSysKey()+"/"+newJob.getStepPath()+":"+newJob.getTransition().getId());
+                currentJobs.addJob(newJob);
+            }
+
+        } catch (Throwable ex) {
+            Logger.error("Could not refresh job list.");
+            Logger.error(ex);
+        }
+
+    }
+
+    @Override
+	public void addRole(String roleName) throws CannotManageException, ObjectNotFoundException {
+        RolePath newRole = Gateway.getLookup().getRolePath(roleName);
+        try {
+            newRole.addAgent(new AgentPath(mSystemKey));
+        } catch (InvalidItemPathException ex) {
+            throw new CannotManageException("Invalid syskey for agent: "+mSystemKey, "");
+        } catch (ObjectCannotBeUpdated ex) {
+            throw new CannotManageException("Could not update role");
+        }
+    }
+
+    @Override
+	public void removeRole(String roleName) throws CannotManageException, ObjectNotFoundException {
+        RolePath rolePath = Gateway.getLookup().getRolePath(roleName);
+        try {
+            rolePath.removeAgent(new AgentPath(mSystemKey));
+        } catch (InvalidItemPathException e) {
+        	throw new CannotManageException("Invalid syskey for agent: "+mSystemKey, "");
+        } catch (ObjectCannotBeUpdated ex) {
+            throw new CannotManageException("Could not update role");
+        }
+    }
+
+}