VCardFiles.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 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.vcard.db;
 
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FilenameFilter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.util.List;
import org.turro.csv.CSVList;
import org.turro.csv.CSVParser;
import org.turro.elephant.context.Application;
import org.turro.elephant.context.ElephantContext;
import org.turro.vcard.VCard;
 
public class VCardFiles {
 
  public static final String 
          VCARD_IMPORT_FOLDER = "/WEB-INF/vCard/in",
          VCARD_EXPORT_FOLDER = "/WEB-INF/vCard/out";
 
  public static File getImportFolder() {
    File root = new File(ElephantContext.getRealPath(VCARD_IMPORT_FOLDER));
    if(!root.exists()) root.mkdir();
    return root;
  }
 
  public static File[] getFiles() {
    File root = new File(ElephantContext.getRealPath(VCARD_IMPORT_FOLDER));
    if(!root.exists()) root.mkdir();
    return root.listFiles(new FilenameFilter() {
      @Override
      public boolean accept(File dir, String name) {
        return name.endsWith(".vcf");
      }
    });
  }
 
  public static File[] getCSVs() {
    File root = new File(ElephantContext.getRealPath(VCARD_IMPORT_FOLDER));
    if(!root.exists()) root.mkdir();
    return root.listFiles(new FilenameFilter() {
      @Override
      public boolean accept(File dir, String name) {
        return name.endsWith(".csv");
      }
    });
  }
 
  public static File[] getCSVRels() {
    File root = new File(ElephantContext.getRealPath(VCARD_IMPORT_FOLDER));
    if(!root.exists()) root.mkdir();
    return root.listFiles(new FilenameFilter() {
      @Override
      public boolean accept(File dir, String name) {
        return name.endsWith(".csvrel");
      }
    });
  }
 
  public static VCardList loadVCards() throws IOException {
    File files[] = getFiles();
    VCardList vl = new VCardList();
    for(File f : files) {
      BufferedReader in = new BufferedReader(
              new InputStreamReader(new FileInputStream(f)));
      List<VCard> l = new VCardParser(in).getVCards();
      if(l != null) {
        vl.addAll(l);
      }
      in.close();
    }
    return vl;
  }
 
  public static CSVList loadCSVs() throws IOException {
    File files[] = getCSVs();
    CSVList cl = new CSVList();
    for(File f : files) {
      BufferedReader in = new BufferedReader(
              new InputStreamReader(new FileInputStream(f)));
      cl = new CSVParser(in).getCSVs();
      in.close();
      // Can't read more than a single CVS at time, fields may differ
      break;
    }
    return cl;
  }
 
  public static CSVList loadCSVRels() throws IOException {
    File files[] = getCSVRels();
    CSVList cl = new CSVList();
    for(File f : files) {
      BufferedReader in = new BufferedReader(
              new InputStreamReader(new FileInputStream(f)));
      cl = new CSVParser(in).getCSVs();
      in.close();
      // Can't read more than a single CVS at time, fields may differ
      break;
    }
    return cl;
  }
 
  public static File saveVCards(VCardList list) throws IOException {
    File root = new File(ElephantContext.getRealPath(VCARD_EXPORT_FOLDER));
    if(!root.exists()) root.mkdirs();
    File output = new File(ElephantContext.getRealPath(VCARD_EXPORT_FOLDER +
            "/" + Application.getUser().getId() + ".vcf"));
    boolean append = false;
    for(VCard vCard : list) {
      Writer out = new OutputStreamWriter(new FileOutputStream(output, append));
      out.write(vCard.getVCard());
      out.close();
      append = true;
    }
    return output;
  }
 
  private VCardFiles() {
  }
 
}