/*
 * 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 org.intellij.lang.xpath.xslt.impl;

import com.intellij.ide.FileIconPatcher;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.util.Key;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiManager;
import gnu.trove.TIntObjectHashMap;
import org.intellij.lang.xpath.xslt.XsltSupport;

import javax.swing.*;

/**
 * @author peter
 */
public class XsltIconProvider implements FileIconPatcher {

  private static final Key<TIntObjectHashMap<Icon>> ICON_KEY = Key.create("XSLT_ICON");

  public Icon patchIcon(Icon baseIcon, VirtualFile file, int flags, Project project) {
    if (project == null) return baseIcon;

    final TIntObjectHashMap<Icon> icons = file.getUserData(ICON_KEY);
    if (icons != null) {
      final Icon icon = icons.get(flags);
      if (icon != null) {
        return icon;
      }
    }

    final PsiFile element = PsiManager.getInstance(project).findFile(file);
    if (element != null) {
      if (XsltSupport.isXsltFile(element)) {
        return cacheIcon(file, flags, icons, XsltSupport.createXsltIcon(baseIcon));
      }
    }
    return baseIcon;
  }

  private static Icon cacheIcon(VirtualFile file, int flags, TIntObjectHashMap<Icon> icons, Icon icon) {
    if (icons == null) {
      file.putUserData(ICON_KEY, icons = new TIntObjectHashMap<Icon>(3));
    }
    icons.put(flags, icon);
    return icon;
  }
}

@@ -728,46 +728,8 @@
 ct;%0A
-import com.intellij.openapi.util.Key;%0A
 impo
@@ -842,44 +842,8 @@
 er;%0A
-import gnu.trove.TIntObjectHashMap;%0A
 impo
@@ -1000,97 +1000,8 @@
  %7B%0A%0A
-  private static final Key%3CTIntObjectHashMap%3CIcon%3E%3E ICON_KEY = Key.create(%22XSLT_ICON%22);%0A%0A
   pu
@@ -1140,222 +1140,23 @@
 nal 
-TIntObjectHashMap%3CIcon%3E icons = file.getUserData(ICON_KEY);%0A    if (icons != null) %7B%0A      final Icon icon = icons.get(flags);%0A      if (icon != null) %7B%0A        return icon;%0A      %7D%0A    %7D%0A%0A    final PsiFile element
+PsiFile psiFile
  = P
@@ -1214,23 +1214,23 @@
 if (
-element
+psiFile
  != null
 ) %7B%0A
@@ -1229,22 +1229,12 @@
 null
-) %7B%0A      if (
+ && 
 Xslt
@@ -1256,22 +1256,20 @@
 ile(
-element
+psiFile
 )) %7B%0A
-  
     
@@ -1280,38 +1280,8 @@
 turn
- cacheIcon(file, flags, icons,
  Xsl
@@ -1317,25 +1317,16 @@
 con)
-)
 ;%0A
-  
     %7D%0A
-    %7D%0A
     
@@ -1351,265 +1351,6 @@
  %7D%0A%0A
-  private static Icon cacheIcon(VirtualFile file, int flags, TIntObjectHashMap%3CIcon%3E icons, Icon icon) %7B%0A    if (icons == null) %7B%0A      file.putUserData(ICON_KEY, icons = new TIntObjectHashMap%3CIcon%3E(3));%0A    %7D%0A    icons.put(flags, icon);%0A    return icon;%0A  %7D%0A
 %7D%0A

remove XsltIconProvider cache: it caches forever (which is wrong) and it's unnecessary because there's already another cache up the stack