CopyrightCheck.ps1

param([string]$SrcRoot,
      [string]$TestRoot)

$rootDir = Split-Path $MyInvocation.MyCommand.Path
$rootDir = [IO.Path]::GetFullPath($rootDir)

if(!$SrcRoot) {
    $SrcRoot = Join-Path $rootDir "src"
}

if(!$TestRoot) {
    $TestRoot = Join-Path $rootDir "tests"
}

$Header = "// ---------------------------------------------------------------------------- 
// Copyright (c) Microsoft Corporation. All rights reserved.
// ----------------------------------------------------------------------------"

function NeedsCopyright([string]$FileName) {
    # Skip designer files bin and org files
    if($FileName.Contains("Designer") -or $FileName.Contains("bin\") -or $FileName.Contains("obj\")) {
        return $false;
    }

    # Check the first line
    $line = Get-Content $FileName -totalCount 1;
    
    # Does it have the header?
    return !$line -or !$line.StartsWith("// ----------------------------------------------------------------------------") 
}

function Get-FilesWithoutCopyright([string]$Directory) {
    return Get-ChildItem $Directory -Recurse -Filter *.cs | Where-Object { NeedsCopyright $_.FullName } | Select-Object FullName
}

function WriteWarnings([string]$Directory, $Files) {
    Write-Warning "There are $($Files.Length) files in '$Directory' without a copyright header:"
    if($Files.Length) {
        $Files | ForEach-Object { Write-Warning $_.FullName }
    }
}

function FixCopyright([string]$FileName) {
    $lines = New-Object System.Collections.ArrayList(,(Get-Content $FileName))
    $lines.Insert(0, "")
    $lines.Insert(0, $Header)

    $lines | Set-Content $FileName -Encoding UTF8
}

$srcFilesWithoutCopyright = Get-FilesWithoutCopyright $SrcRoot

WriteWarnings $SrcRoot $srcFilesWithoutCopyright
$srcFilesWithoutCopyright | ForEach-Object { FixCopyright $_.FullName }