mirror of
https://gitflic.ru/project/openide/openide.git
synced 2026-09-27 10:03:11 +07:00
[peter] WI-2813 download db connector automatically
This commit is contained in:
Gregory Shrago
@@ -1,327 +0,0 @@
|
||||
/*
|
||||
* Copyright 2000-2009 JetBrains s.r.o.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.intellij.facet.impl.ui.libraries;
|
||||
|
||||
import com.intellij.facet.ui.libraries.LibraryDownloadInfo;
|
||||
import com.intellij.facet.ui.libraries.LibraryInfo;
|
||||
import com.intellij.ide.IdeBundle;
|
||||
import com.intellij.openapi.application.PathManager;
|
||||
import com.intellij.openapi.application.Result;
|
||||
import com.intellij.openapi.application.WriteAction;
|
||||
import com.intellij.openapi.fileChooser.FileChooser;
|
||||
import com.intellij.openapi.fileChooser.FileChooserDescriptor;
|
||||
import com.intellij.openapi.fileChooser.FileChooserDescriptorFactory;
|
||||
import com.intellij.openapi.progress.ProcessCanceledException;
|
||||
import com.intellij.openapi.progress.ProgressIndicator;
|
||||
import com.intellij.openapi.progress.ProgressManager;
|
||||
import com.intellij.openapi.project.Project;
|
||||
import com.intellij.openapi.ui.Messages;
|
||||
import com.intellij.openapi.util.Pair;
|
||||
import com.intellij.openapi.util.Ref;
|
||||
import com.intellij.openapi.util.io.FileUtil;
|
||||
import com.intellij.openapi.util.text.StringUtil;
|
||||
import com.intellij.openapi.vfs.LocalFileSystem;
|
||||
import com.intellij.openapi.vfs.VfsUtil;
|
||||
import com.intellij.openapi.vfs.VirtualFile;
|
||||
import com.intellij.openapi.vfs.VirtualFileManager;
|
||||
import com.intellij.util.io.UrlConnectionUtil;
|
||||
import com.intellij.util.net.HttpConfigurable;
|
||||
import com.intellij.util.net.IOExceptionDialog;
|
||||
import org.jetbrains.annotations.NonNls;
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
|
||||
import javax.swing.*;
|
||||
import java.io.*;
|
||||
import java.net.HttpURLConnection;
|
||||
import java.net.URL;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* @author nik
|
||||
*/
|
||||
public class LibraryDownloader {
|
||||
private static final int CONNECTION_TIMEOUT = 60*1000;
|
||||
private static final int READ_TIMEOUT = 60*1000;
|
||||
@NonNls private static final String LIB_SCHEMA = "lib://";
|
||||
|
||||
private LibraryDownloadInfo[] myLibraryInfos;
|
||||
private JComponent myParent;
|
||||
private @Nullable Project myProject;
|
||||
private String myDirectoryForDownloadedLibrariesPath;
|
||||
private final @Nullable String myLibraryPresentableName;
|
||||
|
||||
public LibraryDownloader(final LibraryDownloadInfo[] libraryInfos, final @Nullable Project project, JComponent parent,
|
||||
@Nullable String directoryForDownloadedLibrariesPath, @Nullable String libraryPresentableName) {
|
||||
myProject = project;
|
||||
myLibraryInfos = libraryInfos;
|
||||
myParent = parent;
|
||||
myDirectoryForDownloadedLibrariesPath = directoryForDownloadedLibrariesPath;
|
||||
myLibraryPresentableName = libraryPresentableName;
|
||||
}
|
||||
|
||||
public VirtualFile[] download() {
|
||||
VirtualFile dir = null;
|
||||
if (myDirectoryForDownloadedLibrariesPath != null) {
|
||||
File ioDir = new File(FileUtil.toSystemDependentName(myDirectoryForDownloadedLibrariesPath));
|
||||
ioDir.mkdirs();
|
||||
dir = LocalFileSystem.getInstance().refreshAndFindFileByPath(myDirectoryForDownloadedLibrariesPath);
|
||||
}
|
||||
|
||||
if (dir == null) {
|
||||
dir = chooseDirectoryForLibraries();
|
||||
}
|
||||
|
||||
if (dir != null) {
|
||||
return doDownload(dir);
|
||||
}
|
||||
return VirtualFile.EMPTY_ARRAY;
|
||||
}
|
||||
|
||||
private VirtualFile[] doDownload(final VirtualFile dir) {
|
||||
HttpConfigurable.getInstance().setAuthenticator();
|
||||
final List<Pair<LibraryDownloadInfo, File>> downloadedFiles = new ArrayList<Pair<LibraryDownloadInfo, File>>();
|
||||
final List<VirtualFile> existingFiles = new ArrayList<VirtualFile>();
|
||||
final Ref<Exception> exceptionRef = Ref.create(null);
|
||||
final Ref<LibraryDownloadInfo> currentLibrary = new Ref<LibraryDownloadInfo>();
|
||||
|
||||
String dialogTitle = IdeBundle.message("progress.download.libraries.title");
|
||||
if (myLibraryPresentableName != null) {
|
||||
dialogTitle = IdeBundle.message("progress.download.0.libraries.title", StringUtil.capitalize(myLibraryPresentableName));
|
||||
}
|
||||
|
||||
ProgressManager.getInstance().runProcessWithProgressSynchronously(new Runnable() {
|
||||
public void run() {
|
||||
final ProgressIndicator indicator = ProgressManager.getInstance().getProgressIndicator();
|
||||
try {
|
||||
for (int i = 0; i < myLibraryInfos.length; i++) {
|
||||
LibraryDownloadInfo info = myLibraryInfos[i];
|
||||
currentLibrary.set(info);
|
||||
if (indicator != null) {
|
||||
indicator.checkCanceled();
|
||||
indicator.setText(IdeBundle.message("progress.0.of.1.file.downloaded.text", i, myLibraryInfos.length));
|
||||
}
|
||||
|
||||
final VirtualFile existing = dir.findChild(getExpectedFileName(info));
|
||||
long size = existing != null ? existing.getLength() : -1;
|
||||
|
||||
if (!download(info, size, downloadedFiles)) {
|
||||
existingFiles.add(existing);
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (ProcessCanceledException e) {
|
||||
exceptionRef.set(e);
|
||||
}
|
||||
catch (IOException e) {
|
||||
exceptionRef.set(e);
|
||||
}
|
||||
}
|
||||
}, dialogTitle, true, myProject, myParent);
|
||||
|
||||
Exception exception = exceptionRef.get();
|
||||
if (exception == null) {
|
||||
try {
|
||||
return moveToDir(existingFiles, downloadedFiles, dir);
|
||||
}
|
||||
catch (IOException e) {
|
||||
final String title = IdeBundle.message("progress.download.libraries.title");
|
||||
if (myProject != null) {
|
||||
Messages.showErrorDialog(myProject, title, e.getMessage());
|
||||
}
|
||||
else {
|
||||
Messages.showErrorDialog(myParent, title, e.getMessage());
|
||||
}
|
||||
return VirtualFile.EMPTY_ARRAY;
|
||||
}
|
||||
}
|
||||
|
||||
deleteFiles(downloadedFiles);
|
||||
if (exception instanceof IOException) {
|
||||
String message = IdeBundle.message("error.library.download.failed", exception.getMessage());
|
||||
if (currentLibrary.get() != null) {
|
||||
message += ": " + currentLibrary.get().getDownloadUrl();
|
||||
}
|
||||
final boolean tryAgain = IOExceptionDialog.showErrorDialog(IdeBundle.message("progress.download.libraries.title"), message);
|
||||
if (tryAgain) {
|
||||
return doDownload(dir);
|
||||
}
|
||||
}
|
||||
return VirtualFile.EMPTY_ARRAY;
|
||||
}
|
||||
|
||||
private @Nullable VirtualFile chooseDirectoryForLibraries() {
|
||||
final FileChooserDescriptor descriptor = FileChooserDescriptorFactory.createSingleFolderDescriptor();
|
||||
descriptor.setTitle(IdeBundle.message("dialog.directory.for.libraries.title"));
|
||||
|
||||
final VirtualFile[] files;
|
||||
if (myProject != null) {
|
||||
files = FileChooser.chooseFiles(myProject, descriptor, myProject.getBaseDir());
|
||||
}
|
||||
else {
|
||||
files = FileChooser.chooseFiles(myParent, descriptor);
|
||||
}
|
||||
|
||||
return files.length > 0 ? files[0] : null;
|
||||
}
|
||||
|
||||
private static VirtualFile[] moveToDir(final List<VirtualFile> existingFiles, final List<Pair<LibraryDownloadInfo, File>> downloadedFiles, final VirtualFile dir) throws IOException {
|
||||
List<VirtualFile> files = new ArrayList<VirtualFile>();
|
||||
|
||||
final File ioDir = VfsUtil.virtualToIoFile(dir);
|
||||
for (Pair<LibraryDownloadInfo, File> pair : downloadedFiles) {
|
||||
final LibraryDownloadInfo info = pair.getFirst();
|
||||
final boolean dontTouch = info.getDownloadUrl().startsWith(LIB_SCHEMA) || info.getDownloadUrl().startsWith(LocalFileSystem.PROTOCOL_PREFIX);
|
||||
final File toFile = dontTouch? pair.getSecond() : generateName(info, ioDir);
|
||||
if (!dontTouch) {
|
||||
FileUtil.rename(pair.getSecond(), toFile);
|
||||
}
|
||||
VirtualFile file = new WriteAction<VirtualFile>() {
|
||||
protected void run(final Result<VirtualFile> result) {
|
||||
final String url = VfsUtil.getUrlForLibraryRoot(toFile);
|
||||
LocalFileSystem.getInstance().refreshAndFindFileByIoFile(toFile);
|
||||
result.setResult(VirtualFileManager.getInstance().refreshAndFindFileByUrl(url));
|
||||
}
|
||||
}.execute().getResultObject();
|
||||
if (file != null) {
|
||||
files.add(file);
|
||||
}
|
||||
}
|
||||
|
||||
for (final VirtualFile file : existingFiles) {
|
||||
VirtualFile libraryRootFile = new WriteAction<VirtualFile>() {
|
||||
protected void run(final Result<VirtualFile> result) {
|
||||
final String url = VfsUtil.getUrlForLibraryRoot(VfsUtil.virtualToIoFile(file));
|
||||
result.setResult(VirtualFileManager.getInstance().refreshAndFindFileByUrl(url));
|
||||
}
|
||||
|
||||
}.execute().getResultObject();
|
||||
if (libraryRootFile != null) {
|
||||
files.add(libraryRootFile);
|
||||
}
|
||||
}
|
||||
|
||||
return VfsUtil.toVirtualFileArray(files);
|
||||
}
|
||||
|
||||
private static String getExpectedFileName(LibraryDownloadInfo info) {
|
||||
return info.getFileNamePrefix() + info.getFileNameSuffix();
|
||||
}
|
||||
|
||||
private static File generateName(LibraryDownloadInfo info, final File dir) {
|
||||
String prefix = info.getFileNamePrefix();
|
||||
String suffix = info.getFileNameSuffix();
|
||||
File file = new File(dir, prefix + suffix);
|
||||
int count = 1;
|
||||
while (file.exists()) {
|
||||
file = new File(dir, prefix + "_" + count + suffix);
|
||||
count++;
|
||||
}
|
||||
return file;
|
||||
}
|
||||
|
||||
private static void deleteFiles(final List<Pair<LibraryDownloadInfo, File>> pairs) {
|
||||
for (Pair<LibraryDownloadInfo, File> pair : pairs) {
|
||||
FileUtil.delete(pair.getSecond());
|
||||
}
|
||||
pairs.clear();
|
||||
}
|
||||
|
||||
private boolean download(final LibraryDownloadInfo libraryInfo, final long existingFileSize, final List<Pair<LibraryDownloadInfo, File>> downloadedFiles) throws IOException {
|
||||
final ProgressIndicator indicator = ProgressManager.getInstance().getProgressIndicator();
|
||||
final String presentableUrl = libraryInfo.getPresentableUrl();
|
||||
final String url = libraryInfo.getDownloadUrl();
|
||||
if (url.startsWith(LIB_SCHEMA)) {
|
||||
indicator.setText2(IdeBundle.message("progress.locate.jar.text", getExpectedFileName(libraryInfo)));
|
||||
final String path = url.substring(LIB_SCHEMA.length()).replace('/', File.separatorChar);
|
||||
final File file = PathManager.findFileInLibDirectory(path);
|
||||
downloadedFiles.add(Pair.create(libraryInfo, file));
|
||||
} else if (url.startsWith(LocalFileSystem.PROTOCOL_PREFIX)) {
|
||||
String path = url.substring(LocalFileSystem.PROTOCOL_PREFIX.length()).replace('/', File.separatorChar).replace('\\', File.separatorChar);
|
||||
File file = new File(path);
|
||||
if (file.exists()) downloadedFiles.add(Pair.create(libraryInfo, file));
|
||||
} else {
|
||||
indicator.setText2(IdeBundle.message("progress.connecting.to.dowload.jar.text", presentableUrl));
|
||||
indicator.setIndeterminate(true);
|
||||
HttpURLConnection connection = (HttpURLConnection)new URL(url).openConnection();
|
||||
connection.setConnectTimeout(CONNECTION_TIMEOUT);
|
||||
connection.setReadTimeout(READ_TIMEOUT);
|
||||
|
||||
InputStream input = null;
|
||||
BufferedOutputStream output = null;
|
||||
|
||||
boolean deleteFile = true;
|
||||
File tempFile = null;
|
||||
try {
|
||||
final int responseCode = connection.getResponseCode();
|
||||
if (responseCode != HttpURLConnection.HTTP_OK) {
|
||||
throw new IOException(IdeBundle.message("error.connection.failed.with.http.code.N", responseCode));
|
||||
}
|
||||
|
||||
final int size = connection.getContentLength();
|
||||
if (size != -1 && size == existingFileSize) {
|
||||
return false;
|
||||
}
|
||||
|
||||
tempFile = FileUtil.createTempFile("downloaded", "jar");
|
||||
input = UrlConnectionUtil.getConnectionInputStreamWithException(connection, indicator);
|
||||
output = new BufferedOutputStream(new FileOutputStream(tempFile));
|
||||
indicator.setText2(IdeBundle.message("progress.download.jar.text", getExpectedFileName(libraryInfo), presentableUrl));
|
||||
indicator.setIndeterminate(size == -1);
|
||||
|
||||
int len;
|
||||
final byte[] buf = new byte[1024];
|
||||
int count = 0;
|
||||
while ((len = input.read(buf)) > 0) {
|
||||
indicator.checkCanceled();
|
||||
count += len;
|
||||
if (size > 0) {
|
||||
indicator.setFraction((double)count / size);
|
||||
}
|
||||
output.write(buf, 0, len);
|
||||
}
|
||||
|
||||
deleteFile = false;
|
||||
downloadedFiles.add(Pair.create(libraryInfo, tempFile));
|
||||
}
|
||||
finally {
|
||||
if (input != null) {
|
||||
input.close();
|
||||
}
|
||||
if (output != null) {
|
||||
output.close();
|
||||
}
|
||||
if (deleteFile && tempFile != null) {
|
||||
FileUtil.delete(tempFile);
|
||||
}
|
||||
connection.disconnect();
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
public static LibraryDownloadInfo[] getDownloadingInfos(final LibraryInfo[] libraries) {
|
||||
List<LibraryDownloadInfo> downloadInfos = new ArrayList<LibraryDownloadInfo>();
|
||||
for (LibraryInfo library : libraries) {
|
||||
LibraryDownloadInfo downloadInfo = library.getDownloadingInfo();
|
||||
if (downloadInfo != null) {
|
||||
downloadInfos.add(downloadInfo);
|
||||
}
|
||||
}
|
||||
return downloadInfos.toArray(new LibraryDownloadInfo[downloadInfos.size()]);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user