Workers.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 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.contacts.relation;
 
import java.util.ArrayList;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.function.Predicate;
import org.amic.util.date.Dates;
import org.turro.contacts.BusinessRelation;
import org.turro.contacts.Contact;
import org.turro.contacts.util.ContactList;
 
public class Workers extends HashSet<BusinessRelation> {
 
  private final Contact contact;
  private final Date now;
 
  public Workers(Contact contact) {
    this.contact = contact;
    this.now = new Date();
    load();
  }
  
  public List<Contact> getWorkers() {
    return getWorkers(now);
  }
  
  public List<Contact> getWorkers(Date date) {
    ContactList cs = new ContactList();
    for(BusinessRelation br : this) {
      if(Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        cs.add(br.getContact());
      }
    }
    return cs;
  }  
  
  public boolean isWorker(Contact contact) {
    return contact != null && getWorkers().contains(contact);
  }
 
  public boolean isWorker(Contact contact, Date date) {
    return contact != null && getWorkers(date).contains(contact);
  }
 
  public boolean isByCondition(Contact contact, Predicate<BusinessRelation> predicate) {
    return contact != null && getByCondition(predicate).contains(contact);
  }
 
  public boolean isByCondition(Contact contact, Predicate<BusinessRelation> predicate, Date date) {
    return contact != null && getByCondition(predicate, date).contains(contact);
  }
 
  public List<Contact> getByCondition(Predicate<BusinessRelation> predicate) {
    return getByCondition(predicate, now);
  }
  
  public List<Contact> getByCondition(Predicate<BusinessRelation> predicate, Date date) {
    ContactList cs = new ContactList();
    for(BusinessRelation br : this) {
      if(predicate.test(br) && Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        cs.add(br.getContact());
      }
    }
    return cs;
  }  
 
  public List<BusinessRelation> getRelations(Contact contact) {
    return getRelations(contact, now);
  }
  
  public List<BusinessRelation> getRelations(Contact contact, Date date) {
    List<BusinessRelation> list = new ArrayList<>();
    for(BusinessRelation br : this) {
      if(Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        list.add(br);
      }
    }
    return list;
  }
  
  private void load() {
    addAll(contact.getWorkerRelations());
  }
  
}