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

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.PsiDirectory;
import com.intellij.psi.PsiFile;
import com.intellij.util.IncorrectOperationException;
import com.maddyhome.idea.copyright.CopyrightManager;
import com.maddyhome.idea.copyright.CopyrightProfile;
import com.maddyhome.idea.copyright.CopyrightUpdaters;
import com.maddyhome.idea.copyright.psi.UpdateCopyrightsProvider;
import com.maddyhome.idea.copyright.psi.UpdatePsiFileCopyright;

public class UpdateCopyrightProcessor extends AbstractFileProcessor {
  private static final Logger logger = Logger.getInstance(UpdateCopyrightProcessor.class.getName());

  public static final String TITLE = "Update Copyright";
  public static final String MESSAGE = "Updating copyrights...";

  private Project project;
  private Module module;

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

  public UpdateCopyrightProcessor(Project project, Module module, PsiDirectory dir, boolean subdirs) {
    super(project, dir, subdirs, TITLE, MESSAGE);
    setup(project, module);
  }

  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, null);
    setup(project, module);
  }

  @Override
  protected Runnable preprocessFile(final PsiFile file) 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();
    }

    UpdateCopyrightsProvider updateCopyrightsProvider = CopyrightUpdaters.INSTANCE.forFileType(file.getFileType());
    if(updateCopyrightsProvider == null) {
      return EmptyRunnable.getInstance();
    }

    CopyrightProfile copyrightProfile = CopyrightManager.getInstance(project).getCopyrightOptions(file);
    if (copyrightProfile != null && CopyrightUpdaters.hasExtension(file)) {
      logger.debug("process " + file);
      final UpdatePsiFileCopyright<?> updateCopyright = updateCopyrightsProvider.createInstance(file, copyrightProfile);

      return new Runnable() {
        @Override
        public void run() {
          try {
            updateCopyright.process();

          }
          catch (Exception e) {
            logger.error(e);
          }
        }
      };
    }
    else {
      return EmptyRunnable.getInstance();
    }
  }

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