/*
   * Copyright (C) 2017 NXP Semiconductors
   * Copyright (C) 2017 Bin Meng <bmeng.cn@gmail.com>
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include <common.h>
  #include <command.h>
  #include <dm.h>
  #include <nvme.h>
  
  static int nvme_curr_dev;
  
  static int do_nvme(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  {
  	int ret;
  
  	if (argc == 2) {
  		if (strncmp(argv[1], "scan", 4) == 0) {
  			ret = nvme_scan_namespace();
  			if (ret)
  				return CMD_RET_FAILURE;
  
  			return ret;
  		}
  		if (strncmp(argv[1], "deta", 4) == 0) {
  			struct udevice *udev;
  
  			ret = blk_get_device(IF_TYPE_NVME, nvme_curr_dev,
  					     &udev);
  			if (ret < 0)
  				return CMD_RET_FAILURE;
  
  			nvme_print_info(udev);
  
  			return ret;
  		}
  	}
  
  	return blk_common_cmd(argc, argv, IF_TYPE_NVME, &nvme_curr_dev);
  }
  
  U_BOOT_CMD(
  	nvme, 8, 1, do_nvme,
  	"NVM Express sub-system",
  	"scan - scan NVMe devices
  "
  	"nvme detail - show details of current NVMe device
  "
  	"nvme info - show all available NVMe devices
  "
  	"nvme device [dev] - show or set current NVMe device
  "
  	"nvme part [dev] - print partition table of one or all NVMe devices
  "
  	"nvme read addr blk# cnt - read `cnt' blocks starting at block
  "
  	"     `blk#' to memory address `addr'
  "
  	"nvme write addr blk# cnt - write `cnt' blocks starting at block
  "
  	"     `blk#' from memory address `addr'"
  );