/*
   * Copyright (C) 2000-2005, DENX Software Engineering
   *		Wolfgang Denk <wd@denx.de>
   * Copyright (C) Procsys. All rights reserved.
   *		Mushtaq Khan <mushtaq_k@procsys.com>
   *			<mushtaqk_921@yahoo.co.in>
   * Copyright (C) 2008 Freescale Semiconductor, Inc.
   *		Dave Liu <daveliu@freescale.com>
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  
  #include <common.h>

  #include <ahci.h>
  #include <dm.h>

  #include <command.h>
  #include <part.h>
  #include <sata.h>

  #include <dm/device-internal.h>
  #include <dm/uclass-internal.h>

  static int sata_curr_device = -1;

  int sata_remove(int devnum)
  {
  #ifdef CONFIG_AHCI
  	struct udevice *dev;
  	int rc;
  
  	rc = uclass_find_device(UCLASS_AHCI, devnum, &dev);
  	if (!rc && !dev)
  		rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  	if (rc || !dev) {
  		printf("Cannot find SATA device %d (err=%d)
  ", devnum, rc);
  		return CMD_RET_FAILURE;
  	}
  
  	rc = device_remove(dev, DM_REMOVE_NORMAL);
  	if (rc) {
  		printf("Cannot remove SATA device '%s' (err=%d)
  ", dev->name,
  		       rc);
  		return CMD_RET_FAILURE;
  	}
  
  	return 0;
  #else
  	return sata_stop();
  #endif
  }
  
  int sata_probe(int devnum)
  {
  #ifdef CONFIG_AHCI
  	struct udevice *dev;
  	struct udevice *blk;
  	int rc;
  
  	rc = uclass_get_device(UCLASS_AHCI, devnum, &dev);
  	if (rc)
  		rc = uclass_find_first_device(UCLASS_AHCI, &dev);
  	if (rc) {
  		printf("Cannot probe SATA device %d (err=%d)
  ", devnum, rc);
  		return CMD_RET_FAILURE;
  	}
  	rc = sata_scan(dev);
  	if (rc) {
  		printf("Cannot scan SATA device %d (err=%d)
  ", devnum, rc);
  		return CMD_RET_FAILURE;
  	}
  
  	rc = blk_get_from_parent(dev, &blk);
  	if (!rc) {
  		struct blk_desc *desc = dev_get_uclass_platdata(blk);
  
  		if (desc->lba > 0 && desc->blksz > 0)
  			part_init(desc);
  	}
  
  	return 0;
  #else
  	return sata_initialize() < 0 ? CMD_RET_FAILURE : CMD_RET_SUCCESS;
  #endif
  }

  static int do_sata(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])

  {
  	int rc = 0;

  	if (argc >= 2) {
  		int devnum = 0;
  
  		if (argc == 3)
  			devnum = (int)simple_strtoul(argv[2], NULL, 10);

  		if (!strcmp(argv[1], "stop")) {
  			sata_curr_device = -1;

  			return sata_remove(devnum);

  		}

  		if (!strcmp(argv[1], "init")) {
  			if (sata_curr_device != -1) {
  				rc = sata_remove(devnum);
  				if (rc)
  					return rc;
  			}

  			rc = sata_probe(devnum);
  			if (rc < 0)
  				return CMD_RET_FAILURE;
  			sata_curr_device = rc;
  			return CMD_RET_SUCCESS;

  		}

  	}

  
  	/* If the user has not yet run `sata init`, do it now */

  	if (sata_curr_device == -1) {

  		rc = sata_probe(0);
  		if (rc < 0)

  			return CMD_RET_FAILURE;

  		sata_curr_device = 0;

  	}

  	return blk_common_cmd(argc, argv, IF_TYPE_SATA, &sata_curr_device);

  }
  
  U_BOOT_CMD(
  	sata, 5, 1, do_sata,

  	"SATA sub system",

  	"init - init SATA sub system
  "

  	"sata stop [dev] - disable SATA sub system or device
  "

  	"sata info - show available SATA devices
  "
  	"sata device [dev] - show or set current device
  "
  	"sata part [dev] - print partition table
  "
  	"sata read addr blk# cnt
  "

  	"sata write addr blk# cnt"
  );