/*
 * Copyright (c) 2016 Intel Corporation
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#include <errno.h>
#include <stdio.h>
#include <string.h>

#include <misc/printk.h>
#include <shell/shell.h>
#include <init.h>
#include <fs.h>
#include <stdio.h>
#include <stdlib.h>
#include <inttypes.h>
#include <limits.h>


/* FAT */
#ifdef CONFIG_FAT_FILESYSTEM_ELM
#include <ff.h>
#define FATFS_MNTP      "/RAM:"
/* FatFs work area */
FATFS fat_fs;
/* mounting info */
static struct fs_mount_t fatfs_mnt = {
	.type = FS_FATFS,
	.fs_data = &fat_fs,
};
#endif

/* NFFS */
#ifdef CONFIG_FILE_SYSTEM_NFFS
#include <nffs/nffs.h>
#define NFFS_MNTP       "/nffs"

/* NFFS work area strcut */
static struct nffs_flash_desc flash_desc;

/* mounting info */
static struct fs_mount_t nffs_mnt = {
	.type = FS_NFFS,
	.fs_data = &flash_desc,
};
#endif

#define BUF_CNT 64

#define MAX_PATH_LEN 128
#define MAX_FILENAME_LEN 128
#define MAX_INPUT_LEN 20


#define SHELL_FS    "fs"


static char cwd[MAX_PATH_LEN] = "/";

static void create_abs_path(const char *name, char *path, size_t len)
{
	if (name[0] == '/') {
		strncpy(path, name, len - 1);
		path[len - 1] = '\0';
	} else {
		if (strcmp(cwd, "/") == 0) {
			snprintf(path, len, "/%s", name);
		} else {
			snprintf(path, len, "%s/%s", cwd, name);
		}
	}
}

static int cmd_mkdir(int argc, char *argv[])
{
	int res;
	char path[MAX_PATH_LEN];

	if (argc < 2) {
		printk("Missing argument\n");
		return 0;
	}

	create_abs_path(argv[1], path, sizeof(path));

	res = fs_mkdir(path);
	if (res) {
		printk("Error creating dir[%d]\n", res);
		return res;
	}

	return 0;
}

static int cmd_rm(int argc, char *argv[])
{
	int err;
	char path[MAX_PATH_LEN];

	if (argc < 2) {
		printk("Missing argument\n");
		return 0;
	}

	create_abs_path(argv[1], path, sizeof(path));

	err = fs_unlink(path);
	if (err) {
		printk("Failed to remove %s (%d)\n", path, err);
		return err;
	}

	return 0;
}

static int cmd_read(int argc, char *argv[])
{
	char path[MAX_PATH_LEN];
	struct fs_dirent dirent;
	struct fs_file_t file;
	int count;
	off_t offset;
	int err;

	if (argc < 2) {
		printk("Missing argument\n");
		return 0;
	}

	create_abs_path(argv[1], path, sizeof(path));

	if (argc > 2) {
		count = strtol(argv[2], NULL, 0);
		if (count <= 0) {
			count = INT_MAX;
		}
	} else {
		count = INT_MAX;
	}

	if (argc > 3) {
		offset = strtol(argv[3], NULL, 0);
	} else {
		offset = 0;
	}

	err = fs_stat(path, &dirent);
	if (err) {
		printk("Failed to stat %s (%d)\n", path, err);
		return err;
	}

	if (dirent.type != FS_DIR_ENTRY_FILE) {
		return -EINVAL;
	}

	printk("File size: %zd\n", dirent.size);

	err = fs_open(&file, path);
	if (err) {
		printk("Failed to open %s (%d)\n", path, err);
		return err;
	}

	if (offset > 0) {
		fs_seek(&file, offset, FS_SEEK_SET);
	}

	while (count > 0) {
		ssize_t read;
		u8_t buf[16];
		int i;

		read = fs_read(&file, buf, min(count, sizeof(buf)));
		if (read <= 0) {
			break;
		}

		printk("%08X  ", (unsigned) offset);

		for (i = 0; i < read; i++) {
			printk("%02X ", buf[i]);
		}
		for (; i < sizeof(buf); i++) {
			printk("   ");
		}

		printk(" ");

		for (i = 0; i < read; i++) {
			printk("%c", buf[i] < 32 ||
			       buf[i] > 127 ? '.' : buf[i]);
		}

		printk("\n");

		offset += read;
		count -= read;
	}

	fs_close(&file);

	return 0;
}

