New-AEScript.psm1

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
function New-AEScript {
    param (
        [string]$scriptName = $(
            $name = $(read-host "ScriptName pls ");
            if (-not $name) { $(throw "no name supplied") }
            $name),
        [string]$developmentDir,
        [switch]$UIScript,
        [string]$template = $(if ($UIScript) { "UI-Script" } else { "selectedLayersTemplate" }),
        [switch]$overWriteOK,
        [switch]$createHardlinkInAEScriptsFolder
    )
    
    $prefs = getPrefs; 
    if ("" -ne $developmentDir) {
        $prefs.developmentDir = $developmentDir
    } 
    if (! (test-path $prefs.developmentDir -ErrorAction SilentlyContinue)) {       
        $prefs.developmentDir = getDDir;
    }
        
    $hardLinkTargetFolder = Join-Path $env:APPDATA "Adobe" "After Effects"
    $aeversions = Get-ChildItem $hardLinkTargetFolder -dir | Where-Object { $_.name -match (".*[0-9]+") }
    
    $latestVersionNum = 0;
    foreach ($n in $aeversions) {
        if ($n.name -match "([0-9]+.)([0-9]*)\.*([0-9]*)$") {
            $version = [float]($Matches[1] + $Matches[2])
            if ([float]$latestVersionNum -le $version) { 
                $latestVersionNum = $version
                $latestVersion = $n.name
            }
        }
    }
    $hardLinkTargetFolder = Join-Path $hardLinkTargetFolder $latestVersion "Scripts"
    
    # append "scriptUI panels" to the path
    if ($UIScript) { 
        $outputDir = Join-Path $prefs.developmentDir "ScriptUI Panels"
        $hardLinkTargetFolder = Join-Path $hardLinkTargetFolder "ScriptUI Panels"
    }
    else {
        $outputDir = $prefs.developmentDir
    }
    
    # give it the extension
    if (-not($scriptName -match "(.*).jsx$")) {
        $scriptName = $scriptName + ".jsx";
    }
    # compile the file and hardlink names
    $newScriptPath = Join-Path $outputDir $scriptName
    if ($createHardlinkInAEScriptsFolder) {
        $hardLinkPath = (Join-Path $hardLinkTargetFolder $scriptName)
    }
    
    #get the template
    $templatePath = Join-Path $PSScriptRoot  "$template.txt"; #templates are in the same dir as the PS script
    if (-not(Test-Path $templatePath)) {
        throw "Template does not exist" #Silly rabbit
    }
    
    $templateText = Get-Content $templatePath
    $scriptText = @()
    $templateText | ForEach-Object {
        $scriptText += $_.replace("#scriptname#", $scriptName.replace(".jsx", ""))
    }
    # check for pre-existing files
    $okToGo = $true
    if ((Test-Path $newScriptPath) -and (-not $overWriteOK)) {
        $okToGo = $(
            Write-Host "$scriptName already exists in $outputDir" -ForegroundColor DarkYellow;
            write-host "Overwite? " -NoNewline -ForegroundColor Yellow;
            read-host "(y/N)"
        ).ToLower() -match "^y$"
    }
    # do the things
    if ($okToGo) {
        try {
            Set-Content -Path $newScriptPath -Value $scriptText
            Write-Host "Created $scriptName in $outputDir" -ForegroundColor Green
            if ($createHardlinkInAEScriptsFolder) {
                cmd /c mklink /h "$newScriptPath"  "$hardLinkPath" 
                New-Hardlink $hardLinkPath $newScriptPath
                Write-host "Hardlinked script to AE Scripts folder $hardLinkTargetFolder" -ForegroundColor Green
            }
        }
        catch {
            Write-Host "An error occured" -ForegroundColor Red
            Write-Host $_ -ForegroundColor DarkRed
        }
    }
    else {
        Write-host "Script wasn't written." -ForegroundColor DarkYellow
    }
    savePrefs($prefs)
    
}
function savePrefs($prefs){
    mkdir (join-path $env:APPDATA "pureandapplied") -ErrorAction SilentlyContinue  2>&1 >$null
    Export-Clixml -Path  (join-path $env:APPDATA "pureandapplied" "newAeScript.dat") -InputObject $prefs
}

function getPrefs() {
    $prefs = @{}
    $prefFile = join-path $env:APPDATA "pureandapplied" "newAeScript.dat" 
    if (Test-Path $prefFile) {
        $prefs = import-clixml  $prefFile
    }
    return $prefs
}

function getDDir() {
    Add-Type -AssemblyName System.Windows.Forms
    $browser = New-Object System.Windows.Forms.FolderBrowserDialog
    $browser.description = "Select the Scripts folder for new AE scripts"
    $null = $browser.ShowDialog()
    return $browser.SelectedPath
}