Batch, Remove unneeded Classes, refine, etc
[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / reports / bodies / NotifyBody.java
index 429ea6d..ae7daa0 100644 (file)
@@ -29,32 +29,45 @@ import java.net.URISyntaxException;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.Collection;
+import java.util.Enumeration;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.TreeMap;
+import java.util.jar.JarEntry;
+import java.util.jar.JarFile;
 
+import org.onap.aaf.auth.batch.helpers.LastNotified;
 import org.onap.aaf.auth.batch.reports.Notify;
 import org.onap.aaf.auth.env.AuthzTrans;
 import org.onap.aaf.cadi.Access;
 import org.onap.aaf.misc.env.APIException;
 
 public abstract class NotifyBody {
+       private static final String DUPL = "<td style=\"text-indent: 4em;\">''</td>";
        private static final Map<String,NotifyBody> bodyMap = new HashMap<>();
 
        protected Map<String,List<List<String>>> rows;
+       protected final String env;
+       protected final String gui_url;
+       
        private final String name;
        private final String type;
        private String date;
        private int escalation;
+       private int count;
        
-       public NotifyBody(final String type, final String name) {
+       public NotifyBody(Access access, final String type, final String name) {
                rows = new TreeMap<>();
                this.name = name;
                this.type = type;
                date="";
                escalation = 1;
+               count = 0;
+               env = access.getProperty("CASS_ENV","DEVL");
+               gui_url = access.getProperty("GUI_URL", "");
        }
        
        public void store(List<String> row) {
@@ -85,6 +98,10 @@ public abstract class NotifyBody {
                return name;
        }
        
+       public String type() {
+               return type;
+       }
+       
        public String date() {
                return date;
        }
@@ -105,7 +122,7 @@ public abstract class NotifyBody {
         * @param row
         * @return
         */
-       public abstract String body(AuthzTrans trans, Notify n, String id);
+       public abstract boolean body(AuthzTrans trans, StringBuilder sb, int indent, Notify n, String id);
        
        /**
         * Return "null" if user not found in row... Code will handle.
@@ -114,6 +131,22 @@ public abstract class NotifyBody {
         */
        protected abstract String user(List<String> row);
        
+       /**
+        * Provide a context-sensitive Subject, which includes ENV as well as details
+        * 
+        * @return
+        */
+       public abstract String subject();
+
+       /**
+        * Record the fact that a particular Notification was marked as "sent" by Emailer.
+        * 
+        * @param trans
+        * @param approver
+        * @param ln
+        */
+       public abstract void record(AuthzTrans trans, StringBuilder query, String id, List<String> notified, LastNotified ln);
+       
        /**
         * Get Notify Body based on key of
         * type|name
@@ -127,7 +160,11 @@ public abstract class NotifyBody {
         * 
         */
        public static Collection<NotifyBody> getAll() {
-               return bodyMap.values();
+               // Note: The same Notify Body is entered several times with different keys.
+               // Therefore, need a Set of Values, not all the Values.
+               Set<NotifyBody> set = new HashSet<>();
+               set.addAll(bodyMap.values());
+               return set;
        }
        
        /**
@@ -140,46 +177,86 @@ public abstract class NotifyBody {
                ClassLoader cl = Thread.currentThread().getContextClassLoader();
                Package pkg = NotifyBody.class.getPackage();
                String path = pkg.getName().replace('.', '/');
-//             Enumeration<URL> urls = cl.getResources(path);
-//             while(urls.hasMoreElements()) {
-//                     URL url = urls.nextElement();
                URL url = cl.getResource(path);
-               if(url == null) {
-                       throw new APIException("Cannot load resources from " + path);
-               }
-               System.out.println(url);
-               File dir;
-               try {
-                       dir = new File(url.toURI());
-               } catch (URISyntaxException e) {
-                       throw new APIException(e);
+               List<String> classNames = new ArrayList<>();
+               String urlString = url.toString();
+               if(urlString.startsWith("jar:file:")) {
+                       int exclam = urlString.lastIndexOf('!');
+                       JarFile jf = new JarFile(urlString.substring(9,exclam));
+                       try {
+                               Enumeration<JarEntry> jfe = jf.entries();
+                               while(jfe.hasMoreElements()) {
+                                       String name = jfe.nextElement().getName();
+                                       if(name.startsWith(path) && name.endsWith(".class")) {
+                                               classNames.add(name.substring(0,name.length()-6).replace('/', '.'));
+                                       }
+                               }
+                       } finally {
+                               jf.close();
+                       }
+               } else {
+                       File dir = new File(url.getFile());
+                       for( String f : dir.list()) {
+                               if(f.endsWith(".class")) {
+                                       classNames.add(pkg.getName()+'.'+f.substring(0,f.length()-6));
+                               }
+                       }
                }
-               if(dir.exists()) {
-                       String[] files = dir.list();
-                       if(files!=null) {
-                               for(String sf : files) {
-                                       int dot = sf.indexOf('.');
-                                       if(dot>=0) {
-                                               String cls = pkg.getName()+'.'+sf.substring(0,dot);
-                                               try {
-                                                       Class<?> c = cl.loadClass(cls);
-                                                       if(c!=null) {
-                                                               if(!Modifier.isAbstract(c.getModifiers())) {
-                                                                       Constructor<?> cst = c.getConstructor(Access.class);
-                                                                       NotifyBody nb = (NotifyBody)cst.newInstance(access);
-                                                                       if(nb!=null) {
-                                                                               bodyMap.put("info|"+nb.name, nb);
-                                                                               bodyMap.put(nb.type+'|'+nb.name, nb);
-                                                                       }
-                                                               }
-                                                       }
-                                               } catch (ClassNotFoundException | NoSuchMethodException | SecurityException | InstantiationException | IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
-                                                       e.printStackTrace();
+               for(String cls : classNames) {
+                       try {
+                               Class<?> c = cl.loadClass(cls);
+                               if(c!=null) {
+                                       if(!Modifier.isAbstract(c.getModifiers())) {
+                                               Constructor<?> cst = c.getConstructor(Access.class);
+                                               NotifyBody nb = (NotifyBody)cst.newInstance(access);
+                                               if(nb!=null) {
+                                                       bodyMap.put("info|"+nb.name, nb);
+                                                       bodyMap.put(nb.type+'|'+nb.name, nb);
                                                }
                                        }
                                }
+                       } catch (ClassNotFoundException | NoSuchMethodException | SecurityException | InstantiationException | IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
+                               e.printStackTrace();
                        }
                }
-//             }
+       }
+
+       protected void print(StringBuilder sb, int indent, Object ... objs) {
+               for(int i=0;i<indent;++i) {
+                       sb.append(' ');
+               }
+               for(Object o : objs) {
+                       sb.append(o.toString());
+               }
+       }
+                       
+       protected void println(StringBuilder sb, int indent, Object ... objs) {
+               print(sb,indent,objs);
+               sb.append('\n');
+       }
+
+       protected void printf(StringBuilder sb, int indent, String fmt, Object ... objs) {
+               print(sb,indent,String.format(fmt, objs));
+       }
+
+       protected String printCell(StringBuilder sb, int indent, String current, String prev) {
+               if(current.equals(prev)) {
+                       println(sb,indent,DUPL);
+               } else {
+                       printCell(sb,indent,current);
+               }
+               return current; // use to set prev...
+       }
+       
+       protected void printCell(StringBuilder sb, int indent, String current) {
+               println(sb,indent,"<td>",current,"</td>");
+       }
+       
+       public synchronized void inc() {
+               ++count;
+       }
+       
+       public int count() {
+               return count;
        }
 }