Merge pull request #154 from matokovacik/main

Fix #153

Author: magodo

docs/data-sources/resource.md

@@ -28,6 +28,7 @@ data "restful_resource" "test" {
 ### Optional
 
 - `allow_not_exist` (Boolean) Whether to throw error if the data source being queried doesn't exist (i.e. status code is 404). Defaults to `false`.
+- `body` (Dynamic) The request body that is sent when using `POST` method.
 - `header` (Map of String) The header parameters that are applied to each request. This overrides the `header` set in the provider block.
 - `method` (String) The HTTP Method for the request. Allowed methods are a subset of methods defined in [RFC7231](https://datatracker.ietf.org/doc/html/rfc7231#section-4.3) namely, `GET`, `HEAD`, and `POST`. `POST` support is only intended for read-only URLs, such as submitting a search. Defaults to `GET`.
 - `output_attrs` (Set of String) A set of `output` attribute paths (in [gjson syntax](https://github.com/tidwall/gjson/blob/master/SYNTAX.md)) that will be exported in the `output`. If this is not specified, all attributes will be exported by `output`.

internal/client/client.go

@@ -340,11 +340,17 @@ type ReadOptionDS struct {
 	Header Header
 }
 
-func (c *Client) ReadDS(ctx context.Context, path string, opt ReadOptionDS) (*resty.Response, error) {
+func (c *Client) ReadDS(ctx context.Context, path string, body []byte, opt ReadOptionDS) (*resty.Response, error) {
 	req := c.R().SetContext(ctx)
 	req.SetQueryParamsFromValues(url.Values(opt.Query))
 	req.SetHeaders(opt.Header)
 
+	// Set the body for POST requests if provided
+	if len(body) != 0 && opt.Method == "POST" {
+		req = req.SetHeader("Content-Type", "application/json")
+		req = req.SetBody(body)
+	}
+
 	switch opt.Method {
 	case "", "GET":
 		return req.Get(path)

internal/provider/data_source.go

@@ -26,6 +26,7 @@ type dataSourceData struct {
 	Method        types.String  `tfsdk:"method"`
 	Query         types.Map     `tfsdk:"query"`
 	Header        types.Map     `tfsdk:"header"`
+	Body          types.Dynamic `tfsdk:"body"`
 	Selector      types.String  `tfsdk:"selector"`
 	OutputAttrs   types.Set     `tfsdk:"output_attrs"`
 	AllowNotExist types.Bool    `tfsdk:"allow_not_exist"`
@@ -67,6 +68,11 @@ func (d *DataSource) Schema(ctx context.Context, req datasource.SchemaRequest, r
 				ElementType:         types.StringType,
 				Optional:            true,
 			},
+			"body": schema.DynamicAttribute{
+				Description:         "The request body that is sent when using `POST` method.",
+				MarkdownDescription: "The request body that is sent when using `POST` method.",
+				Optional:            true,
+			},
 			"selector": schema.StringAttribute{
 				Description:         "A selector in gjson query syntax, that is used when `id` represents a collection of resources, to select exactly one member resource of from it",
 				MarkdownDescription: "A selector in [gjson query syntax](https://github.com/tidwall/gjson/blob/master/SYNTAX.md#queries), that is used when `id` represents a collection of resources, to select exactly one member resource of from it",
@@ -93,6 +99,25 @@ func (d *DataSource) Schema(ctx context.Context, req datasource.SchemaRequest, r
 	}
 }
 
+func (d *DataSource) ValidateConfig(ctx context.Context, req datasource.ValidateConfigRequest, resp *datasource.ValidateConfigResponse) {
+	var config dataSourceData
+	diags := req.Config.Get(ctx, &config)
+	resp.Diagnostics.Append(diags...)
+	if diags.HasError() {
+		return
+	}
+	if !config.Body.IsUnknown() && !config.Body.IsNull() {
+		if !config.Method.IsUnknown() {
+			if config.Method.ValueString() != "POST" {
+				resp.Diagnostics.AddError(
+					"Invalid configuration",
+					"`body` is only applicable when `method` is set to `POST`",
+				)
+			}
+		}
+	}
+}
+
 func (d *DataSource) Configure(ctx context.Context, req datasource.ConfigureRequest, resp *datasource.ConfigureResponse) {
 	if req.ProviderData == nil {
 		return
@@ -140,15 +165,31 @@ func (d *DataSource) Read(ctx context.Context, req datasource.ReadRequest, resp
 
 	state := dataSourceData{
 		ID:            config.ID,
+		Method:        config.Method,
 		Query:         config.Query,
 		Header:        config.Header,
+		Body:          config.Body,
 		Selector:      config.Selector,
 		OutputAttrs:   config.OutputAttrs,
 		AllowNotExist: config.AllowNotExist,
 		Precheck:      config.Precheck,
 	}
 
-	response, err := c.ReadDS(ctx, config.ID.ValueString(), *opt)
+	// Set body if provided
+	var body []byte
+	if !config.Body.IsNull() && config.Method.ValueString() == "POST" {
+		var err error
+		body, err = dynamic.ToJSON(config.Body)
+		if err != nil {
+			resp.Diagnostics.AddError(
+				"Error to convert body",
+				err.Error(),
+			)
+			return
+		}
+	}
+
+	response, err := c.ReadDS(ctx, config.ID.ValueString(), body, *opt)
 	if err != nil {
 		resp.Diagnostics.AddError(
 			"Error to call Read",