SymbolicRefs.java

/*
 * Copyright 2000-2010 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.community.intellij.plugins.communitycase.history.browser;

import org.community.intellij.plugins.communitycase.Branch;

import java.util.Collection;
import java.util.TreeSet;

/**
 * @author irengrig
 */
public class SymbolicRefs {
  private Branch myCurrent;
  private final TreeSet<String> myTags;
  private final TreeSet<String> myLocalBranches;
  private final TreeSet<String> myRemoteBranches;
  private String myTrackedRemoteName;

  public SymbolicRefs() {
    myTags = new TreeSet<String>();
    myLocalBranches = new TreeSet<String>();
    myRemoteBranches = new TreeSet<String>();
  }

  public void addRemote(final String branch) {
    myRemoteBranches.add(branch);
  }

  public void addLocal(final String branch) {
    myLocalBranches.add(branch);
  }

  public void addTags(final Collection<String> value) {
    myTags.addAll(value);
  }

  public void addLocals(final Collection<String> value) {
    myLocalBranches.addAll(value);
  }

  public void addRemotes(final Collection<String> value) {
    myRemoteBranches.addAll(value);
  }

  public TreeSet<String> getLocalBranches() {
    return myLocalBranches;
  }

  public TreeSet<String> getRemoteBranches() {
    return myRemoteBranches;
  }

  public TreeSet<String> getTags() {
    return myTags;
  }

  public Branch getCurrent() {
    return myCurrent;
  }

  public void setCurrent(Branch current) {
    myCurrent = current;
  }

  public Kind getKind(final String s) {
    if (myLocalBranches.contains(s)) return Kind.LOCAL;
    if (myRemoteBranches.contains(s)) return Kind.REMOTE;
    return Kind.TAG;
  }

  public void clear() {
    myLocalBranches.clear();
    myRemoteBranches.clear();
    myTags.clear();
  }

  public void setTrackedRemote(String trackedRemoteName) {
    myTrackedRemoteName = trackedRemoteName;
  }

  public String getTrackedRemoteName() {
    return myTrackedRemoteName;
  }

  public static enum Kind {
    TAG,
    LOCAL,
    REMOTE
  }
}