refactor: update variable names and parent org condition

Author: rajat-tomar

data.tf

@@ -1,4 +1,12 @@
+data "aws_organizations_organization" "org" {}
+
 resource "aws_organizations_organizational_unit" "this" {
-  name      = var.name
-  parent_id = data.aws_organizations_organization.org.roots[0].id
+  name      = var.organizational_unit_name
+  parent_id = var.parent_org_id != "" ? var.parent_org_id : data.aws_organizations_organization.org.roots[0].id
+}
+
+resource "aws_organizations_policy_attachment" "policy_attachment" {
+  count     = length(var.attached_policies)
+  policy_id = var.attached_policies[count.index]
+  target_id = aws_organizations_organizational_unit.this.id
 }

main.tf

@@ -1,4 +1,4 @@
-output "id" {
-  description = "The ID of the organizational unit"
+output "organizational_unit_id" {
+  description = "The ID of the created organizational unit."
   value       = aws_organizations_organizational_unit.this.id
 }

outputs.tf

@@ -1,4 +1,16 @@
-variable "name" {
+variable "organizational_unit_name" {
   description = "The name of the organizational unit"
   type        = string
 }
+
+variable "parent_org_id" {
+  description = "The ID of the parent organizational unit."
+  type        = string
+  default     = ""
+}
+
+variable "attached_policies" {
+  description = "List of policy IDs to attach to the organizational unit."
+  type        = list(string)
+  default     = []
+}