controller:
  name: 'Windows-Container-Test'
  host: 'https://vfunction.mycompany.com'
  org_id: '45faddbf-1111-1111-1111-5c3846b4e751'
  app_id: '2bef40e7-1111-1111-1111-b17975a62ca0'
  client_id: '6f809ad3-1111-1111-1111-a86c17f9f9a9'
  client_secret: 'b946d2d2-1111-1111-1111-463dc172b40a'
  type: dotnet
  instrconf_additions:
    inclusions:
#      - a.b.c.
    exclusions:
#      - a.b.c.
  tags:
#    - tag1

server_application:
#  name:
#  include_classes: com.
#  allowed_users:
#  new_user_default_password:

agent:
  # optional versions: framework, dotnet
  version: framework
  # optional environments: iis, service, command
  environment: service
#  application_name:
#  architecture: x86
#  application_command:
### Set the log level for the agent (trace, debug, info, warning, error, critical). The default is debug.
#  log_level: info
### Add process ID to log file name (true/false). Default is false ###
#  add_pid_to_log_file: false
  enable_rejit: true

param (
    [parameter(Mandatory=$true)] [string]$svc
)

Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope Process -Force

# print all parameters
Write-Output "Running with parameters:"
if ($null -ne $svc -and $svc -ne ""){
    Write-Output "instance: $svc"
}

#################################################################
###################### Helper Functions #########################
#################################################################

function Modify-DatadogLoaderConf {
	
	Write-Host "Configuring Datadog Loader.conf"

    $X86_CUSTOM = "CUSTOM;{CD7D4B53-96C8-4552-9C11-6E41DF8EAB8A};win-x86;C:\vfunction\agent\vfagent.net.x86.dll"
	$X64_CUSTOM = "CUSTOM;{CD7D4B53-96C8-4552-9C11-6E41DF8EAB8A};win-x64;C:\vfunction\agent\vfagent.net.dll"
	
	$X86_LOADER = "C:\Program Files\Datadog\.NET Tracer\win-x86\loader.conf"
	$X64_LOADER = "C:\Program Files\Datadog\.NET Tracer\win-x64\loader.conf"

	if (-not (Get-Content $X86_LOADER | Select-String -SimpleMatch $X86_CUSTOM -Quiet)) {
		#Write-Host "X86 Datadog Loader.conf before VFunction modification:"
		#Get-Content $X86_LOADER
		Add-Content -Path $X86_LOADER -Value ""
		Add-Content -Path $X86_LOADER -Value $X86_CUSTOM
		#Write-Host "X86 Datadog Loader.conf after VFunction modification:"
		#Get-Content $X86_LOADER
	}
	if (-not (Get-Content $X64_LOADER | Select-String -SimpleMatch $X64_CUSTOM -Quiet)) {
		#Write-Host "X64 Datadog Loader.conf before VFunction modification:"
		#Get-Content $X64_LOADER
		Add-Content -Path $X64_LOADER -Value ""
		Add-Content -Path $X64_LOADER -Value $X64_CUSTOM
		#Write-Host "X64 Datadog Loader.conf after VFunction modification:"
		#Get-Content $X64_LOADER
	}
}

function Set-WinSvcEnvVars {

	$regPath = "HKLM:\SYSTEM\CurrentControlSet\Services\${svc}"
    $newValues = @(
        "VF_AGENT_CONF_LOCATION=C:\vfunction\config\agent\instances\${svc}\conf.json",
        "COMPlus_TailCallOpt=0"
    )
	
	# Read existing values (default to empty array if none)
	$item = Get-ItemProperty -Path $regPath -ErrorAction SilentlyContinue

    if ($item -and $item.Environment) {
        $existing = $item.Environment
    }
    else {
        $existing = @()
    }
    
	# Merge, filtering out any duplicates
	$merged = ($existing + $newValues) | Select-Object -Unique

	New-ItemProperty `
    -Path $regPath `
    -Name Environment `
    -Value $merged `
    -PropertyType MultiString `
    -Force
}

#################################################################
###################### Datadog Configuration ####################
#################################################################

Modify-DatadogLoaderConf

#################################################################
###################### Win SVC Configuration ####################
#################################################################

Set-WinSvcEnvVars

# escape=`

FROM mcr.microsoft.com/dotnet/framework/runtime:4.8

# Main Content of Dockerfile related to App

# Location where Datadog is added to the Container
RUN powershell -Command `
    # Datadog APM
    Start-Process msiexec.exe -Wait -ArgumentList '/I C:\datadog-dotnet-apm.msi /quiet'; `
    net stop /y was;

# New details for vFunction (VF)
COPY ["vfunction/vfunction-controller-windows*.zip", "C:/vfunction.zip"]
COPY ["vfunction/vfunction-dockerfile-windows-service.ps1", "C:/vfunction.ps1"]
COPY ["vfunction/installation.yaml", "C:/vfunction.yaml"]

# Set Windows Service Name for VF to hook
ENV SVC="CHANGE_ME"

# Expand VF dir
RUN powershell -ExecutionPolicy Bypass Expand-Archive -Path "C:/vfunction.zip" -DestinationPath "C:/";

# Create App-specific VF instance
RUN powershell -ExecutionPolicy Bypass -Command "& C:\vfunction\controller-installation\create-instance.ps1  ` 
    -instance $env:SVC `  
    -agentMode true `  
    -type dotnet"

# Move App-specific configuration to created VF instance
RUN powershell -ExecutionPolicy Bypass Move-Item -Path "C:/vfunction.yaml" `  
    -Destination "C:/vfunction/config/installation/instances/$env:SVC/installation.yaml" -Force; `
    icacls "C:\vfunction" /grant Everyone:F /T

# Set all VF Configuration Files
RUN powershell -ExecutionPolicy Bypass -Command "& C:\vfunction\controller-installation\install.ps1 `
	-instance $env:SVC `
    -noServices true `
	-scriptMode true"

# Modify Datadog Loader.conf and add VF Env Vars to Windows Service's Registry
RUN powershell -ExecutionPolicy Bypass -Command "& C:\vfunction.ps1 -svc $env:SVC"

# End of Dockerfile that starts the Container