Business.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.Date;
import java.util.List;
import java.util.TreeSet;
import java.util.stream.Collectors;
import org.amic.util.date.Dates;
import org.turro.contacts.BusinessRelation;
import org.turro.contacts.Contact;
import org.turro.contacts.ContactType;
import org.turro.contacts.db.ContactsPU;
import org.turro.contacts.util.ContactList;
import org.turro.jpa.Dao;
import org.turro.plugin.contacts.ContactRelations;
import org.turro.plugin.contacts.IContactRelation;
 
public class Business extends TreeSet<BusinessRelation> {
 
  private final Contact contact;
  private final Date now;
 
  public Business(Contact contact) {
    this.contact = contact;
    this.now = new Date();
    load();
  }
  
  public Contact getBusiness() {
    return getBusiness(now);
  }
  
  public Contact getBusiness(Date date) {
    for(BusinessRelation br : this) {
      if(br.isPreferential() && Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        return br.getBusiness();
      }
    }
    for(IContactRelation cr : ContactRelations.getBusiness(contact.getId(), ContactRelations.MODE_OUTER)) {
      if(cr.getStrong() && cr.asWorker()) {
        return getDao().find(Contact.class, cr.getRelatedId());
      }
    }
    for(BusinessRelation br : this) {
      if(Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        return br.getBusiness();
      }
    }
    return null;
  }  
  
  public BusinessRelation getBusinessRelation() {
    return getBusinessRelation(now);
  }
  
  public BusinessRelation getBusinessRelation(Date date) {
    for(BusinessRelation br : this) {
      if(br.isPreferential() && Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        return br;
      }
    }
    for(BusinessRelation br : this) {
      if(Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        return br;
      }
    }
    return null;
  }  
  
  public List<Contact> getCompaniesList() {
    return getBusinessList().stream()
            .filter(c -> ContactType.CONTACT_COMPANY.equals(c.getType()))
            .collect(Collectors.toList());
  }
  
  public List<Contact> getCentersList() {
    return getBusinessList().stream()
            .filter(c -> ContactType.CONTACT_LEARNINGCENTER.equals(c.getType()))
            .collect(Collectors.toList());
  }
  
  public List<Contact> getBusinessList() {
    return getBusinessList(now);
  }
  
  public List<Contact> getBusinessList(Date date) {
    ContactList cs = new ContactList();
    for(BusinessRelation br : this) {
      if(Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        cs.add(br.getBusiness());
      }
    }
    return cs;
  }
  
  public List<Contact> getOutdatedList() {
    return getOutdatedList(now);
  }
  
  public List<Contact> getOutdatedList(Date date) {
    ContactList cs = new ContactList();
    for(BusinessRelation br : this) {
      if(!Dates.inRange(br.getStartDate(), br.getEndDate(), date)) {
        cs.add(br.getBusiness());
      }
    }
    return cs;
  }
  
  public List<Contact> getCoworkers() {
    return getCoworkers(now);
  }
  
  public List<Contact> getCoworkers(Date date) {
    ContactList cs = new ContactList();
    for(BusinessRelation br : this) {
      Workers workers = new Workers(br.getBusiness());
      cs.addAll(workers.getWorkers(date));
    }
    return cs;
  }
  
  private void load() {
    addAll(contact.getBusinessRelations());
  }
  
  /* Dao helpers */
  
  private Dao _dao;
  
  private Dao getDao() {
    if(_dao == null) {
      _dao = new ContactsPU();
    }
    return _dao;
  }
 
}