add reboot

Author: Huan Wang

README.md

@@ -8,6 +8,7 @@ This Go package provides a basic API wrapper for the tp-link M4 mesh router
 - [x] ClientList: Get connected clients
 - [x] Custom: Make a custom request to the router
 - [x] DeviceList: Get connected Decos
+- [x] Reboot: Reboot Deco mesh node(s)
 - [ ] Wan: Get information about wan
 
 `TODO: look through the firmware for more endpoints`

client.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"net/http/cookiejar"
 	"net/url"
+	"strings"
 	"time"
 
 	"github.com/mrmarble/deco/utils"
@@ -51,7 +52,7 @@ type ClientListResp struct {
 			Name           string `json:"name"`
 			Online         bool   `json:"online"`
 			OwnerID        string `json:"owner_id"`
-			RemainTime     int   `json:"remain_time"`
+			RemainTime     int    `json:"remain_time"`
 			SpaceID        string `json:"space_id"`
 			UpSpeed        uint   `json:"up_speed"`
 			WireType       string `json:"wire_type"`
@@ -192,7 +193,7 @@ func (c *Client) ClientList() (*ClientListResp, error) {
 	var result ClientListResp
 	request := request{
 		Operation: "read",
-		Params:    map[string]string{"device_mac": "default"},
+		Params:    map[string]interface{}{"device_mac": "default"},
 	}
 	jsonRequest, _ := json.Marshal(request)
 	err := c.doEncryptedPost(fmt.Sprintf(";stok=%s/admin/client", c.stok), EndpointArgs{form: "client_list"}, jsonRequest, false, &result)
@@ -208,6 +209,33 @@ func (c *Client) ClientList() (*ClientListResp, error) {
 	return &result, nil
 }
 
+// Reboot the deco nodes by mac address
+func (c *Client) Reboot(macAddrs ...string) (map[string]interface{}, error) {
+	var result map[string]interface{}
+	var macList []map[string]string
+
+	for _, mac := range macAddrs {
+		macList = append(macList, map[string]string{
+			"mac": strings.ToUpper(mac),
+		})
+	}
+
+	request := request{
+		Operation: "reboot",
+		Params: map[string]interface{}{
+			"mac_list": macList,
+		},
+	}
+
+	jsonRequest, _ := json.Marshal(request)
+	err := c.doEncryptedPost(fmt.Sprintf(";stok=%s/admin/device", c.stok), EndpointArgs{form: "system"}, jsonRequest, false, &result)
+	if err != nil {
+		return nil, err
+	}
+
+	return result, nil
+}
+
 // Custom lets you make a custom request
 func (c *Client) Custom(path string, params EndpointArgs, body []byte) (interface{}, error) {
 	var result interface{}

deco.go

@@ -53,8 +53,8 @@ type response struct {
 }
 
 type request struct {
-	Operation string            `json:"operation,omitempty"`
-	Params    map[string]string `json:"params,omitempty"`
+	Operation string                 `json:"operation,omitempty"`
+	Params    map[string]interface{} `json:"params,omitempty"`
 }
 
 // EndpointArgs holds the url params to be sent

utils/aes.go

@@ -78,8 +78,8 @@ func AES256Decrypt(encrypted string, key AESKey) (string, error) {
 
 	mode := cipher.NewCBCDecrypter(block, key.Iv)
 	mode.CryptBlocks(cipherText, cipherText)
-	cipherText, _ = pkcs7Unpadding(cipherText, aes.BlockSize)
-	return string(cipherText), nil
+	cipherText, err = pkcs7Unpadding(cipherText, aes.BlockSize)
+	return string(cipherText), err
 }
 
 func pkcs7Padding(b []byte, blocksize int) ([]byte, error) {