1 /*
   2  * Copyright (c) 2007, 2017 Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation. Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 package org.jemmy.control;
  26 
  27 
  28 import java.lang.reflect.Constructor;
  29 import java.lang.reflect.InvocationTargetException;
  30 import org.jemmy.env.Environment;
  31 
  32 
  33 /**
  34  * This is an implementation of the {@code Wrapper} which instantiates a wrap
  35  * of a class returned by {@code getWrapClass(Class)}.
  36  * @author shura
  37  */
  38 public abstract class AbstractWrapper implements Wrapper {
  39 
  40     private Environment env;
  41 
  42     public AbstractWrapper(Environment env) {
  43         this.env = env;
  44     }
  45 
  46     public Environment getEnvironment() {
  47         return env;
  48     }
  49 
  50     protected abstract Class<Wrap> getWrapClass(Class controlClass);
  51 
  52     public <T> Wrap<? extends T> wrap(Class<T> controlClass, T control) {
  53         Class cls = control.getClass();
  54         Class<Wrap> wrp;
  55         do {
  56             wrp = getWrapClass(cls);
  57             if (wrp != null) {
  58                 try {
  59                     return doWrap(control, cls, wrp);
  60                 } catch (InstantiationException ex) {
  61                     throw new WrapperException(cls, wrp, ex);
  62                 } catch (IllegalAccessException ex) {
  63                     throw new WrapperException(cls, wrp, ex);
  64                 } catch (IllegalArgumentException ex) {
  65                     throw new WrapperException(cls, wrp, ex);
  66                 } catch (InvocationTargetException ex) {
  67                     throw new WrapperException(cls, wrp, ex);
  68                 } catch (NoSuchMethodException ex) {
  69                     throw new WrapperException(cls, wrp, ex);
  70                 } catch (SecurityException ex) {
  71                     throw new WrapperException(cls, wrp, ex);
  72                 }
  73             }
  74         } while ((cls = cls.getSuperclass()) != null);
  75         throw new WrapperException(control);
  76     }
  77 
  78     protected <T> Wrap<? extends T> doWrap(T control, Class controlClass, Class wrapperClass) throws NoSuchMethodException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
  79         Constructor cns = null;
  80         Class cls = controlClass;
  81         do {
  82             try {
  83                 cns = wrapperClass.getConstructor(Environment.class, cls);
  84             } catch (NoSuchMethodException e) {
  85             }
  86         } while ((cls = cls.getSuperclass()) != null);
  87         if (cns != null) {
  88             return (Wrap<T>) cns.newInstance(new Environment(env), control);
  89         } else {
  90             throw new WrapperException(controlClass, wrapperClass);
  91         }
  92     }
  93 }