001/*
002 * Copyright (C) 2012 eXo Platform SAS.
003 *
004 * This is free software; you can redistribute it and/or modify it
005 * under the terms of the GNU Lesser General Public License as
006 * published by the Free Software Foundation; either version 2.1 of
007 * the License, or (at your option) any later version.
008 *
009 * This software is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * You should have received a copy of the GNU Lesser General Public
015 * License along with this software; if not, write to the Free
016 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
017 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
018 */
019
020package org.crsh.vfs;
021
022import org.crsh.vfs.spi.FSDriver;
023import org.crsh.vfs.spi.file.FileDriver;
024import org.crsh.vfs.spi.url.URLDriver;
025
026import java.io.IOException;
027import java.net.URISyntaxException;
028import java.net.URL;
029import java.util.ArrayList;
030import java.util.Enumeration;
031import java.util.List;
032
033public class FS {
034
035  /** . */
036  final List<Mount<?>> mounts;
037
038  public FS() {
039    this.mounts = new ArrayList<Mount<?>>();
040  }
041
042  public File get(Path path) throws IOException {
043    return new File(this, path);
044  }
045
046  public <H> FS mount(FSDriver<H> driver) {
047    if (driver == null) {
048      throw new NullPointerException();
049    }
050    mounts.add(Mount.wrap(driver));
051    return this;
052  }
053
054  public FS mount(java.io.File root) {
055    return mount(new FileDriver(root));
056  }
057
058  public FS mount(ClassLoader cl, Path path) throws IOException, URISyntaxException {
059    if (cl == null) {
060      throw new NullPointerException();
061    }
062    if (path == null) {
063      throw new NullPointerException();
064    }
065    if (!path.isDir()) {
066      throw new IllegalArgumentException("Path " + path + " must be a dir");
067    }
068    URLDriver driver = new URLDriver();
069    // Add the resources
070    Enumeration<URL> en = cl.getResources(path.getValue().substring(1));
071    while (en.hasMoreElements()) {
072      URL url = en.nextElement();
073      driver.merge(url);
074    }
075
076    return mount(driver);
077  }
078
079  public FS mount(Class<?> clazz) throws IOException, URISyntaxException {
080    if (clazz == null) {
081      throw new NullPointerException();
082    }
083    URL url = clazz.getProtectionDomain().getCodeSource().getLocation();
084    URLDriver driver = new URLDriver();
085    driver.merge(url);
086    return mount(driver);
087  }
088}