static int cmd_write(int argc, char *argv[])
{
	char path[MAX_PATH_LEN];
	u8_t buf[BUF_CNT];
	u8_t buf_len;
	int arg_offset;
	struct fs_file_t file;
	off_t offset = -1;
	int err;

	if (argc < 3) {
		printk("Missing argument\n");
		return 0;
	}

	create_abs_path(argv[1], path, sizeof(path));

	if (!strcmp(argv[2], "-o")) {
		if (argc < 4) {
			printk("Missing argument\n");
			return 0;
		}

		offset = strtol(argv[3], NULL, 0);

		arg_offset = 4;
	} else {
		arg_offset = 2;
	}

	err = fs_open(&file, path);
	if (err) {
		printk("Failed to open %s (%d)\n", path, err);
		return err;
	}

	if (offset < 0) {
		err = fs_seek(&file, 0, FS_SEEK_END);
	} else {
		err = fs_seek(&file, offset, FS_SEEK_SET);
	}
	if (err) {
		printk("Failed to seek %s (%d)\n", path, err);
		fs_close(&file);
		return err;
	}

	buf_len = 0;
	while (arg_offset < argc) {
		buf[buf_len++] = strtol(argv[arg_offset++], NULL, 16);

		if ((buf_len == BUF_CNT) || (arg_offset == argc)) {
			err = fs_write(&file, buf, buf_len);
			if (err < 0) {
				printk("Failed to write %s (%d)\n", path, err);
				fs_close(&file);
				return err;
			}

			buf_len = 0;
		}
	}

	fs_close(&file);

	return 0;
}

static int cmd_ls(int argc, char *argv[])
{
	char path[MAX_PATH_LEN];
	struct fs_dir_t dir;
	int err;

	if (argc < 2) {
		strcpy(path, cwd);
	} else {
		create_abs_path(argv[1], path, sizeof(path));
	}

	err = fs_opendir(&dir, path);
	if (err) {
		printk("Unable to open %s (err %d)\n", path, err);
		return 0;
	}

	while (1) {
		struct fs_dirent entry;

		err = fs_readdir(&dir, &entry);
		if (err) {
			printk("Unable to read directory\n");
			break;
		}

		/* Check for end of directory listing */
		if (entry.name[0] == '\0') {
			break;
		}

		if (entry.type == FS_DIR_ENTRY_DIR) {
			printk("%s/\n", entry.name);
		} else {
			printk("%s\n", entry.name);
		}
	}

	fs_closedir(&dir);

	return 0;
}

static int cmd_pwd(int argc, char *argv[])
{
	printk("%s\n", cwd);
	return 0;
}

static int cmd_cd(int argc, char *argv[])
{
	char path[MAX_PATH_LEN];
	struct fs_dirent entry;
	int err;

	if (argc < 2) {
		strcpy(cwd, "/");
		return 0;
	}

	if (strcmp(argv[1], "..") == 0) {
		char *prev = strrchr(cwd, '/');

		if (!prev || prev == cwd) {
			strcpy(cwd, "/");
		} else {
			*prev = '\0';
		}

		/* No need to test that a parent exists */
		return 0;
	}

	create_abs_path(argv[1], path, sizeof(path));

	err = fs_stat(path, &entry);
	if (err) {
		printk("%s doesn't exist\n", path);
		return 0;
	}

	if (entry.type != FS_DIR_ENTRY_DIR) {
		printk("%s is not a directory\n", path);
		return 0;
	}

	strcpy(cwd, path);

	return 0;
}

