001    /*
002     * Copyright 2010-2013 JetBrains s.r.o.
003     *
004     * Licensed under the Apache License, Version 2.0 (the "License");
005     * you may not use this file except in compliance with the License.
006     * You may obtain a copy of the License at
007     *
008     * http://www.apache.org/licenses/LICENSE-2.0
009     *
010     * Unless required by applicable law or agreed to in writing, software
011     * distributed under the License is distributed on an "AS IS" BASIS,
012     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013     * See the License for the specific language governing permissions and
014     * limitations under the License.
015     */
016    
017    package org.jetbrains.jet.lang.diagnostics;
018    
019    import com.intellij.openapi.util.TextRange;
020    import com.intellij.psi.PsiElement;
021    import com.intellij.psi.PsiFile;
022    import org.jetbrains.annotations.NotNull;
023    
024    import java.util.List;
025    
026    public abstract class AbstractDiagnostic<E extends PsiElement> implements ParametrizedDiagnostic<E> {
027        private final E psiElement;
028        private final DiagnosticFactoryWithPsiElement<E, ?> factory;
029        private final Severity severity;
030        private List<TextRange> textRanges;
031    
032        public AbstractDiagnostic(@NotNull E psiElement,
033                @NotNull DiagnosticFactoryWithPsiElement<E, ?> factory,
034                @NotNull Severity severity) {
035            this.psiElement = psiElement;
036            this.factory = factory;
037            this.severity = severity;
038        }
039    
040        @NotNull
041        @Override
042        public DiagnosticFactoryWithPsiElement<E, ?> getFactory() {
043            return factory;
044        }
045    
046        @NotNull
047        @Override
048        public PsiFile getPsiFile() {
049            return psiElement.getContainingFile();
050        }
051    
052        @NotNull
053        @Override
054        public Severity getSeverity() {
055            return severity;
056        }
057    
058        @Override
059        @NotNull
060        public E getPsiElement() {
061            return psiElement;
062        }
063    
064        @Override
065        @NotNull
066        public List<TextRange> getTextRanges() {
067            if (textRanges != null) {
068                return textRanges;
069            }
070            return getFactory().getTextRanges(this);
071        }
072    
073        @Override
074        public boolean isValid() {
075            if (!getFactory().isValid(this)) return false;
076            return true;
077        }
078    
079        @NotNull
080        public Diagnostic setTextRanges(@NotNull List<TextRange> textRanges) {
081            this.textRanges = textRanges;
082            return this;
083        }
084    }