1 /*******************************************************************************
2 * Copyright (c) 2016 AT&T Intellectual Property. All rights reserved.
3 *******************************************************************************/
4 package com.att.authz.helpers;
6 import java.util.ArrayList;
9 import java.util.TreeMap;
11 import com.att.authz.actions.Message;
12 import com.att.authz.env.AuthzTrans;
13 import com.att.authz.org.EmailWarnings;
14 import com.att.authz.org.Organization;
15 import com.att.authz.org.Organization.Notify;
16 import com.att.authz.org.Organization.Identity;
17 import com.att.authz.org.OrganizationException;
18 import com.att.authz.org.OrganizationFactory;
19 import org.onap.aaf.inno.env.Env;
20 import org.onap.aaf.inno.env.TimeTaken;
21 import org.onap.aaf.inno.env.Trans;
22 import org.onap.aaf.inno.env.util.Chrono;
23 import com.datastax.driver.core.ResultSet;
24 import com.datastax.driver.core.Row;
25 import com.datastax.driver.core.Session;
26 import com.datastax.driver.core.SimpleStatement;
27 import com.datastax.driver.core.Statement;
29 public class Notification {
31 public static final TreeMap<String,List<Notification>> data = new TreeMap<String,List<Notification>>();
32 public static final long now = System.currentTimeMillis();
34 public final String user;
35 public final Notify type;
36 public final Date last;
37 public final int checksum;
40 public Organization org;
42 private long graceEnds,lastdays;
44 private Notification(String user, int type, Date last, int checksum) {
46 this.type = Notify.from(type);
48 this.checksum = checksum;
53 private Notification(String user, Notify type, Date last, int checksum) {
57 this.checksum = checksum;
62 public static void load(Trans trans, Session session, Creator<Notification> creator ) {
63 trans.info().log( "query: " + creator.select() );
64 TimeTaken tt = trans.start("Load Notify", Env.REMOTE);
68 Statement stmt = new SimpleStatement(creator.select());
69 results = session.execute(stmt);
74 tt = trans.start("Process Notify", Env.SUB);
77 for(Row row : results.all()) {
80 Notification not = creator.create(row);
81 List<Notification> ln = data.get(not.user);
83 ln = new ArrayList<Notification>();
84 data.put(not.user, ln);
93 trans.info().log("Found",count,"Notify Records");
97 public static Notification get(String user, Notify type) {
98 List<Notification> ln = data.get(user);
100 for(Notification n : ln) {
101 if(type.equals(n.type)) {
109 private static Notification getOrCreate(String user, Notify type) {
110 List<Notification> ln = data.get(user);
111 Notification n = null;
113 ln = new ArrayList<Notification>();
116 for(Notification n2 : ln) {
117 if(type.equals(n2.type)) {
124 n = new Notification(user, type, new Date(), 0);
130 public static Notification add(AuthzTrans trans, UserRole ur) {
131 Notification n = getOrCreate(ur.user,Notify.RoleExpiration);
134 n.org = OrganizationFactory.obtain(trans.env(), ur.ns);
135 } catch (OrganizationException e) {
136 trans.error().log(ur.ns, " does not have a Namespace");
141 EmailWarnings ew = n.org.emailWarningPolicy();
142 n.graceEnds = ew.roleEmailInterval();
143 n.lastdays = ew.emailUrgentWarning();
148 StringBuilder sb = new StringBuilder();
153 ouser = n.org.getUser(trans, ur.user);
156 sb.append(ouser.fullName());
159 } catch (Exception e) {
161 sb.append(" Role: ");
163 sb.append(" Expire");
164 if(now<ur.expires.getTime()) {
169 sb.append(Chrono.dateOnlyStamp(ur.expires));
170 sb.append("\n If you wish to extend, type\n");
171 sb.append("\trole user extend ");
175 sb.append("\n If you wish to delete, type\n");
176 sb.append("\trole user del ");
181 n.msg.add(sb.toString());
187 public static Notification addApproval(AuthzTrans trans, Identity ou) {
188 Notification n = getOrCreate(ou.id(),Notify.Approval);
192 if(n.count==0) { // first time.
193 EmailWarnings ew = n.org.emailWarningPolicy();
194 n.graceEnds = ew.apprEmailInterval();
195 n.lastdays = ew.emailUrgentWarning();
201 public static Creator<Notification> v2_0_14 = new Creator<Notification>() {
203 public Notification create(Row row) {
204 return new Notification(row.getString(0), row.getInt(1), row.getDate(2),row.getInt(3));
208 public String select() {
209 return "select user,type,last,checksum from authz.notify";
213 public void set(Message msg) {
217 public int checksum() {
219 for(String l : msg.lines) {
220 for(byte b : l.getBytes()) {
228 public boolean update(AuthzTrans trans, Session session, boolean dryRun) {
229 String update = update();
232 trans.info().log(update);
234 session.execute(update);
236 return true; // Updated info, expect to notify
242 * Returns an Update String for CQL if there is data.
244 * Returns null if nothing to update
247 private String update() {
248 // If this has been done before, there is no change in checkSum and the last time notified is within GracePeriod
249 if(checksum!=0 && checksum()==checksum && now < last.getTime()+graceEnds && now > last.getTime()+lastdays) {
252 return "UPDATE authz.notify SET last = '" +
253 Chrono.dateOnlyStamp(last) +
264 // public void text(Email email) {
265 // for(String s : msg) {
270 public String toString() {
271 return "\"" + user + "\",\"" + type.name() + "\",\"" + Chrono.dateOnlyStamp(last);