| /******************************************************************************* |
| * Copyright (c) 2009, 2017 Tasktop Technologies and others. |
| * |
| * This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License 2.0 |
| * which accompanies this distribution, and is available at |
| * https://www.eclipse.org/legal/epl-2.0/ |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| * Contributors: |
| * Tasktop Technologies - initial API and implementation |
| * Sonatype, Inc. - transport split and caching support |
| *******************************************************************************/ |
| package org.eclipse.equinox.internal.p2.discovery.compatibility.util; |
| |
| import java.io.*; |
| import java.net.URI; |
| import java.nio.charset.StandardCharsets; |
| import java.util.List; |
| import org.eclipse.core.runtime.*; |
| import org.eclipse.equinox.internal.p2.discovery.compatibility.Activator; |
| import org.eclipse.equinox.internal.p2.discovery.compatibility.Messages; |
| import org.eclipse.equinox.internal.p2.repository.AuthenticationFailedException; |
| import org.eclipse.equinox.internal.p2.transport.ecf.RepositoryTransport; |
| |
| /** |
| * A utility for accessing web resources |
| * |
| * @author David Green |
| */ |
| public class TransportUtil { |
| |
| /** |
| * Extend to process character content. |
| * |
| * @see TransportUtil#readResource(URI, TextContentProcessor, IProgressMonitor) |
| */ |
| public interface TextContentProcessor { |
| |
| public void process(Reader reader) throws IOException; |
| |
| } |
| |
| /** |
| * Download an HTTP-based resource |
| * |
| * @param target |
| * the target file to which the content is saved |
| * @param location |
| * the web location of the content |
| * @param monitor |
| * the monitor |
| * @throws IOException |
| * if a network or IO problem occurs |
| * @throws CoreException |
| */ |
| public static void downloadResource(URI location, File target, IProgressMonitor monitor) throws IOException, CoreException { |
| CacheManager cm = Activator.getDefault().getCacheManager(); |
| File cacheFile = cm.createCache(location, monitor); |
| if (cacheFile == null) { |
| throw new CoreException(new Status(IStatus.ERROR, Activator.ID, Messages.TransportUtil_InternalError)); |
| } |
| copyStream(new BufferedInputStream(new FileInputStream(cacheFile)), true, new BufferedOutputStream(new FileOutputStream(target)), true); |
| } |
| |
| public static int copyStream(InputStream in, boolean closeIn, OutputStream out, boolean closeOut) throws IOException { |
| try { |
| int written = 0; |
| byte[] buffer = new byte[16 * 1024]; |
| int len; |
| while ((len = in.read(buffer)) != -1) { |
| out.write(buffer, 0, len); |
| written += len; |
| } |
| return written; |
| } finally { |
| try { |
| if (closeIn) { |
| in.close(); |
| } |
| } finally { |
| if (closeOut) { |
| out.close(); |
| } |
| } |
| } |
| } |
| |
| /** |
| * Read a web-based resource at the specified location using the given processor. |
| * |
| * @param location |
| * the web location of the content |
| * @param processor |
| * the processor that will handle content |
| * @param monitor |
| * the monitor |
| * @throws IOException |
| * if a network or IO problem occurs |
| * @throws CoreException |
| */ |
| public static void readResource(URI location, TextContentProcessor processor, IProgressMonitor monitor) throws IOException, CoreException { |
| CacheManager cm = Activator.getDefault().getCacheManager(); |
| File cacheFile = cm.createCache(location, monitor); |
| if (cacheFile == null) { |
| throw new CoreException(new Status(IStatus.ERROR, Activator.ID, Messages.TransportUtil_InternalError)); |
| } |
| try (InputStream in = new BufferedInputStream(new FileInputStream(cacheFile))) { |
| // FIXME how can the charset be determined? |
| BufferedReader reader = new BufferedReader(new InputStreamReader(in, StandardCharsets.UTF_8)); |
| processor.process(reader); |
| } |
| } |
| |
| /** |
| * Verify availability of resources at the given web locations. Normally this would be done using an HTTP HEAD. |
| * |
| * @param locations |
| * the locations of the resource to verify |
| * @param one |
| * indicate if only one of the resources must exist |
| * @param monitor |
| * the monitor |
| * @return true if the resource exists |
| * @throws CoreException |
| * @throws AuthenticationFailedException |
| */ |
| public static boolean verifyAvailability(List<? extends URI> locations, boolean one, IProgressMonitor monitor) throws IOException, CoreException { |
| if (locations.isEmpty() || locations.size() > 5) { |
| throw new IllegalArgumentException(); |
| } |
| int countFound = 0; |
| for (URI location : locations) { |
| try { |
| new RepositoryTransport().getLastModified(location, monitor); |
| if (one) { |
| return true; |
| } |
| ++countFound; |
| } catch (FileNotFoundException e) { |
| if (!one) { |
| return false; |
| } |
| continue; |
| } |
| } |
| return countFound == locations.size(); |
| } |
| |
| public static String getFileNameFor(String bundleUrl) { |
| if (bundleUrl.charAt(bundleUrl.length() - 1) == '/') { |
| bundleUrl = bundleUrl.substring(0, bundleUrl.length() - 1); |
| } |
| if (bundleUrl.lastIndexOf('/') != -1) { |
| bundleUrl = bundleUrl.substring(bundleUrl.lastIndexOf('/') + 1); |
| } |
| return bundleUrl.replaceAll("[^a-zA-Z0-9_\\.]", "_"); //$NON-NLS-1$ //$NON-NLS-2$ |
| } |
| |
| } |