io.sitoolkit.wt.gui.app.script.ScriptService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-app Show documentation
The newest version!
package io.sitoolkit.wt.gui.app.script;

import java.io.File;
import java.nio.charset.Charset;
import java.nio.file.Path;
import java.util.List;
import java.util.Optional;
import java.util.function.Supplier;
import javax.annotation.Resource;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import io.sitoolkit.wt.app.config.ExtConfig;
import io.sitoolkit.wt.app.ope2script.ChromiumOpener;
import io.sitoolkit.wt.app.page2script.Page2Script;
import io.sitoolkit.wt.app.page2script.Page2ScriptConfig;
import io.sitoolkit.wt.app.test.TestCaseReader;
import io.sitoolkit.wt.app.test.TestScriptGenerator;
import io.sitoolkit.wt.domain.operation.OperationConverter;
import io.sitoolkit.wt.domain.testscript.TestScript;
import io.sitoolkit.wt.domain.testscript.TestScriptDao;
import io.sitoolkit.wt.gui.domain.script.CaseNoCache;
import io.sitoolkit.wt.gui.infra.config.PropertyManager;
import io.sitoolkit.wt.util.infra.concurrent.ExecutorContainer;

public class ScriptService {

  CaseNoCache cache = new CaseNoCache();

  @Resource TestScriptDao dao;

  @Resource io.sitoolkit.wt.infra.PropertyManager runtimePm;

  Page2Script page2script;

  ChromiumOpener chromiumOpener = new ChromiumOpener();

  ConfigurableApplicationContext pageCtx;

  @Resource TestCaseReader testCaseReader;

  OperationConverter operationConverter = new OperationConverter();

  @Resource TestScriptGenerator testScriptGenerator;

  public void loadProject() {
    PropertyManager pm = PropertyManager.get();
    runtimePm.setCsvCharset(pm.getCsvCharset());
    runtimePm.setCsvHasBOM(pm.getCsvHasBOM());
  }

  public TestScript read(File file) {
    while (!initialized()) {}
    return dao.load(file, "TestScript", false);
  }

  public void write(TestScript testScript) {
    dao.write(
        testScript.getScriptFile(), testScript.getTestStepList(), testScript.getHeaders(), true);
  }

  private synchronized boolean initialized() {
    return dao != null;
  }

  public void page2script(String driverType, String baseUrl) {
    pageCtx = new AnnotationConfigApplicationContext(Page2ScriptConfig.class, ExtConfig.class);
    page2script = pageCtx.getBean(Page2Script.class);
    page2script.setOpenScript(false);
    page2script.openBrowser(baseUrl, driverType);
  }

  public void ope2script(String baseUrl) {
    ExecutorContainer.get().execute(() -> chromiumOpener.open(baseUrl));
  }

  public void quitOperating() {
    chromiumOpener.close();
  }

  public List readCaseNo(File testScript) {

    List caseNos = cache.getCaseNosIfNotModified(testScript);

    if (caseNos != null) {
      return caseNos;
    }

    List readCaseNos = testCaseReader.getTestCase(testScript.getAbsolutePath());
    cache.putCaesNos(testScript, readCaseNos);

    return readCaseNos;
  }

  public void write(TestScript testScript, Optional scriptFileType) {
    doWithScriptFileType(scriptFileType, () -> write(testScript));
  }

  public TestScript read(File file, Optional scriptFileType) {
    return doWithScriptFileType(scriptFileType, () -> read(file));
  }

  private  T doWithScriptFileType(Optional scriptFileType, Supplier s) {
    Charset charset = runtimePm.getCsvCharset();
    boolean hasBom = runtimePm.isCsvHasBOM();
    scriptFileType.ifPresent(
        ft -> {
          if (ft.isTextFile()) {
            runtimePm.setCsvCharset(ft.getCharset());
            runtimePm.setCsvHasBOM(ft.isHasBom());
          }
        });
    T result = s.get();
    runtimePm.setCsvCharset(charset);
    runtimePm.setCsvHasBOM(hasBom);
    return result;
  }

  private void doWithScriptFileType(Optional scriptFileType, Runnable r) {
    doWithScriptFileType(
        scriptFileType,
        () -> {
          r.run();
          return null;
        });
  }

  public Path export() {
    page2script.generateFromPage();
    return page2script.getCreateFile();
  }

  public void quitBrowsing() {
    pageCtx.close();
  }

  public void generateNewScript(Path destFile) {
    testScriptGenerator.generateNewScript(destFile);
  }
}