arozos/mod/iot/assits.go

package iot

import (
	"encoding/json"
	"net/http"

	"imuslab.com/arozos/mod/utils"
)

/*
	assits.go
	Author: tobychui

	This script handle assistant functions for iot devices.
	The function implement here should have no effect to the core operation of the iot hub nor the iot pipeline.
*/

//Handle the set and get nickname of a particular IoT device
func (m *Manager) HandleNickName(w http.ResponseWriter, r *http.Request) {
	opr, err := utils.Mv(r, "opr", true)
	if err != nil {
		utils.SendErrorResponse(w, "Invalid operation mode")
		return
	}

	uuid, err := utils.Mv(r, "uuid", true)
	if err != nil {
		utils.SendErrorResponse(w, "Invalid uuid given")
		return
	}

	//Check if the device with the given uuid exists
	deviceExist := false
	for _, dev := range m.cachedDeviceList {
		if dev.DeviceUUID == uuid {
			//Device found. Create a new object and make the pointer point to it
			deviceExist = true
		}
	}

	//Reject operation if device not exists
	if deviceExist == false {
		utils.SendErrorResponse(w, "Target device UUID not exists")
		return
	}

	//Process the required operation
	if opr == "get" {
		if m.db.KeyExists("iot", uuid) {
			deviceNickname := ""
			err := m.db.Read("iot", uuid, &deviceNickname)
			if err != nil {
				utils.SendErrorResponse(w, "Unable to read nickname from database")
				return
			}
			js, _ := json.Marshal(deviceNickname)
			utils.SendJSONResponse(w, string(js))
		} else {
			utils.SendErrorResponse(w, "Nickname not exists")
		}
	} else if opr == "set" {
		//Get name from paramter
		name, err := utils.Mv(r, "name", true)
		if err != nil {
			utils.SendErrorResponse(w, "No nickname was given to the device")
			return
		}

		//Set the name in database
		err = m.db.Write("iot", uuid, name)
		if err != nil {
			utils.SendErrorResponse(w, err.Error())
			return
		}

		utils.SendOK(w)
	} else {
		utils.SendErrorResponse(w, "Unknown operation mode")
		return
	}
}