platform/lang-impl/src/com/intellij/openapi/module/impl/ModulePointerManagerImpl.java

   1 /*
   2  * Copyright 2000-2009 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 com.intellij.openapi.module.impl;
  17
  18 import com.intellij.ProjectTopics;
  19 import com.intellij.openapi.Disposable;
  20 import com.intellij.openapi.module.Module;
  21 import com.intellij.openapi.module.ModulePointer;
  22 import com.intellij.openapi.module.ModulePointerManager;
  23 import com.intellij.openapi.project.ModuleAdapter;
  24 import com.intellij.openapi.project.Project;
  25 import com.intellij.openapi.util.Disposer;
  26
  27 import java.util.HashMap;
  28 import java.util.Map;
  29
  30 /**
  31  * @author nik
  32  */
  33 public class ModulePointerManagerImpl extends ModulePointerManager {
  34   private Map<String, ModulePointerImpl> myUnresolved = new HashMap<String, ModulePointerImpl>();
  35   private Map<Module, ModulePointerImpl> myPointers = new HashMap<Module, ModulePointerImpl>();
  36   private final Project myProject;
  37
  38   public ModulePointerManagerImpl(Project project) {
  39     myProject = project;
  40     project.getMessageBus().connect().subscribe(ProjectTopics.MODULES, new ModuleAdapter() {
  41       @Override
  42       public void beforeModuleRemoved(Project project, Module module) {
  43         unregisterPointer(module);
  44       }
  45
  46       @Override
  47       public void moduleAdded(Project project, Module module) {
  48         final ModulePointerImpl pointer = myUnresolved.remove(module.getName());
  49         if (pointer != null) {
  50           pointer.moduleAdded(module);
  51           registerPointer(module, pointer);
  52         }
  53       }
  54     });
  55   }
  56
  57   private void registerPointer(final Module module, final ModulePointerImpl pointer) {
  58     myPointers.put(module, pointer);
  59     Disposer.register(module, new Disposable() {
  60       public void dispose() {
  61         unregisterPointer(module);
  62       }
  63     });
  64   }
  65
  66   private void unregisterPointer(Module module) {
  67     final ModulePointerImpl pointer = myPointers.remove(module);
  68     if (pointer != null) {
  69       pointer.moduleRemoved(module);
  70       myUnresolved.put(pointer.getModuleName(), pointer);
  71     }
  72   }
  73
  74   @Override
  75   public ModulePointer create(Module module) {
  76     ModulePointerImpl pointer = myPointers.get(module);
  77     if (pointer == null) {
  78       pointer = new ModulePointerImpl(module);
  79       registerPointer(module, pointer);
  80     }
  81     return pointer;
  82   }
  83
  84   @Override
  85   public ModulePointer create(String moduleName) {
  86     final Module module = ModuleManagerImpl.getInstance(myProject).findModuleByName(moduleName);
  87     if (module != null) {
  88       return create(module);
  89     }
  90
  91     ModulePointerImpl pointer = myUnresolved.get(moduleName);
  92     if (pointer == null) {
  93       pointer = new ModulePointerImpl(moduleName);
  94       myUnresolved.put(moduleName, pointer);
  95     }
  96     return pointer;
  97   }
  98 }