/*
* #%L
* gitools-core
* %%
* Copyright (C) 2013 Universitat Pompeu Fabra - Biomedical Genomics group
* %%
* 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, see
* <http://www.gnu.org/licenses/gpl-3.0.html>.
* #L%
*/
package org.gitools.persistence.locators.filters.cache;
import org.apache.commons.io.IOUtils;
import org.gitools.api.PersistenceException;
import org.gitools.api.analysis.IProgressMonitor;
import org.gitools.api.resource.IResourceLocator;
import org.gitools.utils.progressmonitor.ProgressMonitorInputStream;
import java.io.*;
import java.net.URL;
public class CacheResourceLocator implements IResourceLocator {
private boolean cached;
private File cachedFile;
private IResourceLocator originalLocator;
public CacheResourceLocator(File cachedFile, IResourceLocator originalLocator) {
this.cached = false;
this.cachedFile = cachedFile;
this.originalLocator = originalLocator;
}
@Override
public URL getURL() {
return originalLocator.getURL();
}
@Override
public File getReadFile() {
return cachedFile == null ? originalLocator.getReadFile() : cachedFile;
}
public File getWriteFile() {
return originalLocator.getWriteFile();
}
@Override
public String getBaseName() {
return originalLocator.getBaseName();
}
@Override
public String getExtension() {
return originalLocator.getExtension();
}
@Override
public String getName() {
return originalLocator.getName();
}
@Override
public boolean isContainer() {
return false;
}
@Override
public long getContentLength() {
if (cached) {
return cachedFile.length();
}
return originalLocator.getContentLength();
}
@Override
public IResourceLocator getParentLocator() {
return originalLocator;
}
@Override
public IResourceLocator getReferenceLocator(String referenceName) throws PersistenceException {
return originalLocator.getReferenceLocator(referenceName);
}
@Override
public boolean isWritable() {
return originalLocator.isWritable();
}
@Override
public InputStream openInputStream(IProgressMonitor progressMonitor) throws IOException {
if (!cached) {
progressMonitor.begin("Downloading...", originalLocator.getContentLength());
IOUtils.copy(originalLocator.openInputStream(progressMonitor), new FileOutputStream(cachedFile));
cached = true;
progressMonitor.begin("Reading " + getName(), getContentLength());
}
return new ProgressMonitorInputStream(progressMonitor, new FileInputStream(cachedFile));
}
@Override
public OutputStream openOutputStream(IProgressMonitor monitor) throws IOException {
return originalLocator.openOutputStream(monitor);
}
@Override
public void close(IProgressMonitor monitor) {
originalLocator.close(monitor);
}
}