/*

 *   SONEWS News Server

 *   see AUTHORS for the list of contributors

 *

 *   This program 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 3 of the License, or

 *   (at your option) any later version.

 *

 *   This program 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.  See the

 *   GNU General Public License for more details.

 *

 *   You should have received a copy of the GNU General Public License

 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

package org.sonews.daemon;

import java.util.HashMap;

import java.util.Map;

import java.util.Set;

import java.util.concurrent.ConcurrentHashMap;

import org.sonews.daemon.command.Command;

import org.sonews.daemon.command.UnsupportedCommand;

import org.sonews.util.Log;

import org.sonews.util.io.Resource;

/**

 * Selects the correct command processing class.

 * @author Christian Lins

 * @since sonews/1.0

 */

public class CommandSelector

{

  private static Map<Thread, CommandSelector> instances

    = new ConcurrentHashMap<Thread, CommandSelector>();

  private static Map<String, Class<?>> commandClassesMapping

    = new ConcurrentHashMap<String, Class<?>>();

  static

  {

    String[] classes = Resource.getAsString("helpers/commands.list", true).split("\n");

    for(String className : classes)

    {

      if(className.charAt(0) == '#')

      {

        // Skip comments

        continue;

      }

      try

      {

        addCommandHandler(className);

      }

      catch(ClassNotFoundException ex)

      {

        Log.get().warning("Could not load command class: " + ex);

      }

      catch(InstantiationException ex)

      {

        Log.get().severe("Could not instantiate command class: " + ex);

      }

      catch(IllegalAccessException ex)

      {

        Log.get().severe("Could not access command class: " + ex);

      }

    }

  }

  public static void addCommandHandler(String className)

    throws ClassNotFoundException, InstantiationException, IllegalAccessException

  {

    Class<?> clazz = Class.forName(className);

    Command cmd = (Command)clazz.newInstance();

    String[] cmdStrs = cmd.getSupportedCommandStrings();

    for (String cmdStr : cmdStrs)

    {

      commandClassesMapping.put(cmdStr, clazz);

    }

  }

  public static Set<String> getCommandNames()

  {

    return commandClassesMapping.keySet();

  }

  public static CommandSelector getInstance()

  {

    CommandSelector csel = instances.get(Thread.currentThread());

    if(csel == null)

    {

      csel = new CommandSelector();

      instances.put(Thread.currentThread(), csel);

    }

    return csel;

  }

  private Map<String, Command> commandMapping = new HashMap<String, Command>();

  private Command              unsupportedCmd = new UnsupportedCommand();

  private CommandSelector()

  {}

  public Command get(String commandName)

  {

    try

    {

      commandName = commandName.toUpperCase();

      Command cmd = this.commandMapping.get(commandName);

      if(cmd == null)

      {

        Class<?> clazz = commandClassesMapping.get(commandName);

        if(clazz == null)

        {

          cmd = this.unsupportedCmd;

        }

        else

        {

          cmd = (Command)clazz.newInstance();

          this.commandMapping.put(commandName, cmd);

        }

      }

      else if(cmd.isStateful())

      {

        cmd = cmd.getClass().newInstance();

      }

      return cmd;

    }

    catch(Exception ex)

    {

      ex.printStackTrace();

      return this.unsupportedCmd;

    }

  }

}