1 files changed, 182 insertions, 0 deletions
diff --git a/ms_specific_proxyservice/src/main/java/at/asitplus/eidas/specific/proxy/logger/AdvancedStatisticLogger.java b/ms_specific_proxyservice/src/main/java/at/asitplus/eidas/specific/proxy/logger/AdvancedStatisticLogger.java
new file mode 100644
index 00000000..55833b8b
--- /dev/null
+++ b/ms_specific_proxyservice/src/main/java/at/asitplus/eidas/specific/proxy/logger/AdvancedStatisticLogger.java
@@ -0,0 +1,182 @@
+package at.asitplus.eidas.specific.proxy.logger;
+
+import java.io.IOException;
+import java.time.LocalDateTime;
+
+import com.fasterxml.jackson.annotation.JsonFormat;
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.annotation.JsonInclude.Include;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
+import com.fasterxml.jackson.databind.annotation.JsonSerialize;
+import com.fasterxml.jackson.dataformat.javaprop.JavaPropsMapper;
+import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateTimeDeserializer;
+import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer;
+
+import at.asitplus.eidas.specific.modules.msproxyservice.MsProxyServiceConstants;
+import at.gv.egiz.eaaf.core.api.IRequest;
+import at.gv.egiz.eaaf.core.api.IStatusMessenger;
+import at.gv.egiz.eaaf.core.api.data.PvpAttributeDefinitions;
+import at.gv.egiz.eaaf.core.api.idp.IAuthData;
+import at.gv.egiz.eaaf.core.api.idp.IEidAuthData;
+import at.gv.egiz.eaaf.core.api.logging.IStatisticLogger;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+import lombok.Setter;
+import lombok.extern.slf4j.Slf4j;
+
+/**
+ * Extended statistic logger that use a JSON based data model.
+ * 
+ * @author tlenz
+ *
+ */
+@Slf4j
+public class AdvancedStatisticLogger implements IStatisticLogger {
+
+  private static final String DATEFORMATER = "yyyy.MM.dd-HH:mm:ss,SSS";
+  private static final String DEFAULT_NO_IDP_ID = "no idpId available";
+  
+  private static final ObjectMapper mapper = new ObjectMapper();
+  private static final JavaPropsMapper propsMapper = new JavaPropsMapper();
+  
+  private final IStatusMessenger messageService;
+  
+  /**
+   * Build a JSON based statistic logger that uses error messages from a specific source.
+   * 
+   * @param source i18n message source
+   */
+  public AdvancedStatisticLogger(IStatusMessenger source) {
+    this.messageService = source;
+    
+  }
+
+  @Override
+  public void logSuccessOperation(IRequest protocolRequest, IAuthData authData, boolean isSsoSession) {
+    final StatisticLogEntry entry = buildCoreEntry(protocolRequest);
+    entry.setSuccess(SuccessEntry.builder()
+        .spSector(protocolRequest.getServiceProviderConfiguration().getAreaSpecificTargetIdentifier())
+        .spCountry(protocolRequest.getServiceProviderConfiguration().getConfigurationValue(
+            MsProxyServiceConstants.CONIG_PROPS_CONNECTOR_COUNTRYCODE))
+        .citizenCountryCode("AT")
+        .withMandates(selectMandateProcessType(authData))
+        .build());
+    writeEntryToLog(entry);
+    
+  }
+
+  @Override
+  public void logErrorOperation(Throwable throwable) {
+    log.trace("Advanced statistic logger only logs on SP level");
+
+  }
+
+  @Override
+  public void logErrorOperation(Throwable throwable, IRequest errorRequest) {        
+    final StatisticLogEntry entry = buildCoreEntry(errorRequest);                  
+    entry.setError(new ErrorEntry(messageService.getResponseErrorCode(throwable), throwable.getMessage()));
+    writeEntryToLog(entry);
+
+  }
+
+  @Override
+  public void internalTesting() throws Exception {
+    log.trace("Not implemented for a File-based logger");
+
+  }
+
+  private StatisticLogEntry buildCoreEntry(IRequest pendingRequest) {
+    return new StatisticLogEntry(
+        LocalDateTime.now(),
+        pendingRequest.getUniqueTransactionIdentifier(),
+        pendingRequest.getServiceProviderConfiguration() != null 
+            && pendingRequest.getServiceProviderConfiguration().getUniqueIdentifier() != null 
+          ? pendingRequest.getServiceProviderConfiguration().getUniqueIdentifier() : DEFAULT_NO_IDP_ID);  
+    
+  }
+   
+  private MandateProcess selectMandateProcessType(IAuthData authData) {
+    if (authData instanceof IEidAuthData && ((IEidAuthData)authData).isUseMandate()) {
+      return authData.getGenericData(PvpAttributeDefinitions.MANDATE_LEG_PER_SOURCE_PIN_NAME, String.class) != null 
+          ? MandateProcess.FOR_LEGAL_PERSON : MandateProcess.FOR_NATURAL_PERSON;
+      
+    } else {
+      return MandateProcess.NONE; 
+      
+    }
+  }
+      
+  private void writeEntryToLog(StatisticLogEntry entry) {
+    try {
+      log.info(mapper.writeValueAsString(propsMapper.writeValueAsProperties(entry)));
+
+    } catch (final IOException e) {
+      log.error("Can NOT generate statistic entry for logging", e);
+
+    }
+  }
+   
+  public enum MandateProcess { NONE, FOR_NATURAL_PERSON, FOR_LEGAL_PERSON } 
+  
+  @Getter
+  @Setter
+  @RequiredArgsConstructor
+  @JsonInclude(Include.NON_NULL)
+  private static class StatisticLogEntry {
+
+    @JsonSerialize(using = LocalDateTimeSerializer.class)
+    @JsonDeserialize(using = LocalDateTimeDeserializer.class)
+    @JsonFormat(pattern = DATEFORMATER)
+    @JsonProperty("timestamp")
+    private final LocalDateTime timestamp;
+    
+    @JsonProperty("txId")
+    private final String transactionId;
+    
+    @JsonProperty("entityId")
+    private final String idpEntityId;
+    
+    @JsonProperty("result")
+    private SuccessEntry success;
+
+    @JsonProperty("error")
+    private ErrorEntry error;
+    
+  }
+
+  @Getter
+  @Setter
+  @Builder
+  @JsonInclude(Include.NON_NULL)
+  private static class SuccessEntry {
+
+    @JsonProperty("spSector")
+    private final String spSector;
+    
+    @JsonProperty("spCountry")
+    private final String spCountry;
+
+    @JsonProperty("citizenCountryCode")
+    private final String citizenCountryCode;
+
+    @JsonProperty("withMandates")
+    private final MandateProcess withMandates;
+               
+  }
+  
+  @Getter
+  @Setter
+  @RequiredArgsConstructor
+  private static class ErrorEntry {
+
+    @JsonProperty("code")
+    private final String errorCode;
+    
+    @JsonProperty("msg")
+    private final String errorMessage;
+
+  }
+}