UpdateCopyrightProcessor.java

/*
 * 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.maddyhome.idea.copyright.actions;

import com.intellij.copyright.CopyrightManager;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.ProjectRootManager;
import com.intellij.openapi.util.EmptyRunnable;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiFile;
import com.intellij.util.IncorrectOperationException;
import com.maddyhome.idea.copyright.CopyrightProfile;
import com.maddyhome.idea.copyright.psi.UpdateCopyright;
import com.maddyhome.idea.copyright.psi.UpdateCopyrightFactory;
import com.maddyhome.idea.copyright.psi.UpdatePsiFileCopyright;
import com.maddyhome.idea.copyright.util.FileTypeUtil;

public class UpdateCopyrightProcessor extends AbstractFileProcessor
{
    public static final String TITLE = "Update Copyright";
    public static final String MESSAGE = "Updating copyrights...";

    public UpdateCopyrightProcessor(Project project, Module module, PsiFile file)
    {
        super(project, file, TITLE, MESSAGE);
        setup(project, module);
    }

    public UpdateCopyrightProcessor(Project project, Module module, PsiFile[] files)
    {
        super(project, files, TITLE, MESSAGE);
        setup(project, module);
    }

    protected Runnable preprocessFile(final PsiFile file, final boolean allowReplacement) throws IncorrectOperationException
    {
        VirtualFile vfile = file.getVirtualFile();
        if (vfile == null) return EmptyRunnable.getInstance();
        final ProgressIndicator progressIndicator = ProgressManager.getInstance().getProgressIndicator();
        if (progressIndicator != null) {
            progressIndicator.setText2(vfile.getPresentableUrl());
        }
        Module mod = module;
        if (module == null)
        {
            mod = ProjectRootManager.getInstance(project).getFileIndex().getModuleForFile(vfile);
        }

        if (mod == null) return EmptyRunnable.getInstance();

        CopyrightProfile opts = CopyrightManager.getInstance(project).getCopyrightOptions(file);

        if (opts != null && FileTypeUtil.isSupportedFile(file))
        {
            logger.debug("process " + file);
            final UpdateCopyright update = UpdateCopyrightFactory.createUpdateCopyright(project, mod, file, opts);
            if (update == null) return EmptyRunnable.getInstance();
            update.prepare();

            if (update instanceof UpdatePsiFileCopyright && !((UpdatePsiFileCopyright)update).hasUpdates()) return EmptyRunnable.getInstance();

            return () -> {
                try
                {
                  if (update instanceof UpdatePsiFileCopyright) {
                    ((UpdatePsiFileCopyright)update).complete(allowReplacement);
                  } else {
                    update.complete();
                  }
                }
                catch (Exception e)
                {
                    logger.error(e);
                }
            };
        }
        else
        {
            return EmptyRunnable.getInstance();
        }
    }

    private void setup(Project project, Module module)
    {
        this.project = project;
        this.module = module;

    }

    private Project project;
    private Module module;

    private static final Logger logger = Logger.getInstance(UpdateCopyrightProcessor.class.getName());
}