feat: implement faces api with resource and data source

This commit is contained in:
2026-06-01 12:21:02 -04:00
parent 07dea33680
commit d9281bcde9
8 changed files with 614 additions and 0 deletions
+123
View File
@@ -0,0 +1,123 @@
package client
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
)
type Face struct {
ID string `json:"id"`
AssetId string `json:"assetId"`
PersonId string `json:"personId"`
BoundingBoxX1 float64 `json:"boundingBoxX1"`
BoundingBoxY1 float64 `json:"boundingBoxY1"`
BoundingBoxX2 float64 `json:"boundingBoxX2"`
BoundingBoxY2 float64 `json:"boundingBoxY2"`
ImageHeight int `json:"imageHeight"`
ImageWidth int `json:"imageWidth"`
}
type CreateFaceRequest struct {
AssetId string `json:"assetId"`
PersonId string `json:"personId"`
BoundingBoxX1 float64 `json:"boundingBoxX1"`
BoundingBoxY1 float64 `json:"boundingBoxY1"`
BoundingBoxX2 float64 `json:"boundingBoxX2"`
BoundingBoxY2 float64 `json:"boundingBoxY2"`
ImageHeight int `json:"imageHeight"`
ImageWidth int `json:"imageWidth"`
}
type UpdateFaceRequest struct {
PersonId string `json:"personId"`
}
func (c *Client) GetFaces(assetId string) ([]Face, error) {
url := fmt.Sprintf("%s/faces?id=%s", c.HostURL, assetId)
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
}
body, err := c.doRequest(req)
if err != nil {
return nil, err
}
var faces []Face
err = json.Unmarshal(body, &faces)
if err != nil {
return nil, err
}
return faces, nil
}
func (c *Client) CreateFace(face CreateFaceRequest) (*Face, error) {
rb, err := json.Marshal(face)
if err != nil {
return nil, err
}
req, err := http.NewRequest("POST", fmt.Sprintf("%s/faces", c.HostURL), bytes.NewBuffer(rb))
if err != nil {
return nil, err
}
body, err := c.doRequest(req)
if err != nil {
return nil, err
}
var newFace Face
err = json.Unmarshal(body, &newFace)
if err != nil {
return nil, err
}
return &newFace, nil
}
func (c *Client) UpdateFace(id string, update UpdateFaceRequest) (*Face, error) {
// Bulk reassign endpoint is PUT /faces, but usually for a single face it might be PUT /faces/{id}
// or we use the bulk one with one ID.
// Documentation said PUT /faces with a list of ids and a personId.
type BulkUpdateFaceRequest struct {
Ids []string `json:"ids"`
PersonId string `json:"personId"`
}
rb, err := json.Marshal(BulkUpdateFaceRequest{
Ids: []string{id},
PersonId: update.PersonId,
})
if err != nil {
return nil, err
}
req, err := http.NewRequest("PUT", fmt.Sprintf("%s/faces", c.HostURL), bytes.NewBuffer(rb))
if err != nil {
return nil, err
}
_, err = c.doRequest(req)
if err != nil {
return nil, err
}
// The API might not return the updated face, so we might need to fetch it or just return a placeholder.
// But wait, GET /faces requires assetId. If we don't have it, we can't easily fetch it back by ID alone if GET /faces/{id} doesn't exist.
return &Face{ID: id, PersonId: update.PersonId}, nil
}
func (c *Client) DeleteFace(id string) error {
req, err := http.NewRequest("DELETE", fmt.Sprintf("%s/faces/%s", c.HostURL, id), bytes.NewBufferString(`{"force": true}`))
if err != nil {
return err
}
_, err = c.doRequest(req)
return err
}