/*
   * (C) Copyright 2008
   * Stefan Roese, DENX Software Engineering, sr@denx.de.
   *

   * SPDX-License-Identifier:	GPL-2.0+

   */
  
  
  /*
   * UBIFS command support
   */
  
  #undef DEBUG
  
  #include <common.h>
  #include <config.h>
  #include <command.h>

  #include <ubifs_uboot.h>

  static int ubifs_initialized;
  static int ubifs_mounted;

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

  {
  	char *vol_name;
  	int ret;

  	if (argc != 2)

  		return CMD_RET_USAGE;

  	vol_name = argv[1];
  	debug("Using volume %s
  ", vol_name);
  
  	if (ubifs_initialized == 0) {
  		ubifs_init();
  		ubifs_initialized = 1;
  	}

  	ret = uboot_ubifs_mount(vol_name);

  	if (ret)
  		return -1;
  
  	ubifs_mounted = 1;
  
  	return 0;
  }

  int ubifs_is_mounted(void)
  {
  	return ubifs_mounted;
  }
  
  void cmd_ubifs_umount(void)
  {

  	uboot_ubifs_umount();

  	ubifs_mounted = 0;
  	ubifs_initialized = 0;
  }

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

  {
  	if (argc != 1)

  		return CMD_RET_USAGE;

  
  	if (ubifs_initialized == 0) {
  		printf("No UBIFS volume mounted!
  ");
  		return -1;
  	}

  	cmd_ubifs_umount();

  
  	return 0;
  }

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

  {
  	char *filename = "/";
  	int ret;
  
  	if (!ubifs_mounted) {

  		printf("UBIFS not mounted, use ubifsmount to mount volume first!
  ");

  		return -1;
  	}
  
  	if (argc == 2)
  		filename = argv[1];
  	debug("Using filename %s
  ", filename);
  
  	ret = ubifs_ls(filename);

  	if (ret) {
  		printf("** File not found %s **
  ", filename);
  		ret = CMD_RET_FAILURE;
  	}

  
  	return ret;
  }

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

  {
  	char *filename;

  	char *endp;

  	int ret;
  	u32 addr;
  	u32 size = 0;
  
  	if (!ubifs_mounted) {
  		printf("UBIFS not mounted, use ubifs mount to mount volume first!
  ");
  		return -1;
  	}

  	if (argc < 3)

  		return CMD_RET_USAGE;

  	addr = simple_strtoul(argv[1], &endp, 16);

  	if (endp == argv[1])

  		return CMD_RET_USAGE;

  	filename = argv[2];

  	if (argc == 4) {
  		size = simple_strtoul(argv[3], &endp, 16);

  		if (endp == argv[3])

  			return CMD_RET_USAGE;

  	}

  	debug("Loading file '%s' to address 0x%08x (size %d)
  ", filename, addr, size);
  
  	ret = ubifs_load(filename, addr, size);

  	if (ret) {
  		printf("** File not found %s **
  ", filename);
  		ret = CMD_RET_FAILURE;
  	}

  
  	return ret;
  }
  
  U_BOOT_CMD(
  	ubifsmount, 2, 0, do_ubifs_mount,

  	"mount UBIFS volume",

  	"<volume-name>
  "
  	"    - mount 'volume-name' volume"

  );

  U_BOOT_CMD(

  	ubifsumount, 1, 0, do_ubifs_umount,
  	"unmount UBIFS volume",
  	"    - unmount current volume"
  );
  
  U_BOOT_CMD(

  	ubifsls, 2, 0, do_ubifs_ls,

  	"list files in a directory",
  	"[directory]
  "
  	"    - list files in a 'directory' (default '/')"
  );

  U_BOOT_CMD(
  	ubifsload, 4, 0, do_ubifs_load,

  	"load file from an UBIFS filesystem",
  	"<addr> <filename> [bytes]
  "
  	"    - load file 'filename' to address 'addr'"
  );