CompanyTree.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2013 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.financials.business;
 
import java.util.ArrayList;
import java.util.List;
import java.util.Set;
import org.turro.financials.entity.Company;
import org.turro.financials.entity.Contract;
import org.turro.financials.entity.Department;
import org.turro.financials.entity.Headquarters;
import org.turro.financials.entity.Service;
import org.turro.financials.model.business.CompanyWrapper;
import org.turro.i18n.I_;
import org.zkoss.zk.ui.ext.AfterCompose;
import org.zkoss.zul.Tree;
import org.zkoss.zul.Treechildren;
import org.zkoss.zul.Treeitem;
 
public class CompanyTree extends Tree implements AfterCompose {
 
  public CompanyTree() {
    super();
  }
 
  @Override
  public void afterCompose() {
    appendChild(new Treechildren());
    Company company = CompanyWrapper.getDefaultCompany();
    Treeitem li = new Treeitem(company.getName(), company);
    li.appendChild(new Treechildren());
    getTreechildren().appendChild(li);
    fillServices(li);
    fillHeadquarters(li);
  }
  
  public List<Contract> getContracts() {
    List<Contract> contracts = new ArrayList<>();
    processItem(getSelectedItem(), contracts);
    return contracts;
  }
 
  private void processItem(Treeitem selectedItem, List<Contract> contracts) {
    if(selectedItem != null) {
      Object value = selectedItem.getValue();
      if(value instanceof Contract) {
        contracts.add((Contract) value);
      } else if(selectedItem.getTreechildren() != null) {
        for(Treeitem ti : selectedItem.getTreechildren().getItems()) {
          processItem(ti, contracts);
        }
      }
    }
  }
 
  private void fillServices(Treeitem root) {
    Company company = root.getValue();
    Treeitem li = new Treeitem(I_.get("Services"));
    li.appendChild(new Treechildren());
    root.getTreechildren().appendChild(li);
    for(Service service : company.getServices()) {
      Treeitem lis = new Treeitem(service.getName(), service);
      lis.setTooltiptext(lis.getLabel());
      li.getTreechildren().appendChild(lis);
      fillContracts(lis, service.getStores());
    }
  }
 
  private void fillHeadquarters(Treeitem root) {
    Company company = root.getValue();
    Treeitem li = new Treeitem(I_.get("Headquarters"));
    li.appendChild(new Treechildren());
    root.getTreechildren().appendChild(li);
    for(Headquarters headquarters : company.getHeadquarters()) {
      Treeitem lis = new Treeitem(headquarters.getName(), headquarters);
      lis.setTooltiptext(lis.getLabel());
      li.getTreechildren().appendChild(lis);
      fillDepartments(lis, headquarters.getDepartments());
    }
  }
 
  private void fillDepartments(Treeitem root, Set<Department> departments) {
    root.appendChild(new Treechildren());
    Treeitem li = new Treeitem(I_.get("Departments"));
    li.appendChild(new Treechildren());
    root.getTreechildren().appendChild(li);
    for(Department department : departments) {
      Treeitem lis = new Treeitem(department.getName(), department);
      lis.setTooltiptext(lis.getLabel());
      li.getTreechildren().appendChild(lis);
      fillContracts(lis, department.getStores());
    }
  }
 
  private void fillContracts(Treeitem root, Set<Contract> stores) {
    root.appendChild(new Treechildren());
    for(Contract contract : stores) {
      Treeitem lis = new Treeitem(contract.getPartialDescription(), contract);
      lis.setTooltiptext(lis.getLabel());
      root.getTreechildren().appendChild(lis);
    }
  }
 
}