/**
* Copyright (c) <2013-2016>, Wind River Systems, Inc.
*
* Redistribution and use in source and binary forms, with or without modification, are
* permitted provided that the following conditions are met:
*
* 1) Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
*
* 2) Redistributions in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation and/or
* other materials provided with the distribution.
*
* 3) Neither the name of Wind River Systems nor the names of its contributors may be
* used to endorse or promote products derived from this software without specific
* prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
*  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
* SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
* CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
* USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <json-c/json.h>
#include "misc.h"


/* Print a range of numbers in a consistent way. */
char print_range(int start, int end, char *str)
{
    int len;
    if (start == end)
        len = sprintf(str, "%d,", start);
    else
        len = sprintf(str, "%d-%d,", start, end);
    return len;
}

void print_array(char *buf, int *array, int len) {
    int i;
    for(i = 0; i < len; i++) {
       sprintf(buf+strlen(buf),"%d,", array[i]);
    }
    // remove the last comma
    buf[strlen(buf)-1]='\0';
}

#define BUFLEN 1024

/* Takes as input a string representation of online cpus of the form
 * "0,1,3-5", and allocates and returns a struct representing whether
 * each given cpu is online.
 */
struct online_cpus *range_to_array(const char *range)
{
    struct online_cpus *cpuarray = (struct online_cpus *) malloc(BUFLEN);
    int start, end;
    int inrange = 0;
    char *token, *tmp, *tobe_free;
    int done = 0;
    tobe_free = strdup(range);
    token = tmp = tobe_free;

    if (*tmp == '\0') {
        /* empty string, no online cpus */
        cpuarray->numcpus = 0;
        free(tobe_free);
        return cpuarray;
    }
        
    while (1) {
        tmp++;
        if (*tmp == '\0')
            done = 1;
        if (done || (*tmp == ',')) {
            /* expect single value or ending a range */
            if (!token) {
                ERR_LOG("format error, missing token, unable to parse range\n");
                goto error;
            }
            *tmp = '\0';
            end = atoi(token);
            token = 0;
            if (inrange) {
                int i;
                for (i=start; i<= end; i++)
                    cpuarray->status[i] = 1;
                inrange = 0;
            } else {
                cpuarray->status[end] = 1;
            }
        } else if (*tmp == '-') {
            if (inrange) {
                ERR_LOG("format error, unable to parse range\n");
                goto error;
            }
            if (!token) {
                ERR_LOG("format error, missing token, unable to parse range\n");
                goto error;
            }
            *tmp = '\0';
            start = atoi(token);
            token = 0;
            inrange = 1;
        } else {
            /* expect a numerical value */
            if ((*tmp < '0') || (*tmp > '9')) {
                ERR_LOG("format error, expected a numerical value, unable to parse range\n");
                goto error;
            }
                
            if (!token)
                token = tmp;
        }
        if (done)
            break;
    }
    cpuarray->numcpus = end+1;
    free(tobe_free);
    return cpuarray;
error:
    free(cpuarray);
    free(tobe_free);
    return 0;
}


struct json_object *new_json_obj_from_array (struct online_cpus *cpuarray)
{
    int i;

    struct json_object *jobj_array = json_object_new_array();
    if (jobj_array == NULL) {
        return NULL;
    }

    for (i=0;i<cpuarray->numcpus;i++) {
        if (cpuarray->status[i]) {
            json_object_array_add(jobj_array, json_object_new_int(i));
        }
    }
    return jobj_array;
}