/*
   * (C) Copyright 2010
   * Jason Kridner <jkridner@beagleboard.org>
   *
   * Based on cmd_led.c patch from:
   * http://www.mail-archive.com/u-boot@lists.denx.de/msg06873.html
   * (C) Copyright 2008
   * Ulf Samuelsson <ulf.samuelsson@atmel.com>
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  
  #include <common.h>
  #include <config.h>
  #include <command.h>
  #include <status_led.h>
  
  struct led_tbl_s {
  	char		*string;	/* String for use in the command */
  	led_id_t	mask;		/* Mask used for calling __led_set() */

  	void		(*off)(void);	/* Optional function for turning LED off */
  	void		(*on)(void);	/* Optional function for turning LED on */

  	void		(*toggle)(void);/* Optional function for toggling LED */

  };
  
  typedef struct led_tbl_s led_tbl_t;
  
  static const led_tbl_t led_commands[] = {
  #ifdef CONFIG_BOARD_SPECIFIC_LED
  #ifdef STATUS_LED_BIT

  	{ "0", STATUS_LED_BIT, NULL, NULL, NULL },

  #endif
  #ifdef STATUS_LED_BIT1

  	{ "1", STATUS_LED_BIT1, NULL, NULL, NULL },

  #endif
  #ifdef STATUS_LED_BIT2

  	{ "2", STATUS_LED_BIT2, NULL, NULL, NULL },

  #endif
  #ifdef STATUS_LED_BIT3

  	{ "3", STATUS_LED_BIT3, NULL, NULL, NULL },

  #endif
  #endif
  #ifdef STATUS_LED_GREEN

  	{ "green", STATUS_LED_GREEN, green_led_off, green_led_on, NULL },

  #endif
  #ifdef STATUS_LED_YELLOW

  	{ "yellow", STATUS_LED_YELLOW, yellow_led_off, yellow_led_on, NULL },

  #endif
  #ifdef STATUS_LED_RED

  	{ "red", STATUS_LED_RED, red_led_off, red_led_on, NULL },

  #endif
  #ifdef STATUS_LED_BLUE

  	{ "blue", STATUS_LED_BLUE, blue_led_off, blue_led_on, NULL },

  #endif

  	{ NULL, 0, NULL, NULL, NULL }

  };

  enum led_cmd { LED_ON, LED_OFF, LED_TOGGLE };
  
  enum led_cmd get_led_cmd(char *var)

  {
  	if (strcmp(var, "off") == 0) {

  		return LED_OFF;

  	}
  	if (strcmp(var, "on") == 0) {

  		return LED_ON;

  	}

  	if (strcmp(var, "toggle") == 0)
  		return LED_TOGGLE;

  	return -1;
  }
  
  int do_led (cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {

  	int i, match = 0;
  	enum led_cmd cmd;

  
  	/* Validate arguments */
  	if ((argc != 3)) {

  		return CMD_RET_USAGE;

  	}

  	cmd = get_led_cmd(argv[2]);
  	if (cmd < 0) {

  		return CMD_RET_USAGE;

  	}
  
  	for (i = 0; led_commands[i].string; i++) {

  		if ((strcmp("all", argv[1]) == 0) ||

  		    (strcmp(led_commands[i].string, argv[1]) == 0)) {

  			match = 1;
  			switch (cmd) {
  			case LED_ON:
  				if (led_commands[i].on)

  					led_commands[i].on();

  				else

  					__led_set(led_commands[i].mask,
  							  STATUS_LED_ON);

  				break;
  			case LED_OFF:
  				if (led_commands[i].off)

  					led_commands[i].off();

  				else

  					__led_set(led_commands[i].mask,
  							  STATUS_LED_OFF);

  				break;
  			case LED_TOGGLE:
  				if (led_commands[i].toggle)
  					led_commands[i].toggle();
  				else
  					__led_toggle(led_commands[i].mask);

  			}

  			/* Need to set only 1 led if led_name wasn't 'all' */
  			if (strcmp("all", argv[1]) != 0)
  				break;

  		}
  	}
  
  	/* If we ran out of matches, print Usage */

  	if (!match) {

  		return CMD_RET_USAGE;

  	}
  
  	return 0;
  }
  
  U_BOOT_CMD(
  	led, 3, 1, do_led,

  	"["

  #ifdef CONFIG_BOARD_SPECIFIC_LED
  #ifdef STATUS_LED_BIT
  	"0|"
  #endif
  #ifdef STATUS_LED_BIT1
  	"1|"
  #endif
  #ifdef STATUS_LED_BIT2
  	"2|"
  #endif
  #ifdef STATUS_LED_BIT3
  	"3|"
  #endif
  #endif
  #ifdef STATUS_LED_GREEN
  	"green|"
  #endif
  #ifdef STATUS_LED_YELLOW
  	"yellow|"
  #endif
  #ifdef STATUS_LED_RED
  	"red|"
  #endif
  #ifdef STATUS_LED_BLUE
  	"blue|"
  #endif

  	"all] [on|off|toggle]",
  	"[led_name] [on|off|toggle] sets or clears led(s)"

  );