faceprint check

Author: dahernan

facebox/facebox_faceprint.go

@@ -70,3 +70,58 @@ func (c *Client) parseCompareFacesResponse(r io.Reader) ([]float64, error) {
 	}
 	return compareFaceprintsResponse.Confidences, nil
 }
+
+type checkFaceprintRequest struct {
+	Faceprints []string `json:"faceprints"`
+}
+
+func (c *Client) CheckFaceprints(faceprints []string) ([]Face, error) {
+	if len(faceprints) == 0 {
+		return nil, errors.New("faceprints can not be empty")
+	}
+	u, err := url.Parse(c.addr + "/facebox/faceprint/check")
+	if err != nil {
+		return nil, err
+	}
+	if !u.IsAbs() {
+		return nil, errors.New("box address must be absolute")
+	}
+	request := checkFaceprintRequest{
+		Faceprints: faceprints,
+	}
+	var buf bytes.Buffer
+	if err := json.NewEncoder(&buf).Encode(request); err != nil {
+		return nil, errors.Wrap(err, "encoding request body")
+	}
+	req, err := http.NewRequest(http.MethodPost, u.String(), &buf)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Accept", "application/json; charset=utf-8")
+	req.Header.Set("Content-Type", "application/json; charset=utf-8")
+
+	resp, err := c.HTTPClient.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
+		return nil, errors.New(resp.Status)
+	}
+	return c.parseCheckFaceprintResponse(resp.Body)
+}
+
+func (c *Client) parseCheckFaceprintResponse(r io.Reader) ([]Face, error) {
+	var checkResponse struct {
+		Success    bool
+		Error      string
+		Faceprints []Face
+	}
+	if err := json.NewDecoder(r).Decode(&checkResponse); err != nil {
+		return nil, errors.Wrap(err, "decoding response")
+	}
+	if !checkResponse.Success {
+		return nil, ErrFacebox(checkResponse.Error)
+	}
+	return checkResponse.Faceprints, nil
+}