command.go

package main

import (
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"sync"
	"time"

	jsoniter "github.com/json-iterator/go"
	log "github.com/sirupsen/logrus"

	"github.com/gorilla/websocket"
)

const (
	RTTY_CMD_ERR_INVALID       = 1001
	RTTY_CMD_ERR_OFFLINE       = 1002
	RTTY_CMD_ERR_BUSY          = 1003
	RTTY_CMD_ERR_TIMEOUT       = 1004
	RTTY_CMD_ERR_PENDING       = 1005
	RTTY_CMD_ERR_INVALID_TOKEN = 1006
)

var cmdErrMsg = map[int]string{
	RTTY_CMD_ERR_INVALID:       "invalid format",
	RTTY_CMD_ERR_OFFLINE:       "device offline",
	RTTY_CMD_ERR_BUSY:          "server is busy",
	RTTY_CMD_ERR_TIMEOUT:       "timeout",
	RTTY_CMD_ERR_PENDING:       "pending",
	RTTY_CMD_ERR_INVALID_TOKEN: "invalid token",
}

type commandStatus struct {
	token string
	resp  string
	t     *time.Timer
}

type CommandInfo struct {
	Devid    string `json:"devid"`
	Cmd      string `json:"cmd"`
	Sid      string `json:"sid"`
	Username string `json:"username"`
	Password string `json:"password"`
}

var commands sync.Map

func handleCmdResp(data []byte) {
	token := jsoniter.Get(data, "token").ToString()

	if cmd, ok := commands.Load(token); ok {
		cmd := cmd.(*commandStatus)
		cmd.resp = jsoniter.Get(data, "attrs").ToString()
	}
}

func cmdErrReply(err int, w http.ResponseWriter) {
	fmt.Fprintf(w, `{"err": %d, "msg":"%s"}`, err, cmdErrMsg[err])
}

func serveCmd(br *Broker, w http.ResponseWriter, r *http.Request) {
	token := r.URL.Query().Get("token")
	if token != "" {
		cmd, ok := commands.Load(token)
		if ok {
			cmd := cmd.(*commandStatus)
			if len(cmd.resp) == 0 {
				cmdErrReply(RTTY_CMD_ERR_PENDING, w)
			} else {
				commands.Delete(token)
				io.WriteString(w, cmd.resp)
				cmd.t.Stop()
			}
		} else {
			cmdErrReply(RTTY_CMD_ERR_INVALID_TOKEN, w)
		}
		return
	}

	body, err := ioutil.ReadAll(r.Body)
	if err != nil {
		log.Error(err)
		cmdErrReply(RTTY_CMD_ERR_INVALID, w)
		return
	}

	cmdInfo := CommandInfo{}
	err = jsoniter.Unmarshal(body, &cmdInfo)
	if _, ok := httpSessions.Get(cmdInfo.Sid); err != nil || cmdInfo.Cmd == "" || cmdInfo.Devid == "" || ok == false {
		cmdErrReply(RTTY_CMD_ERR_INVALID, w)
		return
	}

	dev, ok := br.devices[cmdInfo.Devid]
	if !ok {
		cmdErrReply(RTTY_CMD_ERR_OFFLINE, w)
		return
	}

	token = genUniqueID("cmd")

	cmd := &commandStatus{
		token: token,
		t: time.AfterFunc(30*time.Second, func() {
			commands.Delete(token)
		}),
	}

	commands.Store(token, cmd)

	msg := fmt.Sprintf(`{"type":"cmd","token":"%s","attrs":%s}`, token, body)
	dev.wsWrite(websocket.TextMessage, []byte(msg))

	fmt.Fprintf(w, `{"token":"%s"}`, token)
}