#!/bin/sh

username=$1
password=$2
baseurl=$3
site=$4

cookie=$(mktemp)
curl_cmd="curl --tlsv1 --silent --cookie ${cookie} --cookie-jar ${cookie} --insecure "

# Check arguments
unifi_requires() {
    if [ -z "$username" -o -z "$password" -o -z "$baseurl" -o -z "$site" ] ; then
        echo "Usage! unifi_count_site username password baseurl site"
        return
    fi
}

# Authenticate against unifi controller
unifi_login() {
    ${curl_cmd} --data "{'username':'$username', 'password':'$password'}" $baseurl/api/login >/dev/null 2>/dev/null
}

# Logout
unifi_logout() {
    ${curl_cmd} $baseurl/logout
}

# Restart APs
unifi_restart_site_ap() {
     for i in `${curl_cmd} --data "json={}" $baseurl/api/s/$site/stat/device | jq '.data[].mac'`
     do
        mac=`echo $i | sed -e 's/^"//' -e 's/"$//'`
        ${curl_cmd} --data "json={'cmd':'restart', 'mac':'${mac}'}" $baseurl/api/s/$site/cmd/devmgr >/dev/null 2>/dev/null
        sleep 5
     done
}

unifi_requires
unifi_login
unifi_restart_site_ap
unifi_logout