#!/bin/sh

. ../common/ajax_common

if [ "${PALANG}" = "en" ]; then
	LANG001="Add_Virtual_Data_link"	#ͨ
	LANG002="Success"	#ɹ
	LANG003="Error"		#ʧ
	LANG004="Set_Virtual_Data_link"	#༭ͨ
	LANG005="Del_Virtual_Data_link"	#ɾͨ
fi


list_data_pie()
{
    ${FLOWEYE} policy listbwo | grep "${CGI_keyword}" | awk \
    -v status="${CGI_status}" \
    'BEGIN{
        dot = "";
        printf "[";
    }{
        if(status != "" && $10 != status) next;

        col = 1;
        printf "%s{", dot;
        printf "\"id\":\"%s\",", $(col++);
        printf "\"name\":\"%s\",", $(col++);
        printf "\"rate\":\"%s\",", $(col++);
        printf "\"bytes\":\"%s\",", $(col++);
        printf "\"size\":\"%s\",", $(col++);
        printf "\"outbps\":\"%s\",", $(col++);
        printf "\"dropbps\":\"%s\",", $(col++);
        printf "\"quota\":\"%s\",", $(col++);
        printf "\"left\":\"%s\",", $(col++);
        printf "\"active\":\"%s\"", $(col++);
        printf "}";
        if(dot == "") dot = ",";
    }END{
        printf "]";
    }'
}


get_data_pie()
{
	printf "{"

    for line in `${FLOWEYE} policy getbwo name=${CGI_name}`
    do
        key=`echo ${line} | cut -d"=" -f1`
        val=`echo ${line} | cut -d"=" -f2`
        printf "\"${key}\":\"${val}\","
    done

    ${FLOWEYE} policy gethtb ${CGI_name} | awk \
    'BEGIN{
        dot="";
        printf "\"htb\":[";
    }{
        if($1=="pri") next;
        printf "%s{", dot;
        printf "\"pri\":\"%s\",", $1;
        printf "\"gbw\":\"%s\",", $2;
        printf "\"bytes\":\"%s\",", $3;
        printf "\"outbps\":\"%s\",", $4;
        printf "\"dropbps\":\"%s\",", $5;
        printf "\"desc\":\"%s\"", $6;
        printf "}";
        if(dot=="") dot=","
    }END{
        printf "]";
    }'

    printf "}"
}


add_data_pie()
{
    [ "${CGI_quota}" = "" ] && CGI_quota=0

	errmsg=`${FLOWEYE} policy addbwo name=${CGI_name} rate=${CGI_rate} quota=${CGI_quota}`

	if [ $? -ne 0 ]; then
		retjson 1 "${errmsg}"
	else
        sync_floweye "policy addbwo name=${CGI_name} rate=${CGI_rate} quota=${CGI_quota}"
		WEB_LOGGER "${LANG001:=ͨ}" "name=${CGI_pname} rate=${CGI_rate} quota=${CGI_quota}"
		retjson 0 "${LANG002:=ɹ}"
	fi
}


edit_data_pie()
{
    num=1
    while [ ${num} -le 6 ]
    do
        eval ${FLOWEYE} policy sethtb name=${CGI_name} pri=${num} gbw=\$CGI_gbw_${num} desc=\$CGI_desc_${num}
        [ $? -ne 0 ] && retjson 1 "${LANG003:=ʧ}"
        eval sync_floweye "policy sethtb name=${CGI_name} pri=${num} gbw=\$CGI_gbw_${num} desc=\$CGI_desc_${num}"
        num=$((${num}+1))
    done

    [ "${CGI_quota}" = "" ] &&CGI_quota=0
	errmsg=`${FLOWEYE} policy setbwo name=${CGI_name} rate=${CGI_rate} quota=${CGI_quota}`

    [ $? -ne 0 ] && retjson 1 "${LANG003:=ʧ}:${errmsg}"

    sync_floweye "policy setbwo name=${CGI_name} rate=${CGI_rate} quota=${CGI_quota}"
    WEB_LOGGER "${LANG004:=༭ͨ}" "name=${CGI_name} rate=${CGI_rate}"
    retjson 0 "${LANG002:=ɹ}"
}


del_data_pie()
{
    errmsg=`${FLOWEYE} policy rmvbwo name=${CGI_name}`

    if [ $? -ne 0 ]; then
        retjson 1 "${errmsg}"
    else
        sync_floweye "policy rmvbwo name=${CGI_name}"
        WEB_LOGGER "${LANG005:=ɾͨ}" "name=${CGI_name}"
		retjson 0 "${LANG002:=ɹ}"
    fi
}


case "${CGI_action}" in
    "list_data_pie")
        retjson 0 "OK" "`list_data_pie`"
        ;;
    
    "get_data_pie")
        retjson 0 "OK" "`get_data_pie`"
        ;;

    "add_data_pie")
        action_check
        add_data_pie
        ;;

    "edit_data_pie")
        action_check
        edit_data_pie
        ;;

    "del_data_pie")
        action_check
        del_data_pie
        ;;

    *)
        retjson 1 "UNKNOW_ACTION"
        ;;
esac
