NotificationMessageGenerator.java

  1. package no.nav.data.team.notify;

  3. import com.github.benmanes.caffeine.cache.Caffeine;
  4. import com.github.benmanes.caffeine.cache.LoadingCache;
  5. import lombok.Data;
  6. import lombok.extern.slf4j.Slf4j;
  7. import no.nav.data.common.auditing.domain.AuditVersion;
  8. import no.nav.data.common.auditing.domain.AuditVersionRepository;
  9. import no.nav.data.common.exceptions.NotFoundException;
  10. import no.nav.data.common.storage.StorageService;
  11. import no.nav.data.team.contact.domain.ContactMessage;
  12. import no.nav.data.team.notify.domain.NotificationTask;
  13. import no.nav.data.team.notify.domain.NotificationTask.AuditTarget;
  14. import no.nav.data.team.notify.dto.MailModels;
  15. import no.nav.data.team.notify.dto.MailModels.InactiveModel;
  16. import no.nav.data.team.notify.dto.MailModels.NudgeModel;
  17. import no.nav.data.team.notify.dto.MailModels.Resource;
  18. import no.nav.data.team.notify.dto.MailModels.TypedItem;
  19. import no.nav.data.team.notify.dto.MailModels.UpdateItem;
  20. import no.nav.data.team.notify.dto.MailModels.UpdateModel;
  21. import no.nav.data.team.notify.dto.MailModels.UpdateModel.TargetType;
  22. import no.nav.data.team.po.domain.ProductArea;
  23. import no.nav.data.team.resource.NomClient;
  24. import no.nav.data.team.shared.Lang;
  25. import no.nav.data.team.shared.domain.Member;
  26. import no.nav.data.team.shared.domain.Membered;
  27. import no.nav.data.team.team.domain.Team;
  28. import org.apache.commons.lang3.StringUtils;
  29. import org.springframework.stereotype.Service;

  31. import java.time.Duration;
  32. import java.util.ArrayList;
  33. import java.util.List;
  34. import java.util.Objects;
  35. import java.util.Optional;
  36. import java.util.UUID;
  37. import java.util.stream.Collectors;

  39. import static no.nav.data.common.utils.StreamUtils.convert;
  40. import static no.nav.data.common.utils.StreamUtils.filterCommonElements;
  41. import static no.nav.data.team.contact.domain.ContactMessage.Paragraph.VarselUrl.url;

  43. @Slf4j
  44. @Service
  45. public class NotificationMessageGenerator {

  47.     private final AuditVersionRepository auditVersionRepository;
  48.     private final LoadingCache<UUID, AuditVersion> auditCache;
  49.     private final LoadingCache<UUID, ProductArea> paCache;
  50.     private final UrlGenerator urlGenerator;
  51.     private final NomClient nomClient;

  53.     public NotificationMessageGenerator(AuditVersionRepository auditVersionRepository,
  54.             StorageService storageService, UrlGenerator urlGenerator, NomClient nomClient) {
  55.         this.auditVersionRepository = auditVersionRepository;
  56.         this.auditCache = Caffeine.newBuilder().recordStats()
  57.                 .expireAfterWrite(Duration.ofMinutes(5))
  58.                 .maximumSize(1000).build(id -> auditVersionRepository.findById(id).orElseThrow());
  59.         this.paCache = Caffeine.newBuilder().recordStats()
  60.                 .expireAfterWrite(Duration.ofMinutes(1))
  61.                 .maximumSize(1000).build(id -> storageService.get(id, ProductArea.class));

  63.         this.urlGenerator = urlGenerator;
  64.         this.nomClient = nomClient;
  65.     }

  67.     public NotificationMessage<UpdateModel> updateSummary(NotificationTask task) {
  68.         var model = new UpdateModel();
  69.         model.setBaseUrl(urlGenerator.getBaseUrl());
  70.         model.setTime(task.getTime());
  71.         task.getTargets().forEach(this::fetchAuditVersions);

  73.         task.getTargets().forEach(t -> {
  74.             if (t.isSilent()) {
  75.                 // target is here only for calculating other diffs, ie. teams in/out of product area
  76.                 return;
  77.             }
  78.             if (t.isCreate()) {
  79.                 AuditVersion auditVersion = t.getCurrAuditVersion();
  80.                 model.getCreated().add(auditToTypedItem(auditVersion, false));
  81.             } else if (t.isDelete()) {
  82.                 AuditVersion auditVersion = t.getPrevAuditVersion();
  83.                 model.getDeleted().add(auditToTypedItem(auditVersion, true));
  84.             } else if (t.isEdit()) { // is edit check -> temp fix due to scheduler bug
  85.                 AuditVersion prevVersion = t.getPrevAuditVersion();
  86.                 AuditVersion currVersion = t.getCurrAuditVersion();
  87.                 UpdateItem diff = diffItem(prevVersion, currVersion, task);
  88.                 if (diff.hasChanged()) {
  89.                     model.getUpdated().add(diff);
  90.                 }
  91.             }
  92.         });

  94.         boolean isEmpty = model.getCreated().isEmpty() && model.getDeleted().isEmpty() && model.getUpdated().isEmpty();
  95.         return new NotificationMessage<>("Teamkatalog oppdatering", model, urlGenerator.isDev(), isEmpty);
  96.     }

  98.     private void fetchAuditVersions(AuditTarget auditTarget) {
  99.         Optional.ofNullable(auditTarget.getPrevAuditId()).ifPresent(id -> auditTarget.setPrevAuditVersion(auditCache.get(id)));
  100.         Optional.ofNullable(auditTarget.getCurrAuditId()).ifPresent(id -> auditTarget.setCurrAuditVersion(auditCache.get(id)));
  101.     }

  103.     private UpdateItem diffItem(AuditVersion prevVersion, AuditVersion currVersion, NotificationTask task) {
  104.         var item = UpdateItem.builder();

  106.         var toName = nameFor(currVersion);
  107.         item.fromName(nameFor(prevVersion));
  108.         item.toName(toName);
  109.         item.item(new TypedItem(typeForAudit(currVersion), currVersion.getTableId(), urlGenerator.urlFor(currVersion), toName));

  111.         if (prevVersion.isTeam()) {
  112.             Team prevData = prevVersion.getTeamData();
  113.             Team currData = currVersion.getTeamData();
  114.             item.fromOwnershipType(Lang.teamOwnershipType(prevData.getTeamOwnershipType()));
  115.             item.toOwnershipType(Lang.teamOwnershipType(currData.getTeamOwnershipType()));
  116.             item.fromTeamType(Lang.teamType(prevData.getTeamType()));
  117.             item.toTeamType(Lang.teamType(currData.getTeamType()));

  119.             if (!Objects.equals(prevData.getProductAreaId(), currData.getProductAreaId())) {
  120.                 Optional.ofNullable(prevData.getProductAreaId()).map(this::getPa).ifPresent(item::oldProductArea);
  121.                 Optional.ofNullable(currData.getProductAreaId()).map(this::getPa).ifPresent(item::newProductArea);
  122.             }
  123.         }
  124.         if (prevVersion.isProductArea()) {
  125.             var newTeams = new ArrayList<AuditTarget>();
  126.             var removedTeams = new ArrayList<AuditTarget>();
  127.             var paId = UUID.fromString(prevVersion.getTableId());
  128.             // Task with productArea targets will contain all it's teams' targets
  129.             var teamTargets = task.getTargets().stream()
  130.                     .filter(AuditTarget::isTeam)
  131.                     .filter(t -> paId.equals(paIdForTeamAudit(t.getPrevAuditVersion())) || paId.equals(paIdForTeamAudit(t.getCurrAuditVersion())))
  132.                     .collect(Collectors.toList());

  134.             log.info("Looking into teams for pa {} {} of {} total targets", paId, teamTargets.size(), task.getTargets().size());

  136.             teamTargets.forEach(t -> {
  137.                 if (t.isCreate() || !paId.equals(paIdForTeamAudit(t.getPrevAuditVersion()))) {
  138.                     log.info("Team added {}", t.getTargetId());
  139.                     newTeams.add(t);
  140.                 } else if (t.isDelete() || !paId.equals(paIdForTeamAudit(t.getCurrAuditVersion()))) {
  141.                     log.info("Team removed {}", t.getTargetId());
  142.                     removedTeams.add(t);
  143.                 }
  144.             });
  145.             item.newTeams(convert(newTeams, this::teamTargetToTypedItem));
  146.             item.removedTeams(convert(removedTeams, this::teamTargetToTypedItem));

  148.             ProductArea prevData = prevVersion.getProductAreaData();
  149.             ProductArea currData = currVersion.getProductAreaData();
  150.             item.fromAreaType(Lang.areaType(prevData.getAreaType()));
  151.             item.toAreaType(Lang.areaType(currData.getAreaType()));
  152.         }
  153.         var fromMembers = members(prevVersion);
  154.         var toMembers = members(currVersion);

  156.         item.removedMembers(convertMember(filterCommonElements(fromMembers, toMembers, Member::getNavIdent)));
  157.         item.newMembers(convertMember(filterCommonElements(toMembers, fromMembers, Member::getNavIdent)));

  159.         return item.build();
  160.     }

  162.     private TypedItem getPa(UUID id) {
  163.         ProductArea pa = null;
  164.         try {
  165.             pa = paCache.get(id);
  166.         } catch (NotFoundException e) {
  167.             log.trace("Product area has been deleted {}", id);
  168.         }
  169.         if (pa == null) {
  170.             pa = auditVersionRepository.findByTableIdOrderByTimeDescLimitOne(id.toString()).getProductAreaData();
  171.             return paToItem(pa, true);
  172.         }
  173.         return paToItem(pa, false);
  174.     }

  176.     private String teamNameFor(AuditTarget teamTarget) {
  177.         return teamTarget.isCreate() || teamTarget.isUpdate() ? teamTarget.getCurrAuditVersion().getTeamData().getName()
  178.                 : teamTarget.getPrevAuditVersion().getTeamData().getName();
  179.     }

  181.     private UUID paIdForTeamAudit(AuditVersion auditVersion) {
  182.         return auditVersion == null ? null : auditVersion.getTeamData().getProductAreaId();
  183.     }

  185.     private List<Resource> convertMember(List<? extends Member> list) {
  186.         return convertIdents(convert(list, Member::getNavIdent));
  187.     }

  189.     private List<Resource> convertIdents(List<String> list) {
  190.         return convert(list, ident -> new Resource(urlGenerator.resourceUrl(ident), nomClient.getNameForIdent(ident).orElse(ident), ident));
  191.     }

  193.     private List<Member> members(AuditVersion version) {
  194.         if (version.isTeam()) {
  195.             return List.copyOf(version.getTeamData().getMembers());
  196.         } else if (version.isProductArea()) {
  197.             return List.copyOf(version.getProductAreaData().getMembers());
  198.         }
  199.         return List.of();
  200.     }

  202.     public ContactMessage nudgeTime(Membered membered, String role) {
  203.         NudgeModel model = NudgeModel.builder()
  204.                 .targetUrl(urlGenerator.urlFor(membered.getClass(), membered.getId()))
  205.                 .targetName(membered.getName())
  206.                 .targetType(Lang.objectType(membered.getClass()))
  207.                 .recipientRole(role.toLowerCase())
  208.                 .cutoffTime(NotificationConstants.NUDGE_TIME_CUTOFF_DESCRIPTION)
  209.                 .build();

  211.         var subject = "Teamkatalog påminnelse for %s %s".formatted(model.getTargetType(), model.getTargetName());

  213.         return new ContactMessage(subject, "nudge")
  214.                 .paragraph("Hei, det har nå gått over %s siden %%s ble sist oppdatert.".formatted(model.getCutoffTime()),
  215.                         url(model.getTargetUrl(), "%s %s".formatted(model.getTargetType(), model.getTargetName())))
  216.                 .paragraph("Som %s mottar du derfor en påminnelse for å sikre at innholdet er korrekt.".formatted(model.getRecipientRole()))
  217.                 .footer(model.getTargetUrl());
  218.     }

  220.     public ContactMessage inactive(Membered membered, String role, List<String> identsInactive) {
  221.         InactiveModel model = InactiveModel.builder()
  222.                 .targetUrl(urlGenerator.urlFor(membered.getClass(), membered.getId()))
  223.                 .targetName(membered.getName())
  224.                 .targetType(Lang.objectType(membered.getClass()))
  225.                 .recipientRole(role.toLowerCase())
  226.                 .members(convertIdents(identsInactive))
  227.                 .build();

  229.         String subject = "Medlemmer av %s %s har blitt inaktive".formatted(model.getTargetType(), model.getTargetName());
  230.         var message = new ContactMessage(subject, "inactive")
  231.                 .paragraph("Hei, %s har nå fått inaktive medlem(mer)", url(model.getTargetUrl(), "%s %s".formatted(model.getTargetType(), model.getTargetName())))
  232.                 .paragraph("Som %s mottar du derfor en påminnelse for å sikre at innholdet er korrekt.".formatted(model.getRecipientRole()))
  233.                 .spacing()
  234.                 .paragraph("Nye inaktive medlemmer:");

  236.         for (MailModels.Resource member : model.getMembers()) {
  237.             message.paragraph(" - %s", url(member.getUrl(), member.getName()));
  238.         }
  239.         message.footer(model.getTargetUrl());

  241.         return message;
  242.     }

  244.     private TargetType typeForAudit(AuditVersion auditVersion) {
  245.         if (auditVersion.isProductArea()) {
  246.             return TargetType.AREA;
  247.         } else if (auditVersion.isTeam()) {
  248.             return TargetType.TEAM;
  249.         }
  250.         return null;
  251.     }

  253.     private String nameFor(AuditVersion auditVersion) {
  254.         if (auditVersion.isTeam()) {
  255.             return auditVersion.getTeamData().getName();
  256.         } else if (auditVersion.isProductArea()) {
  257.             return auditVersion.getProductAreaData().getName();
  258.         }
  259.         return StringUtils.EMPTY;
  260.     }

  262.     private TypedItem auditToTypedItem(AuditVersion auditVersion, boolean deleted) {
  263.         return new TypedItem(typeForAudit(auditVersion), auditVersion.getTableId(), urlGenerator.urlFor(auditVersion), nameFor(auditVersion), deleted);
  264.     }

  266.     private TypedItem teamTargetToTypedItem(AuditTarget target) {
  267.         return new TypedItem(TargetType.TEAM, target.getTargetId().toString(), urlGenerator.urlFor(Team.class, target.getTargetId()), teamNameFor(target), target.isDelete());
  268.     }

  270.     private TypedItem paToItem(ProductArea pa, boolean deleted) {
  271.         return new TypedItem(TargetType.AREA, pa.getId().toString(), urlGenerator.urlFor(pa.getClass(), pa.getId()), pa.getName(), deleted);
  272.     }

  274.     @Data
  275.     public static class NotificationMessage<T> {

  277.         private final String subject;
  278.         private final T model;
  279.         private final boolean empty;
  280.         private final boolean dev;

  282.         NotificationMessage(String subject, T model, boolean dev, boolean isEmpty) {
  283.             this.subject = subject + (dev ? " [DEV]" : "");
  284.             this.model = model;
  285.             this.empty = isEmpty;
  286.             this.dev = dev;
  287.         }

  289.         NotificationMessage(String subject, T model, boolean dev) {
  290.             this(subject, model, dev, false);
  291.         }
  292.     }

  294. }
Created with JaCoCo 0.8.12.202403310830