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.spi.ram;
021
022import org.crsh.util.Utils;
023import org.crsh.vfs.Path;
024import org.crsh.vfs.spi.AbstractFSDriver;
025
026import java.io.ByteArrayInputStream;
027import java.io.IOException;
028import java.io.InputStream;
029import java.lang.reflect.UndeclaredThrowableException;
030import java.net.MalformedURLException;
031import java.net.URL;
032import java.util.ArrayList;
033import java.util.Collections;
034import java.util.HashMap;
035import java.util.Iterator;
036import java.util.List;
037
038public class RAMDriver extends AbstractFSDriver<Path> {
039
040  /** . */
041  private final Path root;
042
043  /** . */
044  final HashMap<Path, String> entries;
045
046  /** . */
047  URL baseURL;
048
049  public RAMDriver() {
050    try {
051      this.root = Path.get("/");
052      this.entries = new HashMap<Path, String>();
053      this.baseURL = new URL("ram", null, 0, "/", new RAMURLStreamHandler(this));
054    }
055    catch (MalformedURLException e) {
056      throw new UndeclaredThrowableException(e);
057    }
058  }
059
060  public void add(String path, String file) {
061    add(Path.get(path), file);
062  }
063
064  public void add(Path path, String file) {
065    entries.put(path, file);
066  }
067
068  public Path root() throws IOException {
069    return root;
070  }
071
072  public String name(Path handle) throws IOException {
073    return handle.getName();
074  }
075
076  public boolean isDir(Path handle) throws IOException {
077    return handle.isDir();
078  }
079
080  public Iterable<Path> children(Path handle) throws IOException {
081    List<Path> children = Collections.emptyList();
082    for (Path entry : entries.keySet()) {
083      if (entry.isChildOf(handle)) {
084        if (children.isEmpty()) {
085          children = new ArrayList<Path>();
086        }
087        children.add(entry);
088      }
089    }
090    return children;
091  }
092
093  public long getLastModified(Path handle) throws IOException {
094    return 0;
095  }
096
097  public Iterator<InputStream> open(Path handle) throws IOException {
098    return Utils.<InputStream>iterator(new ByteArrayInputStream(entries.get(handle).getBytes("UTF-8")));
099  }
100}