Merge pull request #11 from dhlin/snapshot

add snapshot support

Author: Benehiko

internal/pkg/api/image_mixin.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"github.com/ReolinkCameraAPI/reolinkapigo/internal/pkg/enum"
 	"github.com/ReolinkCameraAPI/reolinkapigo/internal/pkg/network/rest"
+	"net/url"
 )
 
 type ImageMixin struct {
@@ -199,6 +200,22 @@ func (im *ImageMixin) SetImageSettings(imageOptions ...OptionImageSetting) func(
 	}
 }
 
+func (im *ImageMixin) Snap() func(handler *rest.RestHandler) ([]byte, error) {
+	return func(handler *rest.RestHandler) ([]byte, error) {
+		params := url.Values{}
+		params.Add("cmd", "Snap")
+		params.Add("channel", "0")
+
+		snapshot, err := handler.RequestRaw("GET", "", params)
+
+		if err != nil {
+			return nil, err
+		}
+
+		return snapshot, nil
+	}
+}
+
 // Set Image Brightness
 // Default: 128
 func ImageOptionBrightness(brightness int) OptionImageSetting {

internal/pkg/network/rest/rest_handler.go

@@ -182,6 +182,27 @@ func NewRestHandler(host string, opts ...OptionRestHandler) *RestHandler {
 // auth: alters the request to include auth token on true
 func (rh *RestHandler) Request(method string, payload interface{}, command string) (*GeneralData, error) {
 
+	params := url.Values{}
+	params.Add("cmd", command)
+
+	respBody, err := rh.RequestRaw(method, payload, params)
+
+	if err != nil {
+		return nil, err
+	}
+
+	var result []*GeneralData
+
+	err = json.Unmarshal(respBody, &result)
+
+	if err != nil {
+		return nil, err
+	}
+
+	return result[0], nil
+}
+
+func (rh *RestHandler) RequestRaw(method string, payload interface{}, params url.Values) ([]byte, error) {
 	var urlConcat string
 	if rh.port > 0 {
 		urlConcat = fmt.Sprintf("%s:%d/%s", rh.host, rh.port, rh.endpoint)
@@ -191,14 +212,12 @@ func (rh *RestHandler) Request(method string, payload interface{}, command strin
 
 	urlConcat = fmt.Sprintf("%s://%s", rh.scheme.String(), urlConcat)
 
-	var data []byte
-
-	params := url.Values{}
 	params.Add("token", rh.token)
-	params.Add("cmd", command)
 
 	urlConcat = fmt.Sprintf("%s?%s", urlConcat, params.Encode())
 
+	var data []byte
+
 	data, err := json.Marshal([]interface{}{payload})
 
 	if err != nil {
@@ -309,20 +328,13 @@ func (rh *RestHandler) Request(method string, payload interface{}, command strin
 		return nil, err
 	}
 
-	var result []*GeneralData
 	body, err := ioutil.ReadAll(resp.Body)
 
 	if err != nil {
 		return nil, err
 	}
 
-	err = json.Unmarshal(body, &result)
-
-	if err != nil {
-		return nil, err
-	}
-
-	return result[0], nil
+	return body, nil
 }
 
 // Set the current token