static int cmd_trunc(int argc, char *argv[])
{
	char path[MAX_PATH_LEN];
	struct fs_file_t file;
	int length;
	int err;

	if (argc < 2) {
		printk("Missing argument\n");
		return 0;
	}

	if (argv[1][0] == '/') {
		strncpy(path, argv[1], sizeof(path) - 1);
		path[MAX_PATH_LEN - 1] = '\0';
	} else {
		if (strcmp(cwd, "/") == 0) {
			snprintf(path, sizeof(path), "/%s", argv[1]);
		} else {
			snprintf(path, sizeof(path), "%s/%s", cwd, argv[1]);
		}
	}

	if (argc > 2) {
		length = strtol(argv[2], NULL, 0);
	} else {
		length = 0;
	}

	err = fs_open(&file, path);
	if (err) {
		printk("Failed to open %s (%d)\n", path, err);
		return err;
	}

	err = fs_truncate(&file, length);
	if (err) {
		printk("Failed to truncate %s (%d)\n", path, err);
		fs_close(&file);
		return err;
	}

	fs_close(&file);

	return 0;
}


#if defined(CONFIG_FILE_SYSTEM_NFFS) || defined(CONFIG_FAT_FILESYSTEM_ELM)
static int cmd_mount(int argc, char *argv[])
{
	int type, res;
	char *mntpt;

	if (argc != 3) {
		return -EINVAL;
	}

	if (!strcmp(argv[1], "fat")) {
		type = FS_FATFS;
	} else if (!strcmp(argv[1], "nffs")) {
		type = FS_NFFS;
	} else {
		return -EINVAL;
	}

	if (argv[2] != NULL) {
		if (type == FS_FATFS) {
#ifdef CONFIG_FAT_FILESYSTEM_ELM
			mntpt = k_malloc(strlen(argv[2]) + 1);
			if (mntpt == NULL) {
				return -ENOMEM;
			}
			((u8_t *)mntpt)[strlen(argv[2])] = '\0';
			memcpy(mntpt, argv[2], strlen(argv[2]));
			fatfs_mnt.mnt_point = (const char *)mntpt;

			res = fs_mount(&fatfs_mnt);
			if (res < 0) {
				printk("Error mounting fat fs.Error Code [%d]\n", res);
				return res;
			}
			printk("Successfully mounted fat fs:%s\n", fatfs_mnt.mnt_point);
#endif
		} else if (type == FS_NFFS) {
#ifdef CONFIG_FILE_SYSTEM_NFFS
			mntpt = k_malloc(strlen(argv[2]) + 1);
			if (mntpt == NULL) {
				return -ENOMEM;
			}
			((u8_t *)mntpt)[strlen(argv[2])] = '\0';
			memcpy(mntpt, argv[2], strlen(argv[2]));
			nffs_mnt.mnt_point = (const char *)mntpt;

			struct device *flash_dev;

			flash_dev = device_get_binding(CONFIG_FS_NFFS_FLASH_DEV_NAME);
			if (!flash_dev) {
				printk("Error in device_get_binding, while mounting nffs fs\n");
				return -ENODEV;
			}

			nffs_mnt.storage_dev = flash_dev;

			res = fs_mount(&nffs_mnt);
			if (res < 0) {
				printk("Error mounting nffs fs.Error code [%d]\n", res);
				return res;
			}
			printk("Successfully mounted nffs:%s\n", nffs_mnt.mnt_point);
#endif
		}
	}
	return 0;
}
#endif


struct shell_cmd fs_commands[] = {
	{ "ls",    cmd_ls,    "List files in current directory" },
	{ "cd",    cmd_cd,    "Change working directory" },
	{ "pwd",   cmd_pwd,   "Print current working directory" },
	{ "mkdir", cmd_mkdir, "Create directory" },
	{ "rm",    cmd_rm,    "Remove file" },
	{ "read",  cmd_read,  "Read from file" },
	{ "write", cmd_write, "Write to file" },
	{ "trunc", cmd_trunc, "Truncate file" },
#if defined(CONFIG_FILE_SYSTEM_NFFS) || defined(CONFIG_FAT_FILESYSTEM_ELM)
	{ "mount", cmd_mount, "<fs e.g: fat/nffs> <mount-point>" },
#endif
	{ NULL, NULL }
};


SHELL_REGISTER(SHELL_FS, fs_commands);