Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.resourcesstorage;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStream;

import java.net.MalformedURLException;

import java.net.URL;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.tools.util.HasAFile;

/**

 *

 * @author jjdelcerro

 */

public class FilesResourcesStorage extends AbstractResourcesStorage {

    public static class FileResource 

            extends AbstractResourcesStorage.AbstractResource

            implements ResourcesStorage.Resource, HasAFile 

      {

        private class ResourceInputStream extends InputStream {

            @Override

            public int read() throws IOException {

                return in.read();

            }

            @Override

            public void close() throws IOException {

                FileResource.this.close();

            }

        }

        private class ResourceOutputStream extends OutputStream {

            @Override

            public void write(int b) throws IOException {

                out.write(b);

            }

            @Override

            public void flush() throws IOException {

                out.flush();

            }

            @Override

            public void close() throws IOException {

                FileResource.this.close();

            }

        }

        private final File resource;

        private FileInputStream in;

        private FileOutputStream out;

        public FileResource(String name, File resource) {

            super(name);

            this.resource = resource;

        }

        public String getName() {

          return name;

        }

        @Override

        public boolean isReadOnly() {

            return false;

        }

        @Override

        public URL getURL() {

            try {

                return this.resource.toURI().toURL();

            } catch (MalformedURLException ex) {

                return null;

            }

        }

        @Override

        public File getFile() {

            return this.resource;

        }

        @Override

        public boolean exists() {

            return this.resource.exists();

        }

        @Override

        public InputStream asInputStream() throws java.io.FileNotFoundException {

            if (this.out != null || this.in != null) {

                throw new IllegalStateException("Input already open");

            }

            this.in = new FileInputStream(this.resource);

            return new ResourceInputStream();

        }

        @Override

        public OutputStream asOutputStream() throws java.io.FileNotFoundException {

            if (this.out != null || this.in != null) {

                throw new IllegalStateException("Input already open");

            }

            this.out = new FileOutputStream(this.resource);

            return new ResourceOutputStream();

        }

        @Override

        public void close() {

            IOUtils.closeQuietly(out);

            IOUtils.closeQuietly(in);

            this.in = null;

            this.out = null;

        }

        @Override

        public void setFile(File file) {

            throw new UnsupportedOperationException("Not supported yet.");

        }

    }

    private final List<String> resourcesPaths;

    public FilesResourcesStorage(String fullPathName) {

        this.resourcesPaths = new ArrayList<>();

        this.resourcesPaths.add(fullPathName);

    }

    public FilesResourcesStorage(List<String> fullPathNames) {

        this.resourcesPaths = new ArrayList<>();

        this.resourcesPaths.addAll(fullPathNames);

    }

    @Override

    public Resource getResource(String resourceName) {

        File fres;

        for (String pathName : resourcesPaths) {

            File folder = new File(pathName);

            if( folder.isDirectory() ) {

                fres = new File(folder,resourceName);

            } else {

                fres = new File(pathName + this.getSeparator() + resourceName);

            }

            if( fres.exists() ) {

                return new FileResource(resourceName, fres);

            }

        }

        String pathName = this.resourcesPaths.get(0);

        fres = new File(pathName + this.getSeparator() + resourceName);

        return new FileResource(resourceName, fres);

    }

    public List<String> getPaths() {

        return this.resourcesPaths;

    }

    @Override

    public List<String> getResourceNames() {

        List<String> names = new ArrayList<>();

        for (String pathName : resourcesPaths) {

            File folder = new File(pathName);

            if( folder.isDirectory() ) {

              for (File f: folder.listFiles()) {

                if( f.isFile() ) {

                  names.add(f.getName());

                }

              }

            } else {

              File parent = folder.getParentFile();

              String prefix = pathName + this.getSeparator();

              int prefix_len = prefix.length();

              for (File f: parent.listFiles()) {

                if( f.isFile() ) {

                  String fname = f.getName();

                  if( StringUtils.startsWithIgnoreCase(fname, prefix) ) {

                    names.add(fname.substring(prefix_len));

                  }

                }

              }

            }

        }

        return names;

    }

}