merge-toc.ps1

param(
[String] $refDocPath = "docs-ref-autogen",
[String] $conceptDocPath = "docs-ref-conceptual",
[String] $titleMappingFileName = "titleMapping.json"
)


function Sort-RefLines
{
  param($refLines)
  $parents = new-object system.collections.stack
  $refToc = @()
    $TextInfo = (Get-Culture).TextInfo

  $parentItem = $Null
  $previousItem = $Null

  foreach($line in $refLines)
  {
    $tocLine = New-Object System.Object
    $curLevel = 0
    $index = 0;

    while($index -lt $line.Length -and $line[$index] -eq '#')
    {
      ++$index
      ++$curLevel
    }

    $originalTitle = Find-TocTitle $line
    $originalHref = Find-TocHref $line
    $unalteredTitle = $originalTitle



    Write-Host $line -ForegroundColor Red
    if ($previousItem -ne $Null)
    {
      if ($previousItem.Level -eq $curLevel)
      {
        #no change to parent item
        #$parentItem = $parentItem
      }
      elseif ($previousItem.Level -lt $curLevel) {
      #  if ## A
      #  Then ### B
        $parentItem = $previousItem;
        $parents.Push($parentItem);
      }
      elseif ($previousItem.Level -gt $curLevel) {
        
        # # Ref
        # ## A
        # ### B
        # ## C

        # # Ref
        # ## A
        # ## B
        # ### C
        # ### D
        # #### E
        # ### F
        # ## G


        While($parentItem.Level -ge $curLevel -and $parents.Count -gt 0)
        {
            Write-Host "Before Pop: Current Parent " + $parentItem.TocTitle
            $parentItem = $parents.Pop()
        }

        if ($parentItem -ne $null)
        {
            $parents.Push($parentItem)
        }

      }

    }

    # Update Title With New Value
    if($Script:titleMap.ContainsKey($unalteredTitle))
    {
        $mapItem = $Script:titleMap.Get_Item($unalteredTitle)
        $tocTitle = $mapItem.TocTitle

        $originalTitle = $tocTitle

        $line = $line.Replace($unalteredTitle, $tocTitle)
        $line = $line.TrimEnd(')') + ' "' + $unalteredTitle + '")' 
    }
    else {
        $parentText = "az"

        if ($parentItem -ne $Null -and $parentItem.OriginalTitle -ne 'Reference')
        {
            $parentText = $parentItem.OriginalTitle
        }

        #nothing in the mapping, should do *default* replace
        if ($parentText -ne $Null -and $unalteredTitle -match $parentText)
        {
            $newTitle = $unalteredTitle -replace $parentText,''
            $newTitle = $newTitle.Trim()

            $newTitle = $TextInfo.ToTitleCase($newTitle)

            $originalTitle = $newTitle

            $line = $line.Replace($unalteredTitle, $newTitle)
            $line = $line.TrimEnd(')') + ' "' + $unalteredTitle + '")' 
        }
    }

    # if href ends with an anchor, it should be sorted before groups/sub-groups (so add zzz to those titles for sorting)
    if ($originalHref -Match '#')
    {
    }
    else {
        $originalTitle = "zzz" + $originalTitle
    }


    # include parent text, so that sorting doesn't break the nesting
    if ($parentItem -ne $Null) {
      $sortKey = $parentItem.SortKey + "." + $originalTitle
    }
    else {
       #No Parent
      $sortKey = $originalTitle
    }


    $tocLine | Add-Member -type NoteProperty -name SortKey -value $sortKey


    $tocLine | Add-Member -type NoteProperty -name Level -value $curLevel
    $tocLine | Add-Member -type NoteProperty -name TocTitle -value $originalTitle
    $tocLine | Add-Member -type NoteProperty -name OriginalTitle -value $unalteredTitle

    $tocLine | Add-Member -type NoteProperty -name OriginalLine -value $line

    $refToc += $tocLine

    $previousItem = $tocLine
  }



  $refToc = $refToc | Sort-Object -Property SortKey

  $returnLines = New-Object System.Collections.Generic.List[System.String]

  foreach ($item in $refToc) {
    $returnLines.Add($item.OriginalLine)
    Write-Host $item.OriginalLine -ForegroundColor Yellow
  }

  $refLines.Clear()
  $refLines.AddRange($returnLines)

  Write-Host "Finishing sort of TOC"
}


function Insert-RefTOC
{
  param([System.Collections.Generic.List[System.String]] $finalLines,
        [String] $refTocFile,
        [String] $topRefGroupName,
        [String] $prefix)

  $refLines = Get-Content $refTocFile

  $refLineColl = New-Object System.Collections.Generic.List[System.String]

  foreach($line in $refLines)
  {
    $refLineColl.Add($line)
  }

  Sort-RefLines $refLineColl

  $refLines = $refLineColl.ToArray()

  $firstLine = $true

  foreach($line in $refLines)
  {
    if($firstLine)
    {
      $firstLine = $false
      $originalTitle = Find-TocTitle $line
      $line = $line.Replace($originalTitle, $topRefGroupName)
    }
    $finalLines.Add($prefix + $line)
  }
}

