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

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.tools.folders.impl;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.lang.management.ManagementFactory;

import java.lang.management.RuntimeMXBean;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import java.util.Set;

import javax.swing.filechooser.FileSystemView;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.tuple.ImmutablePair;

import org.gvsig.tools.folders.FoldersManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultFoldersManager implements FoldersManager {

    private static final Logger logger = LoggerFactory.getLogger(DefaultFoldersManager.class);

    private File temporaryFolder;

    private Map<String, File> folders;

    private int uniqueCounter = 1;

    public DefaultFoldersManager() {

        String temp = System.getenv("GVSIG_TEMP");

        File f;

        if (StringUtils.isEmpty(temp)) {

            temp = FileUtils.getTempDirectoryPath();

            // f = FileUtils.getFile(temp, "tmp-gvsig"+getPID());

            f = FileUtils.getFile(temp, "tmp-gvsig");

        } else {

            f = new File(temp);

        }

        this.setTemporaryFolder(f);

        Runtime.getRuntime().addShutdownHook(new Thread() {

            @Override

            public void run() {

                cleanTemporaryFiles();

            }

        });

    }

    private Map<String, File> getFolders() {

        if( this.folders == null ) {

            this.folders = new HashMap<>();

        }

        return this.folders;

    }

    @Override

    public boolean isEmpty() {

        return this.folders == null || this.folders.isEmpty();

    }

    @Override

    public void setLastPath(String pathId, File path) {

        if( StringUtils.isBlank(pathId) ) {

            throw new IllegalArgumentException("Invalid argument pathId, it is a blank string.");

        }

        this.getFolders().put("last.path." + pathId, path);

    }

    @Override

    public File getLastPath(String pathId, File defaultValue) {

        if( !StringUtils.isBlank(pathId) ) {

            File path = this.getFolders().get("last.path." + pathId);

            if (path != null) {

                return path;

            }

        }

        if (defaultValue != null) {

            return defaultValue;

        }

        return FileSystemView.getFileSystemView().getHomeDirectory();

    }

    @Override

    public Iterator<Map.Entry<String, File>> lastPathsIterator() {

        final Iterator<Map.Entry<String, File>> it0 = this.getFolders().entrySet().iterator();

        Iterator<Map.Entry<String, File>> it = new Iterator<Map.Entry<String, File>>() {

            Map.Entry<String, File> last = null;

            @Override

            public boolean hasNext() {

                while(it0.hasNext() ) {

                    last = it0.next();

                    if( StringUtils.startsWithIgnoreCase(last.getKey(), "last.path.")) {

                        return true;

                    }

                }

                return false;

            }

            @Override

            public Map.Entry<String, File> next() {

                if( last==null ) {

                    return null;

                }

                return new ImmutablePair<>(

                        last.getKey().substring(10),

                        last.getValue()

                );

            }

        };

        return it;

    }

    private int getPID() {

        try {

            RuntimeMXBean rt = ManagementFactory.getRuntimeMXBean();

            int java_pid = Integer.parseInt(rt.getName().substring(0, rt.getName().indexOf("@")));

            return java_pid;

        } catch (Throwable th) {

            return 0;

        }

    }

    @Override

    public File getTemporaryFolder() {

        return this.temporaryFolder;

    }

    @Override

    public void setTemporaryFolder(File folder) {

        if (folder == null) {

            throw new IllegalArgumentException("Can't set temporary folder to null.");

        }

        this.temporaryFolder = folder.getAbsoluteFile();

        this.createTemporaryFolder();

    }

    @Override

    public void cleanTemporaryFiles() {

        if (this.temporaryFolder == null) {

            logger.warn("Can't clean temporary folder, is null.");

        }

        try {

            FileUtils.cleanDirectory(this.temporaryFolder);

        } catch (IOException ex) {

            logger.warn("Can't clean temporary folder (" + this.temporaryFolder.getAbsolutePath() + ").", ex);

        }

    }

    @Override

    public File createTemporaryFolder() {

        if (this.temporaryFolder == null) {

            throw new RuntimeException("Can't create temporary folder, is null.");

        }

        try {

            FileUtils.forceMkdir(temporaryFolder);

        } catch (IOException ex) {

            throw new RuntimeException("Can't create temporary folder", ex);

        }

        return this.temporaryFolder;

    }

    @Override

    public File getTemporaryFile(String... pathComponents) {

        if (this.temporaryFolder == null) {

            throw new RuntimeException("Can't get temporary file, temporary folder is null.");

        }

        File f = FileUtils.getFile(temporaryFolder, pathComponents);

        return f;

    }

    @Override

    public File getUniqueTemporaryFile(String... pathComponents) {

        File f = getTemporaryFile(pathComponents);

        String fullName = f.getAbsolutePath();

        String name = FilenameUtils.removeExtension(fullName);

        String extension = FilenameUtils.getExtension(fullName);

        if (StringUtils.isEmpty(extension)) {

            fullName = name + "-" + Long.toHexString(System.currentTimeMillis()) + "-" + Integer.toHexString(uniqueCounter);

        } else {

            fullName = name + "-" + Long.toHexString(System.currentTimeMillis()) + "-" + Integer.toHexString(uniqueCounter) + "." + extension;

        }

        uniqueCounter++;

        return new File(fullName);

    }

    @Override

    public File createTemporaryFile(String basename, byte[] data) {

        if (this.temporaryFolder == null) {

            throw new RuntimeException("Can't create temporary file, temporary folder is null.");

        }

        File f = getTemporaryFile(basename);

        FileOutputStream fos = null;

        try {

            fos = new FileOutputStream(f);

            IOUtils.write(data, fos);

            f.deleteOnExit();

            return f;

        } catch (IOException ex) {

            throw new RuntimeException("Can't create temporaru file '" + f.getAbsolutePath() + "'.");

        } finally {

            IOUtils.closeQuietly(fos);

        }

    }

    @Override

    public File createTemporaryFile(String basename, String data) throws IOException {

        return this.createTemporaryFile(basename, data.getBytes());

    }

    @Override

    public void set(String id, File file) {

        this.getFolders().put(id, file);

    }

    @Override

    public File get(String id) {

        return this.getFolders().get(id);

    }

    @Override

    public File get(String id, File defaultValue) {

        if (this.getFolders().containsKey(id)) {

            return this.getFolders().get(id);

        }

        return defaultValue;

    }

    @Override

    public File getHome() {

        File home = new File(System.getProperty("user.home"));

        return home;

    }

    @Override

    public Iterator<String> iterator() {

        return this.getFolders().keySet().iterator();

    }

}