TimeTrackerExporter.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2023 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.timetracker;
 
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.turro.action.IMailSender;
import org.turro.action.MailSenders;
import org.turro.auth.Authentication;
import org.turro.elephant.context.ElephantContext;
import org.turro.elephant.db.ElephantPU;
import org.turro.elephant.entities.db.TimeTrackerType;
import org.turro.entities.Entities;
import org.turro.i18n.I_;
import org.turro.jpa.Dao;
import org.turro.jpa.export.ExportAsStream;
import org.turro.message.MessageQueue;
import org.turro.plugin.contacts.IContact;
import org.turro.sql.SqlClause;
import org.turro.time.Time;
 
public class TimeTrackerExporter {
 
  public static void exportTimeTracker() {
    IContact contact = Authentication.getIContact();
    MailSenders.getHeavy()
            .addContact(contact)
            .onStart(s -> MessageQueue.pushMessage(contact, I_.get("Report will be sent by email")))
            .onCancel(s -> MessageQueue.pushMessage(contact, I_.get("Task already running")))
            .onBuild(s -> timeTrackerToExcel((IMailSender) s))
            .send(I_.get("Export time tracker"), I_.get("Export time tracker"));
  }
 
  private static void timeTrackerToExcel(IMailSender sender) {
    Map<String, String> cache = new HashMap<>();
    List<String> l = Arrays.asList(new String[] {
      I_.get("Entity"),
      I_.get("Tracked"),
      I_.get("Moment"),
      I_.get("Type"),
      I_.get("Hours"),
      I_.get("Validated")
    });
    try {
      Dao dao = new ElephantPU();
      ExportAsStream eq = new ExportAsStream(dao, l, 
            SqlClause.select("tt.entityPath, tt.trackedPath, tt.timeTrack, tt.trackType, tt.hours, tt.validated")
                    .from("TimeTracker tt")
                    .where().greaterOrEqual("timeTrack", Time.now().addYears(-1).getDate())
                    .orderBy("tt.entityPath, tt.trackedPath, tt.timeTrack"));
      eq.fieldReplacement((col, obj) -> {
        if(col == 0 || col == 1) {
          if(cache.containsKey((String) obj)) {
            obj = cache.get((String) obj);
          } else {
            String name = Entities.getController(obj).getName();
            cache.put((String) obj, name);
            obj = name;
          }
        } else if(col == 3) {
          obj = ((TimeTrackerType) obj).toString();
        }
        return obj;
      });
      File result = File.createTempFile("report", "_TimeTracker.xls");
      eq.generateExcel(result);
      sender.addAttachment(result);
      sender.onFinish(x -> sender.removeAttachments());
    } catch (IOException ex) {
      Logger.getLogger(TimeTrackerExporter.class.getName()).log(Level.SEVERE, null, ex);
    }
  }
 
}