#include <common.h>
#include <command.h>
#include <driver.h>
#include <linux/ctype.h>

int cmd_get_data_size(char* arg, int default_size)
{
	/* Check for a size specification .b, .w or .l.
	 */
	int len = strlen(arg);
	if (len > 2 && arg[len-2] == '.') {
		switch(arg[len-1]) {
		case 'b':
			return 1;
		case 'w':
			return 2;
		case 'l':
			return 4;
		case 's':
			return -2;
		default:
			return -1;
		}
	}
	return default_size;
}

static ulong device_handler[MAX_DEVICE_TYPE];

int register_device_type_handler(int(*handler)(struct device_d *), ulong device_type)
{
        device_handler[device_type] = (ulong)handler;

        /* FIXME: cycle through devices */

        return 0;
}

static struct device_d *first_device = NULL;
static struct driver_d *first_driver = NULL;

struct device_d *get_device_by_id(char *id)
{
	struct device_d *d;

	d = first_device;

	while(d) {
		if(!strcmp(id, d->id))
			break;
		d = d->next;
	}

	return d;
}

static int match(struct driver_d *drv, struct device_d *dev)
{
        int(*handler)(struct device_d *);

        if (strcmp(dev->name, drv->name))
                return -1;
        if (dev->type != drv->type)
                return -1;
        if(drv->probe(dev))
                return -1;

        dev->driver = drv;

        handler = device_handler[dev->type];

        if(handler)
                (*handler)(dev);
        return 0;
}

int register_device(struct device_d *new_device)
{
	struct driver_d *drv;
        struct device_d *dev;

	dev = first_device;

//        printf("register_device: %s\n",new_device->name);

	if(!dev) {
		first_device = new_device;
	} else {
		while(dev->next)
			dev = dev->next;
	}

	dev->next = new_device;
	new_device->next = 0;

        drv = first_driver;

        while(drv) {
                if (!match(drv, new_device))
                        break;
                drv = drv->next;
        }

	return 0;
}

void unregister_device(struct device_d *old_dev)
{
        struct device_d *dev;
//        printf("unregister_device: %s\n",old_dev->name);

	dev = first_device;

        while (dev) {
                if (!strcmp(dev->next->name, old_dev->name)) {
                        dev->next = old_dev->next;
                        return;
                }
                dev = dev->next;
        }
}

struct driver_d *get_driver_by_name(char *name)
{
	struct driver_d *d;

	d = first_driver;

	while(d) {
		if(!strcmp(name, d->name))
			break;
		d = d->next;
	}

	return d;
}

static void noinfo(struct device_d *dev)
{
        printf("no info available for %s\n", dev->id);
}

static void noshortinfo(struct device_d *dev)
{
}

int register_driver(struct driver_d *new_driver)
{
	struct driver_d *drv;
        struct device_d *dev = NULL;

	drv = first_driver;

//        printf("register_driver: %s\n",new_driver->name);

	if(!drv) {
		first_driver = new_driver;
	} else {
		while(drv->next)
			drv = drv->next;
	}

	drv->next = new_driver;
	new_driver->next = 0;

        if (!new_driver->info)
                new_driver->info = noinfo;
        if (!new_driver->shortinfo)
                new_driver->shortinfo = noshortinfo;

        dev = first_device;
        while (dev) {
                match(new_driver, dev);
                dev = dev->next;
        }

	return 0;
}

static char devicename_from_spec_str_buf[MAX_DRIVER_NAME];

char *deviceid_from_spec_str(const char *str, char **endp)
{
        char *buf = devicename_from_spec_str_buf;
        const char *end;
	int i = 0;

        if (isdigit(*str)) {
                /* No device name given, use default driver mem */
                sprintf(buf, "mem");
                end = str;
        } else {
                /* OK, we have a device name, parse it */
	        while (*str) {
		        if (*str == ':') {
                                str++;
			        buf[i] = 0;
			        break;
        		}

	        	buf[i++] = *str++;
		        buf[i] = 0;
		        if (i == MAX_DRIVER_NAME)
			        return NULL;
        	}
                end = str;
        }

        if (endp)
                *endp = (char *)end;

        return buf;
}

/* Get a device struct from the beginning of the string. Default to mem if no
 * device is given, return NULL if a unknown device is given.
 * If endp is not NULL, this function stores a pointer to the first character
 * after the device name in *endp.
 */
struct device_d *device_from_spec_str(const char *str, char **endp)
{
        char *name;
        name = deviceid_from_spec_str(str, endp);
        return get_device_by_id(name);
}

unsigned long strtoul_suffix(const char *str, char **endp, int base)
{
        unsigned long val;
        char *end;

        val = simple_strtoul(str, &end, base);

        switch (*end) {
        case 'G':
                val *= 1024;
        case 'M':
		val *= 1024;
        case 'k':
        case 'K':
                val *= 1024;
                end++;
        }

        if (endp)
                *endp = (char *)end;

        return val;
}

/*
 * Parse a string representing a memory area and fill in a struct memarea_info.
 */
