package net.sf.bloodball.fit.system;

import java.awt.Color;

import net.sf.bloodball.DefaultTeam;
import net.sf.bloodball.DefaultTeamSetup;
import net.sf.bloodball.confguration.team.ITeam;
import net.sf.bloodball.confguration.team.ITeamSetup;
import net.sf.bloodball.confguration.team.ITeamSetupFactory;

public class TestTeamSetupFactory implements ITeamSetupFactory {

  public static final String HOME_TEAM = "home"; //$NON-NLS-1$
  public static final String GUEST_TEAM = "guest"; //$NON-NLS-1$
  private String beginningTeam;
  private ITeam homeTeam;
  private ITeam guestTeam;

  public void setBeginningTeam(String team) {
    this.beginningTeam = team;
  }

  public ITeamSetup createNewTeamSetup() {
    homeTeam = new DefaultTeam(HOME_TEAM, Color.black);
    guestTeam = new DefaultTeam(GUEST_TEAM, Color.white);
    return new DefaultTeamSetup(homeTeam, guestTeam, getTeam(beginningTeam));
  }

  public ITeam getTeam(String name) {
    return name.equals(HOME_TEAM) ? homeTeam : guestTeam;
  }
  
}