org.eclipse.osee.ote.rest.client/src/org/eclipse/osee/ote/rest/client/internal/RunTests.java

package org.eclipse.osee.ote.rest.client.internal;

import java.net.URI;
import java.util.concurrent.Callable;
import java.util.logging.Level;
import javax.ws.rs.client.Entity;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.UriBuilder;
import org.eclipse.osee.framework.logging.OseeLog;
import org.eclipse.osee.jaxrs.client.JaxRsClient;
import org.eclipse.osee.ote.rest.client.Progress;
import org.eclipse.osee.ote.rest.client.ProgressWithCancel;
import org.eclipse.osee.ote.rest.model.OTEJobStatus;
import org.eclipse.osee.ote.rest.model.OTETestRun;

public class RunTests implements ProgressWithCancel, Callable<ProgressWithCancel> {

   private final URI uri;
   private final OTETestRun tests;
   private final Progress progress;
   private final JaxRsClient factory;
   private OTEJobStatus status;
   private String id;

   public RunTests(URI uri, OTETestRun tests, Progress progress, JaxRsClient factory) {
      this.uri = uri;
      this.tests = tests;
      this.progress = progress;
      this.factory = factory;
   }

   public void doWork() throws Exception {
      status = sendCommand();
      id = status.getJobId();
      if (!status.isSuccess()) {
         throw new Exception("Failed to submit the run command: " + status.getErrorLog());
      }
   }

   @Override
   public boolean cancelSingle() {
      URI targetUri = UriBuilder.fromUri(uri).path("ote").path("run").path("{run-id}").build(id);
      try {
         OTEJobStatus cancelStatus =
            factory.target(targetUri).request(MediaType.APPLICATION_JSON).put(Entity.json(""), OTEJobStatus.class);
         return cancelStatus.isSuccess();
      } catch (Exception e) {
         OseeLog.log(getClass(), Level.SEVERE, e);
         return false;
      }
   }

   private OTEJobStatus sendCommand() throws Exception {
      URI targetUri = UriBuilder.fromUri(uri).path("ote").path("run").build();
      return factory.target(targetUri).request(MediaType.APPLICATION_JSON).post(Entity.json(tests), OTEJobStatus.class);
   }

   @Override
   public boolean cancelAll() {
      URI targetUri = UriBuilder.fromUri(uri).path("ote").path("run").path("{run-id}").build(id);

      try {
         OTEJobStatus cancelStatus =
            factory.target(targetUri).request(MediaType.APPLICATION_JSON).delete(OTEJobStatus.class);
         return cancelStatus.isSuccess();
      } catch (Exception e) {
         OseeLog.log(getClass(), Level.SEVERE, e);
         return false;
      }
   }

   @Override
   final public ProgressWithCancel call() throws Exception {
      try {
         doWork();
         progress.success();
      } catch (Throwable th) {
         progress.fail(th);
      }
      return this;
   }

   @Override
   public void fail(String string) {
      progress.fail(string);
   }

   @Override
   public void fail(Throwable th) {
      progress.fail(th);
   }

   @Override
   public void setUnitsOfWork(int totalUnitsOfWork) {
      progress.setUnitsOfWork(totalUnitsOfWork);
   }

   @Override
   public void setUnitsWorked(int unitsWorked) {
      progress.setUnitsWorked(unitsWorked);
   }

   @Override
   public void success() {
      progress.success();
   }

}