org.snapscript.agent.ProcessAgentService Maven / Gradle / Ivy

Go to download

package org.snapscript.agent;

import org.snapscript.agent.debug.BreakpointMatcher;
import org.snapscript.agent.event.ProcessEventChannel;
import org.snapscript.core.Model;
import org.snapscript.core.ResourceManager;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.CountDownLatch;

public class ProcessAgentService {
   
   private final Map> breakpoints;
   private final ProcessEventChannel channel;
   private final ProcessResourceExecutor executor;
   private final ProcessContext context;
   
   public ProcessAgentService(ProcessContext context, ProcessEventChannel channel, ProcessResourceExecutor executor, ProcessMode mode, Model model) {
      this.breakpoints = new HashMap>();
      this.executor = executor;
      this.channel = channel;
      this.context = context;
   }

   public String loadScript(String project, String resource) {
      ResourceManager manager = context.getManager();
      String path = ProcessStore.getPath(project, resource);

      return manager.getString(path);
   }
   
   public void createBreakpoint(String resource, int line) {
      Map lines = breakpoints.get(resource);
      BreakpointMatcher matcher = context.getMatcher();
      
      if(lines == null) {
         lines = new HashMap();
         breakpoints.put(resource, lines);
      }
      lines.put(line, Boolean.TRUE);
      matcher.update(breakpoints);
   }
   
   public void removeBreakpoint(String resource, int line){
      Map lines = breakpoints.get(resource);
      BreakpointMatcher matcher = context.getMatcher();
      
      if(lines == null) {
         lines = new HashMap();
         breakpoints.put(resource, lines);
      }
      lines.put(line, Boolean.FALSE);
      matcher.update(breakpoints);
   }
   
   public void execute(String project, String resource) {
      BreakpointMatcher matcher = context.getMatcher();
      ProcessStore store = context.getStore();
      String actual = context.getProcess();

      matcher.update(breakpoints);
      store.update(project); 
      executor.execute(channel, actual, project, resource);
   }

   public boolean join(long time) {
      CountDownLatch latch = context.getLatch();

      try {
         latch.await();
      }catch(Exception e) {
         return false;
      }
      return true;
   }
}