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
020 package org.crsh.vfs;
021
022 import org.crsh.vfs.spi.FSDriver;
023 import org.crsh.vfs.spi.file.FileDriver;
024 import org.crsh.vfs.spi.url.URLDriver;
025
026 import java.io.IOException;
027 import java.net.URISyntaxException;
028 import java.net.URL;
029 import java.util.ArrayList;
030 import java.util.Enumeration;
031 import java.util.List;
032
033 public 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 }