FacetEditorsStateManagerImpl.java

/*
 * Copyright 2000-2017 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 com.intellij.facet.impl.ui;

import com.intellij.facet.FacetType;
import com.intellij.openapi.components.PersistentStateComponent;
import com.intellij.openapi.components.State;
import com.intellij.openapi.components.Storage;
import com.intellij.openapi.components.StoragePathMacros;
import com.intellij.util.xmlb.XmlSerializer;
import com.intellij.util.xmlb.annotations.MapAnnotation;
import com.intellij.util.xmlb.annotations.Property;
import com.intellij.util.xmlb.annotations.Tag;
import org.jdom.Element;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.HashMap;
import java.util.Map;

/**
 * @author nik
 */
@State(name = "FacetEditorsStateManager", storages = @Storage(StoragePathMacros.WORKSPACE_FILE))
public class FacetEditorsStateManagerImpl extends FacetEditorsStateManager implements PersistentStateComponent<FacetEditorsStateManagerImpl.FacetEditorsStateBean>{
  private final Map<String, Object> myFacetTypeStates = new HashMap<>();
  private FacetEditorsStateBean myBean = new FacetEditorsStateBean();

  @Override
  public <T> void saveState(@NotNull final FacetType<?, ?> type, final T state) {
    String id = type.getStringId();
    if (state != null) {
      myFacetTypeStates.put(id, state);
    }
    else {
      myFacetTypeStates.remove(id);
      myBean.getFacetTypeElements().remove(id);
    }
  }

  @Override
  @Nullable
  public <T> T getState(@NotNull final FacetType<?, ?> type, @NotNull final Class<T> aClass) {
    String id = type.getStringId();
    //noinspection unchecked
    T state = (T)myFacetTypeStates.get(id);
    if (state == null) {
      FacetTypeStateBean bean = myBean.getFacetTypeElements().get(id);
      if (bean != null) {
        Element element = bean.getState();
        if (element != null) {
          state = XmlSerializer.deserialize(element, aClass);
        }
      }
    }
    return state;
  }

  @Override
  public FacetEditorsStateBean getState() {
    for (Map.Entry<String, Object> entry : myFacetTypeStates.entrySet()) {
      FacetTypeStateBean bean = new FacetTypeStateBean();
      bean.setState(XmlSerializer.serialize(entry.getValue()));
      myBean.getFacetTypeElements().put(entry.getKey(), bean);
    }
    return myBean;
  }

  @Override
  public void loadState(final FacetEditorsStateBean state) {
    myBean = state;
  }

  public static class FacetEditorsStateBean {
    private Map<String, FacetTypeStateBean> myFacetTypeElements = new HashMap<>();

    @Property(surroundWithTag = false)
    @MapAnnotation(surroundKeyWithTag = false, surroundWithTag = false, surroundValueWithTag = false, entryTagName = "facet-type-state", keyAttributeName = "type")
    public Map<String, FacetTypeStateBean> getFacetTypeElements() {
      return myFacetTypeElements;
    }

    public void setFacetTypeElements(final Map<String, FacetTypeStateBean> elements) {
      myFacetTypeElements = elements;
    }
  }

  @Tag("facet-type")
  public static class FacetTypeStateBean {
    private Element myState;

    @Tag("state")
    public Element getState() {
      return myState;
    }

    public void setState(final Element state) {
      myState = state;
    }
  }
}