Watij 修改Watij源码 解决弹窗问题 二

protected static Wnd waitDialogWndEx(final Wnd startWnd, final String windowClassName, final String[] windowText) throws Exception {
     class WindowReady implements Ready {
         Wnd wnd;
         
         public boolean isReady() throws Exception {
         int i = 0;
         while (i < windowText.length){
         wnd = Wnd.findWindowEx(startWnd, windowClassName, windowText[i++]);
         if (wnd.isWindow())
         break;
         }
             debugWnd(wnd, "findByStartWind");
             return wnd.isWindow();
         }

         public String getNotReadyReason() {
         StringBuffer reason = new StringBuffer();
         reason.append("Could not find Dialog with windowClassName=" + windowClassName + " windowText=");
         reason.append(windowText[0]);
         for (int i = 1; i < windowText.length; i++){
               reason.append("," + windowText[i]);
         }
         return reason.toString();
         }

         public Wnd wnd() {
             return wnd;
         }
     }
     WindowReady windowReady = new WindowReady();
     new WaiterImpl(10000, 200).waitUntil(windowReady);
     Wnd wnd = windowReady.wnd();
     return wnd;
 }

import watij.runtime.ie.IE;
import watij.runtime.ie.IEConfirmDialog;
import watij.time.TimeException;
import watij.time.Timer;
import watij.time.TimerImpl;
import com.jniwrapper.win32.ui.Wnd;

public class HandleDialog {
static final int TIMEOUT_POPUP = 20000;
static final int WAIT_INCREMENT = 200;
static final String DIALOG_CLASSNAME = "#32770";
static final String IE_CLASSNAME = "IEFrame";
static final String[] windowTitle = {"Microsoft Internet Explorer","Windows Internet Explorer",
"Internet Explorer","Explorer 用户提示","添加到收藏夹","文件下载","File Download","安全警报","Security Alert"};
private IE ie;

public static void main(String[] args) {
this.ie = new IE();
this.ie.visible(false);

// 新建线程，处理弹出的对话框
handlePopupDialog(this.ie);
// 加载网页
this.ie.start("http://www.example.com");

}
  
  // handle the popup Dialog
  private void handlePopupDialog(final IE ie){
     // 新建线程守护IE，观察IE发生的事件
     new Thread( new Runnable(){
public void run(){
try{
// 该线程等待IE出现
ie.waitUntilExists();
if (ie.exists()){
class DialogReady implements MyReady {
Wnd wnd = null;
boolean ret = false;
public boolean isReady() throws Exception {
return ret; // 如果ret为true，结束等待；否则继续等待
     }
     public String getNotReadyReason() {
       return "there is no popup dialog";
     }
     public boolean actionWhenWait() throws Exception {
       ret = false;
       // 每次等待时都要判断IE是否还存在，因为IE有可能中途被关掉
if(ie.exists()){
int i = 0;
while (i < windowTitle.length){
wnd = Wnd.findWindow(DIALOG_CLASSNAME,windowTitle[i++]);
// 如果IE弹出对话框
if (wnd.isWindow()){
IEConfirmDialog confirm = new IEConfirmDialog(wnd,ie);
String dialogTitle = confirm.title();
// 如果弹窗是“文件下载”对话框，则关掉对话框；如果关掉对话框后IE为空页面，就关掉IE
if (dialogTitle.contains("文件下载") || dialogTitle.contains("File Download")){
//      confirm.cancel();
     confirm.quit();
     System.out.println(Thread.currentThread().getName()
     + "--->关掉“文件下载”对话框。");
     if (ie.getWebBrowser().getLocationURL().equalsIgnoreCase(Blank_URL)){
     ie.close();
     System.out.println(Thread.currentThread().getName()
     + "--->页面为空，关掉IE。");
     }
     } else if (dialogTitle.contains("安全警报") || dialogTitle.contains("Security Alert")){
//      confirm.cancel();
     confirm.ok();
     System.out.println(Thread.currentThread().getName()
     + "--->发生安全警报，点“确定”继续。");
     } else{ // 如果是其他对话框，则直接关掉弹窗
//      confirm.ok();
     confirm.quit();
     System.out.println(Thread.currentThread().getName()
     + "--->关掉对话框。");
     }
ret = true; // 处理完弹窗，设置“结束等待”标志
}
}
}else{
ret = true; // IE已不存在，设置“结束等待”标志
}
return ret;
}
     }
// 设置等待时间
final WaitReady waitReady = new WaitReady(TIMEOUT_POPUP, WAIT_INCREMENT);
waitReady.waitUntil( new DialogReady() );
}

}catch(Exception e){
System.println(e + getClass().getName()+ "handlePopupDialog()");
}
}
}).start();
  }
     
  public class WaitReady {
long timeout;
long waitIncrement;
     Timer timer;
     public WaitReady(long paramTimeout, long paramWaitIncrement) {
     timeout = paramTimeout;
     waitIncrement = paramWaitIncrement;
     }
     public boolean waitUntil(MyReady ready) throws Exception {
         timer = new TimerImpl();
         while (!ready.isReady()) {
             if (timer.isElapsed(timeout)) {
             System.out.println("Timeout exceeded.  Reason= " + ready.getNotReadyReason());
             actionAfterTimeout();
             return false;
             }
             try {
//              controller.handleOutput("  线程：" + Thread.currentThread().getName() + "正在等待");
             Thread.sleep(waitIncrement);
             ready.actionWhenWait();
             } catch (InterruptedException e) {
             throw new TimeException(e);
             }
         }
         return true;
     }
     public void actionAfterTimeout() throws Exception {
     
     }
     public Timer getTimer(){
     return timer;
     }
}

  public interface MyReady {
     boolean isReady() throws Exception;

     String getNotReadyReason();

     boolean actionWhenWait() throws Exception;
  }
  
}