From f8ac3b5dfd065f1fc4e2d907b31e65324c6f904f Mon Sep 17 00:00:00 2001
From: Seif Eddine GHAZOUANI <labsoft@se.ghazouani>
Date: Fri, 3 May 2024 13:30:31 +0200
Subject: [PATCH] add argument to TruncateNewCalendarStatements

---
 .../impl/TruncateNewCalendarStatements.java   | 29 +++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/onebusaway-gtfs-transformer/src/main/java/org/onebusaway/gtfs_transformer/impl/TruncateNewCalendarStatements.java b/onebusaway-gtfs-transformer/src/main/java/org/onebusaway/gtfs_transformer/impl/TruncateNewCalendarStatements.java
index ef2456493..c4c2bca26 100644
--- a/onebusaway-gtfs-transformer/src/main/java/org/onebusaway/gtfs_transformer/impl/TruncateNewCalendarStatements.java
+++ b/onebusaway-gtfs-transformer/src/main/java/org/onebusaway/gtfs_transformer/impl/TruncateNewCalendarStatements.java
@@ -15,6 +15,7 @@
  */
 package org.onebusaway.gtfs_transformer.impl;
 
+import org.onebusaway.csv_entities.schema.annotations.CsvField;
 import org.onebusaway.gtfs.model.ServiceCalendar;
 import org.onebusaway.gtfs.model.ServiceCalendarDate;
 import org.onebusaway.gtfs.services.GtfsMutableRelationalDao;
@@ -35,6 +36,31 @@
 public class TruncateNewCalendarStatements implements GtfsTransformStrategy {
 
     private final Logger _log = LoggerFactory.getLogger(TruncateNewCalendarStatements.class);
+
+    /*
+     * add two arguments used in the truncated transformation strategy
+     * calendarField  --> calendar_field (json config file)
+     *      Calendar.YEAR           =  1
+     *      Calendar.MONTH          =  2
+     *      Calendar.DAY_OF_MONTH   =  5
+     *      Calendar.DAY_OF_YEAR    =  6
+     * calendarAmount --> calendar_amount (json config file)
+     */
+    @CsvField(optional = true)
+    private int calendarField = Calendar.MONTH;
+
+    @CsvField(optional = true)
+    private int calendarAmount = 1;
+
+
+    public void setCalendarField(int calendarField) {
+        this.calendarField = calendarField;
+    }
+
+    public void setCalendarAmount(int calendarAmount) {
+        this.calendarAmount = calendarAmount;
+    }
+
     @Override
     public String getName() {
         return this.getClass().getSimpleName();
@@ -43,9 +69,8 @@ public String getName() {
     @Override
     public void run(TransformContext transformContext, GtfsMutableRelationalDao gtfsMutableRelationalDao) {
         RemoveEntityLibrary removeEntityLibrary = new RemoveEntityLibrary();
-        // TODO make this an argument -- default to one month from now
         Calendar c = Calendar.getInstance();
-        c.roll(Calendar.MONTH, 1);
+        c.add(calendarField, calendarAmount); 
         java.util.Date oneMonthFromNow = c.getTime();
         Set<ServiceCalendar> serviceCalendarsToRemove = new HashSet<ServiceCalendar>();
         for (ServiceCalendar calendar: gtfsMutableRelationalDao.getAllCalendars()) {