org.sikuliserver.ExecuteSikuliAction.java Source code

Introduction

Here is the source code for org.sikuliserver.ExecuteSikuliAction.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package org.sikuliserver;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.PrintStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.io.FileUtils;
import org.json.JSONException;
import org.json.JSONObject;
import org.sikuli.script.FindFailed;

/**
 *
 * @author bcivel
 */
public class ExecuteSikuliAction extends HttpServlet {

    /**
     * Processes requests for both HTTP <code>GET</code> and <code>POST</code>
     * methods.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    protected void processRequest(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {

        /**
         * Check if picture folder exists to store the picture
         * If, not create it.
         */
        File dir = new File("picture");

        if (!dir.exists()) {
            dir.mkdir();
        } else {
            FileUtils.cleanDirectory(new File("picture"));
        }

        String pictureName = new SimpleDateFormat("YYYY.MM.dd.HH.mm.ss.SSS").format(new Date()) + ".";

        PrintStream os = null;
        try {
            System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date())
                    + " INFO - Received: [Request from " + request.getServerName() + "]");

            BufferedReader is = new BufferedReader(new InputStreamReader(request.getInputStream()));
            os = new PrintStream(response.getOutputStream());
            String line = "";

            StringBuilder sb = new StringBuilder();
            while (!(line = is.readLine()).equals("|ENDS|")) {
                sb.append(line);
            }

            JSONObject obj = new JSONObject(sb.toString());

            String action = obj.getString("action");
            String picture = obj.getString("picture");
            String text = obj.getString("text");
            int defaultWait = obj.getInt("defaultWait");
            String extension = obj.getString("pictureExtension");
            String start = new SimpleDateFormat("HH:mm:ss.SSS").format(new Date());

            long start_time = System.currentTimeMillis();
            long end_time = start_time + defaultWait;
            System.out.println(defaultWait);
            byte[] data = Base64.decodeBase64(picture);

            pictureName += extension;
            String picturePath = "picture" + File.separator + pictureName;

            try (OutputStream stream = new FileOutputStream(picturePath)) {
                stream.write(data);
            }

            System.out.println(start + " INFO - Executing: [" + action + ": on picture " + picturePath + "]");

            int actionResult = 0;
            SikuliAction sikuliAction = new SikuliAction();

            boolean actionSuccess = false;
            while (System.currentTimeMillis() < end_time) {
                try {
                    actionResult = sikuliAction.doAction(action, picturePath, text);
                    if (actionResult == 1) {
                        actionSuccess = true;
                        break;
                    }
                } catch (FindFailed ex) {
                    System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date())
                            + " INFO - Element Not Found : " + ex);
                    System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date())
                            + " INFO - Retrying again during " + (System.currentTimeMillis() - end_time) + "ms");
                }
                System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + " INFO - " + action
                        + ": Still Waiting " + (System.currentTimeMillis() - end_time) + "ms");
            }
            if (!actionSuccess) {
                System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date())
                        + " INFO - Element Not Found : " + picturePath);
                os.println("Failed");
                os.println("|ENDR|");

            }

            String end = new SimpleDateFormat("HH:mm:ss.SSS").format(new Date());
            System.out.println(end + " INFO - Done [" + action + "] with result:" + actionResult);
            os.println(actionResult);
            os.println("|ENDR|");
        } catch (IOException e) {
            System.out.println(new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + " IOException : " + e);
            if (os != null) {
                os.println("Failed");
                os.println("|ENDR|");
            }
        } catch (JSONException ex) {
            System.out.println(
                    new SimpleDateFormat("HH:mm:ss.SSS").format(new Date()) + " JSON Badly formated : " + ex);
            if (os != null) {
                os.println("Failed");
                os.println("|ENDR|");
            }
        }
    }

    // <editor-fold defaultstate="collapsed" desc="HttpServlet methods. Click on the + sign on the left to edit the code.">
    /**
     * Handles the HTTP <code>GET</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Handles the HTTP <code>POST</code> method.
     *
     * @param request servlet request
     * @param response servlet response
     * @throws ServletException if a servlet-specific error occurs
     * @throws IOException if an I/O error occurs
     */
    @Override
    protected void doPost(HttpServletRequest request, HttpServletResponse response)
            throws ServletException, IOException {
        processRequest(request, response);
    }

    /**
     * Returns a short description of the servlet.
     *
     * @return a String containing servlet description
     */
    @Override
    public String getServletInfo() {
        return "Short description";
    }// </editor-fold>
}