WorkingDirectory.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.wd.files;
 
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.text.ParseException;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jdom.input.SAXBuilder;
import org.jdom.output.Format;
import org.jdom.output.XMLOutputter;
 
public class WorkingDirectory extends Directory {
 
  public static final String
          WD_ROOT = "/BBWorkDir",
          WD_CONF_ROOT = "/.BBWDConf",
          WD_CACHE = "/.BBWorkDir.cache";
 
  private String bbUser, bbHost, bbPort, bbContext;
  private File wdHome, wdInternal, wdCache;
 
  public WorkingDirectory(String bbUser, String bbHost, String bbPort, String bbContext) {
    this.bbUser = bbUser;
    this.bbHost = bbHost;
    this.bbPort = bbPort;
    this.bbContext = bbContext;
  }
 
  public String getBbContext() {
    return bbContext == null ? "" : bbContext;
  }
 
  public String getBbHost() {
    return bbHost;
  }
 
  public String getBbPort() {
    return bbPort;
  }
 
  public String getBbUser() {
    return bbUser;
  }
 
  public void initialize(String bbCreateDir, String bbCreatePath) throws ParseException {
    wdHome = new File(System.getProperty("user.home") + WD_ROOT + "/" + bbHost);
    if(!wdHome.exists()) {
      wdHome.mkdirs();
    }
    wdInternal = new File(wdHome.getAbsolutePath() + WD_CONF_ROOT);
    if(!wdInternal.exists()) {
      wdInternal.mkdirs();
    }
    wdCache = new File(wdInternal.getAbsolutePath() + WD_CACHE);
    initializeCache();
    checkExisting();
    loadSystemFiles();
    if(bbCreateDir != null) {
      Directory d = getDirs().getDir(bbCreateDir);
      if(d == null) {
        d = addDirectory(bbCreateDir);
        d.setServerPath(bbCreatePath);
        d.getSystemFile().mkdirs();
      }
    }
    loadServerFiles();
    saveCache();
    countFiles();
  }
 
  public void reload() throws ParseException {
    getDirs().clear();
    getFiles().clear();
    initializeCache();
    checkExisting();
    loadSystemFiles();
    loadServerFiles();
    saveCache();
    countFiles();
  }
 
  @Override
  public String getSystemName() {
    return wdHome.getAbsolutePath();
  }
 
  private void initializeCache() throws ParseException {
    SAXBuilder builder = new SAXBuilder();
    Document doc;
    try {
      if(wdCache.exists()) {
        doc = builder.build(wdCache);
        readXML(doc.getRootElement());
      }
    } catch (IOException ex) {
      Logger.getLogger(WorkingDirectory.class.getName()).log(Level.SEVERE, null, ex);
    } catch (JDOMException ex) {
      Logger.getLogger(WorkingDirectory.class.getName()).log(Level.SEVERE, null, ex);
    }
  }
 
  public void saveCache() {
    Document doc = new Document(new Element("bb-working-directory"));
    writeXML(doc.getRootElement());
    try {
      OutputStreamWriter fw = new OutputStreamWriter(new FileOutputStream(wdCache), "UTF-8");
      Format fm = Format.getPrettyFormat();
      fm.setEncoding("UTF-8");
      XMLOutputter xo = new XMLOutputter(fm);
      xo.output(doc, fw);
      fw.close();
    } catch (IOException ex) {
      Logger.getLogger(WorkingDirectory.class.getName()).log(Level.SEVERE, null, ex);
    }
  }
 
}