From 9241b05241582ece3919502a317258cfd60277bb Mon Sep 17 00:00:00 2001
From: "PC-2NR0VQ3\\wai.wong" <wai.wong@enterprisedb.com>
Date: Mon, 25 Nov 2024 15:48:34 +0000
Subject: [PATCH] feat: data sources

---
 examples/data-sources/biganimal_cluster/data-source.tf        | 4 ++++
 .../data-sources/biganimal_faraway_replica/data-source.tf     | 4 ++++
 pkg/provider/data_source_cluster.go                           | 1 +
 pkg/provider/data_source_fareplica.go                         | 1 +
 pkg/provider/data_source_pgd.go                               | 1 +
 5 files changed, 11 insertions(+)

diff --git a/examples/data-sources/biganimal_cluster/data-source.tf b/examples/data-sources/biganimal_cluster/data-source.tf
index 0931e502..fc0ed105 100644
--- a/examples/data-sources/biganimal_cluster/data-source.tf
+++ b/examples/data-sources/biganimal_cluster/data-source.tf
@@ -89,6 +89,10 @@ output "storage" {
   value = data.biganimal_cluster.this.storage
 }
 
+output "wal_storage" {
+  value = data.biganimal_cluster.this.wal_storage
+}
+
 output "superuser_access" {
   value = coalesce(data.biganimal_cluster.this.superuser_access, false)
 }
diff --git a/examples/data-sources/biganimal_faraway_replica/data-source.tf b/examples/data-sources/biganimal_faraway_replica/data-source.tf
index 92d9b0af..ca60f853 100644
--- a/examples/data-sources/biganimal_faraway_replica/data-source.tf
+++ b/examples/data-sources/biganimal_faraway_replica/data-source.tf
@@ -89,6 +89,10 @@ output "storage" {
   value = data.biganimal_faraway_replica.this.storage
 }
 
+output "wal_storage" {
+  value = data.biganimal_faraway_replica.this.wal_storage
+}
+
 output "volume_snapshot_backup" {
   value = data.biganimal_faraway_replica.this.volume_snapshot_backup
 }
diff --git a/pkg/provider/data_source_cluster.go b/pkg/provider/data_source_cluster.go
index fc876772..5dd02003 100644
--- a/pkg/provider/data_source_cluster.go
+++ b/pkg/provider/data_source_cluster.go
@@ -388,6 +388,7 @@ func (c *clusterDataSource) Schema(ctx context.Context, req datasource.SchemaReq
 				MarkdownDescription: "Cluster connection service name.",
 				Computed:            true,
 			},
+			"wal_storage": resourceWal,
 		},
 	}
 }
diff --git a/pkg/provider/data_source_fareplica.go b/pkg/provider/data_source_fareplica.go
index d3ca9d21..8ff5b3df 100644
--- a/pkg/provider/data_source_fareplica.go
+++ b/pkg/provider/data_source_fareplica.go
@@ -271,6 +271,7 @@ func (c *FAReplicaData) Schema(ctx context.Context, req datasource.SchemaRequest
 					},
 				},
 			},
+			"wal_storage": resourceWal,
 		},
 	}
 }
diff --git a/pkg/provider/data_source_pgd.go b/pkg/provider/data_source_pgd.go
index fb6a5abe..c0f587a9 100644
--- a/pkg/provider/data_source_pgd.go
+++ b/pkg/provider/data_source_pgd.go
@@ -284,6 +284,7 @@ func (p pgdDataSource) Schema(ctx context.Context, req datasource.SchemaRequest,
 							Description: "Is read-only connections enabled.",
 							Computed:    true,
 						},
+						"wal_storage": resourceWal,
 					},
 				},
 			},