com.google.cloud.tools.intellij.settings.ExportSettings.java Source code

Introduction

Here is the source code for com.google.cloud.tools.intellij.settings.ExportSettings.java

Source

/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * 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.google.cloud.tools.intellij.settings;

import com.intellij.ide.IdeBundle;
import com.intellij.ide.plugins.IdeaPluginDescriptor;
import com.intellij.ide.plugins.PluginManager;
import com.intellij.ide.plugins.PluginManagerCore;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.application.PathManager;
import com.intellij.openapi.components.ExportableApplicationComponent;
import com.intellij.openapi.components.ExportableComponent;
import com.intellij.openapi.components.ServiceBean;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.util.io.FileUtil;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.io.ZipUtil;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.jar.JarOutputStream;

/**
 * Provides functionality to export the IDEA settings into a specified jar.
 */
public class ExportSettings {

    // TODO: use ImportSettings.SETTINGS_JAR_MARKER after merge
    public static final String SETTINGS_JAR_MARKER = "IntelliJ IDEA Global Settings";

    /**
     * Exports IDEA settings.
     */
    public static void doExport(String path) {
        final Set<ExportableComponent> exportableComponents = new HashSet<ExportableComponent>(Arrays
                .asList(ApplicationManager.getApplication().getComponents(ExportableApplicationComponent.class)));
        exportableComponents.addAll(
                ServiceBean.loadServicesFromBeans(ExportableComponent.EXTENSION_POINT, ExportableComponent.class));

        if (exportableComponents.isEmpty()) {
            return;
        }

        Set<File> exportFiles = new HashSet<File>();
        for (final ExportableComponent markedComponent : exportableComponents) {
            ContainerUtil.addAll(exportFiles, markedComponent.getExportFiles());
        }

        ApplicationManager.getApplication().saveSettings();

        final File saveFile = new File(path);
        try {
            if (saveFile.exists()) {
                final int ret = Messages.showOkCancelDialog(
                        IdeBundle.message("prompt.overwrite.settings.file",
                                FileUtil.toSystemDependentName(saveFile.getPath())),
                        IdeBundle.message("title.file.already.exists"), Messages.getWarningIcon());
                if (ret != Messages.OK) {
                    return;
                }
            }

            final JarOutputStream output = new JarOutputStream(new FileOutputStream(saveFile));
            try {
                final File configPath = new File(PathManager.getConfigPath());
                final Set<String> writtenItemRelativePaths = new HashSet<String>();
                for (File file : exportFiles) {
                    final String rPath = FileUtil.getRelativePath(configPath, file);
                    assert rPath != null;
                    final String relativePath = FileUtil.toSystemIndependentName(rPath);
                    if (file.exists()) {
                        ZipUtil.addFileOrDirRecursively(output, saveFile, file, relativePath, null,
                                writtenItemRelativePaths);
                    }
                }

                exportInstalledPlugins(saveFile, output, writtenItemRelativePaths);

                final File magicFile = new File(FileUtil.getTempDirectory(), SETTINGS_JAR_MARKER);
                FileUtil.createIfDoesntExist(magicFile);
                magicFile.deleteOnExit();
                ZipUtil.addFileToZip(output, magicFile, SETTINGS_JAR_MARKER, writtenItemRelativePaths, null);
            } finally {
                output.close();
            }
        } catch (IOException e1) {
            Messages.showErrorDialog(IdeBundle.message("error.writing.settings", e1.toString()),
                    IdeBundle.message("title.error.writing.file"));
        }

    }

    private static void exportInstalledPlugins(File saveFile, JarOutputStream output,
            Set<String> writtenItemRelativePaths) throws IOException {
        final List<String> oldPlugins = new ArrayList<String>();
        for (IdeaPluginDescriptor descriptor : PluginManagerCore.getPlugins()) {
            if (!descriptor.isBundled() && descriptor.isEnabled()) {
                oldPlugins.add(descriptor.getPluginId().getIdString());
            }
        }
        if (!oldPlugins.isEmpty()) {
            final File tempFile = File.createTempFile("installed", "plugins");
            tempFile.deleteOnExit();
            PluginManagerCore.savePluginsList(oldPlugins, false, tempFile);
            ZipUtil.addDirToZipRecursively(output, saveFile, tempFile, "/" + PluginManager.INSTALLED_TXT, null,
                    writtenItemRelativePaths);
        }
    }
}