/*
* Copyright 2000-2016 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.application.options.colors;
import com.intellij.application.options.schemes.AbstractSchemeActions;
import com.intellij.application.options.schemes.AbstractSchemesPanel;
import com.intellij.application.options.schemes.SchemeNameGenerator;
import com.intellij.openapi.application.ApplicationBundle;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.editor.colors.EditorColorsScheme;
import com.intellij.openapi.editor.colors.impl.AbstractColorsScheme;
import com.intellij.openapi.editor.colors.impl.DefaultColorsScheme;
import com.intellij.openapi.editor.colors.impl.EditorColorsSchemeImpl;
import com.intellij.openapi.editor.colors.impl.EmptyColorScheme;
import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.options.*;
import com.intellij.openapi.project.DefaultProjectFactory;
import com.intellij.openapi.ui.MessageType;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.ui.popup.JBPopupFactory;
import com.intellij.openapi.ui.popup.ListPopup;
import com.intellij.openapi.ui.popup.PopupStep;
import com.intellij.openapi.ui.popup.util.BaseListPopupStep;
import com.intellij.openapi.util.Pair;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.VirtualFile;
import org.jdom.Element;
import org.jetbrains.annotations.NotNull;
import java.awt.*;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
public abstract class ColorSchemeActions extends AbstractSchemeActions<EditorColorsScheme> {
protected ColorSchemeActions(@NotNull AbstractSchemesPanel<EditorColorsScheme, ?> schemesPanel) {
super(schemesPanel);
}
@Override
protected Collection<String> getSchemeImportersNames() {
List<String> importersNames = new ArrayList<>();
for (ImportHandler importHandler : Extensions.getExtensions(ImportHandler.EP_NAME)) {
importersNames.add(importHandler.getTitle());
}
importersNames.addAll(super.getSchemeImportersNames());
return importersNames;
}
@Override
protected void importScheme(@NotNull String importerName) {
if (tryImportWithImportHandler(importerName)) {
return;
}
final SchemeImporter<EditorColorsScheme> importer = SchemeImporterEP.getImporter(importerName, EditorColorsScheme.class);
if (importer != null) {
VirtualFile importSource =
SchemeImportUtil.selectImportSource(importer.getSourceExtensions(), getSchemesPanel(), null, "Choose " + importerName);
if (importSource != null) {
if ("jar".equals(importSource.getExtension())) {
importFromJar(getSchemesPanel().getToolbar(), importer, importSource);
}
else {
doImport(importer, importSource);
}
}
}
}
private void doImport(@NotNull SchemeImporter<EditorColorsScheme> importer, @NotNull VirtualFile importSource) {
try {
EditorColorsScheme imported =
importer.importScheme(DefaultProjectFactory.getInstance().getDefaultProject(), importSource, getOptions().getSelectedScheme(),
name -> {
String newName = SchemeNameGenerator.getUniqueName(name != null ? name : "Unnamed",
candidate -> getSchemesPanel().getModel()
.containsScheme(candidate, false));
AbstractColorsScheme newScheme = new EditorColorsSchemeImpl(EmptyColorScheme.INSTANCE);
newScheme.setName(newName);
newScheme.setDefaultMetaInfo(EmptyColorScheme.INSTANCE);
return newScheme;
});
if (imported != null) {
getOptions().addImportedScheme(imported);
getSchemesPanel()
.showStatus(
ApplicationBundle.message("settings.editor.scheme.import.success", importSource.getPresentableUrl(), imported.getName()),
MessageType.INFO);
}
}
catch (SchemeImportException e) {
handleError(e, importSource);
}
}
private void handleError(@NotNull SchemeImportException e, @NotNull VirtualFile importSource) {
String details = e.getMessage();
getSchemesPanel()
.showStatus(
ApplicationBundle.message("settings.editor.scheme.import.failure", importSource.getPresentableUrl()) +
(StringUtil.isEmpty(details) ? "" : "\n" + details),
MessageType.ERROR);
}
private void importFromJar(@NotNull Component componentAbove,
@NotNull SchemeImporter<EditorColorsScheme> importer,
@NotNull VirtualFile jarFile) {
try {
List<VirtualFile> schemeFiles = getSchemeFiles(jarFile);
if (schemeFiles.size() == 1 || ApplicationManager.getApplication().isUnitTestMode()) {
doImport(importer, schemeFiles.iterator().next());
return;
}
List<Pair<String, VirtualFile>> fileList = new ArrayList<>(schemeFiles.size());
for (VirtualFile file : schemeFiles) {
Element root = SchemeImportUtil.loadSchemeDom(file);
String name = StringUtil.trimStart(ColorSchemeImporter.getSchemeName(root), SchemeManager.EDITABLE_COPY_PREFIX);
fileList.add(Pair.create(name, file));
}
ListPopup popup = JBPopupFactory.getInstance().createListPopup
(new BaseListPopupStep<Pair<String, VirtualFile>>("Choose scheme to import", fileList) {
@NotNull
@Override
public String getTextFor(Pair<String, VirtualFile> item) {
return item.first;
}
@Override
public PopupStep onChosen(Pair<String, VirtualFile> selectedValue, boolean finalChoice) {
return doFinalStep(() -> doImport(importer, selectedValue.second));
}
});
popup.showUnderneathOf(componentAbove);
}
catch (SchemeImportException e) {
handleError(e, jarFile);
}
}
private static List<VirtualFile> getSchemeFiles(@NotNull VirtualFile jarFile) throws SchemeImportException {
List<VirtualFile> schemeFiles = new ArrayList<>();
for (VirtualFile file : jarFile.getChildren()) {
if (file.isDirectory() && "colors".equals(file.getName())) {
for (VirtualFile schemeFile : file.getChildren()) {
String ext = schemeFile.getExtension();
if ("icls".equals(ext) || "xml".equals(ext)) {
schemeFiles.add(schemeFile);
}
}
break;
}
}
if (schemeFiles.isEmpty()) {
throw new SchemeImportException("The are no color schemes in the chosen file.");
}
return schemeFiles;
}
private boolean tryImportWithImportHandler(@NotNull String importerName) {
for (ImportHandler importHandler : Extensions.getExtensions(ImportHandler.EP_NAME)) {
if (importerName.equals(importHandler.getTitle())) {
importHandler.performImport(getSchemesPanel().getToolbar(), scheme -> {
if (scheme != null) getOptions().addImportedScheme(scheme);
});
return true;
}
}
return false;
}
@Override
protected void resetScheme(@NotNull EditorColorsScheme scheme) {
if (Messages
.showOkCancelDialog(ApplicationBundle.message("color.scheme.reset.message"),
ApplicationBundle.message("color.scheme.reset.title"), Messages.getQuestionIcon()) == Messages.OK) {
getOptions().resetSchemeToOriginal(scheme.getName());
}
}
@Override
protected void duplicateScheme(@NotNull EditorColorsScheme scheme, @NotNull String newName) {
getOptions().saveSchemeAs(scheme, newName);
}
@Override
protected void exportScheme(@NotNull EditorColorsScheme scheme, @NotNull String exporterName) {
EditorColorsScheme schemeToExport = scheme;
if (scheme instanceof AbstractColorsScheme) {
EditorColorsScheme parent = ((AbstractColorsScheme)scheme).getParentScheme();
if (!(parent instanceof DefaultColorsScheme)) {
schemeToExport = parent;
}
}
if (schemeToExport.getName().startsWith(SchemeManager.EDITABLE_COPY_PREFIX)) {
schemeToExport = (EditorColorsScheme)schemeToExport.clone();
schemeToExport.setName(SchemeManager.getDisplayName(schemeToExport));
}
super.exportScheme(schemeToExport, exporterName);
}
@Override
protected Class<EditorColorsScheme> getSchemeType() {
return EditorColorsScheme.class;
}
@NotNull
protected abstract ColorAndFontOptions getOptions();
}