X-Git-Url: https://gerrit.onap.org/r/gitweb?p=aaf%2Fauthz.git;a=blobdiff_plain;f=auth%2Fauth-batch%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Faaf%2Fauth%2Fbatch%2Freports%2FNotify.java;h=189857c9628c05749589a8cdad654f1e55521388;hp=7fd26747cea0f8da82c95bf807b58449ab1e7d21;hb=628b7105ce4d9818aac69a082e515f9275fd46fd;hpb=4b6435c97dc1e05ee6a1e06190e3c431f9d1a8fe

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/reports/Notify.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/reports/Notify.java
index 7fd26747..189857c9 100644
--- a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/reports/Notify.java
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/reports/Notify.java
@@ -19,220 +19,230 @@
  *
  */package org.onap.aaf.auth.batch.reports;
 
-import java.io.BufferedReader;
-import java.io.File;
-import java.io.FileReader;
-import java.io.IOException;
-import java.lang.reflect.Constructor;
-import java.lang.reflect.InvocationTargetException;
-import java.util.ArrayList;
-import java.util.HashSet;
-import java.util.List;
-import java.util.Set;
-
-import org.onap.aaf.auth.batch.Batch;
-import org.onap.aaf.auth.batch.reports.bodies.NotifyBody;
-import org.onap.aaf.auth.env.AuthzTrans;
-import org.onap.aaf.auth.org.Mailer;
-import org.onap.aaf.auth.org.Organization.Identity;
-import org.onap.aaf.auth.org.OrganizationException;
-import org.onap.aaf.cadi.Access;
-import org.onap.aaf.cadi.CadiException;
-import org.onap.aaf.cadi.client.Holder;
-import org.onap.aaf.cadi.util.CSV;
-import org.onap.aaf.misc.env.APIException;
-import org.onap.aaf.misc.env.util.Chrono;
-
-public class Notify extends Batch {
-	private static final String HTML_CSS = "HTML_CSS";
-	private final Mailer mailer;
-	private final String header;
-	private final String footer;
-	private List<File> notifyFile;
-	public final String guiURL;
-	private int maxEmails;
-	private int indent;
-
-	public Notify(AuthzTrans trans) throws APIException, IOException, OrganizationException {
-		super(trans.env());
-		String mailerCls = env.getProperty("MAILER");
-		String mailFrom = env.getProperty("MAIL_FROM");
-		String header_html = env.getProperty("HEADER_HTML");
-		String footer_html = env.getProperty("FOOTER_HTML");
-		String maxEmails = env.getProperty("MAX_EMAIL");
-		guiURL = env.getProperty("GUI_URL");
-		this.maxEmails = maxEmails==null?1:Integer.parseInt(maxEmails);
-		if(mailerCls==null || mailFrom==null || guiURL==null || header_html==null || footer_html==null) {
-			throw new APIException("Notify requires MAILER, MAILER_FROM, GUI_URL, HEADER_HTML and FOOTER_HTML properties");
-		}
-		try {
-			Class<?> mailc = Class.forName(mailerCls);
-			Constructor<?> mailcst = mailc.getConstructor(Access.class);
-			mailer = (Mailer)mailcst.newInstance(env.access());
-		} catch (ClassNotFoundException | NoSuchMethodException | SecurityException | InstantiationException | IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
-			throw new APIException("Unable to construct " + mailerCls,e);
-		}
-		
-		String line;
-		StringBuilder sb = new StringBuilder();
-		BufferedReader br = new BufferedReader(new FileReader(header_html));
-		try {
-			while((line=br.readLine())!=null) {
-				sb.append(line);
-				sb.append('\n');
-			}
-			String html_css = env.getProperty(HTML_CSS);
-			int hc = sb.indexOf(HTML_CSS);
-			if(hc!=0 && html_css!=null) {
-				header = sb.replace(hc,hc+HTML_CSS.length(), html_css).toString();
-			} else {
-				header = sb.toString();
-			}
-		} finally {
-			br.close();
-		}
-		
-		
-		
-		// Establish index from header
-		int lastTag = header.lastIndexOf('<');
-		if(lastTag>0) {
-			int prevCR = header.lastIndexOf('\n',lastTag);
-			if(prevCR>0) {
-				indent = lastTag-prevCR;
-			} else {
-				indent = 6; //arbitrary
-			}
-		}
-
-		
-		sb.setLength(0);
-		br = new BufferedReader(new FileReader(footer_html));
-		try {
-			while((line=br.readLine())!=null) {
-				sb.append(line);
-				sb.append('\n');
-			}
-			footer = sb.toString();
-		} finally {
-			br.close();
-		}
-	
-		// Class Load possible data
-		NotifyBody.load(env.access());
-		
-        // Create Intermediate Output 
-        File logDir = logDir();
-        notifyFile = new ArrayList<>();
-        if(args().length>0) {
-        	for(int i=0;i<args().length;++i) {
-        		notifyFile.add(new File(logDir, args()[i]));
-        	}
-        } else {
-        	String fmt = "%s"+Chrono.dateOnlyStamp()+".csv";
-        	File file;
-        	for(NotifyBody nb : NotifyBody.getAll()) {
-        		file = new File(logDir,String.format(fmt, nb.name()));
-        		if(file.exists()) {
-        			trans.info().printf("Processing %s",file.getCanonicalPath());
-        			notifyFile.add(file);
-        		} else {
-        			trans.info().printf("No Files found for %s",nb.name());
-        		}
-        	}
-        }
-	}
-
-	@Override
-	protected void run(AuthzTrans trans) {
-		List<String> toList = new ArrayList<>();
-		List<String> ccList = new ArrayList<>();
-		AuthzTrans noAvg = trans.env().newTransNoAvg();
-		String subject = "Test Notify";
-		boolean urgent = false;
-		
-
-		
-		final Notify notify = this;
-		final Holder<List<String>> info = new Holder<>(null);
-		final Set<String> errorSet = new HashSet<>();
-		
-		try {
-			EMAILTYPE:
-			for(File f : notifyFile) {
-				CSV csv = new CSV(env.access(),f);
-				try {
-					csv.visit(new CSV.Visitor() {
-						@Override
-						public void visit(List<String> row) throws IOException, CadiException {
-							if("info".equals(row.get(0))) {
-								info.set(row);
-							}
-							if(info.get()==null) {
-								throw new CadiException("First line of Feed MUST contain 'info' record");
-							}
-							String key = row.get(0)+'|'+info.get().get(1);
-							NotifyBody body = NotifyBody.get(key);
-							if(body==null) {
-								errorSet.add("No NotifyBody defined for " + key);
-							} else {
-								body.store(row);
-							}
-						}
-					});
-				} catch (IOException | CadiException e) {
-					e.printStackTrace();
-				}
-				
-				// now create Notification
-				for(NotifyBody nb : NotifyBody.getAll()) {
-					for(String id : nb.users()) {
-						toList.clear();
-						ccList.clear();
-						try {
-							Identity identity = trans.org().getIdentity(noAvg, id);
-							if(!identity.isPerson()) {
-								identity = identity.responsibleTo();
-							}
-							for(int i=1;i<nb.escalation();++i) {
-								if(identity != null) {
-									if(i==1) {
-										toList.add(identity.email());
-									} else {
-										identity=identity.responsibleTo();
-										ccList.add(identity.email());
-									}
-								}
-							}
-							
-							StringBuilder content = new StringBuilder();
-							content.append(String.format(header,version,Identity.mixedCase(identity.firstName())));
-							
-							nb.body(noAvg, content, indent, notify, id);
-							content.append(footer);
-								
-							if(!mailer.sendEmail(noAvg, dryRun, toList, ccList, subject,content.toString(), urgent)) {
-								trans.error().log("Mailer failed to send Mail");
-							}
-							if(maxEmails>0 && mailer.count()>=maxEmails) {
-								break EMAILTYPE;
-							}
-						} catch (OrganizationException e) {
-							trans.error().log(e);
-						}
-					}
-				}
-
-			}
-		} finally {
-			for(String s : errorSet) {
-				trans.audit().log(s);
-			}
-		}
-	}
-	
-	@Override
-	protected void _close(AuthzTrans trans) {
-	}
-
-}
+ import java.io.BufferedReader;
+ import java.io.File;
+ import java.io.FileReader;
+ import java.io.IOException;
+ import java.lang.reflect.Constructor;
+ import java.lang.reflect.InvocationTargetException;
+ import java.util.ArrayList;
+ import java.util.HashSet;
+ import java.util.List;
+ import java.util.Set;
+
+ import org.onap.aaf.auth.batch.Batch;
+ import org.onap.aaf.auth.batch.reports.bodies.NotifyBody;
+ import org.onap.aaf.auth.env.AuthzTrans;
+ import org.onap.aaf.auth.org.Mailer;
+ import org.onap.aaf.auth.org.Organization.Identity;
+ import org.onap.aaf.auth.org.OrganizationException;
+ import org.onap.aaf.cadi.Access;
+ import org.onap.aaf.cadi.CadiException;
+ import org.onap.aaf.cadi.client.Holder;
+ import org.onap.aaf.cadi.util.CSV;
+ import org.onap.aaf.misc.env.APIException;
+ import org.onap.aaf.misc.env.util.Chrono;
+
+ public class Notify extends Batch {
+	 private static final String HTML_CSS = "HTML_CSS";
+	 private final Mailer mailer;
+	 private final String header;
+	 private final String footer;
+	 private Set<File> notifyFile;
+	 public final String guiURL;
+	 private int maxEmails;
+	 private int indent;
+
+	 public Notify(AuthzTrans trans) throws APIException, IOException, OrganizationException {
+		 super(trans.env());
+		 String mailerCls = env.getProperty("MAILER");
+		 String mailFrom = env.getProperty("MAIL_FROM");
+		 String header_html = env.getProperty("HEADER_HTML");
+		 String footer_html = env.getProperty("FOOTER_HTML");
+		 String maxEmails = env.getProperty("MAX_EMAIL");
+		 guiURL = env.getProperty("GUI_URL");
+		 this.maxEmails = maxEmails==null?1:Integer.parseInt(maxEmails);
+		 if(mailerCls==null || mailFrom==null || guiURL==null || header_html==null || footer_html==null) {
+			 throw new APIException("Notify requires MAILER, MAILER_FROM, GUI_URL, HEADER_HTML and FOOTER_HTML properties");
+		 }
+		 try {
+			 Class<?> mailc = Class.forName(mailerCls);
+			 Constructor<?> mailcst = mailc.getConstructor(Access.class);
+			 mailer = (Mailer)mailcst.newInstance(env.access());
+		 } catch (ClassNotFoundException | NoSuchMethodException | SecurityException | InstantiationException | IllegalAccessException | IllegalArgumentException | InvocationTargetException e) {
+			 throw new APIException("Unable to construct " + mailerCls,e);
+		 }
+
+		 String line;
+		 StringBuilder sb = new StringBuilder();
+		 BufferedReader br = new BufferedReader(new FileReader(header_html));
+		 try {
+			 while((line=br.readLine())!=null) {
+				 sb.append(line);
+				 sb.append('\n');
+			 }
+			 String html_css = env.getProperty(HTML_CSS);
+			 int hc = sb.indexOf(HTML_CSS);
+			 if(hc!=0 && html_css!=null) {
+				 header = sb.replace(hc,hc+HTML_CSS.length(), html_css).toString();
+			 } else {
+				 header = sb.toString();
+			 }
+		 } finally {
+			 br.close();
+		 }
+
+		 // Establish index from header
+		 int lastTag = header.lastIndexOf('<');
+		 if(lastTag>0) {
+			 int prevCR = header.lastIndexOf('\n',lastTag);
+			 if(prevCR>0) {
+				 indent = lastTag-prevCR;
+			 } else {
+				 indent = 6; //arbitrary
+			 }
+		 }
+
+
+		 sb.setLength(0);
+		 br = new BufferedReader(new FileReader(footer_html));
+		 try {
+			 while((line=br.readLine())!=null) {
+				 sb.append(line);
+				 sb.append('\n');
+			 }
+			 footer = sb.toString();
+		 } finally {
+			 br.close();
+		 }
+
+		 // Class Load possible data
+		 NotifyBody.load(env.access());
+
+		 // Create Intermediate Output 
+		 File logDir = logDir();
+		 notifyFile = new HashSet<>();
+		 if(args().length>0) {
+			 for(int i=0;i<args().length;++i) {
+				 notifyFile.add(new File(logDir, args()[i]));
+			 }
+		 } else {
+			 String fmt = "%s"+Chrono.dateOnlyStamp()+".csv";
+			 File file;
+			 for(NotifyBody nb : NotifyBody.getAll()) {
+				 file = new File(logDir,String.format(fmt, nb.name()));
+				 if(file.exists()) {
+					 trans.info().printf("Processing '%s' in %s",nb.type(),file.getCanonicalPath());
+					 notifyFile.add(file);
+				 } else {
+					 trans.info().printf("No Files found for %s",nb.name());
+				 }
+			 }
+		 }
+	 }
+
+	 @Override
+	 protected void run(AuthzTrans trans) {
+		 List<String> toList = new ArrayList<>();
+		 List<String> ccList = new ArrayList<>();
+		 AuthzTrans noAvg = trans.env().newTransNoAvg();
+		 String subject = "Test Notify";
+		 boolean urgent = false;
+
+
+
+		 final Notify notify = this;
+		 final Holder<List<String>> info = new Holder<>(null);
+		 final Set<String> errorSet = new HashSet<>();
+
+		 try {
+			 for(File f : notifyFile) {
+				 CSV csv = new CSV(env.access(),f);
+				 try {
+					 csv.visit(new CSV.Visitor() {
+						 @Override
+						 public void visit(List<String> row) throws IOException, CadiException {
+							 if("info".equals(row.get(0))) {
+								 info.set(row);
+							 }
+							 if(info.get()==null) {
+								 throw new CadiException("First line of Feed MUST contain 'info' record");
+							 }
+							 String key = row.get(0)+'|'+info.get().get(1);
+							 NotifyBody body = NotifyBody.get(key);
+							 if(body==null) {
+								 errorSet.add("No NotifyBody defined for " + key);
+							 } else {
+								 body.store(row);
+							 }
+						 }
+					 });
+				 } catch (IOException | CadiException e) {
+					 e.printStackTrace();
+				 }
+
+			 }	
+
+			 // now create Notification
+			 for(NotifyBody nb : NotifyBody.getAll()) {
+				 String run = nb.type()+nb.name();
+				 String test = dryRun?run:null;
+				 ONE_EMAIL:
+					 for(String id : nb.users()) {
+
+						 toList.clear();
+						 ccList.clear();
+						 try {
+							 Identity identity = trans.org().getIdentity(noAvg, id);
+							 if(identity==null) {
+								 trans.warn().printf("%s is invalid for this Organization. Skipping notification.",id);
+							 } else {
+								 if(!identity.isPerson()) {
+									 identity = identity.responsibleTo();
+								 }
+								 for(int i=1;i<nb.escalation();++i) {
+									 if(identity != null) {
+										 if(i==1) {
+											 toList.add(identity.email());
+										 } else {
+											 identity=identity.responsibleTo();
+											 ccList.add(identity.email());
+										 }
+									 }
+								 }
+
+								 StringBuilder content = new StringBuilder();
+								 content.append(String.format(header,version,Identity.mixedCase(identity.firstName())));
+
+								 nb.body(noAvg, content, indent, notify, id);
+								 content.append(footer);
+
+								 if(mailer.sendEmail(noAvg, test, toList, ccList, subject,content.toString(), urgent)) {
+									 nb.inc();
+								 } else {
+									 trans.error().log("Mailer failed to send Mail");
+								 }
+								 if(maxEmails>0 && nb.count()>=maxEmails) {
+									 break ONE_EMAIL;
+								 }
+							 }
+						 } catch (OrganizationException e) {
+							 trans.error().log(e);
+						 }
+					 }
+				 trans.info().printf("Emailed %d for %s",nb.count(),run);
+			 }
+
+
+		 } finally {
+			 for(String s : errorSet) {
+				 trans.audit().log(s);
+			 }
+		 }
+	 }
+
+	 @Override
+	 protected void _close(AuthzTrans trans) {
+	 }
+
+ }