function Find-TocTitle
{
  param([String] $line)

  if([String]::IsNullOrWhiteSpace($line))
  {
    return $null
  }
  $leftPos = $line.IndexOf('[');
  $rightPos = $line.IndexOf(']');
  if($leftPos -ge 0 -and $rightPos -gt $leftPos)
  {
    return $line.Substring($leftPos + 1, $rightPos - $leftPos -1)
  }
  return $null
}

function Find-TocHref
{
  param([String] $line)

  if([String]::IsNullOrWhiteSpace($line))
  {
    return $null
  }
  $leftPos = $line.IndexOf('(');
  $rightPos = $line.IndexOf(')');
  if($leftPos -ge 0 -and $rightPos -gt $leftPos)
  {
    return $line.Substring($leftPos + 1, $rightPos - $leftPos -1)
  }
  return $null
}


function Initialize-TitleMap
{
  param([String] $titleMappingFileName)

  if([String]::IsNullOrEmpty($titleMappingFileName))
  { return }

  $titleMappingSrc = Get-Content -Raw -Path $titleMappingFileName | ConvertFrom-Json
  foreach ($item in $titleMappingSrc.PSObject.Members) {

    $itemKey = $item.Name
    $TocTitle = $item.PSObject.Properties.Value.TocTitle
    $PageTitle = $item.PSObject.Properties.Value.PageTitle
    $newObj = New-Object System.Object
    $newObj | Add-Member -type NoteProperty -name "TocTitle" -value $TocTitle
    $newObj | Add-Member -type NoteProperty -name "PageTitle" -value $PageTitle

    $Script:titleMap.Add($itemKey,$newObj)
  }
}

function Replace-ContentTitle
{
  foreach($name in $Script:titleMap.Keys)
  {
    Write-Host "titleMap Key: " $name

    #find pycliyml file
    if($name -eq 'az')
    {
      $ymlFilePath = "index"
    }
    else
    {
      $ymlFilePath = $name.SubString(3).Split(' ') -join '/'
    }
    $ymlFilePath = "$refDocPath/$ymlFilePath.pycliyml"
    if(Test-Path $ymlFilePath)
    {
      Write-Host "Replacing title for $ymlFilePath"
      $lines = Get-Content $ymlFilePath
      $finalLines = New-Object System.Collections.Generic.List[System.String]
      $originalNameLine = "name: " + $name
      foreach($line in $lines)
      {
        if($line.StartsWith("name: "))
        {
          #Write-Host "Replacing This Line: " $line
          $pageTitle = $Script:titleMap.Get_Item($name).PageTitle
          $line = "name: '$pageTitle - $name'"
          #Write-Host "With this: " $line
        }
        $finalLines.Add($line)
      }
      Set-Content $ymlFilePath $finalLines
    }
  }
}

Write-Host "Start merging TOC in folder: $refDocPath and $conceptDocPath"

$conceptTocFile = [System.IO.Path]::Combine($conceptDocPath, "TOC.md")
if(-not (Test-Path $conceptTocFile))
{
  Write-Host "Conceptual toc file $conceptTocFile doesn't exist" 
  exit(1) 
}

$conceptLines = Get-Content $conceptTocFile
$refTocFile = [System.IO.Path]::Combine($refDocPath, "refTOC.md")

$finalTocLines = New-Object System.Collections.Generic.List[System.String]
$includeRefDoc = $false
$level = 0

foreach($line in $conceptLines)
{
  if([System.String]::IsNullOrWhiteSpace($line))
  { break }

  $curLevel = 0
  $index = 0;
  while($index -lt $line.Length -and $line[$index] -eq '#')
  {
    ++$index
    ++$curLevel
  }
  if($curLevel -eq 0)
  {
    Write-Host "Unexpected toc content: $line"
    exit(1)
  }

  if($level -eq 0 -and $curLevel -ne 1)
  {
    Write-Host "First toc line must be start with only one #: $line"
    exit(1)
  }
  
  if($curLevel -gt $level + 1)
  {
    Write-Host "Invalid toc line: $line"
    exit(1)
  }

  $titleMap = @{}
  Initialize-TitleMap $titleMappingFileName


  $level = $curLevel
  if($line.IndexOf("refTOC.md", [System.StringComparison]::OrdinalIgnoreCase) -ge 0)
  {
    $includeRefDoc = $true
    $prefixIndent = "#"*($level-1)
    $topRefGroupName = Find-TocTitle $line
    Insert-RefTOC $finalTocLines $refTocFile $topRefGroupName $prefixIndent
  }
  else
  {
    #fix the link to point correct conceptual doc, leave absolute link alone
    $line = $line -replace '\][ ]*\([ ]*([a-zA-Z])','](_CONCEPTDOCPATH_$1'
    $line = $line.Replace("_CONCEPTDOCPATH_", "../$conceptDocPath/")
    $finalTocLines.Add($line)
  }
}

Replace-ContentTitle

$tocFile = [System.IO.Path]::Combine($refDocPath, "TOC.md")
Set-Content $tocFile $finalTocLines

Write-Host "Complete merging TOC"