remove obsolete dialog
[idea/community.git] / jps / jps-builders / src / org / jetbrains / jps / incremental / instrumentation / NotNullInstrumentingBuilder.java
1 /*
2  * Copyright 2000-2012 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.jetbrains.jps.incremental.instrumentation;
17
18 import com.intellij.compiler.instrumentation.FailSafeClassReader;
19 import com.intellij.compiler.instrumentation.InstrumentationClassFinder;
20 import com.intellij.compiler.notNullVerification.NotNullVerifyingInstrumenter;
21 import com.intellij.openapi.diagnostic.Logger;
22 import org.jetbrains.annotations.NotNull;
23 import org.jetbrains.annotations.Nullable;
24 import org.jetbrains.jps.ModuleChunk;
25 import org.jetbrains.jps.cmdline.ProjectDescriptor;
26 import org.jetbrains.jps.incremental.BinaryContent;
27 import org.jetbrains.jps.incremental.CompileContext;
28 import org.jetbrains.jps.incremental.CompiledClass;
29 import org.jetbrains.jps.incremental.messages.BuildMessage;
30 import org.jetbrains.jps.incremental.messages.CompilerMessage;
31 import org.jetbrains.jps.model.java.JpsJavaExtensionService;
32 import org.jetbrains.org.objectweb.asm.ClassReader;
33 import org.jetbrains.org.objectweb.asm.ClassWriter;
34 import org.jetbrains.org.objectweb.asm.Opcodes;
35
36 import java.io.File;
37
38 /**
39  * @author Eugene Zhuravlev
40  *         Date: 11/21/12
41  */
42 public class NotNullInstrumentingBuilder extends BaseInstrumentingBuilder{
43   private static final Logger LOG = Logger.getInstance("#org.jetbrains.jps.incremental.instrumentation.NotNullInstrumentingBuilder");
44
45   public NotNullInstrumentingBuilder() {
46   }
47
48   @NotNull
49   @Override
50   public String getPresentableName() {
51     return "NotNull instrumentation";
52   }
53
54   @Override
55   protected String getProgressMessage() {
56     return "Adding @NotNull assertions...";
57   }
58
59   @Override
60   protected boolean isEnabled(CompileContext context, ModuleChunk chunk) {
61     final ProjectDescriptor pd = context.getProjectDescriptor();
62     return JpsJavaExtensionService.getInstance().getOrCreateCompilerConfiguration(pd.getProject()).isAddNotNullAssertions();
63   }
64
65   @Override
66   protected boolean canInstrument(CompiledClass compiledClass, int classFileVersion) {
67     return classFileVersion >= Opcodes.V1_5;
68   }
69
70   // todo: probably instrument other NotNull-like annotations defined in project settings?
71   @Override
72   @Nullable
73   protected BinaryContent instrument(CompileContext context,
74                                      CompiledClass compiledClass,
75                                      ClassReader reader,
76                                      ClassWriter writer,
77                                      InstrumentationClassFinder finder) {
78     try {
79       if (NotNullVerifyingInstrumenter.processClassFile((FailSafeClassReader)reader, writer)) {
80         return new BinaryContent(writer.toByteArray());
81       }
82     }
83     catch (Throwable e) {
84       LOG.error(e);
85       final File sourceFile = compiledClass.getSourceFile();
86       String msg = "Cannot instrument " + sourceFile.getName() + ": " + e.getMessage();
87       context.processMessage(new CompilerMessage(getPresentableName(), BuildMessage.Kind.ERROR, msg, sourceFile.getPath()));
88     }
89     return null;
90   }
91
92 }