from pathlib import Path
from datetime import datetime

from xml_parser import sanitize_name

get_all_code = """
function Get-AllCodeServerPages!NAME! {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory=$true)]
        [string]$Resource,

        [Parameter(Mandatory=$true)]
        [string]$RootProperty,

        [Parameter(Mandatory=$false)]
        [string]$CacheKeyProperty = "conceptCodeId",

        [Parameter(Mandatory=$false)]
        [ScriptBlock]$Converter = ${function:ConvertFrom-CodeServerObject!NAME!},


        [Parameter(Mandatory=$false)]
        [int]$PageSize = 500
    )
    
    begin {
        if ($null -eq $Global:cache) {
            $Global:cache = @{}
        }
        $cacheKey = $Resource.GetHashCode()
        $cache[$cacheKey] = @{}
    }
    
    process {
        $currentPage = 0;      
        $Global:debugOut = @()
        while ($null -eq $data.totalPages -or $currentPage -lt $data.totalPages) {
            $uri = "{0}?pageSize={1}&page={2}" -f $Resource, $PageSize, ++$currentPage
            $data = Invoke-RestMethod -Uri $uri

            write-debug $uri
            if ($null -eq $data.totalPages) {
                throw "Did not get totalPages, abort!"
            } else {
                $Global:debugOut += $data.conceptCodes
                if ($null -ne $Converter) {
                    foreach ($codeObject in $data.$RootProperty) {
                        $convertedObject = Invoke-Command -ScriptBlock $Converter -ArgumentList (,$codeObject)
                        $Global:cache[$cacheKey].Add($convertedObject.$CacheKeyProperty, $convertedObject)
                    }
                } else {
                    foreach ($codeObject in $data.$RootProperty) {
                        $key = ($objAttributes | Where-Object {$_.attributeName -eq $CacheKeyProperty}).attributeValue
                        $Global:cache[$cacheKey].Add($key, $codeObject.attributes)
                    }
                }

            }
        }

    write-output $cache[$cacheKey].Values

    }
    
    end {
        
    }
}
"""

get_code = """
function Get-CodeServerObject!NAME! {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory=$true)]
        [string]$conceptCodeId,

        [Parameter(Mandatory=$true)]
        [string]$Resource,

        [Parameter(Mandatory=$true)]
        [string]$RootProperty,

        [Parameter(Mandatory=$false)]
        [string]$CacheKeyProperty = "conceptCodeId"
    )
    
    begin {

        if ($null -eq $Global:cache) {
            $params = @{
                Resource = $Resource;
                RootProperty = $RootProperty;
            }

            if ($PSBoundParameters.ContainsKey("CacheKeyProperty")) {
                $params.Add("CacheKeyProperty", $CacheKeyProperty)
            }

            [void](Get-AllCodeServerPagesRoles @params)
        }

        $cacheKey = $Resource.GetHashCode()
    }
    
    process {
       Write-Debug $cacheKey
       Write-Output $cache[$cacheKey][$conceptCodeId]
    }
    
    end {
        
    }
}
"""

convert_from = """
Function ConvertFrom-CodeServerObject!NAME!([PSCustomObject[]]$codeServerObject!NAME!) {
    $map = @{
!MAPS!	
    }
    
    if ($null -eq $codeServerObject!NAME!) {
        return
    }

    $obj = New-Object CodeServer!NAME! -ArgumentList $codeServerObject!NAME!
    $Global:meDebug = $codeServerObject!NAME!

    write-output $obj
}
"""

class_definition = """
class CodeServer!NAME! {
    CodeServer!NAME!() {}

    CodeServer!NAME!([psobject]$source) {
        $dic = @{}
        $source.attributes | foreach-object {$dic.Add($_.attributeName, $_.attributevalue)}

        $this.{conceptCodeId} = $source.conceptCodeId
!VARIABLES!

    }

!STRINGS!

}

Export-ModuleMember ConvertFrom-CodeServerObject!NAME!
Export-AllCodeServerPages!NAME!
Export-CodeServerObject!NAME!
"""

def define_maps(items: list[str]) -> str:
    return "\n".join([f"\t'{sanitize_name(item)}' = '{item}'" for item in items])

def define_variables(items: list[str]) -> str:
    return "\n".join([f"\t$this.{{{sanitize_name(item)}}} = $dic['{item}']" for item in items])

def define_strings(items: list[str]) -> str:
    return "\n".join(["\t" + "[string]$" + "{" + sanitize_name(item) + "}" for item in items])

def write_psm(name, items: list[str], filename: Path = Path("Connector.psm1")) -> str:
    from rich import print

    if filename.exists():
        filename.rename(str(filename) + f".{datetime.now().strftime('%Y-%m-%d.%H%M%S')}.bak")
    output = get_all_code.replace("!NAME!", name)
    output += get_code.replace("!NAME!", name)
    output += convert_from.replace("!NAME!", name).replace("!MAPS!", define_maps(items))
    output += class_definition.replace("!NAME!", name).replace("!VARIABLES!", define_variables(items)).replace("!STRINGS!", define_strings(items))
    with open(filename, "w") as f:
        f.write(output)
    print(output)
    pass