terraform {
  required_providers {
    terraprobe = {
      source  = "DonsWayo/terraprobe"
      version = "~> 0.1"
    }
  }
}

provider "terraprobe" {
  default_timeout     = 10
  default_retries     = 3
  default_retry_delay = 5
}

# Test API endpoint
resource "terraprobe_http_test" "api" {
  name               = "API Health Check"
  url                = "https://api.example.com/health"
  expect_status_code = 200
  expect_contains    = "healthy"
}

# Test database port
resource "terraprobe_tcp_test" "database" {
  name = "PostgreSQL Port"
  host = "db.example.com"
  port = 5432
}

# Test DNS resolution
resource "terraprobe_dns_test" "website" {
  name        = "Website DNS"
  hostname    = "www.example.com"
  record_type = "A"
}

# Group tests together
resource "terraprobe_test_suite" "production" {
  name        = "Production Tests"
  description = "All production environment tests"

  http_tests = [terraprobe_http_test.api.id]
  tcp_tests  = [terraprobe_tcp_test.database.id]
  dns_tests  = [terraprobe_dns_test.website.id]
}

# Check results
output "test_results" {
  value = {
    passed = terraprobe_test_suite.production.all_passed
    summary = "${terraprobe_test_suite.production.passed_count}/${terraprobe_test_suite.production.total_count} tests passed"
  }
}

resource "terraprobe_http_test" "example" {
  name               = "API Test"
  url                = "https://api.example.com/v1/status"
  method             = "GET"                    # GET, POST, PUT, DELETE, etc.
  expect_status_code = 200
  expect_contains    = "operational"            # Optional: string to find in response
  timeout            = 30                       # Optional: override default timeout
  retries            = 5                        # Optional: override default retries

  headers = {
    "Authorization" = "Bearer ${var.api_token}"
    "Content-Type"  = "application/json"
  }

  body = jsonencode({                          # Optional: request body
    test = true
  })
}

resource "terraprobe_tcp_test" "example" {
  name    = "Redis Connection"
  host    = "cache.example.com"
  port    = 6379
  timeout = 10
  retries = 3
}

resource "terraprobe_dns_test" "example" {
  name          = "Mail Server DNS"
  hostname      = "mail.example.com"
  record_type   = "MX"                    # A, AAAA, CNAME, MX, TXT, NS
  expect_result = "10 mail.example.com."  # Optional: expected value
  resolver      = "1.1.1.1"               # Optional: custom DNS server
  timeout       = 5
}

resource "terraprobe_db_test" "postgres" {
  name     = "PostgreSQL Test"
  type     = "postgres"              # postgres or mysql
  host     = "db.example.com"
  port     = 5432
  database = "myapp"
  username = var.db_user
  password = var.db_password
  ssl_mode = "require"               # disable, require, verify-ca, verify-full

  query = "SELECT COUNT(*) FROM users WHERE active = true"

  # Connection pool settings
  max_open_conn = 25
  max_idle_conn = 5
  max_lifetime  = 300
}

resource "terraprobe_db_test" "mysql" {
  name     = "MySQL Test"
  type     = "mysql"
  host     = "mysql.example.com"
  port     = 3306
  database = "myapp"
  username = var.mysql_user
  password = var.mysql_password

  query = "SELECT VERSION()"
}

resource "terraprobe_test_suite" "production" {
  name        = "Production Environment"
  description = "All production tests"

  http_tests = [
    terraprobe_http_test.api.id,
    terraprobe_http_test.website.id,
  ]

  tcp_tests = [
    terraprobe_tcp_test.database.id,
    terraprobe_tcp_test.cache.id,
  ]

  dns_tests = [
    terraprobe_dns_test.api.id,
    terraprobe_dns_test.cdn.id,
  ]

  db_tests = [
    terraprobe_db_test.postgres.id,
    terraprobe_db_test.mysql.id,
  ]
}

provider "terraprobe" {
  default_timeout     = 10    # Default timeout in seconds for all tests
  default_retries     = 3     # Number of retry attempts
  default_retry_delay = 5     # Seconds between retries
  user_agent          = "TerraProbe/1.0"  # User agent for HTTP tests
}