# ==========================================
# Azure DevOps Test Plan Automation Script
# 2025 Updated Version
# ==========================================

param(
    [string]$OrgUrl = "https://dev.azure.com/YOURORG",
    [string]$Project = "YOURPROJECT",
    [string]$SprintName = "Sprint 18"
)

# Use System.AccessToken inside Azure DevOps Pipeline
$Token = $env:SYSTEM_ACCESSTOKEN
if (-not $Token) {
    throw "SYSTEM_ACCESSTOKEN is not available. Enable 'Allow scripts to access OAuth token' in the pipeline."
}

$Headers = @{
    Authorization = "Bearer $Token"
    "Content-Type" = "application/json"
}

$IterationPath = "$Project\$SprintName"
$AreaPath = $Project

Write-Host "🔧 Starting automation for $SprintName..."

# ---------------------------------------------------------
# 1. Create Test Plan
# ---------------------------------------------------------
$PlanBody = @{
    name      = "$SprintName - Test Plan"
    area      = @{ path = $AreaPath }
    iteration = $IterationPath
} | ConvertTo-Json -Depth 10

$PlanUri = "$OrgUrl/$Project/_apis/test/plans?api-version=7.1-preview.1"
$Plan = Invoke-RestMethod -Uri $PlanUri -Method Post -Headers $Headers -Body $PlanBody

$PlanId = $Plan.id
Write-Host "📘 Created Test Plan: $PlanId"

# ---------------------------------------------------------
# 2. Query User Stories in Sprint
# ---------------------------------------------------------
$Wiql = @"
SELECT [System.Id], [System.Title]
FROM WorkItems
WHERE [System.WorkItemType] = 'User Story'
AND [System.State] <> 'Closed'
AND [System.IterationPath] = '$IterationPath'
"@

$QueryBody = @{ query = $Wiql } | ConvertTo-Json
$QueryUri = "$OrgUrl/$Project/_apis/wit/wiql?api-version=7.1-preview.2"

$UserStories = Invoke-RestMethod -Uri $QueryUri -Method Post -Headers $Headers -Body $QueryBody

if ($UserStories.workItems.Count -eq 0) {
    Write-Host "⚠️ No active User Stories found for $SprintName."
    exit 0
}

# ---------------------------------------------------------
# 3. Create Test Suites + Link Test Cases
# ---------------------------------------------------------
foreach ($Story in $UserStories.workItems) {

    $StoryId = $Story.id
    $StoryDetails = Invoke-RestMethod -Uri "$OrgUrl/$Project/_apis/wit/workitems/$StoryId?api-version=7.1-preview.3" -Headers $Headers
    $StoryTitle = $StoryDetails.fields.'System.Title'

    Write-Host "🧩 Processing User Story $StoryId - $StoryTitle"

    # Create Static Suite
    $SuiteBody = @{
        suiteType = "StaticTestSuite"
        name      = "US$StoryId - $StoryTitle"
    } | ConvertTo-Json -Depth 10

    $SuiteUri = "$OrgUrl/$Project/_apis/test/plans/$PlanId/suites?api-version=7.1-preview.3"
    $Suite = Invoke-RestMethod -Uri $SuiteUri -Method Post -Headers $Headers -Body $SuiteBody

    $SuiteId = $Suite.id
    Write-Host "📁 Created Suite $SuiteId for User Story $StoryId"

    # Get linked Test Cases
    $Relations = $StoryDetails.relations `
        | Where-Object { $_.rel -eq "System.LinkTypes.Hierarchy-Forward" -and $_.url -match "/workItems/" }

    $TestCaseIds = @()
    foreach ($Rel in $Relations) {
        if ($Rel.url -match "/workItems/(\d+)") {
            $TestCaseIds += $Matches[1]
        }
    }

    if ($TestCaseIds.Count -eq 0) {
        Write-Host "⚠️ No test cases linked to User Story $StoryId"
        continue
    }

    # Add Test Cases to Suite
    foreach ($TcId in $TestCaseIds) {
        $AddUri = "$OrgUrl/$Project/_apis/test/plans/$PlanId/suites/$SuiteId/testcases/$TcId?api-version=7.1-preview.3"
        Invoke-RestMethod -Uri $AddUri -Method Post -Headers $Headers
        Write-Host "🔗 Linked Test Case $TcId → Suite $SuiteId"
    }
}

Write-Host "🎉 Automation complete for $SprintName!"