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.FileUtils;

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) {

        return new FileResource(resourceName, getResourceFile(resourceName));

    }

    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;

    }

    @Override

    public boolean remove(String resourceName) {

        File fileToRemove = this.getResourceFile(resourceName);

        return FileUtils.deleteQuietly(fileToRemove);

    }

    @Override

    public boolean allowRemove() {

        return true;

    }

    private File getResourceFile(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 fres;

            }

        }

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

        File folder = new File(pathName);

        if (folder.isDirectory()) {

            fres = new File(folder, resourceName);

        } else {

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

        }

        return fres;

    }

}