/*
 * Copyright 2000-2014 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.
 */

/*
 * @author max
 */
package com.intellij.ide.highlighter;

import com.intellij.ide.structureView.StructureViewBuilder;
import com.intellij.ide.structureView.StructureViewBuilderProvider;
import com.intellij.ide.structureView.StructureViewModel;
import com.intellij.ide.structureView.TreeBasedStructureViewBuilder;
import com.intellij.ide.structureView.impl.java.JavaFileTreeModel;
import com.intellij.lang.Language;
import com.intellij.lang.LanguageStructureViewBuilder;
import com.intellij.lang.PsiStructureViewFactory;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.extensions.Extensions;
import com.intellij.openapi.fileTypes.ContentBasedFileSubstitutor;
import com.intellij.openapi.fileTypes.FileType;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiJavaFile;
import com.intellij.psi.PsiManager;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

public class JavaClsStructureViewBuilderProvider implements StructureViewBuilderProvider {
  @Override
  @Nullable
  public StructureViewBuilder getStructureViewBuilder(@NotNull final FileType fileType, @NotNull final VirtualFile file, @NotNull final Project project) {

    final PsiFile psiFile = PsiManager.getInstance(project).findFile(file);

    final ContentBasedFileSubstitutor[] processors = Extensions.getExtensions(ContentBasedFileSubstitutor.EP_NAME);
    for (ContentBasedFileSubstitutor processor : processors) {
      if (processor.isApplicable(project, file)) {
        final Language language = processor.obtainLanguageForFile(file);
        if (language != null) {
          final PsiStructureViewFactory factory = LanguageStructureViewBuilder.INSTANCE.forLanguage(language);
          if (factory != null) return factory.getStructureViewBuilder(psiFile);
        }
      }
    }

    final PsiJavaFile javaFile = (PsiJavaFile)psiFile;
    if (javaFile == null) return null;
    return new TreeBasedStructureViewBuilder() {
      @Override
      @NotNull
      public StructureViewModel createStructureViewModel(@Nullable Editor editor) {
        return new JavaFileTreeModel(javaFile, editor);
      }
    };
  }
}
@@ -597,31 +597,8 @@
 %0A */
-%0A%0A/*%0A * @author max%0A */
 %0Apac
@@ -1095,16 +1095,60 @@
 actory;%0A
+import com.intellij.lang.java.JavaLanguage;%0A
 import c
@@ -1465,20 +1465,26 @@
 .psi.Psi
-File
+ClassOwner
 ;%0Aimport
@@ -1504,20 +1504,16 @@
 .psi.Psi
-Java
 File;%0Aim
@@ -1632,16 +1632,39 @@
 lable;%0A%0A
+/**%0A * @author max%0A */%0A
 public c
@@ -1833,22 +1833,16 @@
 NotNull 
-final 
 FileType
@@ -1861,22 +1861,16 @@
 NotNull 
-final 
 VirtualF
@@ -1888,22 +1888,16 @@
 NotNull 
-final 
 Project 
@@ -1907,17 +1907,16 @@
 ject) %7B%0A
-%0A
     fina
@@ -1992,57 +1992,467 @@
     
-final ContentBasedFileSubstitutor%5B%5D processors
+if (fileType == JavaClassFileType.INSTANCE && psiFile != null) %7B%0A      Language language = psiFile.getLanguage();%0A      if (language != JavaLanguage.INSTANCE) %7B%0A        PsiStructureViewFactory factory
  = 
+LanguageStructureViewBuilder.INSTANCE.forLanguage(language);%0A        if (factory != null) %7B%0A          return factory.getStructureViewBuilder(psiFile);%0A        %7D%0A      %7D%0A    %7D%0A%0A    //noinspection deprecation%0A    for (ContentBasedFileSubstitutor processor : 
 Exte
@@ -2512,69 +2512,8 @@
 AME)
-;%0A    for (ContentBasedFileSubstitutor processor : processors
 ) %7B%0A
@@ -2892,98 +2892,50 @@
     
-final PsiJavaFile javaFile = (PsiJavaFile)psiFile;%0A    if (javaFile == null) return null;%0A
+if (psiFile instanceof PsiClassOwner) %7B%0A  
     
@@ -2975,24 +2975,26 @@
 Builder() %7B%0A
+  
       @Overr
@@ -3003,16 +3003,18 @@
 e%0A      
+  
 @NotNull
@@ -3014,16 +3014,18 @@
 NotNull%0A
+  
       pu
@@ -3104,24 +3104,26 @@
 ) %7B%0A        
+  
 return new J
@@ -3139,20 +3139,34 @@
 eeModel(
-java
+(PsiClassOwner)psi
 File, ed
@@ -3178,22 +3178,50 @@
 ;%0A      
+  
 %7D%0A    
-%7D
+  %7D;%0A    %7D%0A%0A    return null
 ;%0A  %7D%0A%7D

correct structure factory for non-Java decompiled files