/FileExtraction/Java_unseen/dlitz_resin/modules/resin/src/com/caucho/server/resin/BootClusterConfig.java
/*
 * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.server.resin;

import java.util.ArrayList;

import javax.annotation.PostConstruct;

import com.caucho.cloud.network.ClusterServerProgram;
import com.caucho.cloud.topology.CloudCluster;
import com.caucho.cloud.topology.CloudPod;
import com.caucho.config.ConfigException;
import com.caucho.config.Configurable;
import com.caucho.config.SchemaBean;
import com.caucho.config.program.ConfigProgram;
import com.caucho.config.program.ContainerProgram;
import com.caucho.util.L10N;

/**
 * The BootClusterConfig is the first-pass configuration of the cluster.
 * 
 * It matches the &lt;cluster> tag in the resin.xml
 */
public class BootClusterConfig implements SchemaBean
{
  private static final L10N L = new L10N(BootClusterConfig.class);
  
  private BootResinConfig _resinConfig;
  private CloudCluster _cloudCluster;
  
  private String _id;

  private ContainerProgram _clusterProgram
    = new ContainerProgram();

  private ContainerProgram _serverDefaultProgram
    = new ContainerProgram();

  private ArrayList<BootPodConfig> _pods
    = new ArrayList<BootPodConfig>();

  /*
  private ArrayList<BootServerConfig> _servers
    = new ArrayList<BootServerConfig>();
    */

  /**
   * Creates a new resin server.
   */
  public BootClusterConfig(BootResinConfig resinConfig)
  {
    _resinConfig = resinConfig;
  }

  /**
   * Returns the relax schema.
   */
  @Override
  public String getSchema()
  {
    return "com/caucho/server/resin/cluster.rnc";
  }
  
  /**
   * Returns the cluster's id
   */
  public String getId()
  {
    return _id;
  }
  
  /**
   * Sets the cluster's id
   */
  @Configurable
  public void setId(String id)
  {
    _id = id;
  }
  
  /**
   * Adds a <server-default> for default server configuration.
   */
  @Configurable
  public void addServerDefault(ContainerProgram program)
  {
    _serverDefaultProgram.addProgram(program);
  }
  
  public ContainerProgram getServerDefault()
  {
    return _serverDefaultProgram;
  }

  @Configurable
  public BootPodConfig createPod()
    throws ConfigException
  {
    BootPodConfig pod = new BootPodConfig(this);
    
    _pods.add(pod);

    return pod;
  }

  @Configurable
  public void addPod(BootPodConfig pod)
  {
    _pods.add(pod);
  }

  @Configurable
  public BootServerConfig createServer()
    throws ConfigException
  {
    if (_pods.size() == 0) {
      addPod(createPod());
    }

    BootPodConfig pod = _pods.get(0);

    return pod.createServer();
  }

  @Configurable
  public void addServer(BootServerConfig server)
  {
    if (_pods.size() == 0) {
      addPod(createPod());
    }
    
    BootPodConfig pod = _pods.get(0);

    pod.addServer(server);
  }

  public ArrayList<BootPodConfig> getPodList()
  {
    return _pods;
  }

  /**
   * Managed by watchdog.
   */
  public void addStdoutLog(ConfigProgram config)
  {
  }
  
  public void addContentProgram(ConfigProgram program)
  {
    _clusterProgram.addProgram(program);
  }
  
  ConfigProgram getProgram()
  {
    return _clusterProgram;
  }
  
  @PostConstruct
  public void init()
  {
    if (_id == null)
      throw new ConfigException(L.l("'id' is a require attribute for <cluster>"));
    
    CloudCluster cluster = getCloudCluster();
    
    cluster.putData(new ClusterServerProgram(_serverDefaultProgram));

    getCloudPod();
  }
  
  CloudCluster getCloudCluster()
  {
    if (_id == null)
      throw new ConfigException(L.l("'id' is a require attribute for <cluster>"));
    
    if (_cloudCluster == null) {
      _cloudCluster = _resinConfig.getCloudSystem().findCluster(_id);
      
      if (_cloudCluster == null)
        _cloudCluster = _resinConfig.getCloudSystem().createCluster(_id);
    }
    
    return _cloudCluster;
  }
  
  CloudPod getCloudPod()
  {
    if (_pods.size() == 0) {
      addPod(createPod());
    }
    
    return _pods.get(0).getCloudPod();
  }
  
  @Override
  public String toString()
  {
    return getClass().getSimpleName() + "[" + _id + "]";
  }
}
