feat: implement faces api with resource and data source
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
---
|
||||
# generated by https://github.com/hashicorp/terraform-plugin-docs
|
||||
page_title: "immich_faces Data Source - immich"
|
||||
subcategory: ""
|
||||
description: |-
|
||||
Retrieves a list of faces for a specific asset.
|
||||
---
|
||||
|
||||
# immich_faces (Data Source)
|
||||
|
||||
Retrieves a list of faces for a specific asset.
|
||||
|
||||
## Example Usage
|
||||
|
||||
```terraform
|
||||
data "immich_faces" "current" {
|
||||
asset_id = "your-asset-uuid"
|
||||
}
|
||||
|
||||
output "detected_faces" {
|
||||
value = data.immich_faces.current.faces
|
||||
}
|
||||
```
|
||||
|
||||
<!-- schema generated by tfplugindocs -->
|
||||
## Schema
|
||||
|
||||
### Required
|
||||
|
||||
- `asset_id` (String) The ID of the asset.
|
||||
|
||||
### Read-Only
|
||||
|
||||
- `faces` (Attributes List) List of faces. (see [below for nested schema](#nestedatt--faces))
|
||||
|
||||
<a id="nestedatt--faces"></a>
|
||||
### Nested Schema for `faces`
|
||||
|
||||
Read-Only:
|
||||
|
||||
- `bounding_box_x1` (Number) The left coordinate of the bounding box.
|
||||
- `bounding_box_x2` (Number) The right coordinate of the bounding box.
|
||||
- `bounding_box_y1` (Number) The top coordinate of the bounding box.
|
||||
- `bounding_box_y2` (Number) The bottom coordinate of the bounding box.
|
||||
- `id` (String) Unique identifier for the face.
|
||||
- `image_height` (Number) The height of the image in pixels.
|
||||
- `image_width` (Number) The width of the image in pixels.
|
||||
- `person_id` (String) The ID of the person associated with this face.
|
||||
@@ -0,0 +1,44 @@
|
||||
---
|
||||
# generated by https://github.com/hashicorp/terraform-plugin-docs
|
||||
page_title: "immich_face Resource - immich"
|
||||
subcategory: ""
|
||||
description: |-
|
||||
Manages an Immich face (detected or manual).
|
||||
---
|
||||
|
||||
# immich_face (Resource)
|
||||
|
||||
Manages an Immich face (detected or manual).
|
||||
|
||||
## Example Usage
|
||||
|
||||
```terraform
|
||||
resource "immich_face" "manual" {
|
||||
asset_id = "your-asset-uuid"
|
||||
person_id = "your-person-uuid"
|
||||
bounding_box_x1 = 100.5
|
||||
bounding_box_y1 = 100.5
|
||||
bounding_box_x2 = 200.5
|
||||
bounding_box_y2 = 200.5
|
||||
image_height = 1080
|
||||
image_width = 1920
|
||||
}
|
||||
```
|
||||
|
||||
<!-- schema generated by tfplugindocs -->
|
||||
## Schema
|
||||
|
||||
### Required
|
||||
|
||||
- `asset_id` (String) The ID of the asset this face belongs to.
|
||||
- `bounding_box_x1` (Number) The left coordinate of the bounding box.
|
||||
- `bounding_box_x2` (Number) The right coordinate of the bounding box.
|
||||
- `bounding_box_y1` (Number) The top coordinate of the bounding box.
|
||||
- `bounding_box_y2` (Number) The bottom coordinate of the bounding box.
|
||||
- `image_height` (Number) The height of the image in pixels.
|
||||
- `image_width` (Number) The width of the image in pixels.
|
||||
- `person_id` (String) The ID of the person associated with this face.
|
||||
|
||||
### Read-Only
|
||||
|
||||
- `id` (String) Unique identifier for the face.
|
||||
@@ -0,0 +1,7 @@
|
||||
data "immich_faces" "current" {
|
||||
asset_id = "your-asset-uuid"
|
||||
}
|
||||
|
||||
output "detected_faces" {
|
||||
value = data.immich_faces.current.faces
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
resource "immich_face" "manual" {
|
||||
asset_id = "your-asset-uuid"
|
||||
person_id = "your-person-uuid"
|
||||
bounding_box_x1 = 100.5
|
||||
bounding_box_y1 = 100.5
|
||||
bounding_box_x2 = 200.5
|
||||
bounding_box_y2 = 200.5
|
||||
image_height = 1080
|
||||
image_width = 1920
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
@@ -0,0 +1,231 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/hashicorp/terraform-plugin-framework/path"
|
||||
"github.com/hashicorp/terraform-plugin-framework/resource"
|
||||
"github.com/hashicorp/terraform-plugin-framework/resource/schema"
|
||||
"github.com/hashicorp/terraform-plugin-framework/resource/schema/planmodifier"
|
||||
"github.com/hashicorp/terraform-plugin-framework/resource/schema/stringplanmodifier"
|
||||
"github.com/hashicorp/terraform-plugin-framework/types"
|
||||
"github.com/tfmm/terraform-provider-immich/internal/client"
|
||||
)
|
||||
|
||||
// Ensure the implementation satisfies the expected interfaces.
|
||||
var _ resource.Resource = &faceResource{}
|
||||
var _ resource.ResourceWithImportState = &faceResource{}
|
||||
|
||||
func NewFaceResource() resource.Resource {
|
||||
return &faceResource{}
|
||||
}
|
||||
|
||||
// faceResource defines the resource implementation.
|
||||
type faceResource struct {
|
||||
client *client.Client
|
||||
}
|
||||
|
||||
// faceResourceModel describes the resource data model.
|
||||
type faceResourceModel struct {
|
||||
ID types.String `tfsdk:"id"`
|
||||
AssetId types.String `tfsdk:"asset_id"`
|
||||
PersonId types.String `tfsdk:"person_id"`
|
||||
BoundingBoxX1 types.Float64 `tfsdk:"bounding_box_x1"`
|
||||
BoundingBoxY1 types.Float64 `tfsdk:"bounding_box_y1"`
|
||||
BoundingBoxX2 types.Float64 `tfsdk:"bounding_box_x2"`
|
||||
BoundingBoxY2 types.Float64 `tfsdk:"bounding_box_y2"`
|
||||
ImageHeight types.Int64 `tfsdk:"image_height"`
|
||||
ImageWidth types.Int64 `tfsdk:"image_width"`
|
||||
}
|
||||
|
||||
func (r *faceResource) Metadata(ctx context.Context, req resource.MetadataRequest, resp *resource.MetadataResponse) {
|
||||
resp.TypeName = req.ProviderTypeName + "_face"
|
||||
}
|
||||
|
||||
func (r *faceResource) Schema(ctx context.Context, req resource.SchemaRequest, resp *resource.SchemaResponse) {
|
||||
resp.Schema = schema.Schema{
|
||||
MarkdownDescription: "Manages an Immich face (detected or manual).",
|
||||
|
||||
Attributes: map[string]schema.Attribute{
|
||||
"id": schema.StringAttribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "Unique identifier for the face.",
|
||||
PlanModifiers: []planmodifier.String{
|
||||
stringplanmodifier.UseStateForUnknown(),
|
||||
},
|
||||
},
|
||||
"asset_id": schema.StringAttribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The ID of the asset this face belongs to.",
|
||||
PlanModifiers: []planmodifier.String{
|
||||
stringplanmodifier.RequiresReplace(),
|
||||
},
|
||||
},
|
||||
"person_id": schema.StringAttribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The ID of the person associated with this face.",
|
||||
},
|
||||
"bounding_box_x1": schema.Float64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The left coordinate of the bounding box.",
|
||||
PlanModifiers: []planmodifier.Float64{
|
||||
// RequiresReplace because we can't update bounding box via PUT /faces
|
||||
},
|
||||
},
|
||||
"bounding_box_y1": schema.Float64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The top coordinate of the bounding box.",
|
||||
},
|
||||
"bounding_box_x2": schema.Float64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The right coordinate of the bounding box.",
|
||||
},
|
||||
"bounding_box_y2": schema.Float64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The bottom coordinate of the bounding box.",
|
||||
},
|
||||
"image_height": schema.Int64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The height of the image in pixels.",
|
||||
},
|
||||
"image_width": schema.Int64Attribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The width of the image in pixels.",
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (r *faceResource) Configure(ctx context.Context, req resource.ConfigureRequest, resp *resource.ConfigureResponse) {
|
||||
if req.ProviderData == nil {
|
||||
return
|
||||
}
|
||||
|
||||
client, ok := req.ProviderData.(*client.Client)
|
||||
|
||||
if !ok {
|
||||
resp.Diagnostics.AddError(
|
||||
"Unexpected Resource Configure Type",
|
||||
fmt.Sprintf("Expected *client.Client, got: %T. Please report this issue to the provider developers.", req.ProviderData),
|
||||
)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
r.client = client
|
||||
}
|
||||
|
||||
func (r *faceResource) Create(ctx context.Context, req resource.CreateRequest, resp *resource.CreateResponse) {
|
||||
var data faceResourceModel
|
||||
|
||||
resp.Diagnostics.Append(req.Plan.Get(ctx, &data)...)
|
||||
|
||||
if resp.Diagnostics.HasError() {
|
||||
return
|
||||
}
|
||||
|
||||
createReq := client.CreateFaceRequest{
|
||||
AssetId: data.AssetId.ValueString(),
|
||||
PersonId: data.PersonId.ValueString(),
|
||||
BoundingBoxX1: data.BoundingBoxX1.ValueFloat64(),
|
||||
BoundingBoxY1: data.BoundingBoxY1.ValueFloat64(),
|
||||
BoundingBoxX2: data.BoundingBoxX2.ValueFloat64(),
|
||||
BoundingBoxY2: data.BoundingBoxY2.ValueFloat64(),
|
||||
ImageHeight: int(data.ImageHeight.ValueInt64()),
|
||||
ImageWidth: int(data.ImageWidth.ValueInt64()),
|
||||
}
|
||||
|
||||
face, err := r.client.CreateFace(createReq)
|
||||
if err != nil {
|
||||
resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to create face, got error: %s", err))
|
||||
return
|
||||
}
|
||||
|
||||
data.ID = types.StringValue(face.ID)
|
||||
|
||||
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
||||
}
|
||||
|
||||
func (r *faceResource) Read(ctx context.Context, req resource.ReadRequest, resp *resource.ReadResponse) {
|
||||
var data faceResourceModel
|
||||
|
||||
resp.Diagnostics.Append(req.State.Get(ctx, &data)...)
|
||||
|
||||
if resp.Diagnostics.HasError() {
|
||||
return
|
||||
}
|
||||
|
||||
// We have to find the face in the list for the asset
|
||||
faces, err := r.client.GetFaces(data.AssetId.ValueString())
|
||||
if err != nil {
|
||||
resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to read faces for asset, got error: %s", err))
|
||||
return
|
||||
}
|
||||
|
||||
var found *client.Face
|
||||
for _, f := range faces {
|
||||
if f.ID == data.ID.ValueString() {
|
||||
found = &f
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if found == nil {
|
||||
resp.State.RemoveResource(ctx)
|
||||
return
|
||||
}
|
||||
|
||||
data.PersonId = types.StringValue(found.PersonId)
|
||||
data.BoundingBoxX1 = types.Float64Value(found.BoundingBoxX1)
|
||||
data.BoundingBoxY1 = types.Float64Value(found.BoundingBoxY1)
|
||||
data.BoundingBoxX2 = types.Float64Value(found.BoundingBoxX2)
|
||||
data.BoundingBoxY2 = types.Float64Value(found.BoundingBoxY2)
|
||||
data.ImageHeight = types.Int64Value(int64(found.ImageHeight))
|
||||
data.ImageWidth = types.Int64Value(int64(found.ImageWidth))
|
||||
|
||||
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
||||
}
|
||||
|
||||
func (r *faceResource) Update(ctx context.Context, req resource.UpdateRequest, resp *resource.UpdateResponse) {
|
||||
var data faceResourceModel
|
||||
|
||||
resp.Diagnostics.Append(req.Plan.Get(ctx, &data)...)
|
||||
|
||||
if resp.Diagnostics.HasError() {
|
||||
return
|
||||
}
|
||||
|
||||
updateReq := client.UpdateFaceRequest{
|
||||
PersonId: data.PersonId.ValueString(),
|
||||
}
|
||||
|
||||
_, err := r.client.UpdateFace(data.ID.ValueString(), updateReq)
|
||||
if err != nil {
|
||||
resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to update face, got error: %s", err))
|
||||
return
|
||||
}
|
||||
|
||||
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
||||
}
|
||||
|
||||
func (r *faceResource) Delete(ctx context.Context, req resource.DeleteRequest, resp *resource.DeleteResponse) {
|
||||
var data faceResourceModel
|
||||
|
||||
resp.Diagnostics.Append(req.State.Get(ctx, &data)...)
|
||||
|
||||
if resp.Diagnostics.HasError() {
|
||||
return
|
||||
}
|
||||
|
||||
err := r.client.DeleteFace(data.ID.ValueString())
|
||||
if err != nil {
|
||||
resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to delete face, got error: %s", err))
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func (r *faceResource) ImportState(ctx context.Context, req resource.ImportStateRequest, resp *resource.ImportStateResponse) {
|
||||
// Expected format: face_id
|
||||
resource.ImportStatePassthroughID(ctx, path.Root("id"), req, resp)
|
||||
}
|
||||
@@ -0,0 +1,149 @@
|
||||
package provider
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/hashicorp/terraform-plugin-framework/datasource"
|
||||
"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
|
||||
"github.com/hashicorp/terraform-plugin-framework/types"
|
||||
"github.com/tfmm/terraform-provider-immich/internal/client"
|
||||
)
|
||||
|
||||
// Ensure the implementation satisfies the expected interfaces.
|
||||
var _ datasource.DataSource = &facesDataSource{}
|
||||
|
||||
func NewFacesDataSource() datasource.DataSource {
|
||||
return &facesDataSource{}
|
||||
}
|
||||
|
||||
// facesDataSource defines the data source implementation.
|
||||
type facesDataSource struct {
|
||||
client *client.Client
|
||||
}
|
||||
|
||||
// facesDataSourceModel describes the data source data model.
|
||||
type facesDataSourceModel struct {
|
||||
AssetId types.String `tfsdk:"asset_id"`
|
||||
Faces []facesModel `tfsdk:"faces"`
|
||||
}
|
||||
|
||||
type facesModel struct {
|
||||
ID types.String `tfsdk:"id"`
|
||||
PersonId types.String `tfsdk:"person_id"`
|
||||
BoundingBoxX1 types.Float64 `tfsdk:"bounding_box_x1"`
|
||||
BoundingBoxY1 types.Float64 `tfsdk:"bounding_box_y1"`
|
||||
BoundingBoxX2 types.Float64 `tfsdk:"bounding_box_x2"`
|
||||
BoundingBoxY2 types.Float64 `tfsdk:"bounding_box_y2"`
|
||||
ImageHeight types.Int64 `tfsdk:"image_height"`
|
||||
ImageWidth types.Int64 `tfsdk:"image_width"`
|
||||
}
|
||||
|
||||
func (d *facesDataSource) Metadata(ctx context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) {
|
||||
resp.TypeName = req.ProviderTypeName + "_faces"
|
||||
}
|
||||
|
||||
func (d *facesDataSource) Schema(ctx context.Context, req datasource.SchemaRequest, resp *datasource.SchemaResponse) {
|
||||
resp.Schema = schema.Schema{
|
||||
MarkdownDescription: "Retrieves a list of faces for a specific asset.",
|
||||
|
||||
Attributes: map[string]schema.Attribute{
|
||||
"asset_id": schema.StringAttribute{
|
||||
Required: true,
|
||||
MarkdownDescription: "The ID of the asset.",
|
||||
},
|
||||
"faces": schema.ListNestedAttribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "List of faces.",
|
||||
NestedObject: schema.NestedAttributeObject{
|
||||
Attributes: map[string]schema.Attribute{
|
||||
"id": schema.StringAttribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "Unique identifier for the face.",
|
||||
},
|
||||
"person_id": schema.StringAttribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The ID of the person associated with this face.",
|
||||
},
|
||||
"bounding_box_x1": schema.Float64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The left coordinate of the bounding box.",
|
||||
},
|
||||
"bounding_box_y1": schema.Float64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The top coordinate of the bounding box.",
|
||||
},
|
||||
"bounding_box_x2": schema.Float64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The right coordinate of the bounding box.",
|
||||
},
|
||||
"bounding_box_y2": schema.Float64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The bottom coordinate of the bounding box.",
|
||||
},
|
||||
"image_height": schema.Int64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The height of the image in pixels.",
|
||||
},
|
||||
"image_width": schema.Int64Attribute{
|
||||
Computed: true,
|
||||
MarkdownDescription: "The width of the image in pixels.",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (d *facesDataSource) Configure(ctx context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
|
||||
if req.ProviderData == nil {
|
||||
return
|
||||
}
|
||||
|
||||
client, ok := req.ProviderData.(*client.Client)
|
||||
|
||||
if !ok {
|
||||
resp.Diagnostics.AddError(
|
||||
"Unexpected Data Source Configure Type",
|
||||
fmt.Sprintf("Expected *client.Client, got: %T. Please report this issue to the provider developers.", req.ProviderData),
|
||||
)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
d.client = client
|
||||
}
|
||||
|
||||
func (d *facesDataSource) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
|
||||
var data facesDataSourceModel
|
||||
|
||||
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
|
||||
|
||||
if resp.Diagnostics.HasError() {
|
||||
return
|
||||
}
|
||||
|
||||
faces, err := d.client.GetFaces(data.AssetId.ValueString())
|
||||
if err != nil {
|
||||
resp.Diagnostics.AddError("Client Error", fmt.Sprintf("Unable to read faces, got error: %s", err))
|
||||
return
|
||||
}
|
||||
|
||||
data.Faces = []facesModel{}
|
||||
for _, f := range faces {
|
||||
fState := facesModel{
|
||||
ID: types.StringValue(f.ID),
|
||||
PersonId: types.StringValue(f.PersonId),
|
||||
BoundingBoxX1: types.Float64Value(f.BoundingBoxX1),
|
||||
BoundingBoxY1: types.Float64Value(f.BoundingBoxY1),
|
||||
BoundingBoxX2: types.Float64Value(f.BoundingBoxX2),
|
||||
BoundingBoxY2: types.Float64Value(f.BoundingBoxY2),
|
||||
ImageHeight: types.Int64Value(int64(f.ImageHeight)),
|
||||
ImageWidth: types.Int64Value(int64(f.ImageWidth)),
|
||||
}
|
||||
data.Faces = append(data.Faces, fState)
|
||||
}
|
||||
|
||||
resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
|
||||
}
|
||||
@@ -100,6 +100,7 @@ func (p *immichProvider) Resources(ctx context.Context) []func() resource.Resour
|
||||
NewTagResource,
|
||||
NewWorkflowResource,
|
||||
NewAdminNotificationResource,
|
||||
NewFaceResource,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -111,6 +112,7 @@ func (p *immichProvider) DataSources(ctx context.Context) []func() datasource.Da
|
||||
NewActivitiesDataSource,
|
||||
NewServerDataSource,
|
||||
NewNotificationsDataSource,
|
||||
NewFacesDataSource,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user