package org.webslinger.ext.commons.vfs.object;

 import java.io.ByteArrayInputStream;

 import java.io.ByteArrayOutputStream;

 import java.io.InputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import java.io.OutputStreamWriter;

 import java.io.PrintWriter;

 import java.util.ArrayList;

 import java.nio.ByteBuffer;

 import java.nio.CharBuffer;

 import java.util.Arrays;

 import java.util.Collection;

 import java.util.Collections;

 import java.util.HashMap;

 import java.util.Iterator;

 import java.util.List;

 import java.util.ListIterator;

 import java.util.Map;

 import org.apache.commons.vfs.Capability;

 import org.apache.commons.vfs.FileName;

 import org.apache.commons.vfs.FileObject;

 import org.apache.commons.vfs.FileSystem;

 import org.apache.commons.vfs.FileSystemException;

 import org.apache.commons.vfs.FileSystemManager;

 import org.apache.commons.vfs.FileSystemOptions;

 import org.apache.commons.vfs.FileType;

 import org.apache.commons.vfs.impl.VirtualFileName;

 import org.apache.commons.vfs.provider.AbstractFileSystem;

 import org.webslinger.collections.CollectionUtil;

 import org.webslinger.io.Charsets;

 import org.webslinger.resolver.ObjectWalker;

 public class ObjectFileSystem extends AbstractFileSystem {

     private final Object rootObject;

     private final ObjectWalker walker = new ObjectWalker();

     public ObjectFileSystem(Object rootObject, FileSystemOptions options) throws FileSystemException {

         super(new VirtualFileName("object",  FileName.ROOT_PATH, FileType.FOLDER), null, options);

         this.rootObject = rootObject;

     }

     public void addCapabilities(Collection caps) {

     }

     public FileObject createFile(FileName name) {

         return new ObjectFileObject(this, name);

     }

     private Object getObject(String path) {

         ListIterator<String> it = CollectionUtil.parse(path, "/");

         Object object = walker.get(rootObject, it);

         if (object == null || it.hasNext()) return null;

         return object;

     }

     public InputStream getInputStream(String path) throws Exception {

         char[] chars = walker.getChars(rootObject, CollectionUtil.parse(path, "/"));

         if (chars == null) return null;

         ByteBuffer buf = Charsets.UTF8.encode(CharBuffer.wrap(chars));

         byte[] bytes = buf.array();

         return new ByteArrayInputStream(bytes, buf.arrayOffset(), buf.limit());

     }

     public OutputStream getOutputStream(final String path) throws Exception {

         return new ByteArrayOutputStream() {

             public void close() throws IOException {

                 super.close();

                 try {

                     CharBuffer buf = Charsets.UTF8.decode(ByteBuffer.wrap(toByteArray()));

                     walker.setChars(rootObject, CollectionUtil.parse(path, "/"), buf.array(), buf.arrayOffset(), buf.limit());

                 } catch (Exception e) {

                     throw (IOException) new IOException(e.getMessage()).initCause(e);

                 }

             }

         };

     }

     public FileType getType(String path) {

         ListIterator<String> it = CollectionUtil.parse(path, "/");

         Object object = walker.get(rootObject, it);

         if (object == null || it.hasNext()) return null;

         FileType type = walker.hasChildren(object, it) ? FileType.FOLDER : FileType.FILE;

         return type;

     }

     public FileObject[] getChildren(String path) throws FileSystemException {

         ListIterator<String> it = CollectionUtil.parse(path, "/");

         String[] names = walker.list(rootObject, it);

         if (names == null || it.hasNext()) return null;

         FileObject[] children = new FileObject[names.length];

         for (int i = 0; i < names.length; i++) {

             children[i] = resolveFile(names[i]);

         }

         return children;

     }

 }