ProductImport.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2017 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.product;
 
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Collection;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.turro.string.Strings;
import org.turro.auth.Authentication;
import org.turro.command.Command;
import org.turro.command.Context;
import org.turro.contacts.Contact;
import org.turro.contacts.GroupIt;
import org.turro.contacts.Grouped;
import org.turro.contacts.db.ContactsPU;
import org.turro.csv.CSVEntry;
import org.turro.csv.CSVList;
import org.turro.csv.CSVParser;
import org.turro.describeit.DescribeItUtil;
import org.turro.elephant.context.ElephantContext;
import org.turro.file.FileWrapper;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.Contract;
import org.turro.financials.entity.Product;
import org.turro.financials.entity.ProductByContractor;
import org.turro.i18n.I_;
import org.turro.jpa.Dao;
import org.turro.path.Path;
import org.zkoss.zk.ui.util.Clients;
 
public class ProductImport {
  
  public static final String 
          PRODUCT_IMPORT_FOLDER = "/WEB-INF/product/in",
          PRODUCT_EXPORT_FOLDER = "/WEB-INF/product/out";
 
  public static void uploadFile() {
    FileWrapper fw = new FileWrapper(new File(ElephantContext.getRealPath(PRODUCT_IMPORT_FOLDER)));
    fw.upload(new Command() {
      @Override
      public Object execute(Context context) {
        Clients.showBusy(I_.get("Import CVS"));
        new ProductImport().importFromCSV();
        Clients.clearBusy();
        return null;
      }
    });
  }
  
  public void importFromCSV() {
    for(File file : getCSVs()) {
      try(BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream(file)))) {
        processFile(new CSVParser(in).getCSVs());     
      } catch (FileNotFoundException ex) {
        Logger.getLogger(ProductImport.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      } catch (IOException ex) {
        Logger.getLogger(ProductImport.class.getName()).log(Level.SEVERE, ElephantContext.logMsg(null), ex);
      }
      file.delete();
    }
  }
  
  public File[] getCSVs() {
    File root = new File(ElephantContext.getRealPath(PRODUCT_IMPORT_FOLDER));
    if(!root.exists()) root.mkdir();
    return root.listFiles(new FilenameFilter() {
      @Override
      public boolean accept(File dir, String name) {
        return name.endsWith(".csv");
      }
    });
  }
 
  private void processFile(CSVList list) {
    for(CSVEntry v : list) {
      GroupIt groupIt = createGroups(v);
      IProduct product = null;
      String productCode = v.getField("Code");
      if(!Strings.isBlank(productCode)) {
        product = ProductFactory.getProduct(productCode);
        if(product == null) {
          product = new Product();
          ((Product) product).setProductCode(productCode);
        }
      }
      String codeByProvider = v.getField("CodeByProvider");
      Long providerCode = v.getLongField("Provider");
      if(providerCode > 0 && !Strings.isBlank(codeByProvider)) {
        ProductByContractor pbc = ProductFactory.getContractorProduct(providerCode, codeByProvider);
        if(pbc == null) {
          pbc = new ProductByContractor();
          pbc.setContract(getDao().find(Contract.class, providerCode));
          pbc.setContractorCode(codeByProvider);
          pbc.setDescription(v.getField("Description"));
          pbc.setPrice(v.getDoubleField("Cost"));
          pbc.setTax(v.getDoubleField("Tax"));
          pbc.setProduct((Product) product);
          ((Product) product).getProductByContractors().add(pbc);
        }
      }
      ((Product) product).setDescription(v.getField("Description"));
      ((Product) product).setKeywords(v.getField("Keywords"));
      ((Product) product).setPrice(v.getDoubleField("Price"));
      ((Product) product).setTax(v.getDoubleField("Tax"));
      ((Product) product).setPublishable(v.getBooleanField("Publishable"));
      ((Product) product).setPromoted(v.getBooleanField("Promoted"));
      ((Product) product).setService(v.getBooleanField("Service"));
      ((Product) product).setDelivery(v.getBooleanField("Delivery"));
      ((Product) product).setConcept(v.getBooleanField("Concept"));
      ((Product) product).setAttachments(v.getBooleanField("Attachments"));
      product = (IProduct) getDao().saveObject((Product) product);
      String productPath = FinancialsPU.getObjectPath((Product) product);
      String web = v.getField("WebDescription");
      DescribeItUtil.addDescription(productPath, "Default", web, (Contact) Authentication.getIContact().getContact(), null);
      if(groupIt != null) {
        Collection<Grouped> groups = Grouped.groups(productPath, v.getField("GroupCat"));
        boolean done = false;
        for(Grouped grouped : groups) {
          if(grouped.getParent().getId().equals(groupIt.getId())) {
            done = true;
          }
        }
        if(!done) {
          Grouped grouped = new Grouped();
          grouped.setParent(groupIt);
          grouped.setCategory(v.getField("GroupCat"));
          grouped.setPath(productPath);
          new ContactsPU().saveObject(grouped);
        }
      }
    }
  }
 
  private GroupIt createGroups(CSVEntry v) {
    GroupIt groupIt = null;
    if(v.getList().getHeader().hasPrefix("Group")) {
      String category = v.getField("GroupCat");
      if(!Strings.isBlank(category)) {
        Path path = new Path(v.getField("Group"));
        if(path.getSize() > 0) {
          for(int i = 0; i < path.getSize(); i++) {
            groupIt = GroupIt.getOrCreate(category, path.getNode(i), groupIt);
          }
        }
      }      
    }
    return groupIt;
  }
 
  private Dao _dao;
  
  private Dao getDao() {
    if(_dao == null) {
      _dao = new FinancialsPU();
    }
    return _dao;
  }
  
}