java/java-impl/src/com/intellij/refactoring/introduceparameterobject/usageInfo/AppendAccessorsUsageInfo.java

/*
 * 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.
 */

/*
 * User: anna
 * Date: 02-Nov-2009
 */
package com.intellij.refactoring.introduceparameterobject.usageInfo;

import com.intellij.codeInsight.generation.GenerateMembersUtil;
import com.intellij.openapi.diagnostic.Logger;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiField;
import com.intellij.psi.PsiParameter;
import com.intellij.refactoring.RefactorJBundle;
import com.intellij.refactoring.introduceparameterobject.IntroduceParameterObjectProcessor;
import com.intellij.refactoring.util.FixableUsageInfo;
import com.intellij.util.Function;
import com.intellij.util.IncorrectOperationException;

import java.util.List;
import java.util.Set;

public class AppendAccessorsUsageInfo extends FixableUsageInfo{
  private final boolean myGenerateAccessors;
  private final Set<PsiParameter> paramsNeedingSetters;
  private final Set<PsiParameter> paramsNeedingGetters;
  private final List<IntroduceParameterObjectProcessor.ParameterChunk> parameters;
  private static final Logger LOGGER = Logger.getInstance("#" + AppendAccessorsUsageInfo.class.getName());


  public AppendAccessorsUsageInfo(PsiElement psiClass, boolean generateAccessors, Set<PsiParameter> paramsNeedingGetters,
                                  Set<PsiParameter> paramsNeedingSetters, List<IntroduceParameterObjectProcessor.ParameterChunk> parameters) {
    super(psiClass);
    myGenerateAccessors = generateAccessors;
    this.paramsNeedingGetters = paramsNeedingGetters;
    this.paramsNeedingSetters = paramsNeedingSetters;
    this.parameters = parameters;
  }

  @Override
  public void fixUsage() throws IncorrectOperationException {
    if (myGenerateAccessors) {
      appendAccessors(paramsNeedingGetters, true);
      appendAccessors(paramsNeedingSetters, false);
    }
  }

  private void appendAccessors(final Set<PsiParameter> params, boolean isGetter) {
    final PsiElement element = getElement();
    if (element != null) {
      for (PsiParameter parameter : params) {
        final IntroduceParameterObjectProcessor.ParameterChunk parameterChunk =
          IntroduceParameterObjectProcessor.ParameterChunk.getChunkByParameter(parameter, parameters);
        LOGGER.assertTrue(parameterChunk != null);
        final PsiField field = parameterChunk.getField();
        if (field != null) {
          element.add(isGetter
                      ? GenerateMembersUtil.generateGetterPrototype(field)
                      : GenerateMembersUtil.generateSetterPrototype(field));
        }

      }
    }
  }

  @Override
  public String getConflictMessage() {
    if (!myGenerateAccessors && (!paramsNeedingSetters.isEmpty() || !paramsNeedingGetters.isEmpty())) {
      final StringBuffer buf = new StringBuffer();
      appendConflicts(buf, paramsNeedingGetters);
      appendConflicts(buf, paramsNeedingSetters);
      return RefactorJBundle.message("cannot.perform.the.refactoring") + buf.toString();
    }
    return null;
  }

  private void appendConflicts(StringBuffer buf, final Set<PsiParameter> paramsNeeding) {
    if (!paramsNeeding.isEmpty()) {
      buf.append(paramsNeeding == paramsNeedingGetters ? "Getters" : "Setters");
      buf.append(" for the following fields are required:\n");
      buf.append(StringUtil.join(paramsNeeding, new Function<PsiParameter, String>() {
        public String fun(PsiParameter psiParameter) {
          final IntroduceParameterObjectProcessor.ParameterChunk chunk =
            IntroduceParameterObjectProcessor.ParameterChunk.getChunkByParameter(psiParameter, parameters);
          if (chunk != null) {
            final PsiField field = chunk.getField();
            if (field != null) {
              return field.getName();
            }
          }
          return psiParameter.getName();
        }
      }, ", "));
      buf.append(".\n");
    }
  }
}