int spec_str_to_info(const char *str, struct memarea_info *info)
{
	char *endp;

	memset(info, 0, sizeof(struct memarea_info));

        info->device = device_from_spec_str(str, &endp);
        if (!info->device) {
                printf("unknown device: %s\n", deviceid_from_spec_str(str, NULL));
                return 1;
        }

        str = endp;

        if (!*str) {
                /* no area specification given, use whole device */
                info->size = info->device->size;
                info->end  = info->size - 1;
                goto out;
        }

        if (*str == '+') {
                /* no beginning given but size so start is 0 */
                info->size = strtoul_suffix(str + 1, &endp, 0);
                info->end  = info->size - 1;
                info->flags = MEMAREA_SIZE_SPECIFIED;
                goto out;
        }

        if (*str == '-') {
                /* no beginning given but end, so start is 0 */
                info->end = strtoul_suffix(str + 1, &endp, 0);
                info->size  = info->end + 1;
                info->flags = MEMAREA_SIZE_SPECIFIED;
                goto out;
        }

        str = endp;
        info->start = strtoul_suffix(str, &endp, 0);
        str = endp;

        if (info->start >= info->device->size) {
                printf("Start address is beyond device\n");
                return -1;
        }

        if (!*str) {
                /* beginning given, but no size, pad to the and of the device */
                info->end = info->device->size - 1;
                info->size = info->end - info->start + 1;
                goto out;
        }

	if (*str == '-') {
                /* beginning and end given */
		info->end = strtoul_suffix(str + 1, NULL, 0);
		info->size = info->end - info->start + 1;
                if (info->end < info->start) {
                        printf("end < start\n");
                        return -1;
                }
                info->flags = MEMAREA_SIZE_SPECIFIED;
                goto out;
	}

	if (*str == '+') {
                /* beginning and size given */
		info->size = strtoul_suffix(str + 1, NULL, 0);
                info->end = info->start + info->size - 1;
                info->flags = MEMAREA_SIZE_SPECIFIED;
                goto out;
	}

out:
        /* check for device boundaries */
        if (info->end > info->device->size) {
                info->end = info->device->size - 1;
                info->size = info->end - info->start + 1;
        }
#if 0
	printf("start: 0x%08x\n",info->start);
	printf("size: 0x%08x\n",info->size);
	printf("end: 0x%08x\n",info->end);
#endif
	return 0;
}

ssize_t read(struct device_d *dev, void *buf, size_t count, unsigned long offset, ulong flags)
{
        if (dev->driver->read)
                return dev->driver->read(dev, buf, count, offset, flags);
        return -1;
}

ssize_t write(struct device_d *dev, void *buf, size_t count, unsigned long offset, ulong flags)
{
        if (dev->driver->write)
                return dev->driver->write(dev, buf, count, offset, flags);
        return -1;
}

ssize_t erase(struct device_d *dev, size_t count, unsigned long offset)
{
        if (dev->driver->erase)
                return dev->driver->erase(dev, count, offset);
        return -1;
}

int dummy_probe(struct device_d *dev)
{
        return 0;
}

static struct param_d *get_param_by_name(struct device_d *dev, char *name)
{
        struct param_d *param = dev->param;

        while (param) {
                if (!strcmp(param->name, name))
                        return param;
                param = param->next;
        }

        return NULL;
}

char *dev_get_param(struct device_d *dev, char *name)
{
        struct param_d *param = get_param_by_name(dev, name);

        if (param && param->get)
                return param->get(dev, param->cookie);

        return NULL;
}

int dev_set_param(struct device_d *dev, char *name, char *val)
{
        struct param_d *param = get_param_by_name(dev, name);

        if (param && param->set)
                return param->set(dev, param->cookie, val);

        return -1;
}

int dev_add_parameter(struct device_d *dev, struct param_d *newparam)
{
        struct param_d *param = dev->param;

        newparam->next = 0;

        if (param) {
                while (param->next)
                        param = param->next;
                        param->next = newparam;
        } else {
                dev->param = newparam;
        }

        return 0;
}

int do_devinfo ( cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
{
        struct device_d *dev = first_device;
        struct driver_d *drv = first_driver;
        struct param_d *param;

        if (argc == 1) {
                printf("devices:\n");

                while(dev) {
                        printf("%10s: base=0x%08x size=0x%08x (driver %s)\n",dev->id, dev->map_base, dev->size, dev->name);
                        dev = dev->next;
                }

                printf("drivers:\n");
                while(drv) {
                        printf("%10s\n",drv->name);
                        drv = drv->next;
                }
        } else {
                struct device_d *dev = get_device_by_id(argv[1]);

                if (!dev) {
                        printf("no such device: %s\n",argv[1]);
                        return -1;
                }
                dev->driver->info(dev);

                param = dev->param;

                printf("%s\n", param ? "Parameters:" : "no parameters available");

                while (param) {
                        printf("%16s = %s\n", param->name, param->get(dev, param->cookie));
                        param = param->next;
                }

        }

        return 0;
}

U_BOOT_CMD(
	devinfo,     2,     0,      do_devinfo,
	"devinfo     - display info about devices and drivers\n",
	""
);