mirror of
https://github.com/ChrisTitusTech/winutil.git
synced 2026-10-08 11:39:10 +11:00
Consolidate background work into one job layer (#5002)
* Add a job layer for long running work - Start-WinUtilJob owns busy state, progress, taskbar, logging and errors - Write-WinUtilJobProgress reports from a job without UI checks in the body - Post UI updates instead of waiting on the dispatcher for each one - Move Invoke-WPFInstall onto it as the first workflow * Move the remaining app and feature workflows onto the job layer - Uninstall, AppX install, Features, OOSU and installed detection - Drop the per workflow busy flag, progress, taskbar and error handling - Rework their tests to check the job and its body instead of runspace internals * Run the WinUtil interface on its own thread main.ps1 now only manages the run: it creates a dedicated STA runspace for the window, waits for it, and reports whatever the interface thread failed with. The interface itself moved into Start-WinUtilUserInterface, so the thread that owns the window does nothing but paint and dispatch. - New-WinUtilSessionState builds one starting point for both the interface runspace and the worker pool, carrying $sync, the compiled script globals and every WinUtil function. The pool previously copied only functions matching winutil|WPF, which is not enough for a runspace that has to build a tab. - Invoke-WPFUIThread hands work to the interface runspace as body text plus parameters instead of marshalling a scriptblock. A scriptblock keeps the session state it was written in; running one across runspaces loses the caller's variables on an async post and costs roughly twenty times as much per command, which turned a checkbox refresh into a multi-minute freeze. - Both helpers stop at a shut-down dispatcher, so a job that outlives the window finishes quietly. * Move every long workflow onto the job layer Tweaks, undo, AppX removal and the five Win11 Creator workflows now go through Start-WinUtilJob like the install workflows already did. That removes the five hand-built STA runspaces and the function-definition injection the ISO code needed to reach its own helpers. - One busy flag: $sync.ActiveJob replaces ProcessRunning and Win11ISOProcessRunning, and only the job layer writes it. - Write-WinUtilJobProgress -Hide absorbs the last use of Set-WinUtilTweaksProgressIndicator, so the progress bar and taskbar item have a single owner. The helper is gone. - Show-WinUtilMessage marshals onto the interface thread and logs the prompt, so a job body can ask a question without knowing which thread it is on. The raw MessageBox calls in the ISO workflows are gone. - Win11 Creator status-log lines also go to the session log, and the per-workflow Log/SetProgress helpers are gone. - Get-WinUtilOscdimgPath and Get-WinUtilFreeDriveLetter are now real functions rather than nested ones, so the pool can resolve them. * Log from every WinUtil thread into one session file Start-Transcript only records the runspace it was started on, so every line a worker or the interface logged was being dropped. Write-WinUtilLog now appends to the session log directly, serialized with a named mutex, and the console transcript gets its own file in the same logs directory. * Document the threading model and the job layer * Stop Invoke-WPFUIThread leaking the body's output to its caller The helper returned whatever the body produced, including a bare $null. Callers written against the old void signature then returned an array instead of their own value: Get-WinUtilSelectedPackages handed back @($null, $split), both package lists read as empty, and Install and Uninstall reported success without installing or removing anything. Output is now suppressed unless -PassThru is asked for, which only Show-WinUtilMessage needs. Covered by tests on both the helper and the package split. * Put every button that changes the system on the job layer Invoke-WPFButton now classifies the press instead of running it. Anything that changes the system gets a job; tab switches, selection helpers, window chrome and the WPFPanel* applet launchers stay on the interface thread. Updates, the Ultimate Performance plan, the Fixes buttons, OpenSSH Server, the system repair scan and the AppX query previously ran inline, which froze the window, produced no progress and interleaved their output with a running job. The job layer also owns the console banner now. Write-WinUtilJobBanner draws it once, so the eleven hand-drawn === boxes are gone and every operation announces its start, not only its end. - The job is named after what the button says, read from the config or the control itself, so there is no second list of labels to keep in step. - Show-WinUtilMessage replaces the last raw MessageBox calls, which could not have worked from a worker thread. - Write-WinUtilJobProgress replaces the last direct Set-WinUtilTaskbaritem calls. * Document button routing and the job banner * Read function bodies from the FunctionInfo instead of the function provider Building the session state is on the path to first paint, and going through function:\ for every function cost about as much as the whole interface runspace saved. Time to first window is back level with upstream. * Render the taskbar overlays after first paint The logo overlay render costs about 55ms and nothing can see it until the window is up, so it no longer sits between the interface being built and being shown. Both the logo and the status overlays are now rendered from the same deferred call once the window has painted. * Make a failed package fail the job Both package helpers ran the manager and moved on regardless of its exit code, so a run in which nothing installed still reported success with a green checkmark. They now emit a result per package, classified from the exit code: succeeded, skipped for WinGet telling us there was nothing to do, or failed. The workflow collects them and Complete-WinUtilPackageRun prints the summary and throws when anything failed, which is what puts the job into its failed state. * Time every pipeline step and report the slowest ones Measure-WinUtilStep wraps a step, passes its output through untouched, logs how long it took and keeps the record. Every job and the interface build end with a summary ranking the slowest steps and their share of the total, so "which tweak is taking forever" and "what is holding up startup" are answerable from the log instead of by guessing. Wired into the interface build, each tweak, each undo, each feature, and each package. Jobs also log their own wall-clock duration, and the interface logs the moment it can first service input. * Render the taskbar overlays on the main thread's runspace The overlays need an STA thread, which the worker pool is not, so they get one of their own. Starting it from the interface thread cost more than it saved: opening the runspace took 154-221ms there against 88ms of rendering. Starting it from the main thread instead is free, because that thread does nothing but wait for the window, and the render then overlaps the interface build. Measured over three runs each, time from start to the interface accepting input: 2071/2105ms before, 2105/2131/2193ms started from the interface thread, 2026/2044/2050ms started from the main thread. Also caches the session state, which two runspaces now share, and moves the runspace cleanup registration into its own function for the second caller. * Cut startup to first interaction roughly in half - wire button clicks by type name against a HashSet, not a pipeline per $sync key: 335ms to 81ms - build no tab content before first paint; Invoke-WPFTab already builds the tab it activates - group apps by category into Lists, not by appending to arrays - interface built ~1460ms to ~465ms, ready for input ~2090ms to ~858ms * Warm the unopened tabs while the interface is idle - queue each remaining tab at ApplicationIdle priority after first paint - one tab per queued operation so input is serviced in between - first click on a tab no longer pays for its build * Report failures with the context needed to act on them - Write-WinUtilErrorRecord logs message, exception type, command, line and script stack - used by the job layer, the button funnel, the interface dispatcher and the main thread - route buttons to the job layer by whitelist, so chrome and popup toggles stop starting empty jobs * Wire the Install tab controls where they are created - ChocoRadioButton, WingetRadioButton and the install action buttons come from appnavigation.json, so they do not exist until the Install tab is built - the interface build wired them anyway, which is the three null-reference errors reported on close since tab content moved behind first paint - Initialize-WinUtilInstallTabControls now does it from the tab build, guarded - offline mode disables the install buttons from there too, for the same reason - new test fails if the interface build touches any config-generated control * Stop losing warnings and non-terminating errors from job bodies - a worker buffers its warning and error streams on an object nobody reads: Write-Warning never reached the log, Write-Error reached nothing at all - the job layer merges both into the log, so all 30+ Write-Warning and 4 Write-Error sites in the helpers are visible without touching each one - the interface runspace warning stream is drained on exit too - $sync.LoggedErrors counts error events, detail lines excluded - a job that logged errors without throwing now finishes as "N error(s)" with a warning overlay instead of a green checkmark * Drive winget through Microsoft.WinGet.Client for real progress The winget CLI hides its progress bar as soon as its output is redirected, so a package could only ever be reported as started and finished. The module reports progress and returns a structured result. - Install-WinUtilWinGetClient installs and imports the module, cached per session - Invoke-WinUtilWinGetCommand runs a cmdlet on a nested PowerShell and polls its progress stream, which cannot be redirected like output or errors - percentages map into the package's slice of the job bar: "7zip.7zip - 1.9 MB / 1.9 MB" - outcome comes from Status and InstallerErrorCode, not an exit code - a package already present is upgraded, not reinstalled: Install-WinGetPackage re-downloads and re-runs the installer even without -Force - detection uses Get-WinGetPackage and matches on name as well as id, so apps installed outside winget are recognised (Brave, and every other ARP entry) - falls back to the command line unchanged when the module cannot be installed Verified against real winget in the eval VM, 12 checks; 545 unit tests pass. * Show the install phase as running instead of finished Measured what the module actually emits: 7 progress records whether the package is 1.9 MB or 57.8 MB, only two of them download samples, and the install phase reports 0 then 100 with nothing between. On VLC the install is 4.3s of the 9.7s. - the download gets the first half of the package's slice, so reaching 100% download no longer fills the bar - the install phase pulses the bar and counts elapsed seconds in the label, because neither winget nor the module exposes installer progress - scan the whole progress collection, not just its last record: a byte sample can be superseded within milliseconds - RoundedProgressBarStyle gained an indeterminate trigger; it had none Fixes uninstall reporting a package that is not installed as a failure, which is what UninstallError after 354ms was, and adds ExtendedErrorCode to the detail. * Say what a winget failure actually was The command line prints a sentence for a failure; the client module returns only an HRESULT, so the same failure read as "COMException (0x8A15007D)". Both report the same number, so one table serves both paths. - Get-WinUtilWinGetErrorMessage explains the codes WinUtil hits, and gives the hex plus the return-code reference for anything else - 0x8A15007D now reads: installed for a single user, cannot be removed while running as administrator, remove it from Settings > Apps - unsigned HRESULTs are wrapped rather than cast, which overflowed Int32 - a shared failure reason is repeated in the thrown message - the banner wraps at 76 columns instead of drawing a box wider than the console * Keep the run position in the progress text during a package - the bar itself was already whole-workflow: 0-12, 25-37, 50-62, 75-87, 100 - but the status read "A.A - 50% downloaded", dropping the (n/total) the old per-package messages carried - callers pass a label, so it now reads "A.A (1/4) - 50% downloaded" * Pulse the progress bar in place instead of filling it - IsIndeterminate makes WPF discard Value and stretch the indicator across the whole track: measured 398px of a 400px track at value 40, against 159px correct - the pulse is driven by Tag instead, so the bar keeps the progress it reached - RemoveStoryboard on exit, because Stop left the indicator at whatever opacity the pulse happened to be on * perf: cut Install tab build and keep the interface answering during warmup - look apps up by hashtable index, not dynamic member: Install tab app area 361ms -> 91ms - cap a render pass at 25 apps so a large category cannot stall the interface - yield between batches when building speculative tab content - claim a tab as initialized before building it, so a click during a yield cannot double build - time each step of a tab switch * xaml: drop layout elements and styles that do nothing - remove 8 single child wrappers from control templates, one per instance of every button, toggle and tweak switch - delete unreferenced labelfortweaks and ScrollVisibilityRectangle styles - verified pixel identical across all five tabs * fix: bring the last workflows into the job layer - upgrade all runs package by package on the worker instead of spawning a console - PS profile setup runs pwsh with output captured, not a Windows Terminal tab - resolve the PS7 profile path from pwsh, so remove targets the file install wrote - treat winget exit 3010 and 1641 as success; a reboot requirement is not a failure - drop the power plan success popups, the job layer already reports the result - load PresentationFramework before a message box on a worker, and log if it cannot show - probe optional commands with -ErrorAction so a missing choco does not throw - refresh PATH after installing chocolatey * fix: keep the runspace handle out of the console and the pipeline - suppress the IAsyncResult Start-WinUtilJob got back, which printed a table on every button press - test fails if any caller leaves Invoke-WPFRunspace unassigned * fix: choco parity with winget, and prompts that cannot hang or assume consent - choco runs one package per call, so progress moves and a failure names the package - add an Upgrade action; upgrade all was building "choco install all", which is not a package - explain a choco failure from its own output instead of reporting a bare exit code - pass a progress slice to choco from install and uninstall, as winget already gets - never show a message box without a window: a modal there never returns - an unanswerable prompt answers No, so it can never stand in for consent - uninstall requires an explicit Yes rather than the absence of a No * feat: headless runs report and finish on their own - progress goes to the console when there is no window, throttled so downloads do not bury it - one entry point for preset and config; a preset can now be a baseline a config adds to - apply selected toggles, which only ever applied themselves from the window - exit code carries the outcome: 0 clean, 1 problems, 2 nothing selected - elevation waits for the elevated run and hands its code back - per step timeout, so an installer that never returns cannot hang the run for good - a step that throws no longer abandons the remaining steps - name an unrecognised config entry instead of failing on a null list, and ignore duplicates - import with no window logs instead of throwing on a message box type it cannot load - temp file cleanup skips files in use rather than reporting each as an error * perf: stop the interface stalling while the app list loads Measured with a new input-priority heartbeat: 2669 ms unresponsive across 18 stalls, worst 399 ms. Now 502 ms across 5, worst 151 ms, and nothing after the first three seconds. - cache app icons on disk and fetch the missing ones on a worker; assigning a remote address to an Image made WPF fetch and decode it, landing back on the interface thread whenever the network answered, for a minute after startup - share the six app entry event handlers instead of building them per entry: 3.18 ms to 1.36 ms per entry, measured - slice rendering and tab building by a deadline rather than an entry count, so a slower machine cannot turn a batch into a stall - yield while building the tab opened at startup, as the warmup already did - reject a ParameterList that is a flattened pair; it silently ran the worker with two characters of the parameter name and did nothing - add Start-WinUtilUIHeartbeat behind WINUTIL_TRACE_UI to measure any of this * fix: build the other tabs before finishing the app list Tab warmup was queued at idle priority while app rendering was queued at background priority, so no tab was warmed until every render pass had run. For the first few seconds every tab except the open one was empty, and clicking one paid for its whole build: Tweaks measures 400-530 ms. Verified from the log: all tabs were ready after 32 of 32 background steps before, and after 5 of 32 now. - warm tabs at background priority so they are not starved by the app list - hold app rendering while any tab is still unbuilt - stand aside for 400 ms after input, and skip drawing entries for a tab that is not on screen, resuming when it is - report the whole latency distribution rather than only gaps over 60 ms; a thread kept busy by short pieces of work showed no stall while every interaction still waited behind the piece in flight * fix: ask before closing over running work, and shut down in order Closing while a job ran tore the worker pool down underneath it. A queued instance then started on a runspace already in Closing, threw on a thread pool thread where nothing catches, and ended the process with an unhandled InvalidRunspaceStateException instead of exiting. - ask whether to wait for the running job or stop it, and keep the window open if the close is cancelled - track what is running so it can be stopped, and stop it before closing the pool rather than pulling the runspaces away from it - refuse to queue new work once shutdown has begun - close the window by itself once an awaited job finishes - bound the wait, so a worker that cannot be stopped does not keep the window open for ever - phrase the question so the buttons answer it in any language; Windows labels them itself and text naming them Yes and No does not match Ja and Nein - treat a missing collection as empty; @($null) is a one element array, which read as one running item when nothing was running * fix: let a running job finish in the console after the window is closed Waiting kept the window open until the job ended, which is not what closing it means. The window now goes at once and the run continues where it can still be seen, ending the process when it is done. - close the window immediately and leave the job on the worker pool - skip the pool shutdown on that path; it would stop the work just kept - wait for the job on the main thread, then close the pool and exit - treat a shut down dispatcher as no window, so progress reaches the console instead of being posted to a dispatcher that drops it - bound that wait, so a job that never returns cannot hold the process open - guard the close post against a window that has already gone * feat: pause button, and progress that rewrites its line - add a pause button beside the progress bar; a run holds at the point every loop reports progress, since a command already started cannot be suspended - hold only inside a job worker: whoever presses pause reports it through the same progress call and would otherwise wait on itself - clear the pause once the body returns, or the finish reporting pauses too and the job stays marked running for ever - release it when the window is closed over the job, since there is then no button left to resume with - rewrite the console progress line in place rather than adding one line per update; a single install scrolled a screenful - keep one line per update when output is redirected, where there is no cursor to move, and throttle harder there - say goodbye where the process actually ends; closing can be declined or leave a job running * feat: stop button for the running action - add a stop button beside pause; it asks first, since stopping an install halfway is not undoable - end the run at the same safe point a pause holds at, so anything already started finishes and nothing is cut in half - report it as stopped rather than failed, through its own cancellation - clear the stop before reporting it, or the report throws it again from inside the handler doing the reporting - release a pause when stopping, or the run would never reach the point where it notices - cut the worker off after a grace period, since a single long step reaches no safe point until it returns - reset both flags per job, so a late stop cannot end the next run * fix: use a stop glyph that does not look like a missing one U+E71A is a hollow square at button size and reads as the empty box a font shows for a code point it lacks. U+E73B is the filled square. - add a test that every private use code point in the markup and the scripts exists in Segoe MDL2 Assets * fix: icon buttons render in the icon font, not a fallback A char assigned to Content is not laid out with the control's own font. It falls back to whatever font claims the code point and is drawn in that font's colour and metrics, which is why the pause icon came out teal and a different size from the cross beside it. Add-SelectedAppsMenuItem already cast to string for the same reason. - cast every icon assignment to string: pause, play, the app entry popup and the theme button - use the cancel cross for stop; a square is a blank block at button size whether it is filled or hollow - test that no icon reaches Content as a char * fix: the font scaling slider goes where it is clicked WPF leaves IsMoveToPointEnabled off, so a click on the track pages by LargeChange instead of moving the thumb to the pointer. LargeChange defaults to 1.0, which over a range of 0.75 to 2.0 is most of the track, so clicking anywhere toggled between 100% and 200%. - follow the click, and page by one tick rather than a full unit - test that every slider does both * fix: draw the logo at the size it is asked for, and give the window its own icon The rasteriser built the bitmap from the canvas rather than the requested size, so every render came out 100 by 100 whatever was asked for: a 32px icon was that downscaled, and anything larger was an upscale. The canvas was also smaller than the artwork, whose paths run to about 108 by 110, so the right and bottom edges were cut off. Between the two, the logo covered about a third of the icon it was drawn into. - rasterise from the geometry's real bounds into a bitmap of the requested size, centred and filling the frame - keep the shapes in one place, so the control and the bitmap draw the same art - set the window icon at the sizes the system reports for this display, small and large, instead of leaving Windows to scale one bitmap - hold the icon handles for the life of the window and free the ones replaced * fix: give the nav logo its square box back Fitting the box to the artwork left no room around it. The logo is taller than it is wide, so a square control filled by it edge to edge sat against the tab buttons next to it. - centre the artwork in a square box with a small margin, so the control fills the size it was given rather than the artwork's own proportions - the bitmap form is unchanged and still fills the frame, which is what an icon wants * refactor: one background queue, one UI-alive check, drop the stall tracer - move the DPI window icon work out to feat/dpi-window-icon and revert it here - drop Start-WinUtilUIHeartbeat: it found the startup stalls, it is not something a normal run should carry - add Start-WinUtilBackgroundQueue and put tab warmup and app-entry rendering on it; they were the same pump written twice - add Test-WinUtilUIAlive, replacing the same dispatcher guard hand-written eleven times in three spellings - carry the queue name as the dispatcher's own argument, not a captured variable, so the posted step still resolves where it was written - give Invoke-WinUtilWhenIdle an -Argument for the same reason - load the WPF assemblies in preferences-theme tests instead of relying on logo-render having loaded them first * docs: bring the agent rules in line with the job layer Section 13 still pointed at Set-WinUtilTweaksProgressIndicator, which the job layer removed, and said nothing about the pieces that replaced it. - point the UI-helper rule at Write-WinUtilJobProgress and Test-WinUtilUIAlive instead of the deleted progress indicator - say that long operations go through Start-WinUtilJob, and that a job body owns neither the busy flag, the banner, nor its own interface handling - send deferred interface work through Start-WinUtilBackgroundQueue rather than a fourth hand-rolled pump - record how values reach a posted scriptblock, and why a closure is the wrong answer: it carries the value but binds command lookup to a copied scope - state that diagnostic scaffolding is measured with and then deleted - require each Pester file to load what it needs; several passed only because an earlier file in alphabetical order had loaded it * refactor: trim the package work back to what the pipeline needs The winget client module earns its place: winget.exe hides its progress bar once its output is redirected, so the CLI can never say how far along an install is and the progress bar has nothing to show. What rode in behind it did not. - keep the module, the per-package progress and the result objects both managers now return; those are what the job layer reports from - take the action from the caller instead of probing the machine first: Install, Uninstall and Upgrade each pick their own cmdlet and verb, matching the set Install-WinUtilProgramChoco already takes - drop the Get-WinGetPackage probe with it, which cost a nested call per package and made the module path mean "install or upgrade" while the command line path still meant "install" - send the upgrade workflow through -Action Upgrade, which is what it was always asking for - revert Invoke-WinUtilCurrentSystem to the command line; reading which apps are installed has nothing to do with reporting progress - cut the winget and choco error tables: the hex code and Microsoft's own list say the same thing without a copy to keep in step, and choco's reason is already in the output that gets logged * test: drop the tests that assert on source text Sixty-six tests read a function file and regex-matched its contents, so they failed on edits that changed no behaviour: ten of them broke during a refactor that only moved code between files. A test that pins how something is spelled is an edit detector, not a test. - remove the It blocks that Get-Content a .ps1 and match against it, and the Describe blocks left with nothing in them - keep the ones that read source to check a set rather than a spelling, such as every WPF handler resolving to a defined function and every $sync member being declared; those catch a real mistake - drop Get-WinUtilFunctionFile, which had no callers left * fix: make the active job slot safe to claim and release Three defects in the job layer's shared state, all of them races that the dispatcher happened to hide. - claim $sync.ActiveJob under the collection's own lock. Interface events are serialised by the dispatcher, but a headless run, a scheduled caller and a job body starting another job are not, and a test-then-assign there let two jobs both believe they owned the slot - identify a run by token rather than by name. A worker the stop watchdog cut off can still be unwinding when the next job starts, and its finally block released the slot unconditionally, wiping the claim the next job had just made. Both the worker and the watchdog now release only what they still own - rename Write-WinUtilJobProgress to Step-WinUtilJob. Write- in PowerShell means adds to a stream, and this blocks while paused and throws OperationCanceledException on stop. The trap was already live: the job layer has to clear both flags before its own finish reporting or that reporting re-raises the stop it is reporting Also corrects the cross-runspace cost noted in the tests. Marshalling a scriptblock is not "roughly twenty times" dearer; measured over 400 command invocations it is 5354 ms against 3 ms rebuilt from text, because every command the body invokes is resolved back through the originating runspace. * refactor: drop the WinGet client module from this branch The module is the one thing here that adds a runtime dependency: 53 MB from PSGallery on first use, fetched at elevated privilege. It buys progress movement inside a single package and nothing else. That is a different kind of change from the rest of this branch, which only moves work between threads, and it deserves to be judged on its own. - remove Install-WinUtilWinGetClient and Invoke-WinUtilWinGetCommand - collapse Install-WinUtilProgramWinget onto the command line path it already had underneath, keeping the per-package result objects, so a failed package still fails the job - drop ProgressBase, ProgressSpan and Label from the winget path, which only existed to slice the bar inside one package. Chocolatey keeps its own: it reports per package from its own loop and never needed the module - read upgradable packages from the winget command line output again Kept on feat/winget-client-module, which branches from here. * docs: follow the Write-WinUtilJobProgress rename in the architecture page * fix: address the review findings on the job layer Ten findings held up on inspection. The first two were breakage this branch introduced. - run a nested Start-WinUtilJob inline instead of refusing it. Install Features reaches the job layer twice, once through its feature.json entry and again from Invoke-WPFFeatureInstall, so the inner call was refused and features never installed - read the package manager preference rather than ChocoRadioButton.IsChecked in Invoke-WPFInstallUpgrade, which runs on a worker where touching a control throws - pass parameters and return values through the Invoke-WPFUIThread fallback. [action] carries neither, so Show-WinUtilMessage lost both its arguments and its answer whenever the dispatch delegate was not yet built - register a shell after BeginInvoke, not before: a NotStarted instance looks finished to the pruning pass and could be dropped before shutdown saw it - clear $sync.ActiveJobToken wherever the slot is released, through one Clear-WinUtilActiveJob helper. Clearing only the name left a token that could match a later run - clear $global:WinUtilIsJobWorker when a worker finishes. Pool runspaces are reused, so the flag outlived the job that set it - set the pause and stop button state through the dispatcher - take a locked snapshot of $sync.ActiveShells before enumerating it, and create the collection under the lock - stop the watchdog waiting on the interface thread. It issues the stop and watches later ticks instead of sleeping for up to ten seconds - throw rather than return on the ISO failure paths, which the job layer was reporting as finished, and check exit codes in Invoke-WPFSystemRepair and Invoke-WPFUltimatePerformance so a failed step fails the job - give the pause and stop buttons AutomationProperties.Name; their content is a private-use glyph * fix: address the remaining review findings Seven more held up. The first is the one that mattered most and this branch introduced it. - separate recycling a runspace pool from shutting down. Initialize-WinUtilRunspacePool replaces a pool that is no longer open by calling Close-WinUtilRunspacePool, which set $sync.ShuttingDown. Nothing resets it, so a single recycle made the job layer refuse every later action for the rest of the session - bound the ISO mount wait at 60 seconds. The loop had no exit but success, and one job runs at a time, so a damaged or already-mounted image blocked every other action and the shutdown wait - check robocopy exit codes in both ISO and USB workflows through one Invoke-WinUtilRobocopy. Codes of 8 and above mean files were not copied, which produced media that reported complete and did not boot - read oscdimg stderr as it arrives. Draining stdout first and stderr afterwards deadlocks once the stderr pipe fills - look for oscdimg on PATH, in both ADK roots and in the per-user WinGet package root, using the same search before and after the install attempt - set the child error preference to Stop for the profile setup, and fail on captured error records. A non-terminating failure exited zero and was reported as installed - confirm each id parsed out of the winget upgrade table against winget before upgrading it, and keep stderr out of the parse. The table is localised and truncated, so a wrapped version or a translated header matched the same shape * fix: address the minor and nitpick review findings The review body carried 38 findings beyond the 20 inline ones, in collapsed sections. These are the ones that held up. Real defects: - the theme hook assigned to $handled instead of $handled.Value, so a WM_SETTINGCHANGE was never marked handled - a declined UAC prompt left $elevated null and exited 0, which a headless caller reads as a successful run - Start-Transcript ran before the log directory existed, so the first run failed before logging started - Write-WinUtilLog appended even when the mutex wait timed out, which is the case with the most contention and the one that interleaves lines - Wait-WinUtilRemainingWork took [int] minutes, so any fractional timeout truncated to zero and the bound meant "do not wait" - the console progress throttle only applied when the text was unchanged, so a job reporting per package wrote on every call - an undecodable icon response stayed in the cache and was skipped on every later run - Start-WinUtilAssetRendering leaked its dedicated STA runspace; the cleanup callback now disposes a runspace it was given - Invoke-WPFFixesWinget could not repair a broken WinGet, because Install-WinUtilWinget returns early when winget is detected. Added -Force - winget's own reboot-required and reboot-initiated codes counted as failures Smaller: - set TLS 1.2 and a timeout before fetching the Chocolatey bootstrap - let the headless queue drain survive one failing item, as the dispatcher path already does - bind the idle timer to the interface dispatcher explicitly - route the export message through Show-WinUtilMessage like the import branch - suppress New-Item output that was reaching the job's output stream Tests that could not fail: - the bounded-wait test passed a fractional timeout that truncated to zero, so the wait never ran - the slider tests cast possibly-absent attributes to [double], and compared two empty strings for the regression they exist to catch * fix: startup crash and icons that only appeared once the list was drawn Both were mine, and both were introduced while addressing review findings. The crash. Register-WinUtilRunspaceCleanup compiles its helper type once and guards that with a type-exists check, so a session already holding the old shape keeps it. Adding a Runspace property to that type therefore failed on every later run with "The property 'Runspace' cannot be found". The type is back to exactly what it was, and the asset rendering runspace lives until the process exits: one STA runspace for the lifetime of the app is the cheaper trade against a helper type that cannot be changed safely in a session. The icons. Start-WinUtilIconFetch ran from Complete-WinUtilInstallAppRendering, so nothing was fetched until every entry had been drawn and no icon appeared for the first several seconds. Upstream assigned a remote address per Image and let WPF fetch them all at once, which is what made them appear promptly. - fetch after each render batch instead, gated so one fetch runs at a time and whatever queues up meanwhile is taken when that fetch ends - clear the gate when the runspace refuses the work, or no fetch would ever run again for the rest of the session - request each wave together rather than one icon after another Measured on a cold cache in a VM: first icons at 1.35s and all of them by 4.0s, against nothing before roughly 7s previously. * fix: address the second review pass All three are in code this branch added. - accept DISM exit code 3010 as success. It is ERROR_SUCCESS_REBOOT_REQUIRED, meaning the image was repaired and the change lands on restart, so the exit code check added in the last pass turned a successful repair into a failed job. Carried per step, since the same number from chkdsk or sfc does not mean that, and logged as a warning so the restart is not silent - say what is happening on the WinGet repair path. Install-WinUtilWinget -Force runs while WinGet is installed, and printed "WinGet is not installed" - append to PATH after installing Chocolatey rather than replacing it. Overwriting with the machine and user values drops whatever this process added earlier in the session * fix: treat chkdsk's own exit codes as the outcomes they are - chkdsk /scan reports 1 and 2 as ordinary results, so aborting the whole repair on them skipped sfc and dism for no reason - 3 stays a failure: the disk could not be checked, and the later steps are not worth running on a disk in that state - drive the decision off each step's SuccessCodes instead of a branch hard-coded to DISM's 3010, so the same number means what that step means - cover chkdsk 0, 1, 2 and 3, and 3010 from the wrong step * fix: address the Codex review findings on the job layer - send WPFPanel buttons that carry a function through the job layer: the system corruption scan waited on chkdsk, sfc and dism on the interface thread, with nothing to pause, stop or report it - stop a timed out headless step before starting the next one, and abandon the run if it will not stop, rather than changing the machine from two runs at once - keep the stop watchdog's slot claimed until the worker has actually gone - skip the install summary reset when the Install tab has not been built, which offline startup does by opening Tweaks first - keep Win11ISO out of the tab warmup: building it runs the existing work check, which reports or prompts while the user is on another tab - pass --include-unknown when upgrading, since the scan that found the packages used it * fix: stop the tab warmup dying on a sync that is still growing - Reset-WPFCheckBoxes enumerated $sync live while setting IsChecked, whose handlers add to $sync, and while the warmup was building controls into it; either one invalidated the enumerator and the warmup reported "Collection was modified" - snapshot both loops - cover it with a checkbox that grows $sync from its own handler, which reproduces the failure without the fix * fix: shrink the pause and stop buttons and take them away when idle - they were sized like the title bar icons, next to a 6px progress bar - give them a size of their own and the smaller icon font, which font scaling still applies to - collapsed unless a job is running: they were only ever disabled, so a finished or failed run left two dead buttons on screen - the progress bar still stays to report how the run ended * fix: colour the progress bar by how the run ended - only the taskbar item carried the state, so a run that finished with errors left a full bar in the normal colour, reading as a clean finish - the fill now follows the bar's Foreground, which the job layer points at an error or warning colour and back again - by resource reference, so switching theme repaints it - add error and warning colours to both themes * fix: stop rendering the app navigation twice, which left Install dead - upstream moved the app navigation render into Initialize-WPFUI, and this branch still rendered it beforehand, so it was built twice - the second pass clears the target grid, and the "already wired" guard goes by name, so the replacement buttons counted as wired and never got a click handler: Install and Uninstall did nothing, with no error anywhere - leave that render to Initialize-WPFUI * fix: uninstall apps that belong to the signed in user - WinGet answers APPINSTALLER_CLI_ERROR_ADMIN_CONTEXT_ACTION_PROHIBITED for anything installed in user scope while it is running elevated, and WinUtil is always elevated, so those uninstalls did nothing - a process cannot drop its own elevation, so hand that one command to a scheduled task running as the interactive user at limited run level - retry only on that code, so everything else is untouched * style: keep the ISO variable names the file already used - new scriptblock parameters were named in PascalCase, which put 20 lines into the diff that are otherwise identical to main - nothing about the behaviour changes * refactor: take the icon cache out of the job layer change It is its own feature, not something consolidating background work needs. Entries go back to assigning the favicon address to the image, as main does. The work is kept on feat/install-icon-cache and follows once this lands. * refactor: take pause and stop out of the job layer change They are a new feature of their own, not part of consolidating background work, and main has nothing like them today. - drop the two buttons, their routing and the checkpoints that served them - keep Stop-WinUtilActiveWork: closing the pool and the headless step timeout both need to end work, and neither is the user pressing a button The work is kept on feat/job-pause-stop and follows once this lands. * refactor: drop changes the job layer does not need - the icon glyph strings are a rendering fix of their own: main renders a char, and whether that is right has nothing to do with background work. Reverted here, kept on fix/ui-glyph-strings - trim the temp cleanup tweak to the part this change forces: errors have to be suppressed because the job layer counts a logged error as a failed step, but counting what was removed is unrelated polish * refactor: cut comments and wrappers that were not earning their place - the Recycle switch explained itself over four lines; two say what a reader needs to know - drop a .PARAMETER that only restated the parameter name - Unregister-WinUtilActiveShell had one caller and cost twenty lines, so it is inlined - the running check was the same try/catch three times over, now one helper * refactor: comments describe the code, not how it got here - remove five comments left pointing at code the icon and pause splits took away, two of them sitting above unrelated lines - cut the ones that recounted a bug rather than describing behaviour: the double rendered navigation, the dropped [action] parameters, the shell registered too early, the Add-Type shape - shorten the rest to what a reader needs * refactor: inline the winget error message at its only call site - Get-WinUtilWinGetErrorMessage was 26 lines for a one line format string - Its zero guard was dead: exit code 0 is handled by an earlier branch - Drop the file and the two test dot-sources of it * refactor: drop the user scope uninstall fix, it has its own branch - Invoke-WinUtilUnelevated and the elevation retry are self contained - They ship on fix/winget-user-scope-uninstall, which does not need the job layer - Removes a file and 3 tests from a PR that is already large * docs: tighten function help and comments - Cut narrative framing from the job layer, queue and shutdown doc blocks - Kept the technical reasons, dropped the restatements around them * docs: drop comments that restate the code - Consent check, powercfg exit codes, shortcut guard, rethrow, single reason - Compact the UI thread helper's help * fix: address async job layer review findings * fix: harden job startup and dependency discovery * fix: close remaining async review gaps * fix: finalize shutdown and ISO recovery paths * fix: guarantee job cleanup and error attribution * fix: preserve job outcomes during cancellation * fix: close async rendering and logging races * fix: bound shutdown and surface UI startup failures * fix: skip blocked user-scope install updates * fix: address final PR review findings * fix: warn on blocked user-scope package actions * fix: propagate file-backed headless exit codes * Address final async job review findings * Resolve latest automated review findings * Preserve clipboard history in activity tweak --------- Co-authored-by: Chris Titus <contact@christitus.com>
This commit is contained in:
@@ -1,18 +1,147 @@
|
||||
function Close-WinUtilRunspacePool {
|
||||
if ($null -eq $sync -or -not $sync.ContainsKey("runspace") -or $null -eq $sync.runspace) {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Stops anything still running and closes the worker pool
|
||||
|
||||
.DESCRIPTION
|
||||
Closing the pool with work still in it is what produced an unhandled
|
||||
InvalidRunspaceStateException: a queued instance starts on a runspace that is already
|
||||
closing, throws on a thread pool thread, and takes the process down. Whatever is in
|
||||
flight is therefore asked to stop, and waited for, before the pool is closed.
|
||||
#>
|
||||
param(
|
||||
[int]$StopTimeoutSeconds = 15,
|
||||
|
||||
# Leaves ShuttingDown clear: nothing resets it, so setting it here would refuse every
|
||||
# later action for the rest of the session
|
||||
[switch]$Recycle
|
||||
)
|
||||
|
||||
if ($null -eq $sync) {
|
||||
return
|
||||
}
|
||||
|
||||
$poolLock = Get-WinUtilRunspacePoolLock
|
||||
[System.Threading.Monitor]::Enter($poolLock)
|
||||
try {
|
||||
if ($sync.runspace.RunspacePoolStateInfo.State -notin @(
|
||||
[System.Management.Automation.Runspaces.RunspacePoolState]::Closed,
|
||||
[System.Management.Automation.Runspaces.RunspacePoolState]::Closing,
|
||||
[System.Management.Automation.Runspaces.RunspacePoolState]::Broken
|
||||
)) {
|
||||
$sync.runspace.Close()
|
||||
# Set before stopping, so nothing that is winding down queues fresh work behind us
|
||||
if (-not $Recycle) {
|
||||
$sync.ShuttingDown = $true
|
||||
}
|
||||
|
||||
if (-not $sync.ContainsKey("runspace") -or $null -eq $sync.runspace) {
|
||||
return
|
||||
}
|
||||
|
||||
$stopped = $true
|
||||
try {
|
||||
$stopped = Stop-WinUtilActiveWork -TimeoutSeconds $StopTimeoutSeconds
|
||||
} catch {
|
||||
$stopped = $false
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Could not stop running work cleanly: $($_.Exception.Message)"
|
||||
}
|
||||
|
||||
$pool = $sync.runspace
|
||||
$cleanupDeferred = $false
|
||||
try {
|
||||
$poolState = $pool.RunspacePoolStateInfo.State
|
||||
$terminalStates = @(
|
||||
[System.Management.Automation.Runspaces.RunspacePoolState]::Closed,
|
||||
[System.Management.Automation.Runspaces.RunspacePoolState]::Broken
|
||||
)
|
||||
|
||||
if (-not $stopped -and $poolState -notin $terminalStates) {
|
||||
# Close and Dispose both wait for an invocation that ignored BeginStop. Hand
|
||||
# cleanup to the thread pool so the timeout above remains a real upper bound.
|
||||
$cleanupDeferred = $true
|
||||
if ($poolState -ne [System.Management.Automation.Runspaces.RunspacePoolState]::Closing) {
|
||||
Register-WinUtilRunspacePoolCleanup -RunspacePool $pool
|
||||
}
|
||||
} elseif ($poolState -notin ($terminalStates + [System.Management.Automation.Runspaces.RunspacePoolState]::Closing)) {
|
||||
$pool.Close()
|
||||
}
|
||||
} catch {
|
||||
# A pool that will not close cleanly must not stop the window from closing
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Worker pool did not close cleanly: $($_.Exception.Message)"
|
||||
} finally {
|
||||
if (-not $cleanupDeferred) {
|
||||
try {
|
||||
$pool.Dispose()
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Worker pool did not dispose cleanly: $($_.Exception.Message)"
|
||||
}
|
||||
}
|
||||
$sync.Remove("runspace")
|
||||
if ($sync.ActiveShells) { $sync.ActiveShells.Clear() }
|
||||
}
|
||||
} finally {
|
||||
$sync.runspace.Dispose()
|
||||
$sync.Remove("runspace")
|
||||
[System.Threading.Monitor]::Exit($poolLock)
|
||||
}
|
||||
}
|
||||
|
||||
function Register-WinUtilRunspacePoolCleanup {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Closes and disposes a worker pool without blocking the calling thread
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[System.Management.Automation.Runspaces.RunspacePool]$RunspacePool
|
||||
)
|
||||
|
||||
if (-not ("WinUtilRunspacePoolCleanup" -as [type])) {
|
||||
Add-Type @"
|
||||
using System;
|
||||
using System.Management.Automation.Runspaces;
|
||||
|
||||
public sealed class WinUtilRunspacePoolCleanupState
|
||||
{
|
||||
public RunspacePool RunspacePool { get; set; }
|
||||
public IAsyncResult Handle { get; set; }
|
||||
}
|
||||
|
||||
public static class WinUtilRunspacePoolCleanup
|
||||
{
|
||||
public static readonly System.Threading.WaitOrTimerCallback Callback = Cleanup;
|
||||
|
||||
public static void Cleanup(object state, bool timedOut)
|
||||
{
|
||||
var cleanupState = state as WinUtilRunspacePoolCleanupState;
|
||||
if (cleanupState == null || cleanupState.RunspacePool == null || cleanupState.Handle == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
cleanupState.RunspacePool.EndClose(cleanupState.Handle);
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
finally
|
||||
{
|
||||
try
|
||||
{
|
||||
cleanupState.RunspacePool.Dispose();
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"@
|
||||
}
|
||||
|
||||
$cleanupState = [WinUtilRunspacePoolCleanupState]::new()
|
||||
$cleanupState.RunspacePool = $RunspacePool
|
||||
$cleanupState.Handle = $RunspacePool.BeginClose($null, $null)
|
||||
[System.Threading.ThreadPool]::RegisterWaitForSingleObject(
|
||||
$cleanupState.Handle.AsyncWaitHandle,
|
||||
[WinUtilRunspacePoolCleanup]::Callback,
|
||||
$cleanupState,
|
||||
-1,
|
||||
$true
|
||||
) | Out-Null
|
||||
}
|
||||
|
||||
@@ -0,0 +1,58 @@
|
||||
function Complete-WinUtilPackageRun {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Reports what a package run actually did and fails the job on unexpected errors
|
||||
|
||||
.DESCRIPTION
|
||||
Package managers report failure through an exit code, which is easy to walk past.
|
||||
Without this the job layer would show a green checkmark for a run in which nothing
|
||||
changed. Unexpected failures terminate the job; expected elevated-context skips
|
||||
raise a warning so the job cannot claim that every requested action completed.
|
||||
|
||||
.PARAMETER Action
|
||||
Install or Uninstall, used in the summary text.
|
||||
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Action,
|
||||
|
||||
[object[]]$Results = @()
|
||||
)
|
||||
|
||||
$succeeded = @($Results | Where-Object { $_.Outcome -eq "Succeeded" })
|
||||
$skipped = @($Results | Where-Object { $_.Outcome -eq "Skipped" })
|
||||
$failed = @($Results | Where-Object { $_.Outcome -eq "Failed" })
|
||||
|
||||
$summary = "$($succeeded.Count) succeeded, $($skipped.Count) skipped, $($failed.Count) failed"
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action summary: $summary"
|
||||
Write-Host "$Action summary: $summary"
|
||||
|
||||
foreach ($result in $skipped) {
|
||||
Write-Host " skipped $($result.Package) - $($result.Detail)"
|
||||
}
|
||||
foreach ($result in $failed) {
|
||||
Write-Host " failed $($result.Package) - $($result.Detail)" -ForegroundColor Red
|
||||
}
|
||||
|
||||
$adminContextSkipped = @($skipped | Where-Object { $_.ExitCode -eq -1978335107 })
|
||||
if ($adminContextSkipped.Count -gt 0) {
|
||||
Write-Warning "$($adminContextSkipped.Count) package action(s) were skipped because elevated WinUtil cannot modify user-scoped installations."
|
||||
}
|
||||
|
||||
if ($failed.Count -gt 0) {
|
||||
$names = ($failed | ForEach-Object { $_.Package }) -join ', '
|
||||
$reasons = @($failed | ForEach-Object { $_.Detail } | Sort-Object -Unique)
|
||||
|
||||
$message = if ($reasons.Count -eq 1) {
|
||||
"$($failed.Count) of $($Results.Count) package(s) failed: $names. $($reasons[0])"
|
||||
} else {
|
||||
"$($failed.Count) of $($Results.Count) package(s) failed: $names. See the lines above for each reason."
|
||||
}
|
||||
# Each failed package was already logged by its package-manager adapter. Carry that fact
|
||||
# with the summary exception so the job wrapper adds context without another error count.
|
||||
$exception = [System.InvalidOperationException]::new($message)
|
||||
$exception.Data["WinUtilErrorReported"] = $true
|
||||
throw $exception
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,60 @@
|
||||
function Get-WinUtilAppEntryHandlers {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
The event handlers shared by every app entry on the Install tab
|
||||
|
||||
.DESCRIPTION
|
||||
A scriptblock literal inside a loop is a new scriptblock every time round, and
|
||||
building six of them per app is the single largest cost of drawing the app list:
|
||||
measured at 2.13 ms per entry against 0.62 ms when they are made once and reused.
|
||||
|
||||
None of them close over anything per entry. They read the sender through $this, so
|
||||
one instance serves every app.
|
||||
#>
|
||||
|
||||
if ($null -ne $script:WinUtilAppEntryHandlers) {
|
||||
return $script:WinUtilAppEntryHandlers
|
||||
}
|
||||
|
||||
$script:WinUtilAppEntryHandlers = @{
|
||||
BorderClick = {
|
||||
# Resolve through $sync because the border's child is a layout Grid for FOSS entries
|
||||
$childCheckbox = $sync.$($this.Tag)
|
||||
$childCheckbox.IsChecked = -not $childCheckbox.IsChecked
|
||||
}
|
||||
MouseEnter = {
|
||||
if (($sync.$($this.Tag).IsChecked) -eq $false) {
|
||||
$this.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallHighlightedColor")
|
||||
}
|
||||
}
|
||||
MouseLeave = {
|
||||
if (($sync.$($this.Tag).IsChecked) -eq $false) {
|
||||
$this.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallUnselectedColor")
|
||||
}
|
||||
}
|
||||
RightClick = {
|
||||
# Store the selected app in a global variable so it can be used in the popup
|
||||
$sync.appPopupSelectedApp = $this.Tag
|
||||
# Set the popup position to the current mouse position
|
||||
$sync.appPopup.PlacementTarget = $this
|
||||
$sync.appPopup.IsOpen = $true
|
||||
}
|
||||
# The checkbox sits inside the entry layout Grid, so the border is one level further up
|
||||
Checked = {
|
||||
Invoke-WPFSelectedCheckboxesUpdate -type "Add" -checkboxName $this.Tag
|
||||
$borderElement = $this.Parent.Parent
|
||||
$borderElement.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallSelectedColor")
|
||||
}
|
||||
Unchecked = {
|
||||
Invoke-WPFSelectedCheckboxesUpdate -type "Remove" -checkboxName $this.Tag
|
||||
$borderElement = $this.Parent.Parent
|
||||
$borderElement.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallUnselectedColor")
|
||||
}
|
||||
ImageFailed = {
|
||||
$this.Visibility = "Collapsed"
|
||||
$this.Parent.Children[0].Visibility = "Visible"
|
||||
}
|
||||
}
|
||||
|
||||
return $script:WinUtilAppEntryHandlers
|
||||
}
|
||||
@@ -4,6 +4,8 @@ function Get-WinUtilEnvironmentReport {
|
||||
Collects the allowlisted data used by the WinUtil environment report.
|
||||
#>
|
||||
|
||||
$reportWarnings = [System.Collections.Generic.List[string]]::new()
|
||||
|
||||
$windows = [ordered]@{
|
||||
edition = $null
|
||||
version = $null
|
||||
@@ -27,7 +29,9 @@ function Get-WinUtilEnvironmentReport {
|
||||
$hardware.totalMemoryGB = [math]::Round(([double]$operatingSystem.TotalVisibleMemorySize / 1MB), 2)
|
||||
}
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to collect Windows/memory info from Win32_OperatingSystem: $($_.Exception.Message)"
|
||||
$message = "Failed to collect Windows/memory info from Win32_OperatingSystem: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
try {
|
||||
@@ -37,7 +41,9 @@ function Get-WinUtilEnvironmentReport {
|
||||
$hardware.logicalProcessorCount = [int](($processors | Measure-Object -Property NumberOfLogicalProcessors -Sum).Sum)
|
||||
}
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to collect CPU info from Win32_Processor: $($_.Exception.Message)"
|
||||
$message = "Failed to collect CPU info from Win32_Processor: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
$powershell = [ordered]@{
|
||||
@@ -49,7 +55,9 @@ function Get-WinUtilEnvironmentReport {
|
||||
try {
|
||||
$powershell.executionPolicy = (Get-ExecutionPolicy).ToString()
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to read PowerShell execution policy: $($_.Exception.Message)"
|
||||
$message = "Failed to read PowerShell execution policy: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
# Re-use built-in functionality
|
||||
@@ -57,14 +65,23 @@ function Get-WinUtilEnvironmentReport {
|
||||
try {
|
||||
$chocolatey.installed = (Test-WinUtilPackageManager -choco 6>$null) -eq "installed"
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to check Chocolatey availability: $($_.Exception.Message)"
|
||||
$message = "Failed to check Chocolatey availability: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
if ($chocolatey.installed) {
|
||||
try {
|
||||
$chocolatey.version = (choco -v 2>&1 | Select-Object -First 1).ToString().Trim()
|
||||
$global:LASTEXITCODE = 0
|
||||
$versionOutput = @(choco -v 2>&1)
|
||||
if ($LASTEXITCODE -ne 0) {
|
||||
throw "Chocolatey version probe exited with code $LASTEXITCODE."
|
||||
}
|
||||
$chocolatey.version = ($versionOutput | Select-Object -First 1).ToString().Trim()
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to read Chocolatey version: $($_.Exception.Message)"
|
||||
$message = "Failed to read Chocolatey version: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -72,18 +89,29 @@ function Get-WinUtilEnvironmentReport {
|
||||
try {
|
||||
$winget.installed = (Test-WinUtilPackageManager -winget 6>$null) -eq "installed"
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to check WinGet availability: $($_.Exception.Message)"
|
||||
$message = "Failed to check WinGet availability: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
if ($winget.installed) {
|
||||
try {
|
||||
$winget.version = (winget -v 2>&1 | Select-Object -First 1).ToString().Trim()
|
||||
$global:LASTEXITCODE = 0
|
||||
$versionOutput = @(winget -v 2>&1)
|
||||
if ($LASTEXITCODE -ne 0) {
|
||||
throw "WinGet version probe exited with code $LASTEXITCODE."
|
||||
}
|
||||
$winget.version = ($versionOutput | Select-Object -First 1).ToString().Trim()
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to read WinGet version: $($_.Exception.Message)"
|
||||
$message = "Failed to read WinGet version: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
}
|
||||
|
||||
$system = [ordered]@{ pendingRebootRequired = $false }
|
||||
# Null means the registry state could not be read. Do not turn an access/provider failure into
|
||||
# a misleading "no reboot required" result.
|
||||
$system = [ordered]@{ pendingRebootRequired = $null }
|
||||
try {
|
||||
$rebootPaths = @(
|
||||
"HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Component Based Servicing\RebootPending",
|
||||
@@ -93,17 +121,30 @@ function Get-WinUtilEnvironmentReport {
|
||||
# A present-but-empty PendingFileRenameOperations value still returns a non-null object, so
|
||||
# check the actual entries rather than just whether the property exists.
|
||||
$pendingFileRenameOperations = @(
|
||||
(Get-ItemProperty -Path "HKLM:\SYSTEM\CurrentControlSet\Control\Session Manager" `
|
||||
-Name "PendingFileRenameOperations" -ErrorAction SilentlyContinue).PendingFileRenameOperations |
|
||||
(Get-ItemProperty -LiteralPath "HKLM:\SYSTEM\CurrentControlSet\Control\Session Manager" `
|
||||
-ErrorAction Stop).PendingFileRenameOperations |
|
||||
Where-Object { -not [string]::IsNullOrWhiteSpace([string]$_) }
|
||||
)
|
||||
$system.pendingRebootRequired = ($rebootPaths | Where-Object { Test-Path $_ }).Count -gt 0 -or
|
||||
$system.pendingRebootRequired = ($rebootPaths | Where-Object {
|
||||
Test-Path -LiteralPath $_ -ErrorAction Stop
|
||||
}).Count -gt 0 -or
|
||||
$pendingFileRenameOperations.Count -gt 0
|
||||
} catch {
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message "Failed to check pending-reboot registry state: $($_.Exception.Message)"
|
||||
$message = "Failed to check pending-reboot registry state: $($_.Exception.Message)"
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
$tweaksState = Get-WinUtilTweaksStateReport
|
||||
if ($tweaksState.collectionStatus -ne "collected") {
|
||||
$message = "Failed to collect the complete tweak state for the environment report."
|
||||
Write-WinUtilLog -Component "EnvironmentReport" -Level "WARN" -Message $message
|
||||
[void]$reportWarnings.Add($message)
|
||||
}
|
||||
|
||||
foreach ($message in $reportWarnings) {
|
||||
Write-Warning $message
|
||||
}
|
||||
|
||||
return [pscustomobject][ordered]@{
|
||||
schemaVersion = "1.0"
|
||||
|
||||
@@ -22,17 +22,17 @@ function Get-WinUtilRecentLogs {
|
||||
$LogDirectory = Join-Path $sync.winutildir "logs"
|
||||
}
|
||||
|
||||
if (-not (Test-Path $LogDirectory)) {
|
||||
if (-not (Test-Path -LiteralPath $LogDirectory -ErrorAction Stop)) {
|
||||
return ""
|
||||
}
|
||||
|
||||
$cutoff = (Get-Date).AddDays(-$Days)
|
||||
$logFiles = Get-ChildItem -Path $LogDirectory -Filter "winutil_*.log" -File -ErrorAction SilentlyContinue |
|
||||
$logFiles = Get-ChildItem -LiteralPath $LogDirectory -Filter "winutil_*.log" -File -ErrorAction Stop |
|
||||
Where-Object { $_.LastWriteTime -ge $cutoff } |
|
||||
Sort-Object LastWriteTime
|
||||
|
||||
$sections = foreach ($logFile in $logFiles) {
|
||||
"=== $($logFile.Name) ===`n$(Get-Content -Path $logFile.FullName -Raw)"
|
||||
"=== $($logFile.Name) ===`n$(Get-Content -LiteralPath $logFile.FullName -Raw -ErrorAction Stop)"
|
||||
}
|
||||
|
||||
return ($sections -join "`n`n")
|
||||
|
||||
@@ -0,0 +1,17 @@
|
||||
function Get-WinUtilRunspacePoolLock {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Returns the lock that serializes worker-pool startup and shutdown
|
||||
#>
|
||||
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
if ($null -eq $sync.RunspacePoolLock) {
|
||||
$sync.RunspacePoolLock = [object]::new()
|
||||
}
|
||||
|
||||
return $sync.RunspacePoolLock
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
}
|
||||
@@ -8,10 +8,9 @@ function Get-WinUtilSelectedPackages {
|
||||
[string] $Preference
|
||||
)
|
||||
|
||||
# A single package has no meaningful percentage to show
|
||||
if ($PackageList.count -eq 1) {
|
||||
Invoke-WPFUIThread -ScriptBlock { Set-WinUtilTaskbaritem -state "Indeterminate" -value 0.01 -overlay "logo" }
|
||||
} else {
|
||||
Invoke-WPFUIThread -ScriptBlock { Set-WinUtilTaskbaritem -state "Normal" -value 0.01 -overlay "logo" }
|
||||
Step-WinUtilJob -State "Indeterminate"
|
||||
}
|
||||
|
||||
$packagesWinget = [System.Collections.ArrayList]::new()
|
||||
|
||||
@@ -13,35 +13,18 @@ function Initialize-InstallAppEntry {
|
||||
$appKey
|
||||
)
|
||||
|
||||
$app = $sync.configs.applicationsHashtable.$appKey
|
||||
$app = $sync.configs.applicationsHashtable[$appKey]
|
||||
$handlers = Get-WinUtilAppEntryHandlers
|
||||
|
||||
# Create the outer Border for the application type
|
||||
$border = New-Object Windows.Controls.Border
|
||||
$border.Style = $sync.Form.Resources.AppEntryBorderStyle
|
||||
$border.Tag = $appKey
|
||||
$border.ToolTip = Get-WinUtilEntryToolTip -Description $app.description -Key $appKey
|
||||
$border.Add_MouseLeftButtonUp({
|
||||
# Resolve through $sync because the border's child is a layout Grid for FOSS entries
|
||||
$childCheckbox = $sync.$($this.Tag)
|
||||
$childCheckbox.IsChecked = -not $childCheckbox.IsChecked
|
||||
})
|
||||
$border.Add_MouseEnter({
|
||||
if (($sync.$($this.Tag).IsChecked) -eq $false) {
|
||||
$this.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallHighlightedColor")
|
||||
}
|
||||
})
|
||||
$border.Add_MouseLeave({
|
||||
if (($sync.$($this.Tag).IsChecked) -eq $false) {
|
||||
$this.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallUnselectedColor")
|
||||
}
|
||||
})
|
||||
$border.Add_MouseRightButtonUp({
|
||||
# Store the selected app in a global variable so it can be used in the popup
|
||||
$sync.appPopupSelectedApp = $this.Tag
|
||||
# Set the popup position to the current mouse position
|
||||
$sync.appPopup.PlacementTarget = $this
|
||||
$sync.appPopup.IsOpen = $true
|
||||
})
|
||||
$border.Add_MouseLeftButtonUp($handlers.BorderClick)
|
||||
$border.Add_MouseEnter($handlers.MouseEnter)
|
||||
$border.Add_MouseLeave($handlers.MouseLeave)
|
||||
$border.Add_MouseRightButtonUp($handlers.RightClick)
|
||||
|
||||
$checkBox = New-Object Windows.Controls.CheckBox
|
||||
# Sanitize the name for WPF
|
||||
@@ -49,18 +32,8 @@ function Initialize-InstallAppEntry {
|
||||
# Store the original appKey in Tag
|
||||
$checkBox.Tag = $appKey
|
||||
$checkbox.Style = $sync.Form.Resources.AppEntryCheckboxStyle
|
||||
# The checkbox sits inside the entry layout Grid, so the border is one level further up
|
||||
$checkbox.Add_Checked({
|
||||
Invoke-WPFSelectedCheckboxesUpdate -type "Add" -checkboxName $this.Tag
|
||||
$borderElement = $this.Parent.Parent
|
||||
$borderElement.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallSelectedColor")
|
||||
})
|
||||
|
||||
$checkbox.Add_Unchecked({
|
||||
Invoke-WPFSelectedCheckboxesUpdate -type "Remove" -checkboxName $this.Tag
|
||||
$borderElement = $this.Parent.Parent
|
||||
$borderElement.SetResourceReference([Windows.Controls.Control]::BackgroundProperty, "AppInstallUnselectedColor")
|
||||
})
|
||||
$checkbox.Add_Checked($handlers.Checked)
|
||||
$checkbox.Add_Unchecked($handlers.Unchecked)
|
||||
|
||||
$contentPanel = New-Object Windows.Controls.StackPanel
|
||||
$contentPanel.Orientation = "Horizontal"
|
||||
@@ -73,15 +46,16 @@ function Initialize-InstallAppEntry {
|
||||
$fallback = New-Object Windows.Controls.TextBlock
|
||||
$fallback.Text = $app.content.TrimStart(".").Substring(0, 1).ToUpper()
|
||||
$fallback.FontWeight = "Bold"; $fallback.HorizontalAlignment = "Center"; $fallback.VerticalAlignment = "Center"
|
||||
if ($app.link) { $fallback.Visibility = "Collapsed" }
|
||||
$fallback.SetResourceReference([Windows.Controls.TextBlock]::FontSizeProperty, "AppEntryFontSize")
|
||||
$fallback.SetResourceReference([Windows.Controls.TextBlock]::ForegroundProperty, "ToggleButtonOnColor")
|
||||
[void]$icon.Children.Add($fallback)
|
||||
if ($app.link) {
|
||||
$fallback.Visibility = "Collapsed"
|
||||
$logo = New-Object Windows.Controls.Image
|
||||
$logo.Stretch = [Windows.Media.Stretch]::Uniform
|
||||
$logo.Source = "https://www.google.com/s2/favicons?sz=64&domain_url=$([uri]::EscapeDataString($app.link))"
|
||||
$logo.Add_ImageFailed({ $this.Visibility = "Collapsed"; $this.Parent.Children[0].Visibility = "Visible" })
|
||||
$logo.Add_ImageFailed($handlers.ImageFailed)
|
||||
|
||||
[void]$icon.Children.Add($logo)
|
||||
}
|
||||
[void]$contentPanel.Children.Add($icon)
|
||||
@@ -90,6 +64,8 @@ function Initialize-InstallAppEntry {
|
||||
$appName = New-Object Windows.Controls.TextBlock
|
||||
$appName.Style = $sync.Form.Resources.AppEntryNameStyle
|
||||
$appName.Text = $app.content
|
||||
|
||||
# Add FOSS label after the name if FOSS
|
||||
[void]$contentPanel.Children.Add($appName)
|
||||
$checkBox.Content = $contentPanel
|
||||
|
||||
@@ -110,6 +86,7 @@ function Initialize-InstallAppEntry {
|
||||
|
||||
[void]$entryLayout.Children.Add($fossBadge)
|
||||
}
|
||||
|
||||
$border.Child = $entryLayout
|
||||
if ($sync.selectedApps -contains $appKey) {
|
||||
$checkBox.IsChecked = $true
|
||||
|
||||
@@ -16,14 +16,17 @@ function Initialize-InstallCategoryAppList {
|
||||
$Apps
|
||||
)
|
||||
|
||||
# Pre-group apps by category before creating WPF controls.
|
||||
# Pre-group apps by category before creating WPF controls. Lists, because appending to
|
||||
# an array copies it and there are several hundred apps.
|
||||
$appsByCategory = @{}
|
||||
# Indexed, not dynamic member, lookup: the latter goes through the PSObject adapter and
|
||||
# costs about seventy times as much per app.
|
||||
foreach ($appKey in $Apps.Keys) {
|
||||
$category = $Apps.$appKey.Category
|
||||
$category = $Apps[$appKey].Category
|
||||
if (-not $appsByCategory.ContainsKey($category)) {
|
||||
$appsByCategory[$category] = @()
|
||||
$appsByCategory[$category] = [System.Collections.Generic.List[string]]::new()
|
||||
}
|
||||
$appsByCategory[$category] += $appKey
|
||||
$appsByCategory[$category].Add($appKey)
|
||||
}
|
||||
$sync.InstallAppRenderQueue = [System.Collections.Queue]::new()
|
||||
|
||||
|
||||
@@ -0,0 +1,35 @@
|
||||
function Initialize-WinUtilInstallTabControls {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Wires the Install tab controls that are generated from config rather than declared
|
||||
in XAML
|
||||
|
||||
.DESCRIPTION
|
||||
The package manager radio buttons and the install action buttons are created by
|
||||
Invoke-WPFUIElements, so they do not exist until the Install tab is built. Setting
|
||||
them up anywhere other than immediately after that build makes the code depend on
|
||||
when the tab happens to be created.
|
||||
#>
|
||||
|
||||
if ($sync.ChocoRadioButton) {
|
||||
$sync.ChocoRadioButton.Add_Checked({
|
||||
$sync.preferences.packagemanager = "Choco"
|
||||
})
|
||||
}
|
||||
if ($sync.WingetRadioButton) {
|
||||
$sync.WingetRadioButton.Add_Checked({
|
||||
$sync.preferences.packagemanager = "Winget"
|
||||
})
|
||||
}
|
||||
|
||||
switch ($sync.preferences.packagemanager) {
|
||||
"Choco" { if ($sync.ChocoRadioButton) { $sync.ChocoRadioButton.IsChecked = $true }; break }
|
||||
"Winget" { if ($sync.WingetRadioButton) { $sync.WingetRadioButton.IsChecked = $true }; break }
|
||||
}
|
||||
|
||||
if ($PARAM_OFFLINE) {
|
||||
foreach ($name in "WPFInstall", "WPFUninstall", "WPFInstallUpgrade", "WPFGetInstalled") {
|
||||
if ($sync.$name) { $sync.$name.IsEnabled = $false }
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,38 +1,34 @@
|
||||
function Initialize-WinUtilRunspacePool {
|
||||
if ($sync.runspace -and $sync.runspace.RunspacePoolStateInfo.State -eq [System.Management.Automation.Runspaces.RunspacePoolState]::Opened) {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Opens the shared worker pool that Start-WinUtilJob runs job bodies in
|
||||
#>
|
||||
|
||||
$poolLock = Get-WinUtilRunspacePoolLock
|
||||
[System.Threading.Monitor]::Enter($poolLock)
|
||||
try {
|
||||
if ($sync.runspace -and $sync.runspace.RunspacePoolStateInfo.State -eq [System.Management.Automation.Runspaces.RunspacePoolState]::Opened) {
|
||||
return $sync.runspace
|
||||
}
|
||||
|
||||
if ($sync.runspace) {
|
||||
# A replacement, not a shutdown
|
||||
Close-WinUtilRunspacePool -Recycle
|
||||
}
|
||||
|
||||
# Set the maximum number of threads for the RunspacePool to the number of threads on the machine.
|
||||
$maxthreads = [Math]::Max([int]$env:NUMBER_OF_PROCESSORS, 1)
|
||||
|
||||
$sync.runspace = [runspacefactory]::CreateRunspacePool(
|
||||
1, # Minimum thread count
|
||||
$maxthreads, # Maximum thread count
|
||||
(New-WinUtilSessionState), # Initial session state
|
||||
$Host # Machine to create runspaces on
|
||||
)
|
||||
|
||||
$sync.runspace.Open()
|
||||
return $sync.runspace
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($poolLock)
|
||||
}
|
||||
|
||||
if ($sync.runspace) {
|
||||
Close-WinUtilRunspacePool
|
||||
}
|
||||
|
||||
# Set the maximum number of threads for the RunspacePool to the number of threads on the machine.
|
||||
$maxthreads = [Math]::Max([int]$env:NUMBER_OF_PROCESSORS, 1)
|
||||
|
||||
# Create a new session state for parsing variables into our runspace.
|
||||
$hashVars = New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList 'sync', $sync, $null
|
||||
$offlineVar = New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList 'PARAM_OFFLINE', $PARAM_OFFLINE, $null
|
||||
$initialSessionState = [System.Management.Automation.Runspaces.InitialSessionState]::CreateDefault()
|
||||
|
||||
$initialSessionState.Variables.Add($hashVars)
|
||||
$initialSessionState.Variables.Add($offlineVar)
|
||||
|
||||
# Get every WinUtil/WPF function and add it to the session state.
|
||||
$functions = Get-ChildItem function:\ | Where-Object { $_.Name -imatch 'winutil|WPF' }
|
||||
foreach ($function in $functions) {
|
||||
$functionDefinition = Get-Content function:\$($function.Name)
|
||||
$functionEntry = New-Object System.Management.Automation.Runspaces.SessionStateFunctionEntry -ArgumentList $function.Name, $functionDefinition
|
||||
$initialSessionState.Commands.Add($functionEntry)
|
||||
}
|
||||
|
||||
$sync.runspace = [runspacefactory]::CreateRunspacePool(
|
||||
1, # Minimum thread count
|
||||
$maxthreads, # Maximum thread count
|
||||
$initialSessionState, # Initial session state
|
||||
$Host # Machine to create runspaces on
|
||||
)
|
||||
|
||||
$sync.runspace.Open()
|
||||
return $sync.runspace
|
||||
}
|
||||
|
||||
@@ -1,7 +1,11 @@
|
||||
function Initialize-WinUtilTabContent {
|
||||
param(
|
||||
[Parameter(Mandatory = $true)]
|
||||
[string]$TabName
|
||||
[string]$TabName,
|
||||
|
||||
# Build in batches, letting the interface answer in between. Used by the warmup, which
|
||||
# nobody is waiting on. A tab the user just clicked is built in one go.
|
||||
[switch]$Yield
|
||||
)
|
||||
|
||||
if ($null -eq $sync.InitializedTabs) {
|
||||
@@ -12,30 +16,42 @@ function Initialize-WinUtilTabContent {
|
||||
return
|
||||
}
|
||||
|
||||
switch ($TabName) {
|
||||
"Install" {
|
||||
Initialize-WPFUI -targetGridName "appscategory"
|
||||
|
||||
Initialize-WPFUI -targetGridName "appspanel"
|
||||
}
|
||||
"Tweaks" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.tweaks -targetGridName "tweakspanel" -columncount 2
|
||||
}
|
||||
"Config" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.feature -targetGridName "featurespanel" -columncount 2
|
||||
}
|
||||
"AppX" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.appx -targetGridName "appxpanel" -columncount 2
|
||||
}
|
||||
"Win11ISO" {
|
||||
if ($sync.Form -and $sync.Form.Dispatcher) {
|
||||
$sync.Form.Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{ Invoke-WinUtilISOCheckExistingWork }) | Out-Null
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
# Claimed before building, not after: a yielding build lets a click through, and that click
|
||||
# would otherwise start building the same tab a second time.
|
||||
$sync.InitializedTabs[$TabName] = $true
|
||||
|
||||
# Sync freshly built controls to any selections already in $sync.selected* (import/preset).
|
||||
Reset-WPFCheckBoxes -doToggles $true
|
||||
try {
|
||||
switch ($TabName) {
|
||||
"Install" {
|
||||
Measure-WinUtilStep -Scope "UI" -Name "Install tab: category area" -ScriptBlock {
|
||||
Initialize-WPFUI -targetGridName "appscategory"
|
||||
}
|
||||
Measure-WinUtilStep -Scope "UI" -Name "Install tab: app area" -ScriptBlock {
|
||||
Initialize-WPFUI -targetGridName "appspanel"
|
||||
}
|
||||
Initialize-WinUtilInstallTabControls
|
||||
}
|
||||
"Tweaks" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.tweaks -targetGridName "tweakspanel" -columncount 2 -Yield:$Yield
|
||||
}
|
||||
"Config" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.feature -targetGridName "featurespanel" -columncount 2 -Yield:$Yield
|
||||
}
|
||||
"AppX" {
|
||||
Invoke-WPFUIElements -configVariable $sync.configs.appx -targetGridName "appxpanel" -columncount 2 -Yield:$Yield
|
||||
}
|
||||
"Win11ISO" {
|
||||
if (Test-WinUtilUIAlive) {
|
||||
$sync.Form.Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{ Invoke-WinUtilISOCheckExistingWork }) | Out-Null
|
||||
}
|
||||
}
|
||||
}
|
||||
# Controls built just now start unchecked, so anything already chosen by an import or a
|
||||
# preset has to be applied to them once they exist
|
||||
Reset-WPFCheckBoxes -doToggles $true
|
||||
} catch {
|
||||
# A half built tab must be allowed to rebuild rather than staying empty forever
|
||||
$sync.InitializedTabs[$TabName] = $false
|
||||
throw
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,15 +4,30 @@ function Initialize-WinUtilTaskbarOverlayAssets {
|
||||
[bool]$IncludeStatusAssets = $true
|
||||
)
|
||||
|
||||
if ($IncludeLogo -and -not $sync["logorender"]) {
|
||||
$sync["logorender"] = (Invoke-WinUtilAssets -Type "Logo" -Size 90 -Render)
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
if ($null -eq $sync.AssetRenderLock) {
|
||||
$sync.AssetRenderLock = [object]::new()
|
||||
}
|
||||
$assetRenderLock = $sync.AssetRenderLock
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
|
||||
if ($IncludeStatusAssets -and -not $sync["checkmarkrender"]) {
|
||||
$sync["checkmarkrender"] = (Invoke-WinUtilAssets -Type "checkmark" -Size 512 -Render)
|
||||
}
|
||||
[System.Threading.Monitor]::Enter($assetRenderLock)
|
||||
try {
|
||||
if ($IncludeLogo -and -not $sync["logorender"]) {
|
||||
$sync["logorender"] = (Invoke-WinUtilAssets -Type "Logo" -Size 90 -Render)
|
||||
}
|
||||
|
||||
if ($IncludeStatusAssets -and -not $sync["warningrender"]) {
|
||||
$sync["warningrender"] = (Invoke-WinUtilAssets -Type "warning" -Size 512 -Render)
|
||||
if ($IncludeStatusAssets -and -not $sync["checkmarkrender"]) {
|
||||
$sync["checkmarkrender"] = (Invoke-WinUtilAssets -Type "checkmark" -Size 512 -Render)
|
||||
}
|
||||
|
||||
if ($IncludeStatusAssets -and -not $sync["warningrender"]) {
|
||||
$sync["warningrender"] = (Invoke-WinUtilAssets -Type "warning" -Size 512 -Render)
|
||||
}
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($assetRenderLock)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -58,7 +58,14 @@ function Install-WinUtilAPPX {
|
||||
$manifestPath = ($manifestOutput | Select-Object -Last 1).ToString().Trim()
|
||||
if (-not [string]::IsNullOrWhiteSpace($manifestPath)) {
|
||||
Write-WinUtilLog -Component "AppX" -Message "Registered local AppX manifest for $Name`: $manifestPath"
|
||||
return
|
||||
return [pscustomobject]@{
|
||||
Package = $Name
|
||||
Manager = "appx"
|
||||
Action = "Install"
|
||||
ExitCode = 0
|
||||
Outcome = "Succeeded"
|
||||
Detail = "registered local manifest"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -70,10 +77,12 @@ function Install-WinUtilAPPX {
|
||||
if ([string]::IsNullOrWhiteSpace($StoreId)) {
|
||||
$errorMessage = "Unable to install $Name because no local manifest or Microsoft Store ID is available."
|
||||
Write-WinUtilLog -Level "ERROR" -Component "AppX" -Message $errorMessage
|
||||
throw $errorMessage
|
||||
$exception = [System.InvalidOperationException]::new($errorMessage)
|
||||
$exception.Data["WinUtilErrorReported"] = $true
|
||||
throw $exception
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "AppX" -Message "No usable local manifest found for $Name. Installing Microsoft Store product $StoreId."
|
||||
Install-WinUtilWinget
|
||||
$null = Install-WinUtilWinget
|
||||
Install-WinUtilProgramWinget -Action Install -Programs @("msstore:$StoreId")
|
||||
}
|
||||
|
||||
@@ -1,7 +1,39 @@
|
||||
function Install-WinUtilChoco {
|
||||
if (-not (Get-Command -Name choco)) {
|
||||
Write-Host "Chocolatey is not installed. Installing now..."
|
||||
$installScript = Invoke-WebRequest -Uri https://community.chocolatey.org/install.ps1 -UseBasicParsing
|
||||
Invoke-Command -ScriptBlock ([scriptblock]::Create($installScript.Content))
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Installs Chocolatey if it is not already present
|
||||
#>
|
||||
|
||||
if (Get-Command -Name choco -ErrorAction SilentlyContinue) {
|
||||
return
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "Package" -Message "Chocolatey is not installed, installing it now."
|
||||
Step-WinUtilJob -Status "Installing Chocolatey" -State "Indeterminate"
|
||||
|
||||
# Windows PowerShell 5.1 can negotiate a protocol the site refuses, which the official
|
||||
# bootstrap sets explicitly for the same reason
|
||||
[System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor [System.Net.SecurityProtocolType]::Tls12
|
||||
$installScript = Invoke-WebRequest -Uri https://community.chocolatey.org/install.ps1 -UseBasicParsing -TimeoutSec 60
|
||||
Invoke-Command -ScriptBlock ([scriptblock]::Create($installScript.Content))
|
||||
|
||||
# The installer extends PATH for new processes, which this one is not. Appended rather than
|
||||
# replaced: overwriting drops whatever this process added earlier in the session, and a
|
||||
# later step looking for that tool would no longer find it.
|
||||
$existing = $env:PATH -split ';' | Where-Object { $_ }
|
||||
$persisted = @(
|
||||
[System.Environment]::GetEnvironmentVariable("Path", "Machine")
|
||||
[System.Environment]::GetEnvironmentVariable("Path", "User")
|
||||
) -join ';' -split ';' | Where-Object { $_ }
|
||||
|
||||
$missing = $persisted | Where-Object { $existing -notcontains $_ }
|
||||
if ($missing) {
|
||||
$env:PATH = (@($existing) + @($missing)) -join ';'
|
||||
}
|
||||
|
||||
if (-not (Get-Command -Name choco -ErrorAction SilentlyContinue)) {
|
||||
throw "Chocolatey was installed but choco is still not on PATH."
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "Package" -Message "Chocolatey installed."
|
||||
}
|
||||
|
||||
@@ -1,20 +1,116 @@
|
||||
function Install-WinUtilProgramChoco {
|
||||
<#
|
||||
|
||||
.SYNOPSIS
|
||||
Installs, upgrades or uninstalls packages with Chocolatey and reports each outcome
|
||||
|
||||
.DESCRIPTION
|
||||
One package per call to choco, so the progress bar moves through the list and a failure
|
||||
names the package that failed rather than the whole batch. Choco's own output goes to the
|
||||
log instead of the console, the way the WinGet path reports.
|
||||
|
||||
.PARAMETER Action
|
||||
Install, Upgrade or Uninstall.
|
||||
|
||||
.PARAMETER Programs
|
||||
The package names. For Upgrade, the single entry "all" upgrades everything.
|
||||
|
||||
.PARAMETER ProgressBase
|
||||
Where this call starts within the job's overall progress bar.
|
||||
|
||||
.PARAMETER ProgressSpan
|
||||
How much of the overall bar these packages account for. Zero reports nothing.
|
||||
|
||||
#>
|
||||
param (
|
||||
[Parameter(Mandatory=$true)]
|
||||
[ValidateSet("Install", "Uninstall")]
|
||||
[ValidateSet("Install", "Uninstall", "Upgrade")]
|
||||
[string]$Action,
|
||||
|
||||
[Parameter(Mandatory=$true)]
|
||||
[string[]]$Programs
|
||||
[string[]]$Programs,
|
||||
|
||||
[int]$ProgressBase = 0,
|
||||
|
||||
[int]$ProgressSpan = 0
|
||||
)
|
||||
|
||||
if ($Action -eq 'Install') {
|
||||
$arguments = "install $Programs -y"
|
||||
} else {
|
||||
$arguments = "uninstall $Programs -y"
|
||||
# Chocolatey reports "nothing needed doing" and "it worked, now reboot" through exit codes
|
||||
# rather than as failures
|
||||
$rebootCodes = @{
|
||||
1641 = "installed, the installer started a restart"
|
||||
3010 = "installed, a restart is needed to finish"
|
||||
}
|
||||
$nothingToDo = @{
|
||||
2 = "nothing to do"
|
||||
}
|
||||
$verb = $Action.ToLowerInvariant()
|
||||
$chocoAvailable = $null -ne (Get-Command choco -ErrorAction SilentlyContinue)
|
||||
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action choco package(s): $($Programs -join ', ')"
|
||||
$process = Start-Process -FilePath choco -ArgumentList $arguments -NoNewWindow -Wait -PassThru
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action choco package(s) completed: $($Programs -join ', ') (exit code: $($process.ExitCode))"
|
||||
$packages = @($Programs | Where-Object { -not [string]::IsNullOrWhiteSpace($_) })
|
||||
$total = $packages.Count
|
||||
$index = 0
|
||||
|
||||
foreach ($program in $packages) {
|
||||
$index++
|
||||
if ($ProgressSpan -gt 0 -and $total -gt 0) {
|
||||
$percent = $ProgressBase + [int]((($index - 1) / $total) * $ProgressSpan)
|
||||
Step-WinUtilJob -Status "$Action $program ($index/$total)" -Percent $percent
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action choco package: $program"
|
||||
|
||||
# --no-progress stops choco redrawing a percentage line that only makes sense on a
|
||||
# console nobody is watching
|
||||
$arguments = @($verb, $program, "-y", "--no-progress")
|
||||
# Each worker runspace has its own global scope. Reset the native-command result there so
|
||||
# command-not-found cannot inherit a successful code from earlier work in the same pool.
|
||||
$global:LASTEXITCODE = $null
|
||||
if (-not $chocoAvailable) {
|
||||
$output = "Chocolatey is not installed or is not available on PATH."
|
||||
$exitCode = -1
|
||||
} else {
|
||||
$output = & choco @arguments 2>&1
|
||||
$exitCode = if ($null -eq $global:LASTEXITCODE) { -1 } else { [int]$global:LASTEXITCODE }
|
||||
}
|
||||
|
||||
if ($exitCode -eq 0) {
|
||||
$outcome = "Succeeded"
|
||||
$detail = "exit code 0"
|
||||
} elseif ($rebootCodes.ContainsKey($exitCode)) {
|
||||
$outcome = "Succeeded"
|
||||
$detail = $rebootCodes[$exitCode]
|
||||
} elseif ($nothingToDo.ContainsKey($exitCode)) {
|
||||
$outcome = "Skipped"
|
||||
$detail = $nothingToDo[$exitCode]
|
||||
} else {
|
||||
$outcome = "Failed"
|
||||
$detail = if ($exitCode -eq -1) { "Chocolatey command did not start" } else { "exit code $exitCode" }
|
||||
}
|
||||
|
||||
$level = if ($outcome -eq "Failed") { "ERROR" } else { "INFO" }
|
||||
Write-WinUtilLog -Level $level -Component "Package" -Message "$Action choco package $($outcome.ToLowerInvariant()): $program ($detail)"
|
||||
|
||||
if ($outcome -eq "Failed") {
|
||||
# The reason is somewhere in choco's output, and without it the log says only that
|
||||
# a number came back
|
||||
foreach ($line in @($output | Select-Object -Last 15)) {
|
||||
$text = ([string]$line).Trim()
|
||||
if ($text) { Write-WinUtilLog -Level "WARN" -Component "Package" -Detail -Message $text }
|
||||
}
|
||||
}
|
||||
|
||||
if ($ProgressSpan -gt 0 -and $total -gt 0) {
|
||||
Step-WinUtilJob -Status "$Action $program ($index/$total)" -Percent ($ProgressBase + [int](($index / $total) * $ProgressSpan))
|
||||
}
|
||||
|
||||
[pscustomobject]@{
|
||||
Package = $program
|
||||
Manager = "choco"
|
||||
Action = $Action
|
||||
ExitCode = $exitCode
|
||||
Outcome = $outcome
|
||||
Detail = $detail
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,32 +1,116 @@
|
||||
Function Install-WinUtilProgramWinget {
|
||||
<#
|
||||
|
||||
.SYNOPSIS
|
||||
Installs or uninstalls packages with WinGet and reports the outcome of each one
|
||||
|
||||
.DESCRIPTION
|
||||
Emits one result object per package so the caller can tell what actually happened
|
||||
rather than assuming the run succeeded.
|
||||
|
||||
Runs one winget command per package so a failure names the package that failed rather
|
||||
than the whole batch. Progress moves per package: winget hides its own progress bar once
|
||||
its output is redirected, so there is nothing to report from inside a single install.
|
||||
|
||||
#>
|
||||
param (
|
||||
[Parameter(Mandatory=$true)]
|
||||
[ValidateSet("Install", "Uninstall")]
|
||||
[ValidateSet("Install", "Uninstall", "Upgrade")]
|
||||
[string]$Action,
|
||||
|
||||
[Parameter(Mandatory=$true)]
|
||||
[string[]]$Programs
|
||||
)
|
||||
|
||||
# APPINSTALLER_CLI_ERROR_ADMIN_CONTEXT_ACTION_PROHIBITED. WinGet refuses to act on a package
|
||||
# that was installed in user scope while it is running elevated, and WinUtil is always
|
||||
# elevated, so every per-user app answers this and nothing happens.
|
||||
$adminContextProhibited = -1978335107
|
||||
|
||||
# WinGet reports "there was nothing to do" through the exit code rather than as success
|
||||
$nothingToDo = @{
|
||||
-1978335135 = "already installed"
|
||||
-1978335189 = "no applicable update"
|
||||
}
|
||||
# The installer worked and wants a restart to finish. Windows reports that as its own exit
|
||||
# code rather than as zero, and treating it as a failure marks working installs as broken.
|
||||
$rebootExitCodes = @{
|
||||
3010 = "installed, a restart is needed to finish"
|
||||
1641 = "installed, the installer started a restart"
|
||||
# WinGet's own equivalents. -1978334966 is deliberately absent: it means a reboot is
|
||||
# required before the install can proceed, which is not a completed install.
|
||||
-1978334967 = "installed, a restart is needed to finish"
|
||||
-1978334965 = "installed, the installer started a restart"
|
||||
}
|
||||
|
||||
foreach ($program in $Programs) {
|
||||
if ([string]::IsNullOrWhiteSpace($program) -or $program -eq "na") {
|
||||
continue
|
||||
}
|
||||
|
||||
$source = "winget"
|
||||
if ($program.StartsWith("msstore:", [System.StringComparison]::OrdinalIgnoreCase)) {
|
||||
$upgradeAll = $Action -eq "Upgrade" -and $program -eq "all"
|
||||
$source = if ($upgradeAll) { "all configured sources" } else { "winget" }
|
||||
if (-not $upgradeAll -and $program.StartsWith("msstore:", [System.StringComparison]::OrdinalIgnoreCase)) {
|
||||
$source = "msstore"
|
||||
$program = $program.Substring("msstore:".Length)
|
||||
}
|
||||
|
||||
if ($Action -eq 'Install') {
|
||||
$arguments = @("install", "--id", $program, "--accept-package-agreements", "--accept-source-agreements", "--source", $source, "--silent")
|
||||
} else {
|
||||
$arguments = @("uninstall", "--id", $program, "--source", $source, "--silent")
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action winget package: $program (source: $source)"
|
||||
|
||||
$outcome = "Failed"
|
||||
$detail = "no result"
|
||||
$exitCode = -1
|
||||
|
||||
$arguments = switch ($Action) {
|
||||
"Uninstall" { @("uninstall", "--id", $program, "--source", $source, "--silent") }
|
||||
# --include-unknown because the scan that found these ran with it: without it winget
|
||||
# refuses every package whose installed version it could not read
|
||||
"Upgrade" {
|
||||
if ($upgradeAll) {
|
||||
@("upgrade", "--all", "--accept-package-agreements", "--accept-source-agreements", "--include-unknown", "--silent")
|
||||
} else {
|
||||
@("upgrade", "--id", $program, "--accept-package-agreements", "--accept-source-agreements", "--source", $source, "--include-unknown", "--silent")
|
||||
}
|
||||
}
|
||||
default { @("install", "--id", $program, "--accept-package-agreements", "--accept-source-agreements", "--source", $source, "--silent") }
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action winget package: $program (source: $source)"
|
||||
$process = Start-Process -FilePath winget -ArgumentList $arguments -NoNewWindow -Wait -PassThru
|
||||
Write-WinUtilLog -Component "Package" -Message "$Action winget package completed: $program (exit code: $($process.ExitCode))"
|
||||
$exitCode = $process.ExitCode
|
||||
|
||||
if ($exitCode -eq 0) {
|
||||
$outcome = "Succeeded"
|
||||
$detail = "exit code 0"
|
||||
} elseif ($rebootExitCodes.ContainsKey($exitCode)) {
|
||||
$outcome = "Succeeded"
|
||||
$detail = $rebootExitCodes[$exitCode]
|
||||
} elseif ($nothingToDo.ContainsKey($exitCode)) {
|
||||
$outcome = "Skipped"
|
||||
$detail = $nothingToDo[$exitCode]
|
||||
} elseif ($exitCode -eq $adminContextProhibited) {
|
||||
$outcome = "Skipped"
|
||||
$detail = switch ($Action) {
|
||||
"Install" { "already installed for the current user; elevated WinUtil cannot update it" }
|
||||
"Upgrade" { "not upgraded; installed for the current user and elevated WinUtil cannot modify it" }
|
||||
"Uninstall" { "remains installed for the current user; elevated WinUtil cannot uninstall it" }
|
||||
}
|
||||
} else {
|
||||
$outcome = "Failed"
|
||||
# The client module reports the same failure as a bare HRESULT, so the hex form and
|
||||
# Microsoft's own list serve both paths
|
||||
$detail = "WinGet reported 0x{0:X8}. See https://learn.microsoft.com/windows/package-manager/winget/returnCodes" -f $exitCode
|
||||
}
|
||||
|
||||
$level = if ($outcome -eq "Failed") { "ERROR" } else { "INFO" }
|
||||
Write-WinUtilLog -Level $level -Component "Package" -Message "$Action winget package $($outcome.ToLowerInvariant()): $program ($detail)"
|
||||
|
||||
[pscustomobject]@{
|
||||
Package = $program
|
||||
Manager = "winget"
|
||||
Action = $Action
|
||||
ExitCode = $exitCode
|
||||
Outcome = $outcome
|
||||
Detail = $detail
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -7,11 +7,21 @@ function Install-WinUtilWinget {
|
||||
.DESCRIPTION
|
||||
installs winGet if needed
|
||||
#>
|
||||
if ((Test-WinUtilPackageManager -winget) -eq "installed") {
|
||||
param(
|
||||
[switch]$Force
|
||||
)
|
||||
|
||||
# The repair action needs Repair-WinGetPackageManager to run even when winget is detected,
|
||||
# which is the case a broken installation presents
|
||||
if (-not $Force -and (Test-WinUtilPackageManager -winget) -eq "installed") {
|
||||
return
|
||||
}
|
||||
|
||||
Write-Host "WinGet is not installed. Installing now..." -ForegroundColor Red
|
||||
if ($Force) {
|
||||
Write-Host "Repairing the WinGet installation..." -ForegroundColor Yellow
|
||||
} else {
|
||||
Write-Host "WinGet is not installed. Installing now..." -ForegroundColor Red
|
||||
}
|
||||
|
||||
Install-PackageProvider -Name NuGet -Force
|
||||
Install-Module -Name Microsoft.WinGet.Client -Force
|
||||
|
||||
@@ -7,7 +7,7 @@ function Invoke-WinUtilAssets {
|
||||
|
||||
if ($render -and $null -ne $sync) {
|
||||
if ($null -eq $sync.RenderedAssetCache) {
|
||||
$sync.RenderedAssetCache = @{}
|
||||
$sync.RenderedAssetCache = [Hashtable]::Synchronized(@{})
|
||||
}
|
||||
|
||||
$cacheKey = "$(([string]$type).ToLowerInvariant())|$Size"
|
||||
|
||||
@@ -0,0 +1,116 @@
|
||||
function Invoke-WinUtilCloseRequest {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Asks what to do about work that is still running when the window is closed
|
||||
|
||||
.DESCRIPTION
|
||||
A half finished install or tweak run is not ended without asking. Either it finishes
|
||||
without the window, reporting to the console and then exiting, or it is stopped and
|
||||
everything closes now.
|
||||
|
||||
.PARAMETER RunningJob
|
||||
The name of the job in flight, so the question names what is at stake.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$RunningJob
|
||||
)
|
||||
|
||||
# The question carries the meaning rather than naming buttons: Windows labels them in its own
|
||||
# language, so "Yes" in the text would not match a button reading "Ja".
|
||||
$answer = Show-WinUtilMessage -Button "YesNoCancel" -Icon "Warning" -Title "$RunningJob is still running" -Message @"
|
||||
$RunningJob has not finished yet.
|
||||
|
||||
Close the window and let it finish in the console?
|
||||
|
||||
WinUtil will exit on its own once it is done. If you do not, it will be
|
||||
stopped and everything closes now. Cancel keeps WinUtil open.
|
||||
"@
|
||||
|
||||
switch ("$answer") {
|
||||
"Yes" {
|
||||
Write-WinUtilLog -Component "UI" -Message "Close requested: closing the window, $RunningJob continues in the console."
|
||||
$sync.FinishInConsole = $true
|
||||
$sync.ForceClose = $true
|
||||
|
||||
Write-Host ""
|
||||
Write-Host "WinUtil's window is closed. $RunningJob is still running here, and this window will close when it finishes." -ForegroundColor Cyan
|
||||
Write-Host ""
|
||||
|
||||
# Posted rather than closed from inside the handler that is already unwinding
|
||||
Request-WinUtilWindowClose
|
||||
}
|
||||
"No" {
|
||||
Write-WinUtilLog -Component "UI" -Message "Close requested: stopping $RunningJob."
|
||||
Step-WinUtilJob -Status "Stopping $RunningJob" -State "Indeterminate"
|
||||
$sync.ForceClose = $true
|
||||
|
||||
# Close the window first. The main thread owns pool shutdown after ShowDialog
|
||||
# returns, so the worker can finish its UI-dispatching finally block before the UI
|
||||
# runspace is disposed. Waiting for it here would deadlock the dispatcher.
|
||||
Request-WinUtilWindowClose
|
||||
}
|
||||
default {
|
||||
Write-WinUtilLog -Component "UI" -Message "Close cancelled, $RunningJob is still running."
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function Request-WinUtilWindowClose {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Closes the window from outside the handler that is currently cancelling the close
|
||||
|
||||
#>
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
return
|
||||
}
|
||||
|
||||
$sync.Form.Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{
|
||||
$sync.Form.Close()
|
||||
}) | Out-Null
|
||||
}
|
||||
|
||||
function Wait-WinUtilRemainingWork {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Waits for work that outlived the window, reporting to the console
|
||||
|
||||
.DESCRIPTION
|
||||
Runs on the main thread once the interface has gone. The job is still on the worker
|
||||
pool and keeps logging, so this only waits and keeps the wait visible.
|
||||
|
||||
.PARAMETER TimeoutMinutes
|
||||
Upper bound, so a worker that never returns cannot keep the process alive.
|
||||
#>
|
||||
param(
|
||||
# Double rather than int: an int silently truncates a fractional value to zero, which
|
||||
# turns the bound into "do not wait at all"
|
||||
[double]$TimeoutMinutes = 120
|
||||
)
|
||||
|
||||
if (-not $sync.FinishInConsole -or -not $sync.ActiveJob) {
|
||||
return
|
||||
}
|
||||
|
||||
$job = $sync.ActiveJob
|
||||
Write-WinUtilLog -Component "UI" -Message "Window closed, waiting for $job to finish."
|
||||
Write-Host "Waiting for $job to finish..." -ForegroundColor Cyan
|
||||
|
||||
$clock = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
while ($sync.ActiveJob -and $clock.Elapsed.TotalMinutes -lt $TimeoutMinutes) {
|
||||
Start-Sleep -Milliseconds 250
|
||||
}
|
||||
|
||||
# The job's last progress line is still open, so anything after it needs a fresh line
|
||||
Complete-WinUtilConsoleProgress
|
||||
|
||||
if ($sync.ActiveJob) {
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "$job did not finish within $TimeoutMinutes minutes, exiting anyway."
|
||||
Write-Host "$job is taking longer than $TimeoutMinutes minutes. Exiting." -ForegroundColor Yellow
|
||||
return
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "UI" -Message "$job finished after the window closed, in $([int]$clock.Elapsed.TotalSeconds)s."
|
||||
Write-Host "$job finished. Closing." -ForegroundColor Green
|
||||
}
|
||||
@@ -117,7 +117,17 @@ Function Invoke-WinUtilCurrentSystem {
|
||||
|
||||
Foreach ($tweaks in $serviceKeys) {
|
||||
Foreach ($tweak in $tweaks) {
|
||||
$Service = Get-Service -Name $tweak.Name
|
||||
try {
|
||||
$Service = Get-Service -Name $tweak.Name -ErrorAction $readErrorAction
|
||||
} catch {
|
||||
if ($StopOnReadError -and $_.FullyQualifiedErrorId -like "NoServiceFoundForGivenName*") {
|
||||
# A removed optional service means this tweak is not applied; it does
|
||||
# not make the registry and service state for every other tweak unknown.
|
||||
$values += $False
|
||||
continue
|
||||
}
|
||||
throw
|
||||
}
|
||||
|
||||
if ($Service) {
|
||||
$actualValue = $Service.StartType
|
||||
@@ -125,6 +135,8 @@ Function Invoke-WinUtilCurrentSystem {
|
||||
if ($expectedValue -ne $actualValue) {
|
||||
$values += $False
|
||||
}
|
||||
} elseif ($StopOnReadError) {
|
||||
$values += $False
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -8,7 +8,9 @@ function Invoke-WinUtilExplorerUpdate {
|
||||
)
|
||||
|
||||
if ($action -eq "refresh") {
|
||||
Invoke-WPFRunspace -ScriptBlock {
|
||||
# The handle is of no use to the caller, and leaving it in the pipeline puts it into
|
||||
# whatever result the calling workflow returns
|
||||
$null = Invoke-WPFRunspace -ScriptBlock {
|
||||
# Define the Win32 type only if it doesn't exist
|
||||
if (-not ([System.Management.Automation.PSTypeName]'Win32').Type) {
|
||||
Add-Type -TypeDefinition @"
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
@@ -21,25 +21,34 @@ function Invoke-WinUtilISORefreshUSBDrives {
|
||||
$sync["Win11ISOUSBDisks"] = $removable
|
||||
}
|
||||
|
||||
function Get-WinUtilFreeDriveLetter {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Returns the first unused drive letter between D and Z, or $null when there is none.
|
||||
#>
|
||||
|
||||
$used = (Get-PSDrive -PSProvider FileSystem).Name
|
||||
foreach ($c in [char[]](68..90)) {
|
||||
if ($used -notcontains [string]$c) { return $c }
|
||||
}
|
||||
return $null
|
||||
}
|
||||
|
||||
function Invoke-WinUtilISOWriteUSB {
|
||||
$contentsDir = $sync["Win11ISOContentsDir"]
|
||||
$usbDisks = $sync["Win11ISOUSBDisks"]
|
||||
|
||||
if (-not $contentsDir -or -not (Test-Path $contentsDir)) {
|
||||
[System.Windows.MessageBox]::Show("No modified ISO content found. Please complete Steps 1-3 first.", "Not Ready", "OK", "Warning")
|
||||
Show-WinUtilMessage -Message "No modified ISO content found. Please complete Steps 1-3 first." -Title "Not Ready" -Button "OK" -Icon "Warning" | Out-Null
|
||||
return
|
||||
}
|
||||
|
||||
$installWim = Join-Path $contentsDir "sources\install.wim"
|
||||
$installEsd = Join-Path $contentsDir "sources\install.esd"
|
||||
if (Test-Path $installEsd) {
|
||||
$installEsdFile = Get-Item $installEsd
|
||||
$esdSizeBytes = $installEsdFile.Length
|
||||
$esdSizeMB = [math]::Ceiling($esdSizeBytes / 1MB)
|
||||
$esdSizeBytes = (Get-Item $installEsd).Length
|
||||
if ($esdSizeBytes -ge 4GB) {
|
||||
[System.Windows.MessageBox]::Show(
|
||||
"This ISO uses an install.esd file that is $esdSizeMB MB. WinUtil's FAT32 USB format cannot store files larger than 4 GB.`n`nExport an ISO instead or use media with install.wim.",
|
||||
"USB Creation Not Supported", "OK", "Warning")
|
||||
$esdSizeMB = [math]::Ceiling($esdSizeBytes / 1MB)
|
||||
Show-WinUtilMessage -Message "This ISO uses an install.esd file that is $esdSizeMB MB. WinUtil's FAT32 USB format cannot store files larger than 4 GB.`n`nExport an ISO instead or use media with install.wim." -Title "USB Creation Not Supported" -Button "OK" -Icon "Warning" | Out-Null
|
||||
return
|
||||
}
|
||||
}
|
||||
@@ -58,95 +67,59 @@ function Invoke-WinUtilISOWriteUSB {
|
||||
}
|
||||
|
||||
if (-not $targetDisk) {
|
||||
[System.Windows.MessageBox]::Show("Please select a USB drive from the dropdown.", "No Drive Selected", "OK", "Warning")
|
||||
Show-WinUtilMessage -Message "Please select a USB drive from the dropdown." -Title "No Drive Selected" -Button "OK" -Icon "Warning" | Out-Null
|
||||
return
|
||||
}
|
||||
|
||||
$diskNum = $targetDisk.Number
|
||||
$sizeGB = [math]::Round($targetDisk.Size / 1GB, 1)
|
||||
|
||||
$confirm = [System.Windows.MessageBox]::Show(
|
||||
"ALL data on Disk $diskNum ($($targetDisk.FriendlyName), $sizeGB GB) will be PERMANENTLY ERASED.`n`nAre you sure you want to continue?",
|
||||
"Confirm USB Erase", "YesNo", "Warning")
|
||||
$diskNum = $targetDisk.Number
|
||||
$sizeGB = [math]::Round($targetDisk.Size / 1GB, 1)
|
||||
|
||||
$confirm = Show-WinUtilMessage -Message "ALL data on Disk $diskNum ($($targetDisk.FriendlyName), $sizeGB GB) will be PERMANENTLY ERASED.`n`nAre you sure you want to continue?" -Title "Confirm USB Erase" -Button "YesNo" -Icon "Warning"
|
||||
if ($confirm -ne "Yes") {
|
||||
Write-WinUtilISOLog "USB write cancelled by user."
|
||||
return
|
||||
}
|
||||
|
||||
$sync["WPFWin11ISOWriteUSBButton"].IsEnabled = $false
|
||||
$sync["Win11ISOProcessRunning"] = $true
|
||||
Write-WinUtilISOLog "Starting USB write to Disk $diskNum..."
|
||||
Start-WinUtilJob -Name "USB write" -Description "Writing USB drive" -Parameters @{
|
||||
DiskNumber = $diskNum
|
||||
ContentsDir = $contentsDir
|
||||
} -ScriptBlock {
|
||||
param($DiskNumber, $contentsDir)
|
||||
|
||||
$runspace = [Management.Automation.Runspaces.RunspaceFactory]::CreateRunspace()
|
||||
$runspace.ApartmentState = "STA"
|
||||
$runspace.ThreadOptions = "ReuseThread"
|
||||
$runspace.Open()
|
||||
$runspace.SessionStateProxy.SetVariable("sync", $sync)
|
||||
$runspace.SessionStateProxy.SetVariable("diskNum", $diskNum)
|
||||
$runspace.SessionStateProxy.SetVariable("contentsDir", $contentsDir)
|
||||
|
||||
$script = [Management.Automation.PowerShell]::Create()
|
||||
$script.Runspace = $runspace
|
||||
$script.AddScript({
|
||||
|
||||
function Log($msg) {
|
||||
$ts = (Get-Date).ToString("HH:mm:ss")
|
||||
$sync["WPFWin11ISOStatusLog"].Dispatcher.Invoke([action]{
|
||||
$sync["WPFWin11ISOStatusLog"].Text += "`n[$ts] $msg"
|
||||
$sync["WPFWin11ISOStatusLog"].CaretIndex = $sync["WPFWin11ISOStatusLog"].Text.Length
|
||||
$sync["WPFWin11ISOStatusLog"].ScrollToEnd()
|
||||
})
|
||||
}
|
||||
|
||||
function SetProgress($label, $pct) {
|
||||
$sync["WPFWin11ISOStatusLog"].Dispatcher.Invoke([action]{
|
||||
$sync["WPFTweaksProgressBar"].Visibility = "Visible"
|
||||
$sync["WPFTweaksProgressLabel"].Text = $label
|
||||
$sync["WPFTweaksProgressLabel"].ToolTip = $label
|
||||
$sync["WPFTweaksProgressValue"].Value = [Math]::Max($pct, 5)
|
||||
})
|
||||
}
|
||||
|
||||
function Get-FreeDriveLetter {
|
||||
$used = (Get-PSDrive -PSProvider FileSystem).Name
|
||||
foreach ($c in [char[]](68..90)) {
|
||||
if ($used -notcontains [string]$c) { return $c }
|
||||
}
|
||||
return $null
|
||||
}
|
||||
Invoke-WPFUIThread -ScriptBlock { $sync["WPFWin11ISOWriteUSBButton"].IsEnabled = $false }
|
||||
|
||||
try {
|
||||
SetProgress "Formatting USB drive..." 10
|
||||
Write-WinUtilISOLog "Starting USB write to Disk $DiskNumber..."
|
||||
Step-WinUtilJob -Status "Formatting USB drive..." -Percent 10
|
||||
|
||||
# Phase 1: Clean disk via diskpart (retry once if the drive is not yet ready)
|
||||
$dpFile1 = Join-Path $env:TEMP "winutil_diskpart_$(Get-Random).txt"
|
||||
"select disk $diskNum`nclean`nexit" | Set-Content -Path $dpFile1 -Encoding ASCII
|
||||
Log "Running diskpart clean on Disk $diskNum..."
|
||||
"select disk $DiskNumber`nclean`nexit" | Set-Content -Path $dpFile1 -Encoding ASCII
|
||||
Write-WinUtilISOLog "Running diskpart clean on Disk $DiskNumber..."
|
||||
$dpCleanOut = diskpart /s $dpFile1
|
||||
$dpCleanOut | Where-Object { $_ -match '\S' } | ForEach-Object { Log " diskpart: $_" }
|
||||
$dpCleanOut | Where-Object { $_ -match '\S' } | ForEach-Object { Write-WinUtilISOLog " diskpart: $_" }
|
||||
Remove-Item $dpFile1 -Force
|
||||
|
||||
if (($dpCleanOut -join ' ') -match 'device is not ready') {
|
||||
Log "Disk $diskNum was not ready; waiting 5 seconds and retrying clean..."
|
||||
Write-WinUtilISOLog "Disk $DiskNumber was not ready; waiting 5 seconds and retrying clean..."
|
||||
Start-Sleep -Seconds 5
|
||||
Update-Disk -Number $diskNum
|
||||
Update-Disk -Number $DiskNumber
|
||||
$dpFile1b = Join-Path $env:TEMP "winutil_diskpart_$(Get-Random).txt"
|
||||
"select disk $diskNum`nclean`nexit" | Set-Content -Path $dpFile1b -Encoding ASCII
|
||||
diskpart /s $dpFile1b | Where-Object { $_ -match '\S' } | ForEach-Object { Log " diskpart: $_" }
|
||||
"select disk $DiskNumber`nclean`nexit" | Set-Content -Path $dpFile1b -Encoding ASCII
|
||||
diskpart /s $dpFile1b | Where-Object { $_ -match '\S' } | ForEach-Object { Write-WinUtilISOLog " diskpart: $_" }
|
||||
Remove-Item $dpFile1b -Force
|
||||
}
|
||||
|
||||
# Phase 2: Initialize as GPT
|
||||
Start-Sleep -Seconds 2
|
||||
Update-Disk -Number $diskNum
|
||||
$diskObj = Get-Disk -Number $diskNum
|
||||
Update-Disk -Number $DiskNumber
|
||||
$diskObj = Get-Disk -Number $DiskNumber
|
||||
if ($diskObj.PartitionStyle -eq 'RAW') {
|
||||
Initialize-Disk -Number $diskNum -PartitionStyle GPT
|
||||
Log "Disk $diskNum initialized as GPT."
|
||||
Initialize-Disk -Number $DiskNumber -PartitionStyle GPT
|
||||
Write-WinUtilISOLog "Disk $DiskNumber initialized as GPT."
|
||||
} else {
|
||||
Set-Disk -Number $diskNum -PartitionStyle GPT
|
||||
Log "Disk $diskNum converted to GPT (was $($diskObj.PartitionStyle))."
|
||||
Set-Disk -Number $DiskNumber -PartitionStyle GPT
|
||||
Write-WinUtilISOLog "Disk $DiskNumber converted to GPT (was $($diskObj.PartitionStyle))."
|
||||
}
|
||||
|
||||
# Phase 3: Create FAT32 partition via diskpart, then format with Format-Volume
|
||||
@@ -154,64 +127,64 @@ function Invoke-WinUtilISOWriteUSB {
|
||||
$volLabel = "W11-" + (Get-Date).ToString('yyMMdd')
|
||||
$dpFile2 = Join-Path $env:TEMP "winutil_diskpart2_$(Get-Random).txt"
|
||||
$maxFat32PartitionMB = 32768
|
||||
$diskSizeMB = [int][Math]::Floor((Get-Disk -Number $diskNum).Size / 1MB)
|
||||
$diskSizeMB = [int][Math]::Floor((Get-Disk -Number $DiskNumber).Size / 1MB)
|
||||
$createPartitionCommand = "create partition primary"
|
||||
if ($diskSizeMB -gt $maxFat32PartitionMB) {
|
||||
$createPartitionCommand = "create partition primary size=$maxFat32PartitionMB"
|
||||
Log "Disk $diskNum is $diskSizeMB MB; creating FAT32 partition capped at $maxFat32PartitionMB MB (32 GB)."
|
||||
Write-WinUtilISOLog "Disk $DiskNumber is $diskSizeMB MB; creating FAT32 partition capped at $maxFat32PartitionMB MB (32 GB)."
|
||||
}
|
||||
|
||||
@(
|
||||
"select disk $diskNum"
|
||||
"select disk $DiskNumber"
|
||||
$createPartitionCommand
|
||||
"exit"
|
||||
) | Set-Content -Path $dpFile2 -Encoding ASCII
|
||||
Log "Creating partitions on Disk $diskNum..."
|
||||
diskpart /s $dpFile2 | Where-Object { $_ -match '\S' } | ForEach-Object { Log " diskpart: $_" }
|
||||
Write-WinUtilISOLog "Creating partitions on Disk $DiskNumber..."
|
||||
diskpart /s $dpFile2 | Where-Object { $_ -match '\S' } | ForEach-Object { Write-WinUtilISOLog " diskpart: $_" }
|
||||
Remove-Item $dpFile2 -Force
|
||||
|
||||
SetProgress "Formatting USB partition..." 25
|
||||
Step-WinUtilJob -Status "Formatting USB partition..." -Percent 25
|
||||
Start-Sleep -Seconds 3
|
||||
Update-Disk -Number $diskNum
|
||||
Update-Disk -Number $DiskNumber
|
||||
|
||||
$partitions = Get-Partition -DiskNumber $diskNum
|
||||
Log "Partitions on Disk $diskNum after creation: $($partitions.Count)"
|
||||
$partitions = Get-Partition -DiskNumber $DiskNumber
|
||||
Write-WinUtilISOLog "Partitions on Disk $DiskNumber after creation: $($partitions.Count)"
|
||||
foreach ($p in $partitions) {
|
||||
Log " Partition $($p.PartitionNumber) Type=$($p.Type) Letter=$($p.DriveLetter) Size=$([math]::Round($p.Size/1MB))MB"
|
||||
Write-WinUtilISOLog " Partition $($p.PartitionNumber) Type=$($p.Type) Letter=$($p.DriveLetter) Size=$([math]::Round($p.Size/1MB))MB"
|
||||
}
|
||||
|
||||
$winpePart = $partitions | Where-Object { $_.Type -eq "Basic" } | Select-Object -Last 1
|
||||
if (-not $winpePart) {
|
||||
throw "Could not find the Basic partition on Disk $diskNum after creation."
|
||||
throw "Could not find the Basic partition on Disk $DiskNumber after creation."
|
||||
}
|
||||
|
||||
# Format using Format-Volume (reliable on fresh drives; diskpart format fails
|
||||
# with 'no volume selected' when the partition has never been formatted before)
|
||||
Log "Formatting Partition $($winpePart.PartitionNumber) as FAT32 (label: $volLabel)..."
|
||||
Get-Partition -DiskNumber $diskNum -PartitionNumber $winpePart.PartitionNumber |
|
||||
Write-WinUtilISOLog "Formatting Partition $($winpePart.PartitionNumber) as FAT32 (label: $volLabel)..."
|
||||
Get-Partition -DiskNumber $DiskNumber -PartitionNumber $winpePart.PartitionNumber |
|
||||
Format-Volume -FileSystem FAT32 -NewFileSystemLabel $volLabel -Force -Confirm:$false
|
||||
Log "Partition $($winpePart.PartitionNumber) formatted as FAT32."
|
||||
Write-WinUtilISOLog "Partition $($winpePart.PartitionNumber) formatted as FAT32."
|
||||
|
||||
SetProgress "Assigning drive letters..." 30
|
||||
Step-WinUtilJob -Status "Assigning drive letters..." -Percent 30
|
||||
Start-Sleep -Seconds 2
|
||||
Update-Disk -Number $diskNum
|
||||
Update-Disk -Number $DiskNumber
|
||||
|
||||
try { Remove-PartitionAccessPath -DiskNumber $diskNum -PartitionNumber $winpePart.PartitionNumber -AccessPath "$($winpePart.DriveLetter):" } catch { Log "Warning: could not remove existing partition access path: $_" }
|
||||
$usbLetter = Get-FreeDriveLetter
|
||||
try { Remove-PartitionAccessPath -DiskNumber $DiskNumber -PartitionNumber $winpePart.PartitionNumber -AccessPath "$($winpePart.DriveLetter):" } catch { Write-WinUtilISOLog -Level "WARN" -Message "Could not remove existing partition access path: $_" }
|
||||
$usbLetter = Get-WinUtilFreeDriveLetter
|
||||
if (-not $usbLetter) { throw "No free drive letters (D-Z) available to assign to the USB data partition." }
|
||||
Set-Partition -DiskNumber $diskNum -PartitionNumber $winpePart.PartitionNumber -NewDriveLetter $usbLetter
|
||||
Log "Assigned drive letter $usbLetter to WINPE partition (Partition $($winpePart.PartitionNumber))."
|
||||
Set-Partition -DiskNumber $DiskNumber -PartitionNumber $winpePart.PartitionNumber -NewDriveLetter $usbLetter
|
||||
Write-WinUtilISOLog "Assigned drive letter $usbLetter to WINPE partition (Partition $($winpePart.PartitionNumber))."
|
||||
Start-Sleep -Seconds 2
|
||||
|
||||
$usbDrive = "${usbLetter}:"
|
||||
$retries = 0
|
||||
while (-not (Test-Path $usbDrive) -and $retries -lt 6) {
|
||||
$retries++
|
||||
Log "Waiting for $usbDrive to become accessible (attempt $retries/6)..."
|
||||
Write-WinUtilISOLog "Waiting for $usbDrive to become accessible (attempt $retries/6)..."
|
||||
Start-Sleep -Seconds 2
|
||||
}
|
||||
if (-not (Test-Path $usbDrive)) { throw "Drive $usbDrive is not accessible after letter assignment." }
|
||||
Log "USB data partition: $usbDrive"
|
||||
Write-WinUtilISOLog "USB data partition: $usbDrive"
|
||||
|
||||
$contentSizeBytes = (Get-ChildItem -LiteralPath $contentsDir -File -Recurse -Force | Measure-Object -Property Length -Sum).Sum
|
||||
if (-not $contentSizeBytes) { $contentSizeBytes = 0 }
|
||||
@@ -223,7 +196,7 @@ function Invoke-WinUtilISOWriteUSB {
|
||||
$partitionCapacityGB = [math]::Round($partitionCapacityBytes / 1GB, 2)
|
||||
$partitionFreeGB = [math]::Round($partitionFreeBytes / 1GB, 2)
|
||||
|
||||
Log "Source content size: $contentSizeGB GB. USB partition capacity: $partitionCapacityGB GB, free: $partitionFreeGB GB."
|
||||
Write-WinUtilISOLog "Source content size: $contentSizeGB GB. USB partition capacity: $partitionCapacityGB GB, free: $partitionFreeGB GB."
|
||||
|
||||
if ($contentSizeBytes -gt $partitionCapacityBytes) {
|
||||
throw "ISO content ($contentSizeGB GB) is larger than the USB partition capacity ($partitionCapacityGB GB). Use a larger USB drive or reduce image size."
|
||||
@@ -233,55 +206,41 @@ function Invoke-WinUtilISOWriteUSB {
|
||||
throw "Insufficient free space on USB partition. Required: $contentSizeGB GB, available: $partitionFreeGB GB."
|
||||
}
|
||||
|
||||
SetProgress "Copying Windows 11 files to USB..." 45
|
||||
Step-WinUtilJob -Status "Copying Windows 11 files to USB..." -Percent 45
|
||||
|
||||
# Copy files; split install.wim if > 4 GB (FAT32 limit)
|
||||
$installWim = Join-Path $contentsDir "sources\install.wim"
|
||||
if (Test-Path $installWim) {
|
||||
$wimSizeMB = [math]::Round((Get-Item $installWim).Length / 1MB)
|
||||
if ($wimSizeMB -gt 3800) {
|
||||
Log "install.wim is $wimSizeMB MB - splitting for FAT32 compatibility... This will take several minutes."
|
||||
Write-WinUtilISOLog "install.wim is $wimSizeMB MB - splitting for FAT32 compatibility... This will take several minutes."
|
||||
Set-ItemProperty -LiteralPath $installWim -Name IsReadOnly -Value $false
|
||||
$splitDest = Join-Path $usbDrive "sources\install.swm"
|
||||
New-Item -ItemType Directory -Path (Split-Path $splitDest) -Force
|
||||
New-Item -ItemType Directory -Path (Split-Path $splitDest) -Force | Out-Null
|
||||
Split-WindowsImage -ImagePath $installWim -SplitImagePath $splitDest -FileSize 3800 -CheckIntegrity
|
||||
Log "install.wim split complete."
|
||||
Log "Copying remaining files to USB..."
|
||||
& robocopy $contentsDir $usbDrive /E /XF install.wim /NFL /NDL /NJH /NJS
|
||||
Write-WinUtilISOLog "install.wim split complete."
|
||||
Write-WinUtilISOLog "Copying remaining files to USB..."
|
||||
Invoke-WinUtilRobocopy -Source $contentsDir -Destination $usbDrive -Arguments @("/E","/XF","install.wim","/NFL","/NDL","/NJH","/NJS")
|
||||
} else {
|
||||
& robocopy $contentsDir $usbDrive /E /NFL /NDL /NJH /NJS
|
||||
Invoke-WinUtilRobocopy -Source $contentsDir -Destination $usbDrive -Arguments @("/E","/NFL","/NDL","/NJH","/NJS")
|
||||
}
|
||||
} else {
|
||||
& robocopy $contentsDir $usbDrive /E /NFL /NDL /NJH /NJS
|
||||
Invoke-WinUtilRobocopy -Source $contentsDir -Destination $usbDrive -Arguments @("/E","/NFL","/NDL","/NJH","/NJS")
|
||||
}
|
||||
|
||||
SetProgress "Finalising USB drive..." 90
|
||||
Log "Files copied to USB."
|
||||
SetProgress "USB write complete" 100
|
||||
Log "USB drive is ready for use."
|
||||
Step-WinUtilJob -Status "Finalising USB drive..." -Percent 90
|
||||
Write-WinUtilISOLog "Files copied to USB."
|
||||
Step-WinUtilJob -Status "USB write complete" -Percent 100
|
||||
Write-WinUtilISOLog "USB drive is ready for use."
|
||||
|
||||
$sync["WPFWin11ISOStatusLog"].Dispatcher.Invoke([action]{
|
||||
[System.Windows.MessageBox]::Show(
|
||||
"USB drive created successfully!`n`nYou can now boot from this drive to install Windows 11.",
|
||||
"USB Ready", "OK", "Info")
|
||||
})
|
||||
Show-WinUtilMessage -Message "USB drive created successfully!`n`nYou can now boot from this drive to install Windows 11." -Title "USB Ready" -Button "OK" -Icon "Info" | Out-Null
|
||||
} catch {
|
||||
Log "ERROR during USB write: $_"
|
||||
$sync["WPFWin11ISOStatusLog"].Dispatcher.Invoke([action]{
|
||||
[System.Windows.MessageBox]::Show("USB write failed:`n`n$_", "USB Write Error", "OK", "Error")
|
||||
})
|
||||
Write-WinUtilISOLog -Level "ERROR" -Message "USB write failed: $_"
|
||||
$_.Exception.Data["WinUtilErrorReported"] = $true
|
||||
Show-WinUtilMessage -Message "USB write failed:`n`n$_" -Title "USB Write Error" -Button "OK" -Icon "Error" | Out-Null
|
||||
throw
|
||||
} finally {
|
||||
Start-Sleep -Milliseconds 800
|
||||
$sync["Win11ISOProcessRunning"] = $false
|
||||
$sync["WPFWin11ISOStatusLog"].Dispatcher.Invoke([action]{
|
||||
$sync["WPFTweaksProgressBar"].Visibility = "Collapsed"
|
||||
$sync["WPFTweaksProgressLabel"].Text = ""
|
||||
$sync["WPFTweaksProgressLabel"].ToolTip = ""
|
||||
$sync["WPFTweaksProgressValue"].Value = 0
|
||||
$sync["WPFWin11ISOWriteUSBButton"].IsEnabled = $true
|
||||
})
|
||||
Invoke-WPFUIThread -ScriptBlock { $sync["WPFWin11ISOWriteUSBButton"].IsEnabled = $true }
|
||||
}
|
||||
})
|
||||
|
||||
$script.BeginInvoke()
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,15 +1,68 @@
|
||||
function Invoke-WinUtilInstallPSProfile {
|
||||
if (-not (Get-Command wt)) {
|
||||
Write-Host "Windows Terminal not found. Installing..."
|
||||
Install-WinUtilWinget
|
||||
winget install Microsoft.WindowsTerminal --source winget --silent
|
||||
function Get-WinUtilPowerShell7Path {
|
||||
$command = Get-Command pwsh -CommandType Application -ErrorAction SilentlyContinue
|
||||
if ($command) { return $command.Source }
|
||||
|
||||
foreach ($candidate in @(
|
||||
"$env:ProgramFiles\PowerShell\7\pwsh.exe",
|
||||
"$env:LOCALAPPDATA\Microsoft\WindowsApps\pwsh.exe")) {
|
||||
if (Test-Path -LiteralPath $candidate) { return $candidate }
|
||||
}
|
||||
|
||||
if (-not (Get-Command pwsh)) {
|
||||
Write-Host "PowerShell 7 not found. Installing..."
|
||||
return $null
|
||||
}
|
||||
|
||||
function Invoke-WinUtilInstallPSProfile {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Installs the CTT PowerShell profile
|
||||
|
||||
.DESCRIPTION
|
||||
The profile targets PowerShell 7, so its setup script has to run under pwsh rather than
|
||||
the runspace this job is on. It runs as a child process with its output captured, so the
|
||||
job log records what happened instead of it scrolling past in a terminal nobody kept.
|
||||
#>
|
||||
|
||||
$pwshPath = Get-WinUtilPowerShell7Path
|
||||
if (-not $pwshPath) {
|
||||
Step-WinUtilJob -Status "Installing PowerShell 7" -State "Indeterminate"
|
||||
Write-WinUtilLog -Component "Feature" -Message "PowerShell 7 not found, installing it first."
|
||||
|
||||
Install-WinUtilWinget
|
||||
winget install Microsoft.PowerShell --source winget --installer-type wix --silent
|
||||
Install-WinUtilProgramWinget -Action Install -Programs @("Microsoft.PowerShell") | Out-Null
|
||||
|
||||
# WinGet updates the persisted PATH, not this already-running process. Resolve the
|
||||
# standard install locations as well as the current PATH before deciding it failed.
|
||||
$pwshPath = Get-WinUtilPowerShell7Path
|
||||
if (-not $pwshPath) {
|
||||
throw "PowerShell 7 could not be installed, so the profile cannot be set up."
|
||||
}
|
||||
}
|
||||
|
||||
Step-WinUtilJob -Status "Running the profile setup" -State "Indeterminate"
|
||||
|
||||
$setupUrl = "https://github.com/ChrisTitusTech/powershell-profile/raw/main/setup.ps1"
|
||||
# Stop in the child, so a setup failure is a nonzero exit rather than a logged error and a
|
||||
# exit code of zero
|
||||
$output = & $pwshPath -NoProfile -NonInteractive -Command "`$ErrorActionPreference = 'Stop'; irm '$setupUrl' | iex" 2>&1
|
||||
$exitCode = $LASTEXITCODE
|
||||
|
||||
$failures = 0
|
||||
foreach ($line in @($output)) {
|
||||
if ($line -is [System.Management.Automation.ErrorRecord]) {
|
||||
$failures++
|
||||
Write-WinUtilErrorRecord -ErrorRecord $line -Component "Feature" -Context "PowerShell profile setup"
|
||||
} elseif (-not [string]::IsNullOrWhiteSpace($line)) {
|
||||
Write-WinUtilLog -Component "Feature" -Message ([string]$line).Trim()
|
||||
}
|
||||
}
|
||||
|
||||
if ($exitCode -ne 0) {
|
||||
throw "The profile setup script exited with code $exitCode."
|
||||
}
|
||||
|
||||
if ($failures -gt 0) {
|
||||
throw "The profile setup script reported $failures error(s); see the log."
|
||||
}
|
||||
|
||||
wt new-tab pwsh -NoExit -Command "irm https://github.com/ChrisTitusTech/powershell-profile/raw/main/setup.ps1 | iex"
|
||||
Write-WinUtilLog -Component "Feature" -Message "CTT PowerShell profile installed. Open a new PowerShell 7 session to use it."
|
||||
}
|
||||
|
||||
@@ -20,9 +20,13 @@ function Invoke-WinUtilSSHServer {
|
||||
|
||||
#Adding Firewall rule for port 22
|
||||
Write-Host "Setting up firewall rules"
|
||||
if (-not ((Get-NetFirewallRule -Name 'sshd').Enabled)) {
|
||||
$firewallRule = Get-NetFirewallRule -Name 'sshd' -ErrorAction SilentlyContinue
|
||||
if ($null -eq $firewallRule) {
|
||||
New-NetFirewallRule -Name sshd -DisplayName 'OpenSSH Server (sshd)' -Enabled True -Direction Inbound -Protocol TCP -Action Allow -LocalPort 22
|
||||
Write-Host "Firewall rule for OpenSSH Server created and enabled."
|
||||
} elseif (-not $firewallRule.Enabled) {
|
||||
Set-NetFirewallRule -Name 'sshd' -Enabled True
|
||||
Write-Host "Firewall rule for OpenSSH Server enabled."
|
||||
}
|
||||
|
||||
# An SSH logon for a member of the administrators group gets a full token
|
||||
|
||||
@@ -1,10 +1,36 @@
|
||||
function Invoke-WinUtilUninstallPSProfile {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Restores the PowerShell 7 profile the CTT profile replaced
|
||||
|
||||
if (Test-Path ($Profile + ".bak")) {
|
||||
Move-Item -Path ($Profile + ".bak") -Destination $Profile
|
||||
} else {
|
||||
Remove-Item -Path $Profile
|
||||
.DESCRIPTION
|
||||
The profile path has to come from pwsh itself. $PROFILE inside this job is the worker's
|
||||
own Windows PowerShell profile, which is not the file the install wrote.
|
||||
#>
|
||||
|
||||
$pwshPath = Get-WinUtilPowerShell7Path
|
||||
if (-not $pwshPath) {
|
||||
throw "PowerShell 7 is not installed, so there is no CTT profile to remove."
|
||||
}
|
||||
|
||||
Write-Host "Successfully uninstalled CTT PowerShell Profile." -ForegroundColor Green
|
||||
$profilePath = (& $pwshPath -NoProfile -NonInteractive -Command '$PROFILE' | Select-Object -First 1)
|
||||
if ([string]::IsNullOrWhiteSpace($profilePath)) {
|
||||
throw "Could not determine the PowerShell 7 profile path."
|
||||
}
|
||||
$profilePath = $profilePath.Trim()
|
||||
$backupPath = "$profilePath.bak"
|
||||
|
||||
if (Test-Path $backupPath) {
|
||||
Move-Item -Path $backupPath -Destination $profilePath -Force
|
||||
Write-WinUtilLog -Component "Feature" -Message "Restored the profile that was in place before: $profilePath"
|
||||
return
|
||||
}
|
||||
|
||||
if (Test-Path $profilePath) {
|
||||
Remove-Item -Path $profilePath -Force
|
||||
Write-WinUtilLog -Component "Feature" -Message "Removed the CTT PowerShell profile: $profilePath"
|
||||
return
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Level "WARN" -Component "Feature" -Message "No PowerShell 7 profile found at $profilePath, nothing to remove."
|
||||
}
|
||||
|
||||
@@ -0,0 +1,98 @@
|
||||
function Measure-WinUtilStep {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Times one step of a pipeline and records it for the timing summary
|
||||
|
||||
.DESCRIPTION
|
||||
Output passes through untouched, so this can wrap an existing expression without
|
||||
changing what the caller receives. Each step is logged as a "timing:" line and kept
|
||||
in $sync.StepTimings for the summary to rank.
|
||||
|
||||
.PARAMETER Name
|
||||
What the step is, as it should read in the log.
|
||||
|
||||
.PARAMETER ScriptBlock
|
||||
The work to time.
|
||||
|
||||
.PARAMETER Scope
|
||||
Groups steps that belong to the same run, normally a job name or "UI".
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory, Position = 0)]
|
||||
[string]$Name,
|
||||
|
||||
[Parameter(Mandatory, Position = 1)]
|
||||
[scriptblock]$ScriptBlock,
|
||||
|
||||
[string]$Scope = "WinUtil"
|
||||
)
|
||||
|
||||
$stopwatch = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
try {
|
||||
& $ScriptBlock
|
||||
} finally {
|
||||
$stopwatch.Stop()
|
||||
|
||||
if ($null -ne $sync.StepTimings) {
|
||||
$null = $sync.StepTimings.Add([pscustomobject]@{
|
||||
Scope = $Scope
|
||||
Step = $Name
|
||||
Milliseconds = $stopwatch.ElapsedMilliseconds
|
||||
})
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component $Scope -Message "timing: $Name took $($stopwatch.ElapsedMilliseconds) ms"
|
||||
}
|
||||
}
|
||||
|
||||
function Write-WinUtilTimingSummary {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Logs the slowest steps of a scope, so the log answers "what took so long"
|
||||
|
||||
.PARAMETER Scope
|
||||
Which group of steps to report on.
|
||||
|
||||
.PARAMETER Top
|
||||
How many of the slowest steps to list.
|
||||
|
||||
.PARAMETER TotalMilliseconds
|
||||
Wall clock total. Without it the summary sums the steps, missing whatever happened
|
||||
between them.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Scope,
|
||||
|
||||
[int]$Top = 5,
|
||||
|
||||
[long]$TotalMilliseconds = -1,
|
||||
|
||||
[int]$StartIndex = 0
|
||||
)
|
||||
|
||||
if ($null -eq $sync.StepTimings) {
|
||||
return
|
||||
}
|
||||
|
||||
[System.Threading.Monitor]::Enter($sync.StepTimings.SyncRoot)
|
||||
try {
|
||||
$timingSnapshot = @($sync.StepTimings.ToArray())
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.StepTimings.SyncRoot)
|
||||
}
|
||||
|
||||
$steps = @($timingSnapshot | Select-Object -Skip $StartIndex | Where-Object { $_.Scope -eq $Scope })
|
||||
if ($steps.Count -eq 0) {
|
||||
return
|
||||
}
|
||||
|
||||
$measured = ($steps | Measure-Object -Property Milliseconds -Sum).Sum
|
||||
$total = if ($TotalMilliseconds -ge 0) { $TotalMilliseconds } else { $measured }
|
||||
|
||||
Write-WinUtilLog -Component $Scope -Message "timing summary: $($steps.Count) step(s), $measured ms measured of $total ms total"
|
||||
foreach ($step in ($steps | Sort-Object Milliseconds -Descending | Select-Object -First $Top)) {
|
||||
$share = if ($total -gt 0) { [int](($step.Milliseconds / $total) * 100) } else { 0 }
|
||||
Write-WinUtilLog -Component $Scope -Message "timing summary: $($step.Milliseconds) ms ($share%) $($step.Step)"
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,55 @@
|
||||
function New-WinUtilSessionState {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Builds the InitialSessionState every WinUtil runspace is created from
|
||||
|
||||
.DESCRIPTION
|
||||
The interface runspace and the worker pool start from the same state: the shared
|
||||
$sync hashtable, the compiled script's globals, and every WinUtil function. That is
|
||||
what lets the interface build a tab and a job body call any helper without injecting
|
||||
definitions by hand. PowerShell's own functions are skipped, the default session
|
||||
state already carries them.
|
||||
|
||||
Cached: an InitialSessionState is a template any number of runspaces are created
|
||||
from, and building it is not free.
|
||||
#>
|
||||
|
||||
if ($sync.SessionState) {
|
||||
return $sync.SessionState
|
||||
}
|
||||
|
||||
$initialSessionState = [System.Management.Automation.Runspaces.InitialSessionState]::CreateDefault()
|
||||
|
||||
$variables = @(
|
||||
@{ Name = "sync"; Value = $sync },
|
||||
@{ Name = "PARAM_OFFLINE"; Value = $PARAM_OFFLINE },
|
||||
@{ Name = "inputXML"; Value = $inputXML },
|
||||
@{ Name = "WinUtilAutounattendXml"; Value = $WinUtilAutounattendXml }
|
||||
)
|
||||
|
||||
foreach ($variable in $variables) {
|
||||
$initialSessionState.Variables.Add(
|
||||
(New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList $variable.Name, $variable.Value, $null)
|
||||
)
|
||||
}
|
||||
|
||||
$builtInFunctions = [System.Collections.Generic.HashSet[string]]::new(
|
||||
[string[]]@($initialSessionState.Commands |
|
||||
Where-Object { $_ -is [System.Management.Automation.Runspaces.SessionStateFunctionEntry] } |
|
||||
ForEach-Object { $_.Name }),
|
||||
[StringComparer]::OrdinalIgnoreCase
|
||||
)
|
||||
|
||||
foreach ($function in (Get-ChildItem function:\)) {
|
||||
if ($builtInFunctions.Contains($function.Name)) {
|
||||
continue
|
||||
}
|
||||
|
||||
$initialSessionState.Commands.Add(
|
||||
(New-Object System.Management.Automation.Runspaces.SessionStateFunctionEntry -ArgumentList $function.Name, $function.Definition)
|
||||
)
|
||||
}
|
||||
|
||||
$sync.SessionState = $initialSessionState
|
||||
return $initialSessionState
|
||||
}
|
||||
@@ -0,0 +1,131 @@
|
||||
function Register-WinUtilRunspaceCleanup {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Disposes a PowerShell instance and any owned runspace once its work has finished
|
||||
|
||||
.DESCRIPTION
|
||||
Ends the invocation and disposes the instance from a thread pool callback, so
|
||||
nothing has to wait for a fire-and-forget runspace to complete just to clean it up.
|
||||
|
||||
.PARAMETER PowerShell
|
||||
The instance to dispose.
|
||||
|
||||
.PARAMETER Handle
|
||||
The handle returned by its BeginInvoke.
|
||||
|
||||
.PARAMETER Runspace
|
||||
A dedicated runspace owned by the invocation. Shared pool invocations omit it.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
$PowerShell,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
$Handle,
|
||||
|
||||
$Runspace
|
||||
)
|
||||
|
||||
# Version the CLR helper because Add-Type definitions survive repeated in-memory WinUtil runs.
|
||||
# Older sessions can already contain the V1 type, whose state object has no Runspace property.
|
||||
if (-not ("WinUtilRunspaceCleanupV3" -as [type])) {
|
||||
Add-Type @"
|
||||
using System;
|
||||
using System.Management.Automation;
|
||||
using System.Management.Automation.Runspaces;
|
||||
using System.Threading;
|
||||
|
||||
public sealed class WinUtilRunspaceCleanupStateV3
|
||||
{
|
||||
public PowerShell PowerShell { get; set; }
|
||||
public IAsyncResult Handle { get; set; }
|
||||
public Runspace Runspace { get; set; }
|
||||
}
|
||||
|
||||
public static class WinUtilRunspaceCleanupV3
|
||||
{
|
||||
public static readonly System.Threading.WaitOrTimerCallback Callback = Cleanup;
|
||||
|
||||
public static bool Register(WinUtilRunspaceCleanupStateV3 state)
|
||||
{
|
||||
try
|
||||
{
|
||||
ThreadPool.RegisterWaitForSingleObject(state.Handle.AsyncWaitHandle, Callback, state, -1, true);
|
||||
return true;
|
||||
}
|
||||
catch
|
||||
{
|
||||
// Registration is normally infallible, but work has already started. A background
|
||||
// waiter preserves asynchronous cleanup without blocking the WPF dispatcher.
|
||||
try
|
||||
{
|
||||
var thread = new Thread(() =>
|
||||
{
|
||||
try
|
||||
{
|
||||
state.Handle.AsyncWaitHandle.WaitOne();
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
Cleanup(state, false);
|
||||
});
|
||||
thread.IsBackground = true;
|
||||
thread.Name = "WinUtil runspace cleanup fallback";
|
||||
thread.Start();
|
||||
return true;
|
||||
}
|
||||
catch
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static void Cleanup(object state, bool timedOut)
|
||||
{
|
||||
var cleanupState = state as WinUtilRunspaceCleanupStateV3;
|
||||
if (cleanupState == null || cleanupState.PowerShell == null || cleanupState.Handle == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
cleanupState.PowerShell.EndInvoke(cleanupState.Handle);
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
finally
|
||||
{
|
||||
cleanupState.PowerShell.Dispose();
|
||||
if (cleanupState.Runspace != null)
|
||||
{
|
||||
try
|
||||
{
|
||||
cleanupState.Runspace.Close();
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
finally
|
||||
{
|
||||
cleanupState.Runspace.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
"@
|
||||
}
|
||||
|
||||
$cleanupState = [WinUtilRunspaceCleanupStateV3]::new()
|
||||
$cleanupState.PowerShell = $PowerShell
|
||||
$cleanupState.Handle = $Handle
|
||||
$cleanupState.Runspace = $Runspace
|
||||
$registered = [WinUtilRunspaceCleanupV3]::Register($cleanupState)
|
||||
if (-not $registered) {
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Could not register asynchronous cleanup for background work; it will be reclaimed when later work or shutdown inspects it."
|
||||
}
|
||||
}
|
||||
@@ -55,7 +55,9 @@ function Remove-WinUtilProvisionedAPPX {
|
||||
$failureDetails = ($removalOutput | Out-String).Trim()
|
||||
$errorMessage = "AppX provisioned package removal failed: $failureDetails"
|
||||
Write-WinUtilLog -Level "ERROR" -Component "AppX" -Message $errorMessage
|
||||
throw $errorMessage
|
||||
$exception = [System.InvalidOperationException]::new($errorMessage)
|
||||
$exception.Data["WinUtilErrorReported"] = $true
|
||||
throw $exception
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "AppX" -Message "AppX provisioned package removal completed."
|
||||
|
||||
@@ -22,19 +22,30 @@ function Reset-WPFCheckBoxes {
|
||||
)
|
||||
$selectedSet = [System.Collections.Generic.HashSet[string]]::new([string[]]@($sync.selectedApps + $sync.selectedTweaks + $sync.selectedFeatures + $sync.selectedAppx), [StringComparer]::OrdinalIgnoreCase)
|
||||
|
||||
foreach ($syncEntry in $sync.GetEnumerator()) {
|
||||
# A synchronized Hashtable protects individual operations, not enumeration. Materialize the
|
||||
# snapshot under its lock, then release it before handlers run and mutate $sync.
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
$syncEntries = @($sync.GetEnumerator())
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
foreach ($syncEntry in $syncEntries) {
|
||||
if ($syncEntry.Value -is [System.Windows.Controls.CheckBox] -and $syncEntry.Name -notlike "WPFToggle*" -and $syncEntry.Name -like $checkboxfilterpattern) {
|
||||
$checkboxName = $syncEntry.Key
|
||||
$sync.$checkboxName.IsChecked = $selectedSet.Contains($checkboxName)
|
||||
}
|
||||
}
|
||||
|
||||
# Update Installs tab UI values
|
||||
$count = $sync.SelectedApps.Count
|
||||
$sync.WPFselectedAppsButton.Content = "Selected Apps: $count"
|
||||
# On every change, remove all entries inside the Popup Menu. This is done, so we can keep the alphabetical order even if elements are selected in a random way
|
||||
$sync.selectedAppsstackPanel.Children.Clear()
|
||||
$sync.selectedApps | Foreach-Object { Add-SelectedAppsMenuItem -name $($sync.configs.applicationsHashtable.$_.Content) -key $_ }
|
||||
# Update Installs tab UI values. These are built with the Install tab, and this runs for
|
||||
# whichever tab is built first: offline starts on Tweaks, so they are not there yet.
|
||||
if ($sync.selectedAppsstackPanel) {
|
||||
$count = $sync.SelectedApps.Count
|
||||
$sync.WPFselectedAppsButton.Content = "Selected Apps: $count"
|
||||
# On every change, remove all entries inside the Popup Menu. This is done, so we can keep the alphabetical order even if elements are selected in a random way
|
||||
$sync.selectedAppsstackPanel.Children.Clear()
|
||||
$sync.selectedApps | Foreach-Object { Add-SelectedAppsMenuItem -name $($sync.configs.applicationsHashtable.$_.Content) -key $_ }
|
||||
}
|
||||
|
||||
if($doToggles) {
|
||||
# Restore toggle switch states from imported config.
|
||||
@@ -42,7 +53,13 @@ function Reset-WPFCheckBoxes {
|
||||
# from the export file were not part of the saved config and should keep whatever
|
||||
# state the live system already has (set during UI initialisation via Get-WinUtilToggleStatus).
|
||||
$importedToggles = [System.Collections.Generic.HashSet[string]]::new([string[]]@($sync.selectedToggles), [StringComparer]::OrdinalIgnoreCase)
|
||||
foreach ($toggle in $sync.GetEnumerator()) {
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
$toggleEntries = @($sync.GetEnumerator())
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
foreach ($toggle in $toggleEntries) {
|
||||
if ($toggle.Key -like "WPFToggle*" -and $toggle.Value -is [System.Windows.Controls.CheckBox] -and $importedToggles.Contains($toggle.Key)) {
|
||||
$sync[$toggle.Key].IsChecked = $true
|
||||
}
|
||||
|
||||
@@ -1,39 +0,0 @@
|
||||
function Set-WinUtilTweaksProgressIndicator {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Shows, updates, or hides the window-level progress indicator used by long-running
|
||||
workflows such as app management, Tweaks, AppX management, and Win11 Creator.
|
||||
It lives outside the TabControl, so it stays visible no matter which tab is active.
|
||||
.PARAMETER Visible
|
||||
Whether the indicator should be shown or hidden.
|
||||
.PARAMETER Label
|
||||
The text to display above the progress bar.
|
||||
.PARAMETER Percent
|
||||
The percentage of the progress bar that should be filled (0-100).
|
||||
#>
|
||||
param(
|
||||
[bool]$Visible,
|
||||
[string]$Label,
|
||||
[ValidateRange(0,100)]
|
||||
[int]$Percent
|
||||
)
|
||||
|
||||
if ($null -eq $sync.form -or $null -eq $sync.form.Dispatcher) {
|
||||
return
|
||||
}
|
||||
|
||||
$indicatorVisible = if ($Visible) { [Windows.Visibility]::Visible } else { [Windows.Visibility]::Collapsed }
|
||||
$indicatorLabel = $Label
|
||||
$hasLabel = $PSBoundParameters.ContainsKey('Label')
|
||||
$hasPercent = $PSBoundParameters.ContainsKey('Percent')
|
||||
|
||||
Invoke-WPFUIThread -ScriptBlock {
|
||||
$sync.WPFTweaksProgressBar.Visibility = $indicatorVisible
|
||||
if ($hasLabel) {
|
||||
$sync.WPFTweaksProgressLabel.Text = $indicatorLabel
|
||||
}
|
||||
if ($hasPercent) {
|
||||
$sync.WPFTweaksProgressValue.Value = $Percent
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -2,6 +2,14 @@ function Show-WinUtilMessage {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Shows a WinUtil message box and returns the selected result.
|
||||
|
||||
.DESCRIPTION
|
||||
Message boxes need the interface thread, so this marshals onto it and can therefore be
|
||||
called from a job body as well as from an event handler. Every prompt is also written to
|
||||
the session log so the log shows what the user was asked and not just what happened next.
|
||||
|
||||
With no window there is nobody to click, so nothing is shown. A modal put up in that
|
||||
state never returns and takes the worker with it.
|
||||
#>
|
||||
param (
|
||||
[string]$Message,
|
||||
@@ -10,5 +18,24 @@ function Show-WinUtilMessage {
|
||||
$Icon = "Information"
|
||||
)
|
||||
|
||||
[System.Windows.MessageBox]::Show($Message, $Title, $Button, $Icon)
|
||||
Write-WinUtilLog -Component "Dialog" -Message "$Title : $($Message -replace '\r?\n', ' ')"
|
||||
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
# Anything with a choice is answered with the one that does not go ahead, so a prompt
|
||||
# nobody saw can never stand in for consent
|
||||
$unattended = if ("$Button" -eq "OK") { "OK" } else { "No" }
|
||||
Write-WinUtilLog -Level "WARN" -Component "Dialog" -Message "No window to ask on, answering '$unattended' for: $Title"
|
||||
return $unattended
|
||||
}
|
||||
|
||||
return Invoke-WPFUIThread -PassThru -Parameters @{
|
||||
Message = $Message
|
||||
Title = $Title
|
||||
Button = $Button
|
||||
Icon = $Icon
|
||||
} -ScriptBlock {
|
||||
param($Message, $Title, $Button, $Icon)
|
||||
|
||||
[System.Windows.MessageBox]::Show($Message, $Title, $Button, $Icon)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,40 @@
|
||||
function Start-WinUtilAssetRendering {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Renders the taskbar overlay bitmaps on a thread of their own
|
||||
|
||||
.DESCRIPTION
|
||||
Rasterising the overlays costs the interface thread time it could spend getting the
|
||||
window up. The bitmaps are frozen before publication, so they can be built anywhere.
|
||||
|
||||
Nothing waits on this: if the render has not finished when an overlay is asked for,
|
||||
Set-WinUtilTaskbaritem renders it in place.
|
||||
|
||||
Needs STA for RenderTargetBitmap, which the shared worker pool is not.
|
||||
#>
|
||||
|
||||
$runspace = [runspacefactory]::CreateRunspace((New-WinUtilSessionState))
|
||||
$runspace.ApartmentState = "STA"
|
||||
$runspace.ThreadOptions = "ReuseThread"
|
||||
$runspace.Open()
|
||||
|
||||
$shell = [powershell]::Create()
|
||||
$shell.Runspace = $runspace
|
||||
[void]$shell.AddScript({
|
||||
# A new runspace does not inherit assemblies loaded by the interface runspace. On a cold
|
||||
# process these types otherwise fail before the off-thread render can do any work.
|
||||
Add-Type -AssemblyName WindowsBase
|
||||
Add-Type -AssemblyName PresentationCore
|
||||
Add-Type -AssemblyName PresentationFramework
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "render taskbar overlays (off thread)" -ScriptBlock {
|
||||
Initialize-WinUtilTaskbarOverlayAssets -IncludeLogo $true -IncludeStatusAssets $true
|
||||
}
|
||||
})
|
||||
|
||||
$handle = $shell.BeginInvoke()
|
||||
|
||||
Register-WinUtilRunspaceCleanup -PowerShell $shell -Handle $handle -Runspace $runspace
|
||||
|
||||
return $handle
|
||||
}
|
||||
@@ -0,0 +1,156 @@
|
||||
function Start-WinUtilBackgroundQueue {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Drains a queue of interface work one item at a time, between the things the user does
|
||||
|
||||
.DESCRIPTION
|
||||
For work that must run on the interface thread but that nobody waits on: unopened
|
||||
tabs, app list entries. One item per queued operation, so input is answered between
|
||||
them instead of after the whole list.
|
||||
|
||||
Re-posted rather than looped: only returning to the dispatcher lets it service input.
|
||||
Posted as a compiled action rather than through Invoke-WPFUIThread, whose body
|
||||
crosses runspaces as text and would recompile on each of the hundreds of posts a full
|
||||
app list costs.
|
||||
|
||||
.PARAMETER Name
|
||||
Identifies the queue in $sync so a re-posted pump finds its state.
|
||||
|
||||
.PARAMETER Queue
|
||||
The queue to drain. Items mean whatever Step says they mean.
|
||||
|
||||
.PARAMETER Step
|
||||
Runs one item. Receives the dequeued item.
|
||||
|
||||
.PARAMETER OnComplete
|
||||
Runs once on the interface thread after the last item.
|
||||
|
||||
.PARAMETER RequiresTab
|
||||
Work drawing into this tab waits while another tab is shown.
|
||||
|
||||
.PARAMETER DeferWhile
|
||||
Extra reason to hold off, tested each round. Lets a more urgent queue go first.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Name,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
$Queue,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
[scriptblock]$Step,
|
||||
|
||||
[scriptblock]$OnComplete,
|
||||
|
||||
[string]$RequiresTab,
|
||||
|
||||
[scriptblock]$DeferWhile
|
||||
)
|
||||
|
||||
if ($null -eq $sync.BackgroundQueues) {
|
||||
$sync.BackgroundQueues = [hashtable]::Synchronized(@{})
|
||||
}
|
||||
|
||||
$sync.BackgroundQueues[$Name] = @{
|
||||
Queue = $Queue
|
||||
Step = $Step
|
||||
OnComplete = $OnComplete
|
||||
RequiresTab = $RequiresTab
|
||||
DeferWhile = $DeferWhile
|
||||
}
|
||||
|
||||
# No window means no dispatcher to spread over and nothing competing for the thread
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
while ($Queue.Count -gt 0) {
|
||||
# One failing item must not abandon the rest or strand the state, matching the
|
||||
# dispatcher path
|
||||
try {
|
||||
& $Step $Queue.Dequeue()
|
||||
} catch {
|
||||
Write-WinUtilErrorRecord -ErrorRecord $_ -Component "UI" -Context "Background queue '$Name'"
|
||||
}
|
||||
}
|
||||
if ($OnComplete) { & $OnComplete }
|
||||
$sync.BackgroundQueues.Remove($Name)
|
||||
return
|
||||
}
|
||||
|
||||
Request-WinUtilBackgroundQueueStep -Name $Name
|
||||
}
|
||||
|
||||
function Request-WinUtilBackgroundQueueStep {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Posts the next step of a queue at background priority
|
||||
|
||||
.PARAMETER Name
|
||||
Which queue to advance.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Name
|
||||
)
|
||||
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
return
|
||||
}
|
||||
|
||||
# The name travels as the dispatcher's argument, not captured: this function has returned by
|
||||
# the time the block runs, and a closure would bind command lookup to a copied scope.
|
||||
$null = $sync.Form.Dispatcher.BeginInvoke(
|
||||
[System.Windows.Threading.DispatcherPriority]::Background,
|
||||
[System.Windows.Threading.DispatcherOperationCallback]{
|
||||
param($QueueName)
|
||||
Invoke-WinUtilBackgroundQueueStep -Name $QueueName
|
||||
return $null
|
||||
},
|
||||
$Name)
|
||||
}
|
||||
|
||||
function Invoke-WinUtilBackgroundQueueStep {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Runs one item of a queue and asks for the next, or finishes
|
||||
|
||||
.PARAMETER Name
|
||||
Which queue to advance.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Name
|
||||
)
|
||||
|
||||
$state = $sync.BackgroundQueues[$Name]
|
||||
if ($null -eq $state) {
|
||||
return
|
||||
}
|
||||
|
||||
if ($state.Queue.Count -gt 0) {
|
||||
$defer = (Test-WinUtilDeferBackgroundWork -RequiresTab $state.RequiresTab) -or
|
||||
($state.DeferWhile -and (& $state.DeferWhile))
|
||||
|
||||
# Waits rather than competing with whatever the user is doing
|
||||
if ($defer) {
|
||||
Invoke-WinUtilWhenIdle -Argument $Name -Callback {
|
||||
param($QueueName)
|
||||
Invoke-WinUtilBackgroundQueueStep -Name $QueueName
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
try {
|
||||
& $state.Step $state.Queue.Dequeue()
|
||||
} catch {
|
||||
Write-WinUtilErrorRecord -ErrorRecord $_ -Component "UI" -Context "Background queue '$Name'"
|
||||
}
|
||||
}
|
||||
|
||||
if ($state.Queue.Count -gt 0) {
|
||||
Request-WinUtilBackgroundQueueStep -Name $Name
|
||||
return
|
||||
}
|
||||
|
||||
$sync.BackgroundQueues.Remove($Name)
|
||||
if ($state.OnComplete) { & $state.OnComplete }
|
||||
}
|
||||
@@ -4,8 +4,30 @@ function Invoke-WinUtilInstallAppRenderBatch {
|
||||
$CategoryBatch
|
||||
)
|
||||
|
||||
foreach ($appKey in $CategoryBatch.AppKeys) {
|
||||
# A count is not a unit of time. How long a fixed number of entries takes depends on the
|
||||
# machine and on the category, so the pass runs to a deadline instead and hands back
|
||||
# whatever it did not reach. That caps how long a click can be left waiting.
|
||||
$budgetMs = 25
|
||||
$keys = @($CategoryBatch.AppKeys)
|
||||
|
||||
# This runs on the dispatcher, so keep the slice free of logging and other disk I/O.
|
||||
$clock = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
$rendered = 0
|
||||
foreach ($appKey in $keys) {
|
||||
$sync.$appKey = Initialize-InstallAppEntry -TargetElement $CategoryBatch.TargetElement -AppKey $appKey
|
||||
$rendered++
|
||||
# at least one per pass, or a slow machine would never finish the list
|
||||
if ($clock.ElapsedMilliseconds -ge $budgetMs) {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if ($rendered -lt $keys.Count) {
|
||||
$sync.InstallAppRenderQueue.Enqueue([pscustomobject]@{
|
||||
Category = $CategoryBatch.Category
|
||||
TargetElement = $CategoryBatch.TargetElement
|
||||
AppKeys = @($keys[$rendered..($keys.Count - 1)])
|
||||
})
|
||||
}
|
||||
|
||||
# Entries render in batches, so a filter that is already active has to be applied to each new
|
||||
@@ -21,23 +43,7 @@ function Invoke-WinUtilInstallAppRenderBatch {
|
||||
|
||||
function Complete-WinUtilInstallAppRendering {
|
||||
$sync.InstallAppEntriesRendered = $true
|
||||
}
|
||||
|
||||
function Invoke-WinUtilInstallAppRenderNextBatch {
|
||||
if ($sync.InstallAppRenderQueue.Count -gt 0) {
|
||||
$categoryBatch = $sync.InstallAppRenderQueue.Dequeue()
|
||||
Invoke-WinUtilInstallAppRenderBatch -CategoryBatch $categoryBatch
|
||||
}
|
||||
|
||||
if ($sync.InstallAppRenderQueue.Count -gt 0) {
|
||||
$sync.Form.Dispatcher.BeginInvoke(
|
||||
[System.Windows.Threading.DispatcherPriority]::Background,
|
||||
[action]{ Invoke-WinUtilInstallAppRenderNextBatch }
|
||||
) | Out-Null
|
||||
return
|
||||
}
|
||||
|
||||
Complete-WinUtilInstallAppRendering
|
||||
}
|
||||
|
||||
function Start-WinUtilInstallAppRendering {
|
||||
@@ -47,18 +53,15 @@ function Start-WinUtilInstallAppRendering {
|
||||
|
||||
$sync.InstallAppEntriesRendered = $false
|
||||
|
||||
if ($sync.Form -and $sync.Form.Dispatcher) {
|
||||
$sync.Form.Dispatcher.BeginInvoke(
|
||||
[System.Windows.Threading.DispatcherPriority]::Background,
|
||||
[action]{ Invoke-WinUtilInstallAppRenderNextBatch }
|
||||
) | Out-Null
|
||||
return
|
||||
}
|
||||
|
||||
while ($sync.InstallAppRenderQueue.Count -gt 0) {
|
||||
$categoryBatch = $sync.InstallAppRenderQueue.Dequeue()
|
||||
Invoke-WinUtilInstallAppRenderBatch -CategoryBatch $categoryBatch
|
||||
}
|
||||
|
||||
Complete-WinUtilInstallAppRendering
|
||||
Start-WinUtilBackgroundQueue -Name "InstallAppRender" -Queue $sync.InstallAppRenderQueue `
|
||||
-RequiresTab "Install" `
|
||||
-Step { param($CategoryBatch) Invoke-WinUtilInstallAppRenderBatch -CategoryBatch $CategoryBatch } `
|
||||
-OnComplete { Complete-WinUtilInstallAppRendering } `
|
||||
-DeferWhile {
|
||||
# Tabs that have never been built come first. This list is already on screen and
|
||||
# filling in, while another tab is empty until it is built, so a click on one costs
|
||||
# the whole build. The list finishing a little later is not felt; a tab that takes
|
||||
# half a second to open is.
|
||||
$sync.TabWarmupQueue -and $sync.TabWarmupQueue.Count -gt 0
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,258 @@
|
||||
function Start-WinUtilJob {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Runs a long operation off the UI thread with the shared progress, taskbar, log and
|
||||
error handling applied around it
|
||||
|
||||
.DESCRIPTION
|
||||
One job at a time. Owns the busy flag, the progress bar, the taskbar item, the
|
||||
console banner and the log lines for the job's lifetime. The body does the work and
|
||||
calls Step-WinUtilJob; it must not print a banner or set the busy flag itself. A body
|
||||
that throws is caught and the interface restored in a finally, so a failure cannot
|
||||
leave the UI stuck busy.
|
||||
|
||||
.PARAMETER Name
|
||||
Log component and progress text, for example Install.
|
||||
|
||||
.PARAMETER ScriptBlock
|
||||
The work. Receives Parameters as named parameters.
|
||||
|
||||
.PARAMETER Parameters
|
||||
Values passed to the body by name.
|
||||
|
||||
.PARAMETER Description
|
||||
Progress text shown while the job starts. Defaults to the job name.
|
||||
|
||||
.PARAMETER DisableAppList
|
||||
Greys out the app list for the duration, for jobs that change what is installed.
|
||||
|
||||
.EXAMPLE
|
||||
Start-WinUtilJob -Name "Install" -Parameters @{ Packages = $packages } -ScriptBlock {
|
||||
param($Packages)
|
||||
Step-WinUtilJob -Status "Installing" -Percent 10
|
||||
}
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Name,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
[scriptblock]$ScriptBlock,
|
||||
|
||||
[hashtable]$Parameters = @{},
|
||||
|
||||
[string]$Description,
|
||||
|
||||
[switch]$DisableAppList
|
||||
)
|
||||
|
||||
if ($sync.ShuttingDown -or $sync.FinishInConsole) {
|
||||
Write-WinUtilLog -Level "WARN" -Component $Name -Message "Refused to start $Name, WinUtil is closing."
|
||||
return $null
|
||||
}
|
||||
|
||||
# A nested job runs inline: the outer already owns the slot and the reporting, so claiming
|
||||
# again would refuse it and skip its work. Feature installs arrive twice, from feature.json
|
||||
# and from Invoke-WPFFeatureInstall.
|
||||
if ($global:WinUtilIsJobWorker) {
|
||||
& $ScriptBlock @Parameters
|
||||
return $null
|
||||
}
|
||||
|
||||
# Locked: a headless or scheduled caller is not serialised by the dispatcher, where
|
||||
# test-then-assign lets two jobs both own the slot. The token identifies the run, so a worker
|
||||
# still unwinding cannot release a slot the next job holds.
|
||||
$jobToken = [guid]::NewGuid().ToString()
|
||||
$blockedBy = $null
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
if ($sync.ActiveJob) {
|
||||
$blockedBy = $sync.ActiveJob
|
||||
} else {
|
||||
$sync.ActiveJob = $Name
|
||||
$sync.ActiveJobToken = $jobToken
|
||||
$sync.LastJobResult = $null
|
||||
}
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
|
||||
if ($blockedBy) {
|
||||
Show-WinUtilMessage -Message "$blockedBy is still running. Wait for it to finish before starting another action." -Title "WinUtil" -Button "OK" -Icon "Warning" | Out-Null
|
||||
return $null
|
||||
}
|
||||
|
||||
$label = if ($Description) { $Description } else { $Name }
|
||||
try {
|
||||
$timingStartIndex = if ($sync.StepTimings) { $sync.StepTimings.Count } else { 0 }
|
||||
|
||||
Write-WinUtilLog -Component $Name -Message "$Name job started."
|
||||
Write-WinUtilJobBanner -Message $label
|
||||
Step-WinUtilJob -Status "$label..." -Percent 0 -State "Normal" -Overlay "logo"
|
||||
|
||||
if ($DisableAppList -and (Test-WinUtilUIAlive)) {
|
||||
Invoke-WPFUIThread -ScriptBlock {
|
||||
if ($null -ne $sync.ItemsControl) { $sync.ItemsControl.IsEnabled = $false }
|
||||
}
|
||||
}
|
||||
|
||||
# Rebuilt from its text inside the runspace: a scriptblock carries the session state it was
|
||||
# defined in, and recreating it there binds it to the worker. The handle is discarded,
|
||||
# printing it puts an IAsyncResult table on the console on every button press.
|
||||
$null = Invoke-WPFRunspace -ParameterList @(
|
||||
("JobName", $Name),
|
||||
("JobLabel", $label),
|
||||
("JobBody", $ScriptBlock.ToString()),
|
||||
("JobParameters", $Parameters),
|
||||
("JobRestoresAppList", [bool]$DisableAppList),
|
||||
("JobToken", $jobToken),
|
||||
("TimingStartIndex", $timingStartIndex)
|
||||
) -ScriptBlock {
|
||||
param($JobName, $JobLabel, $JobBody, $JobParameters, $JobRestoresAppList, $JobToken, $TimingStartIndex)
|
||||
|
||||
# Marks this runspace as the one doing the work, so a pause holds here and not in
|
||||
# whoever asked for it
|
||||
$global:WinUtilIsJobWorker = $true
|
||||
$global:WinUtilJobErrorCount = 0
|
||||
$global:WinUtilJobWarningCount = 0
|
||||
|
||||
$jobClock = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
try {
|
||||
$body = [scriptblock]::Create($JobBody)
|
||||
|
||||
# A worker's warning and error streams buffer on a PowerShell object nobody reads.
|
||||
# Merging them into the output stream is what gets them to the log.
|
||||
& $body @JobParameters 2>&1 3>&1 | ForEach-Object {
|
||||
if ($_ -is [System.Management.Automation.WarningRecord]) {
|
||||
$global:WinUtilJobWarningCount++
|
||||
Write-WinUtilLog -Level "WARN" -Component $JobName -Message $_.Message
|
||||
} elseif ($_ -is [System.Management.Automation.ErrorRecord]) {
|
||||
Write-WinUtilErrorRecord -ErrorRecord $_ -Component $JobName -Context "Non-terminating error"
|
||||
}
|
||||
}
|
||||
|
||||
$jobClock.Stop()
|
||||
|
||||
# A step can fail without throwing, for example a registry write refused by policy.
|
||||
# The counter belongs to this worker runspace, so unrelated UI errors cannot change
|
||||
# this job's result while it is running.
|
||||
$newErrors = $global:WinUtilJobErrorCount
|
||||
$newWarnings = $global:WinUtilJobWarningCount
|
||||
if ($newErrors -gt 0) {
|
||||
Write-WinUtilLog -Level "WARN" -Component $JobName -Message "$JobName job finished in $($jobClock.ElapsedMilliseconds) ms with $newErrors error(s)."
|
||||
Write-WinUtilJobBanner -Message "$JobLabel finished with $newErrors error(s), see the log" -Level "ERROR"
|
||||
Step-WinUtilJob -Status "$JobName finished with $newErrors error(s)" -Percent 100 -State "Paused" -Overlay "warning"
|
||||
} elseif ($newWarnings -gt 0) {
|
||||
Write-WinUtilLog -Level "WARN" -Component $JobName -Message "$JobName job finished in $($jobClock.ElapsedMilliseconds) ms with $newWarnings warning(s)."
|
||||
Write-WinUtilJobBanner -Message "$JobLabel finished with $newWarnings warning(s), see the log"
|
||||
Step-WinUtilJob -Status "$JobName finished with $newWarnings warning(s)" -Percent 100 -State "Paused" -Overlay "warning"
|
||||
} else {
|
||||
Write-WinUtilLog -Component $JobName -Message "$JobName job finished in $($jobClock.ElapsedMilliseconds) ms."
|
||||
Write-WinUtilJobBanner -Message "$JobLabel finished"
|
||||
Step-WinUtilJob -Status "$JobName finished" -Percent 100 -State "None" -Overlay "checkmark"
|
||||
}
|
||||
} catch {
|
||||
$jobClock.Stop()
|
||||
# A leaf that logs before rethrowing marks that exact exception. Preserve the outer
|
||||
# context and stack without counting it twice; unrelated earlier errors do not qualify.
|
||||
$errorAlreadyReported = $_.Exception.Data["WinUtilErrorReported"] -eq $true
|
||||
Write-WinUtilErrorRecord -ErrorRecord $_ -Component $JobName -Context "$JobName failed after $($jobClock.ElapsedMilliseconds) ms" -DetailOnly:$errorAlreadyReported
|
||||
Write-WinUtilJobBanner -Message "$JobLabel failed: $($_.Exception.Message)" -Level "ERROR"
|
||||
Step-WinUtilJob -Status "$JobName failed" -Percent 100 -State "Error" -Overlay "warning"
|
||||
} finally {
|
||||
$jobResult = [pscustomobject]@{
|
||||
Token = $JobToken
|
||||
Errors = $global:WinUtilJobErrorCount
|
||||
Warnings = $global:WinUtilJobWarningCount
|
||||
}
|
||||
|
||||
# Pool runspaces are reused, so leaving this set would make the next piece of
|
||||
# background work on this runspace believe it is a job worker
|
||||
$global:WinUtilIsJobWorker = $false
|
||||
$global:WinUtilJobErrorCount = 0
|
||||
$global:WinUtilJobWarningCount = 0
|
||||
|
||||
Write-WinUtilTimingSummary -Scope $JobName -TotalMilliseconds $jobClock.ElapsedMilliseconds -StartIndex $TimingStartIndex
|
||||
|
||||
# A worker the watchdog cut off can reach here after the next job claimed the slot,
|
||||
# and everything below releases shared state.
|
||||
$stillOwns = $false
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
$stillOwns = $sync.ActiveJobToken -eq $JobToken
|
||||
if ($stillOwns) {
|
||||
$sync.LastJobResult = $jobResult
|
||||
}
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
|
||||
if ($stillOwns) {
|
||||
try {
|
||||
if ($JobRestoresAppList -and (Test-WinUtilUIAlive)) {
|
||||
Invoke-WPFUIThread -ScriptBlock {
|
||||
if ($null -ne $sync.ItemsControl) { $sync.ItemsControl.IsEnabled = $true }
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component $JobName -Message "Could not restore the app list after $JobName finished: $($_.Exception.Message)"
|
||||
} finally {
|
||||
# Dispatcher shutdown can race the alive check and abort the restore call.
|
||||
# The worker is still finished, so its slot must always be released.
|
||||
$null = Clear-WinUtilActiveJob -Token $JobToken
|
||||
}
|
||||
} else {
|
||||
Write-WinUtilLog -Level "WARN" -Component $JobName -Message "$JobName unwound after another job had started; leaving its state alone."
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
$scheduleError = $_
|
||||
try {
|
||||
Write-WinUtilErrorRecord -ErrorRecord $scheduleError -Component $Name -Context "Could not schedule $Name"
|
||||
$sync.LastJobResult = [pscustomobject]@{ Token = $jobToken; Errors = 1; Warnings = 0 }
|
||||
Write-WinUtilJobBanner -Message "$label could not start" -Level "ERROR"
|
||||
Step-WinUtilJob -Status "$Name could not start" -Percent 100 -State "Error" -Overlay "warning"
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component $Name -Message "Could not report that $Name failed to start: $($_.Exception.Message)"
|
||||
} finally {
|
||||
try {
|
||||
if ($DisableAppList -and (Test-WinUtilUIAlive)) {
|
||||
Invoke-WPFUIThread -ScriptBlock {
|
||||
if ($null -ne $sync.ItemsControl) { $sync.ItemsControl.IsEnabled = $true }
|
||||
}
|
||||
}
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component $Name -Message "Could not restore the app list after $Name failed to start: $($_.Exception.Message)"
|
||||
} finally {
|
||||
$null = Clear-WinUtilActiveJob -Token $jobToken
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function Clear-WinUtilActiveJob {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Releases the active job slot, clearing its name and its token together
|
||||
|
||||
.DESCRIPTION
|
||||
A token left set still matches a later run and blocks new work.
|
||||
|
||||
.PARAMETER Token
|
||||
Release only if this run still owns the slot. Omit to release unconditionally.
|
||||
#>
|
||||
param([string]$Token)
|
||||
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
if (-not $Token -or $sync.ActiveJobToken -eq $Token) {
|
||||
$sync.ActiveJobToken = $null
|
||||
$sync.ActiveJob = $null
|
||||
return $true
|
||||
}
|
||||
return $false
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
function Start-WinUtilTabWarmup {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Builds the tabs the user has not opened yet, while the interface is idle
|
||||
|
||||
.DESCRIPTION
|
||||
Tab content has to be built on the interface thread, so a tab that is still empty
|
||||
when it is first clicked makes that click pay for the build. Queueing the builds
|
||||
moves that cost to where nothing is waiting on it.
|
||||
|
||||
Queued at background priority rather than idle priority. At idle priority this never
|
||||
ran until the app list had finished, which is the exact window in which a tab the
|
||||
user clicks is still empty and costs a full build to open.
|
||||
#>
|
||||
|
||||
# Win11ISO is left out: building it runs the existing work check, which raises the resume
|
||||
# prompt while the user is on another tab. That check belongs to opening the tab, not warming
|
||||
# it.
|
||||
$pending = [System.Collections.Queue]::new()
|
||||
foreach ($tab in @("Tweaks", "Config", "AppX")) {
|
||||
if (-not $sync.InitializedTabs[$tab]) {
|
||||
$pending.Enqueue($tab)
|
||||
}
|
||||
}
|
||||
|
||||
if ($pending.Count -eq 0) {
|
||||
return
|
||||
}
|
||||
|
||||
$sync.TabWarmupQueue = $pending
|
||||
Start-WinUtilBackgroundQueue -Name "TabWarmup" -Queue $pending -Step {
|
||||
param($Tab)
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "warm $Tab tab" -ScriptBlock {
|
||||
Initialize-WinUtilTabContent -TabName $Tab -Yield
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,530 @@
|
||||
function Start-WinUtilUserInterface {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Builds the WinUtil window, wires its event handlers and runs it to completion
|
||||
|
||||
.DESCRIPTION
|
||||
This is the whole interface. It runs on the dedicated STA interface runspace that
|
||||
main.ps1 starts, so the thread that owns the window does nothing but paint and
|
||||
dispatch: every long operation goes to the worker pool through Start-WinUtilJob.
|
||||
|
||||
The call blocks until the window is closed, and the interface runspace is the only
|
||||
place that is allowed to touch controls directly.
|
||||
#>
|
||||
|
||||
$buildClock = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "load WPF assemblies" -ScriptBlock {
|
||||
[void][System.Reflection.Assembly]::LoadWithPartialName('presentationframework')
|
||||
}
|
||||
|
||||
[xml]$XAML = $inputXML
|
||||
|
||||
# Read the XAML file
|
||||
$readerOperationSuccessful = $false # There's more cases of failure then success.
|
||||
$readerFailure = $null
|
||||
$reader = (New-Object System.Xml.XmlNodeReader $xaml)
|
||||
try {
|
||||
Measure-WinUtilStep -Scope "UI" -Name "parse XAML" -ScriptBlock {
|
||||
$sync["Form"] = [Windows.Markup.XamlReader]::Load( $reader )
|
||||
}
|
||||
$readerOperationSuccessful = $true
|
||||
} catch [System.Management.Automation.MethodInvocationException] {
|
||||
$readerFailure = $_
|
||||
Write-Host "We ran into a problem with the XAML code. Check the syntax for this control..." -ForegroundColor Red
|
||||
Write-Host $error[0].Exception.Message -ForegroundColor Red
|
||||
|
||||
If ($error[0].Exception.Message -like "*button*") {
|
||||
write-Host "Ensure your <button in the `$inputXML does NOT have a Click=ButtonClick property. PS can't handle this`n`n`n`n" -ForegroundColor Red
|
||||
}
|
||||
} catch {
|
||||
$readerFailure = $_
|
||||
Write-Host "Unable to load Windows.Markup.XamlReader. Double-check syntax and ensure .net is installed." -ForegroundColor Red
|
||||
}
|
||||
|
||||
if (-NOT ($readerOperationSuccessful)) {
|
||||
Write-Host "Failed to parse xaml content using Windows.Markup.XamlReader's Load Method." -ForegroundColor Red
|
||||
Write-Host "Quitting WinUtil..." -ForegroundColor Red
|
||||
Write-WinUtilLog -Level "ERROR" -Component "UI" -Message "Failed to parse the XAML content. WinUtil cannot start."
|
||||
throw [System.InvalidOperationException]::new("Failed to parse the XAML content. WinUtil cannot start.", $readerFailure.Exception)
|
||||
}
|
||||
|
||||
# Setup the Window to follow listen for windows Theme Change events and update the winutil theme
|
||||
# throttle logic needed, because windows seems to send more than one theme change event per change
|
||||
$themeState = @{ LastChange = [datetime]::MinValue }
|
||||
$debounceInterval = [timespan]::FromSeconds(2)
|
||||
$sync.Form.Add_Loaded({
|
||||
$interopHelper = New-Object System.Windows.Interop.WindowInteropHelper $sync.Form
|
||||
$hwndSource = [System.Windows.Interop.HwndSource]::FromHwnd($interopHelper.Handle)
|
||||
$hwndSource.AddHook({
|
||||
param (
|
||||
[System.IntPtr]$hwnd,
|
||||
[int]$msg,
|
||||
[System.IntPtr]$wParam,
|
||||
[System.IntPtr]$lParam,
|
||||
[ref]$handled
|
||||
)
|
||||
$null = $hwnd, $wParam, $lParam
|
||||
# Check for the Event WM_SETTINGCHANGE (0x1001A) and validate that Button shows the icon for "Auto" => [char]0xF08C
|
||||
if (($msg -eq 0x001A) -and $sync.ThemeButton.Content -eq [char]0xF08C) {
|
||||
$currentTime = [datetime]::Now
|
||||
if ($currentTime - $themeState.LastChange -gt $debounceInterval) {
|
||||
Invoke-WinutilThemeChange -theme "Auto"
|
||||
$themeState.LastChange = $currentTime
|
||||
# [ref] out-parameter: assigning to $handled would only replace the local
|
||||
$handled.Value = $true
|
||||
}
|
||||
}
|
||||
return 0
|
||||
})
|
||||
})
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "apply theme" -ScriptBlock {
|
||||
Invoke-WinutilThemeChange -theme $sync.preferences.theme
|
||||
}
|
||||
|
||||
# No tab content is built before first paint. Invoke-WPFTab builds whichever tab it
|
||||
# activates, and ContentRendered activates the default one.
|
||||
$sync.InitializedTabs = @{}
|
||||
|
||||
#===========================================================================
|
||||
# Store Form Objects In PowerShell
|
||||
#===========================================================================
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "map named controls" -ScriptBlock {
|
||||
$xaml.SelectNodes("//*[@Name]") | ForEach-Object {$sync["$("$($psitem.Name)")"] = $sync["Form"].FindName($psitem.Name)}
|
||||
}
|
||||
|
||||
# Built here so it carries this runspace's session state: posted work then runs as ordinary
|
||||
# interface code, not a much slower cross-runspace nested pipeline. Invoke-WPFUIThread is the
|
||||
# caller-facing side.
|
||||
$sync.UIDispatchDelegate = [System.Func[object, object]]{
|
||||
param($Work)
|
||||
|
||||
try {
|
||||
$body = [scriptblock]::Create($Work.Body)
|
||||
$parameters = $Work.Parameters
|
||||
if ($parameters -and $parameters.Count -gt 0) {
|
||||
& $body @parameters
|
||||
} else {
|
||||
& $body
|
||||
}
|
||||
} catch {
|
||||
if ($Work.PropagateErrors) {
|
||||
throw
|
||||
}
|
||||
# Fire-and-forget work has no waiting caller to report its failure.
|
||||
Write-WinUtilErrorRecord -ErrorRecord $_ -Component "UI" -Context "Interface work"
|
||||
}
|
||||
}
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "wire static button clicks" -ScriptBlock {
|
||||
# CheckBox and RadioButton also derive from ButtonBase, so the exact type name is what
|
||||
# decides, not -is
|
||||
$clickableTypes = [System.Collections.Generic.HashSet[string]]::new([string[]]@("Button", "ToggleButton"), [StringComparer]::OrdinalIgnoreCase)
|
||||
$alreadyWired = [System.Collections.Generic.HashSet[string]]::new([string[]]@($sync.Buttons), [StringComparer]::OrdinalIgnoreCase)
|
||||
|
||||
$clickHandler = {
|
||||
[System.Object]$Sender = $args[0]
|
||||
Invoke-WPFButton $Sender.name
|
||||
}
|
||||
|
||||
foreach ($entry in @($sync.GetEnumerator())) {
|
||||
$control = $entry.Value
|
||||
if ($null -eq $control -or -not $clickableTypes.Contains($control.GetType().Name)) {
|
||||
continue
|
||||
}
|
||||
if (-not $alreadyWired.Add([string]$entry.Key)) {
|
||||
continue
|
||||
}
|
||||
|
||||
$control.Add_Click($clickHandler)
|
||||
$sync.Buttons.Add($entry.Key) | Out-Null
|
||||
}
|
||||
}
|
||||
|
||||
#===========================================================================
|
||||
# Setup and Show the Form
|
||||
#===========================================================================
|
||||
|
||||
# Progress bar in taskbaritem > Set-WinUtilProgressbar
|
||||
$sync["Form"].TaskbarItemInfo = New-Object System.Windows.Shell.TaskbarItemInfo
|
||||
Set-WinUtilTaskbaritem -state "None"
|
||||
|
||||
# Wired before the window is shown, so work queued during startup already knows to stand
|
||||
# aside for anything the user does
|
||||
Register-WinUtilInputWatch
|
||||
|
||||
# Set the titlebar
|
||||
$sync["Form"].title = $sync["Form"].title + " " + $sync.version
|
||||
# Set the commands that will run when the form is closed
|
||||
$sync["Form"].Add_Closing({
|
||||
param($eventSender, $closingArgs)
|
||||
$null = $eventSender
|
||||
|
||||
# The pool cannot be torn down under work that is still running: the runspace error that
|
||||
# follows is unhandled and ends the process
|
||||
if ($sync.ActiveJob -and -not $sync.ForceClose) {
|
||||
$closingArgs.Cancel = $true
|
||||
Invoke-WinUtilCloseRequest -RunningJob $sync.ActiveJob
|
||||
return
|
||||
}
|
||||
|
||||
# Work that is meant to outlive the window needs the pool it is running on. main.ps1
|
||||
# waits for it and shuts the pool down once it is done.
|
||||
if ($sync.FinishInConsole) {
|
||||
Write-WinUtilLog -Component "UI" -Message "Window closing, leaving $($sync.ActiveJob) to finish in the console."
|
||||
return
|
||||
}
|
||||
|
||||
# main.ps1 owns pool shutdown after the window has finished closing. Doing it from this
|
||||
# dispatcher callback can deadlock with a worker that is in its UI-thread cleanup block.
|
||||
Write-WinUtilLog -Component "UI" -Message "Window closing; the main thread will shut down the worker pool."
|
||||
})
|
||||
|
||||
# Attach the event handler to the Click event
|
||||
$sync.SearchBarClearButton.Add_Click({
|
||||
$sync.SearchBar.Text = ""
|
||||
$sync.SearchBarClearButton.Visibility = "Collapsed"
|
||||
|
||||
# Focus the search bar after clearing the text
|
||||
$sync.SearchBar.Focus()
|
||||
$sync.SearchBar.SelectAll()
|
||||
})
|
||||
|
||||
# add some shortcuts for people that don't like clicking
|
||||
function Invoke-WinUtilFontScaleStep([double]$Step) { $sync.FontScalingSlider.Value = [math]::Max(0.75, [math]::Min(2.0, $sync.FontScalingSlider.Value + $Step)); Invoke-WinUtilFontScaling -ScaleFactor $sync.FontScalingSlider.Value }
|
||||
|
||||
$commonKeyEvents = {
|
||||
if ($sync.ActiveJob) {
|
||||
return
|
||||
}
|
||||
|
||||
# Handle key presses of single keys
|
||||
switch ($_.Key) {
|
||||
"Escape" { $sync.SearchBar.Text = "" }
|
||||
}
|
||||
# Handle Alt key combinations for navigation
|
||||
if ($_.KeyboardDevice.Modifiers -eq "Alt") {
|
||||
$keyEventArgs = $_
|
||||
switch ($_.SystemKey) {
|
||||
"I" { Invoke-WPFButton "WPFTab1BT"; $keyEventArgs.Handled = $true } # Navigate to Install tab and suppress Windows Warning Sound
|
||||
"T" { Invoke-WPFButton "WPFTab2BT"; $keyEventArgs.Handled = $true } # Navigate to Tweaks tab
|
||||
"C" { Invoke-WPFButton "WPFTab3BT"; $keyEventArgs.Handled = $true } # Navigate to Config tab
|
||||
"U" { Invoke-WPFButton "WPFTab4BT"; $keyEventArgs.Handled = $true } # Navigate to Updates tab
|
||||
"W" { Invoke-WPFButton "WPFTab5BT"; $keyEventArgs.Handled = $true } # Navigate to Win11ISO tab
|
||||
}
|
||||
}
|
||||
# Handle Ctrl key combinations for specific actions
|
||||
if ($_.KeyboardDevice.Modifiers -eq "Ctrl") {
|
||||
$keyEventArgs = $_
|
||||
switch ($_.Key) {
|
||||
"F" { $sync.SearchBar.Focus() } # Focus on the search bar
|
||||
"Q" { $this.Close() } # Close the application
|
||||
}
|
||||
}
|
||||
$ctrlShiftModifiers = [Windows.Input.ModifierKeys]::Control -bor [Windows.Input.ModifierKeys]::Shift
|
||||
if ($_.KeyboardDevice.Modifiers -eq "Ctrl" -or $_.KeyboardDevice.Modifiers -eq $ctrlShiftModifiers) {
|
||||
$keyEventArgs = $_
|
||||
switch ($_.Key) {
|
||||
{ $_ -in "OemPlus", "Add" } { Invoke-WinUtilFontScaleStep 0.05; $keyEventArgs.Handled = $true }
|
||||
{ $_ -in "OemMinus", "Subtract" } { Invoke-WinUtilFontScaleStep -0.05; $keyEventArgs.Handled = $true }
|
||||
}
|
||||
}
|
||||
}
|
||||
$sync["Form"].Add_PreViewKeyDown($commonKeyEvents)
|
||||
$sync["Form"].Add_PreviewMouseWheel({
|
||||
if ([Windows.Input.Keyboard]::Modifiers -eq "Ctrl") { Invoke-WinUtilFontScaleStep $(if ($_.Delta -gt 0) { 0.05 } else { -0.05 }); $_.Handled = $true }
|
||||
})
|
||||
|
||||
$sync["Form"].Add_MouseLeftButtonDown({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings", "Theme", "FontScaling")
|
||||
$sync["Form"].DragMove()
|
||||
})
|
||||
|
||||
$sync["Form"].Add_MouseDoubleClick({
|
||||
if ($_.OriginalSource.Name -eq "NavDockPanel" -or
|
||||
$_.OriginalSource.Name -eq "GridBesideNavDockPanel") {
|
||||
if ($sync["Form"].WindowState -eq [Windows.WindowState]::Normal) {
|
||||
[Windows.SystemCommands]::MaximizeWindow($sync.Form)
|
||||
}
|
||||
else{
|
||||
[Windows.SystemCommands]::RestoreWindow($sync.Form)
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
$sync["Form"].Add_Deactivated({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings", "Theme", "FontScaling")
|
||||
})
|
||||
|
||||
$sync["Form"].Add_ContentRendered({
|
||||
# Load the Windows Forms assembly
|
||||
Add-Type -AssemblyName System.Windows.Forms
|
||||
$primaryScreen = [System.Windows.Forms.Screen]::PrimaryScreen
|
||||
# Check if the primary screen is found
|
||||
if ($primaryScreen) {
|
||||
# Extract screen width and height for the primary monitor
|
||||
$screenWidth = $primaryScreen.Bounds.Width
|
||||
$screenHeight = $primaryScreen.Bounds.Height
|
||||
$sync.Form.MinWidth = [Math]::Min([double]$sync.Form.MinWidth, [double]$screenWidth)
|
||||
|
||||
# Compare with the primary monitor size
|
||||
if ($sync.Form.ActualWidth -gt $screenWidth -or $sync.Form.ActualHeight -gt $screenHeight) {
|
||||
$sync.Form.Left = 0
|
||||
$sync.Form.Top = 0
|
||||
$sync.Form.Width = $screenWidth
|
||||
$sync.Form.Height = $screenHeight
|
||||
}
|
||||
}
|
||||
|
||||
if ($PARAM_OFFLINE) {
|
||||
# Show offline banner
|
||||
$sync.WPFOfflineBanner.Visibility = [System.Windows.Visibility]::Visible
|
||||
|
||||
# Disable the install tab
|
||||
$sync.WPFTab1BT.IsEnabled = $false
|
||||
$sync.WPFTab1BT.Opacity = 0.5
|
||||
$sync.WPFTab1BT.ToolTip = "Internet connection required for installing applications."
|
||||
|
||||
# The install action buttons are generated with the Install tab, so
|
||||
# Initialize-WinUtilInstallTabControls disables them when that tab is built
|
||||
|
||||
# Show offline indicator
|
||||
Write-Host "Offline mode detected - Install tab disabled." -ForegroundColor Yellow
|
||||
|
||||
# Optionally switch to a different tab if install tab was going to be default
|
||||
Invoke-WPFTab "WPFTab2BT" -Yield # Switch to Tweaks tab instead
|
||||
}
|
||||
else {
|
||||
# Online - ensure install tab is enabled
|
||||
$sync.WPFTab1BT.IsEnabled = $true
|
||||
$sync.WPFTab1BT.Opacity = 1.0
|
||||
$sync.WPFTab1BT.ToolTip = $null
|
||||
Invoke-WPFTab "WPFTab1BT" -Yield # Default to install tab
|
||||
}
|
||||
|
||||
$sync["Form"].Focus()
|
||||
$sync["Form"].Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{ Initialize-WinUtilRunspacePool | Out-Null }) | Out-Null
|
||||
$sync["Form"].Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{
|
||||
Set-WinUtilTaskbaritem -overlay "logo"
|
||||
}) | Out-Null
|
||||
$sync["Form"].Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Background, [action]{ Start-WinUtilTabWarmup }) | Out-Null
|
||||
})
|
||||
|
||||
# The SearchBarTimer is used to delay the search operation until the user has stopped typing for a short period
|
||||
# This prevents the ui from stuttering when the user types quickly as it dosnt need to update the ui for every keystroke
|
||||
|
||||
$searchBarTimer = New-Object System.Windows.Threading.DispatcherTimer
|
||||
$searchBarTimer.Interval = [TimeSpan]::FromMilliseconds(300)
|
||||
$searchBarTimer.IsEnabled = $false
|
||||
|
||||
$searchBarTimer.add_Tick({
|
||||
$searchBarTimer.Stop()
|
||||
switch ($sync.currentTab) {
|
||||
"Install" {
|
||||
Find-AppsByNameOrDescription -SearchString $sync.SearchBar.Text -Categories $sync.SelectedAppCategories.ToArray()
|
||||
}
|
||||
"Tweaks" {
|
||||
Find-TweaksByNameOrDescription -SearchString $sync.SearchBar.Text
|
||||
}
|
||||
"AppX" {
|
||||
Find-TweaksByNameOrDescription -SearchString $sync.SearchBar.Text
|
||||
}
|
||||
}
|
||||
})
|
||||
$sync["SearchBar"].Add_TextChanged({
|
||||
if ($sync.SearchBar.Text -ne "") {
|
||||
$sync.SearchBarClearButton.Visibility = "Visible"
|
||||
$sync.SearchBarIcon.Visibility = "Collapsed"
|
||||
} else {
|
||||
$sync.SearchBarClearButton.Visibility = "Collapsed"
|
||||
$sync.SearchBarIcon.Visibility = "Visible"
|
||||
}
|
||||
|
||||
if ($searchBarTimer.IsEnabled) {
|
||||
$searchBarTimer.Stop()
|
||||
}
|
||||
$searchBarTimer.Start()
|
||||
})
|
||||
|
||||
# Category filter chips. The chip carries its category in Tag, so one handler covers all of them.
|
||||
$sync.AppCategoryChips = @(
|
||||
@{ Name = "WPFSearchChipAll"; Category = "" }
|
||||
@{ Name = "WPFSearchChipBrowsers"; Category = "Browsers" }
|
||||
@{ Name = "WPFSearchChipCommunications"; Category = "Communications" }
|
||||
@{ Name = "WPFSearchChipDevelopment"; Category = "Development" }
|
||||
@{ Name = "WPFSearchChipDocument"; Category = "Document" }
|
||||
@{ Name = "WPFSearchChipGames"; Category = "Games" }
|
||||
@{ Name = "WPFSearchChipMicrosoftTools"; Category = "Microsoft Tools" }
|
||||
@{ Name = "WPFSearchChipMultimediaTools"; Category = "Multimedia Tools" }
|
||||
@{ Name = "WPFSearchChipProTools"; Category = "Pro Tools" }
|
||||
@{ Name = "WPFSearchChipSelfhostedTools"; Category = "Selfhosted Tools" }
|
||||
@{ Name = "WPFSearchChipUtilities"; Category = "Utilities" }
|
||||
)
|
||||
$sync.SelectedAppCategories = [System.Collections.Generic.List[string]]::new()
|
||||
|
||||
foreach ($appCategoryChip in $sync.AppCategoryChips) {
|
||||
$sync[$appCategoryChip.Name].Tag = $appCategoryChip.Category
|
||||
$sync[$appCategoryChip.Name].Add_Click({ Invoke-WinUtilAppCategoryChip -Chip $this })
|
||||
}
|
||||
|
||||
$sync["Form"].Add_Loaded({
|
||||
param($e)
|
||||
$null = $e
|
||||
$sync.Form.MinWidth = "1150"
|
||||
$sync["Form"].MaxWidth = [Double]::PositiveInfinity
|
||||
$sync["Form"].MaxHeight = [Double]::PositiveInfinity
|
||||
})
|
||||
|
||||
Measure-WinUtilStep -Scope "UI" -Name "build nav logo" -ScriptBlock {
|
||||
$NavLogoPanel = $sync["Form"].FindName("NavLogoPanel")
|
||||
$NavLogoPanel.Children.Add((Invoke-WinUtilAssets -Type "logo" -Size 25)) | Out-Null
|
||||
}
|
||||
|
||||
$sync["Form"].Add_Activated({
|
||||
Set-WinUtilTaskbaritem -overlay "logo"
|
||||
})
|
||||
|
||||
$sync["ThemeButton"].Add_Click({
|
||||
Invoke-WPFPopup -PopupActionTable @{ "Settings" = "Hide"; "Theme" = "Toggle"; "FontScaling" = "Hide" }
|
||||
})
|
||||
$sync["AutoThemeMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Theme")
|
||||
Invoke-WinutilThemeChange -theme "Auto"
|
||||
})
|
||||
$sync["DarkThemeMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Theme")
|
||||
Invoke-WinutilThemeChange -theme "Dark"
|
||||
})
|
||||
$sync["LightThemeMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Theme")
|
||||
Invoke-WinutilThemeChange -theme "Light"
|
||||
})
|
||||
|
||||
$sync["SettingsButton"].Add_Click({
|
||||
Invoke-WPFPopup -PopupActionTable @{ "Settings" = "Toggle"; "Theme" = "Hide"; "FontScaling" = "Hide" }
|
||||
})
|
||||
$sync["ImportMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
Invoke-WPFImpex -type "import"
|
||||
})
|
||||
$sync["ExportMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
Invoke-WPFImpex -type "export"
|
||||
})
|
||||
$sync["ExportEnvironmentReportMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
Invoke-WPFExportEnvironmentReport
|
||||
})
|
||||
$sync["AboutMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
|
||||
$authorInfo = @"
|
||||
Author : <a href="https://github.com/ChrisTitusTech">@ChrisTitusTech</a>
|
||||
UI : <a href="https://github.com/MyDrift-user">@MyDrift-user</a>, <a href="https://github.com/Marterich">@Marterich</a>
|
||||
Runspace : <a href="https://github.com/DeveloperDurp">@DeveloperDurp</a>, <a href="https://github.com/Marterich">@Marterich</a>
|
||||
GitHub : <a href="https://github.com/ChrisTitusTech/winutil">ChrisTitusTech/winutil</a>
|
||||
Version : <a href="https://github.com/ChrisTitusTech/winutil/releases/tag/$($sync.version)">$($sync.version)</a>
|
||||
"@
|
||||
Show-CustomDialog -Title "About" -Message $authorInfo
|
||||
})
|
||||
$sync["DocumentationMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
Start-Process "https://winutil.christitus.com/"
|
||||
})
|
||||
$sync["SponsorMenuItem"].Add_Click({
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("Settings")
|
||||
|
||||
$authorInfo = @"
|
||||
<a href="https://github.com/sponsors/ChrisTitusTech">Current sponsors for ChrisTitusTech:</a>
|
||||
"@
|
||||
$authorInfo += "`n"
|
||||
try {
|
||||
$sponsors = Invoke-WinUtilSponsors
|
||||
foreach ($sponsor in $sponsors) {
|
||||
$authorInfo += "<a href=`"https://github.com/sponsors/ChrisTitusTech`">$sponsor</a>`n"
|
||||
}
|
||||
} catch {
|
||||
$authorInfo += "An error occurred while fetching or processing the sponsors: $_`n"
|
||||
}
|
||||
Show-CustomDialog -Title "Sponsors" -Message $authorInfo -EnableScroll $true
|
||||
})
|
||||
|
||||
# Font Scaling Event Handlers
|
||||
$sync["FontScalingButton"].Add_Click({
|
||||
Invoke-WPFPopup -PopupActionTable @{ "Settings" = "Hide"; "Theme" = "Hide"; "FontScaling" = "Toggle" }
|
||||
})
|
||||
|
||||
$sync["FontScalingSlider"].Add_ValueChanged({
|
||||
param($slider)
|
||||
$percentage = [math]::Round($slider.Value * 100)
|
||||
$sync.FontScalingValue.Text = "$percentage%"
|
||||
})
|
||||
|
||||
$sync["FontScalingResetButton"].Add_Click({
|
||||
$sync.FontScalingSlider.Value = 1.0
|
||||
$sync.FontScalingValue.Text = "100%"
|
||||
})
|
||||
|
||||
$sync["FontScalingApplyButton"].Add_Click({
|
||||
$scaleFactor = $sync.FontScalingSlider.Value
|
||||
Invoke-WinUtilFontScaling -ScaleFactor $scaleFactor
|
||||
Invoke-WPFPopup -Action "Hide" -Popups @("FontScaling")
|
||||
})
|
||||
|
||||
# Win11ISO Tab button handlers
|
||||
$sync["WPFWin11ISOBrowseButton"].Add_Click({
|
||||
Invoke-WinUtilISOBrowse
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISODownloadLink"].Add_Click({
|
||||
Start-Process "https://www.microsoft.com/software-download/windows11"
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOMountButton"].Add_Click({
|
||||
Invoke-WinUtilISOMountAndVerify
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOModifyButton"].Add_Click({
|
||||
Invoke-WinUtilISOModify
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOChooseISOButton"].Add_Click({
|
||||
$sync["WPFWin11ISOOptionUSB"].Visibility = "Collapsed"
|
||||
Invoke-WinUtilISOExport
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOChooseUSBButton"].Add_Click({
|
||||
$sync["WPFWin11ISOOptionUSB"].Visibility = "Visible"
|
||||
Invoke-WinUtilISORefreshUSBDrives
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISORefreshUSBButton"].Add_Click({
|
||||
Invoke-WinUtilISORefreshUSBDrives
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOWriteUSBButton"].Add_Click({
|
||||
Invoke-WinUtilISOWriteUSB
|
||||
})
|
||||
|
||||
$sync["WPFWin11ISOCleanResetButton"].Add_Click({
|
||||
Invoke-WinUtilISOCleanAndReset
|
||||
})
|
||||
|
||||
$buildClock.Stop()
|
||||
Write-WinUtilLog -Component "UI" -Message "Interface built in $($buildClock.ElapsedMilliseconds) ms, showing the window."
|
||||
Write-WinUtilTimingSummary -Scope "UI" -TotalMilliseconds $buildClock.ElapsedMilliseconds
|
||||
|
||||
# Input priority runs behind everything already queued, so this fires at the first moment
|
||||
# the window could actually service a click
|
||||
$sync["Form"].Dispatcher.BeginInvoke([System.Windows.Threading.DispatcherPriority]::Input, [action]{
|
||||
$sinceStart = [int]((Get-Date) - $sync.StartedAt).TotalMilliseconds
|
||||
Write-WinUtilLog -Component "UI" -Message "timing: interface ready for input $sinceStart ms after start."
|
||||
}) | Out-Null
|
||||
|
||||
$sync["Form"].ShowDialog() | Out-Null
|
||||
|
||||
# ShowDialog returns once the window is gone; stop the dispatcher so this runspace can close
|
||||
[System.Windows.Threading.Dispatcher]::CurrentDispatcher.InvokeShutdown()
|
||||
}
|
||||
@@ -0,0 +1,104 @@
|
||||
function Step-WinUtilJob {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Advances a job to its next reportable point, honouring a pause or stop on the way
|
||||
|
||||
.DESCRIPTION
|
||||
Every loop calls this, so it is the one point a run reliably passes between steps and
|
||||
therefore the only place it can be held or ended without cutting into a command in
|
||||
flight. It blocks while the run is paused and throws OperationCanceledException once
|
||||
a stop is asked for, so calling it from a finally, or from a catch already reporting
|
||||
a failure, re-raises that stop. The job layer clears the flags before its own finish
|
||||
reporting for that reason.
|
||||
|
||||
Drives the progress bar and taskbar item together and does nothing without a window,
|
||||
so job bodies need no UI checks. The update is posted rather than waited on: a job
|
||||
reporting per item would otherwise stall on the interface thread each time.
|
||||
|
||||
.PARAMETER Status
|
||||
Text for the progress label
|
||||
|
||||
.PARAMETER Percent
|
||||
Completion between 0 and 100
|
||||
|
||||
.PARAMETER State
|
||||
Taskbar state. Normal while working, Error on failure, None when finished.
|
||||
|
||||
.PARAMETER Overlay
|
||||
Taskbar overlay icon: logo, checkmark, warning or None
|
||||
|
||||
.PARAMETER Hide
|
||||
Clears and hides the progress bar. Used when leaving a finished job behind rather
|
||||
than while one is running.
|
||||
#>
|
||||
param(
|
||||
[string]$Status,
|
||||
[int]$Percent = -1,
|
||||
[ValidateSet("Normal", "Error", "Paused", "Indeterminate", "None")]
|
||||
[string]$State,
|
||||
[string]$Overlay,
|
||||
[switch]$Hide
|
||||
)
|
||||
|
||||
# With no window every update is thrown away, and a window closed over running work counts
|
||||
# as none: its dispatcher accepts posts and discards them. The console is what is left.
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
if (-not $Hide) {
|
||||
Write-WinUtilConsoleProgress -Status $Status -Percent $Percent
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
Invoke-WPFUIThread -Async -Parameters @{
|
||||
Status = $Status
|
||||
Percent = [Math]::Min([Math]::Max($Percent, -1), 100)
|
||||
State = $State
|
||||
Overlay = $Overlay
|
||||
HideBar = [bool]$Hide
|
||||
HasStatus = $PSBoundParameters.ContainsKey('Status')
|
||||
HasState = $PSBoundParameters.ContainsKey('State')
|
||||
HasOverlay = $PSBoundParameters.ContainsKey('Overlay')
|
||||
} -ScriptBlock {
|
||||
param($Status, $Percent, $State, $Overlay, $HideBar, $HasStatus, $HasState, $HasOverlay)
|
||||
|
||||
if ($HideBar) {
|
||||
$sync.WPFTweaksProgressBar.Visibility = [Windows.Visibility]::Collapsed
|
||||
$sync.WPFTweaksProgressLabel.Text = ""
|
||||
$sync.WPFTweaksProgressLabel.ToolTip = $null
|
||||
$sync.WPFTweaksProgressValue.Value = 0
|
||||
return
|
||||
}
|
||||
|
||||
$hasPercent = $Percent -ge 0
|
||||
|
||||
if ($HasStatus -or $hasPercent) {
|
||||
$sync.WPFTweaksProgressBar.Visibility = [Windows.Visibility]::Visible
|
||||
}
|
||||
if ($HasStatus) {
|
||||
$sync.WPFTweaksProgressLabel.Text = $Status
|
||||
$sync.WPFTweaksProgressLabel.ToolTip = $Status
|
||||
}
|
||||
if ($hasPercent) {
|
||||
$sync.WPFTweaksProgressValue.Value = $Percent
|
||||
$sync.Form.TaskbarItemInfo.ProgressValue = $Percent / 100
|
||||
}
|
||||
if ($HasState) {
|
||||
# Pulse in place at whatever progress has been reached. IsIndeterminate would make
|
||||
# WPF discard Value and fill the whole bar, which reads as finished.
|
||||
$sync.WPFTweaksProgressValue.Tag = if ($State -eq "Indeterminate") { "Pulse" } else { $null }
|
||||
|
||||
# By resource reference rather than a fixed brush, so switching theme repaints it
|
||||
$barColor = switch ($State) {
|
||||
"Error" { "ProgressBarErrorColor" }
|
||||
"Paused" { "ProgressBarWarningColor" }
|
||||
default { "ProgressBarForegroundColor" }
|
||||
}
|
||||
$sync.WPFTweaksProgressValue.SetResourceReference([Windows.Controls.Control]::ForegroundProperty, $barColor)
|
||||
|
||||
Set-WinUtilTaskbaritem -state $State
|
||||
}
|
||||
if ($HasOverlay) {
|
||||
Set-WinUtilTaskbaritem -overlay $Overlay
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,150 @@
|
||||
function Test-WinUtilShellRunning {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Whether one instance is still running, treating a disposed one as finished
|
||||
#>
|
||||
param($PowerShell)
|
||||
|
||||
try {
|
||||
return $PowerShell.InvocationStateInfo.State -in @(
|
||||
[System.Management.Automation.PSInvocationState]::Running,
|
||||
[System.Management.Automation.PSInvocationState]::Stopping
|
||||
)
|
||||
} catch {
|
||||
return $false
|
||||
}
|
||||
}
|
||||
|
||||
function Register-WinUtilActiveShell {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Records a PowerShell instance that is running on the worker pool
|
||||
|
||||
.DESCRIPTION
|
||||
An instance still queued when the pool closes starts on a closing runspace, throws on
|
||||
a thread pool thread where nothing catches, and takes the process down with it.
|
||||
Tracking what is in flight is what lets those be stopped first.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
$PowerShell
|
||||
)
|
||||
|
||||
# Synchronized protects one operation, not a test followed by an assignment, so the
|
||||
# collection is created under the shared lock
|
||||
[System.Threading.Monitor]::Enter($sync.SyncRoot)
|
||||
try {
|
||||
if ($null -eq $sync.ActiveShells) {
|
||||
$sync.ActiveShells = [System.Collections.ArrayList]::Synchronized([System.Collections.ArrayList]::new())
|
||||
}
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.SyncRoot)
|
||||
}
|
||||
|
||||
# Nothing disposes these on the way out, so finished ones are dropped here instead of
|
||||
# accumulating for the life of the session
|
||||
foreach ($finished in (Get-WinUtilActiveShell)) {
|
||||
if (-not (Test-WinUtilShellRunning $finished)) {
|
||||
try {
|
||||
$sync.ActiveShells.Remove($finished)
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Could not remove a completed worker from the active set: $_"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$null = $sync.ActiveShells.Add($PowerShell)
|
||||
}
|
||||
|
||||
function Get-WinUtilActiveShell {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
A snapshot of the tracked instances, copied under the collection's own lock
|
||||
|
||||
.DESCRIPTION
|
||||
Enumerating a synchronized ArrayList is not itself synchronized; a concurrent Add or
|
||||
Remove throws mid-loop. SyncRoot is the documented fix.
|
||||
#>
|
||||
|
||||
if ($null -eq $sync.ActiveShells) {
|
||||
return @()
|
||||
}
|
||||
|
||||
[System.Threading.Monitor]::Enter($sync.ActiveShells.SyncRoot)
|
||||
try {
|
||||
return @($sync.ActiveShells.ToArray())
|
||||
} finally {
|
||||
[System.Threading.Monitor]::Exit($sync.ActiveShells.SyncRoot)
|
||||
}
|
||||
}
|
||||
|
||||
function Stop-WinUtilActiveWork {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Asks everything running on the worker pool to stop, and waits for it
|
||||
|
||||
.DESCRIPTION
|
||||
Stop is a request, not a kill: a command already inside an installer runs until it
|
||||
returns. The wait is bounded so a worker that never returns cannot hold the window
|
||||
open.
|
||||
|
||||
.PARAMETER TimeoutSeconds
|
||||
How long to wait before giving up on it.
|
||||
#>
|
||||
param(
|
||||
[int]$TimeoutSeconds = 15,
|
||||
|
||||
# Issue the stop and return. The caller polls Test-WinUtilActiveWorkRunning instead of
|
||||
# blocking here, which matters on the interface thread where a wait freezes the window.
|
||||
[switch]$NoWait
|
||||
)
|
||||
|
||||
$shells = Get-WinUtilActiveShell
|
||||
if ($shells.Count -eq 0) {
|
||||
return $true
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Component "UI" -Message "Stopping $($shells.Count) running item(s) before closing."
|
||||
|
||||
foreach ($shell in $shells) {
|
||||
if (Test-WinUtilShellRunning $shell) {
|
||||
try {
|
||||
$null = $shell.BeginStop($null, $null)
|
||||
} catch {
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Could not request that a worker stop: $_"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if ($NoWait) {
|
||||
return $false
|
||||
}
|
||||
|
||||
$clock = [System.Diagnostics.Stopwatch]::StartNew()
|
||||
while ($clock.Elapsed.TotalSeconds -lt $TimeoutSeconds) {
|
||||
$stillRunning = @(Get-WinUtilActiveShell | Where-Object { Test-WinUtilShellRunning $_ }).Count
|
||||
|
||||
if ($stillRunning -eq 0) {
|
||||
Write-WinUtilLog -Component "UI" -Message "Everything stopped after $($clock.ElapsedMilliseconds) ms."
|
||||
return $true
|
||||
}
|
||||
|
||||
Start-Sleep -Milliseconds 100
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Level "WARN" -Component "UI" -Message "Gave up waiting for work to stop after $TimeoutSeconds seconds, closing anyway."
|
||||
return $false
|
||||
}
|
||||
|
||||
function Test-WinUtilActiveWorkRunning {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Whether any tracked instance is still running
|
||||
#>
|
||||
|
||||
foreach ($shell in (Get-WinUtilActiveShell)) {
|
||||
if (Test-WinUtilShellRunning $shell) { return $true }
|
||||
}
|
||||
|
||||
return $false
|
||||
}
|
||||
@@ -0,0 +1,97 @@
|
||||
function Register-WinUtilInputWatch {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Records when the user last did something, so background work can step aside
|
||||
|
||||
.DESCRIPTION
|
||||
Preview events run before the control handles the input, so the timestamp is set
|
||||
even for a click the control then spends time on.
|
||||
#>
|
||||
|
||||
$sync.LastInputAt = [datetime]::MinValue
|
||||
|
||||
$stamp = { $sync.LastInputAt = [datetime]::Now }
|
||||
$sync.Form.Add_PreviewMouseDown($stamp)
|
||||
$sync.Form.Add_PreviewKeyDown($stamp)
|
||||
$sync.Form.Add_PreviewMouseWheel($stamp)
|
||||
}
|
||||
|
||||
function Test-WinUtilDeferBackgroundWork {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Whether speculative work should wait rather than run now
|
||||
|
||||
.DESCRIPTION
|
||||
Background priority queues work behind input but does not make it interruptible:
|
||||
whatever is running must finish before a click is looked at, which is why the pieces
|
||||
are kept short. Waits while the user is active, or while the work draws into a tab
|
||||
that is not on screen.
|
||||
|
||||
.PARAMETER RequiresTab
|
||||
The tab this work draws into. Work for a hidden tab waits.
|
||||
#>
|
||||
param(
|
||||
[string]$RequiresTab
|
||||
)
|
||||
|
||||
if ($sync.LastInputAt) {
|
||||
$sinceInput = ([datetime]::Now - $sync.LastInputAt).TotalMilliseconds
|
||||
# long enough to cover a click and the work it starts, short enough not to be noticed
|
||||
if ($sinceInput -lt 400) {
|
||||
return $true
|
||||
}
|
||||
}
|
||||
|
||||
if ($RequiresTab -and $sync.currentTab -and $sync.currentTab -ne $RequiresTab) {
|
||||
return $true
|
||||
}
|
||||
|
||||
return $false
|
||||
}
|
||||
|
||||
function Invoke-WinUtilWhenIdle {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Runs a callback once the interface is not being used
|
||||
|
||||
.DESCRIPTION
|
||||
A one shot timer, not a dispatcher post: a post at background priority runs straight
|
||||
away and the point is to leave a gap.
|
||||
|
||||
.PARAMETER Callback
|
||||
What to run once the wait is over.
|
||||
|
||||
.PARAMETER Argument
|
||||
Passed to the callback. Carried on the timer rather than captured, so the callback
|
||||
resolves commands where it was written, not in a copied scope.
|
||||
|
||||
.PARAMETER DelayMilliseconds
|
||||
How long to wait before looking again.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[scriptblock]$Callback,
|
||||
|
||||
$Argument,
|
||||
|
||||
[int]$DelayMilliseconds = 150
|
||||
)
|
||||
|
||||
if (-not (Test-WinUtilUIAlive)) {
|
||||
return
|
||||
}
|
||||
|
||||
# Bound to the interface dispatcher explicitly: the default picks up the calling thread's,
|
||||
# which is only correct while every caller reaches here through a UI post
|
||||
$timer = New-Object System.Windows.Threading.DispatcherTimer([System.Windows.Threading.DispatcherPriority]::Background, $sync.Form.Dispatcher)
|
||||
$timer.Interval = [timespan]::FromMilliseconds($DelayMilliseconds)
|
||||
$timer.Tag = @{ Callback = $Callback; Argument = $Argument }
|
||||
# Sender taken from the argument, matching how the rest of this codebase handles timer ticks
|
||||
$timer.Add_Tick({
|
||||
param($eventSender)
|
||||
$ticked = [System.Windows.Threading.DispatcherTimer]$eventSender
|
||||
$ticked.Stop()
|
||||
& $ticked.Tag.Callback $ticked.Tag.Argument
|
||||
})
|
||||
$timer.Start()
|
||||
}
|
||||
@@ -19,28 +19,16 @@ function Test-WinUtilPackageManager {
|
||||
|
||||
if ($winget) {
|
||||
if (Get-Command winget -ErrorAction SilentlyContinue) {
|
||||
Write-Host "===========================================" -ForegroundColor Green
|
||||
Write-Host "--- WinGet is installed ---" -ForegroundColor Green
|
||||
Write-Host "===========================================" -ForegroundColor Green
|
||||
$status = "installed"
|
||||
} else {
|
||||
Write-Host "===========================================" -ForegroundColor Red
|
||||
Write-Host "--- WinGet is not installed ---" -ForegroundColor Red
|
||||
Write-Host "===========================================" -ForegroundColor Red
|
||||
$status = "not-installed"
|
||||
}
|
||||
}
|
||||
|
||||
if ($choco) {
|
||||
if (Get-Command choco -ErrorAction SilentlyContinue) {
|
||||
Write-Host "===========================================" -ForegroundColor Green
|
||||
Write-Host "--- Chocolatey is installed ---" -ForegroundColor Green
|
||||
Write-Host "===========================================" -ForegroundColor Green
|
||||
$status = "installed"
|
||||
} else {
|
||||
Write-Host "===========================================" -ForegroundColor Red
|
||||
Write-Host "--- Chocolatey is not installed ---" -ForegroundColor Red
|
||||
Write-Host "===========================================" -ForegroundColor Red
|
||||
$status = "not-installed"
|
||||
}
|
||||
}
|
||||
|
||||
@@ -77,7 +77,11 @@ function Update-WinUtilSelections {
|
||||
|
||||
foreach ($listName in $nextSelections.Keys) {
|
||||
foreach ($cbkey in $nextSelections[$listName]) {
|
||||
$sync.$listName.Add($cbkey)
|
||||
# Appending, so the same entry can already be there: a preset and a config that both
|
||||
# name it would otherwise select it twice
|
||||
if ($sync.$listName -notcontains $cbkey) {
|
||||
$sync.$listName.Add($cbkey)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,91 @@
|
||||
function Write-WinUtilConsoleProgress {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Reports job progress on the console for runs that have no window
|
||||
|
||||
.DESCRIPTION
|
||||
The progress bar is the only thing telling a user how far along a job is, so a
|
||||
headless run needs the same information in the only place it has.
|
||||
|
||||
On a console the line is rewritten in place, because a package that reports every
|
||||
few hundred milliseconds would otherwise scroll a screenful for one install. When
|
||||
output is redirected there is no cursor to move, so each update is its own line and
|
||||
they are throttled hard instead.
|
||||
#>
|
||||
param(
|
||||
[string]$Status,
|
||||
[int]$Percent = -1
|
||||
)
|
||||
|
||||
if ([string]::IsNullOrWhiteSpace($Status) -and $Percent -lt 0) {
|
||||
return
|
||||
}
|
||||
|
||||
if ($null -eq $sync.ConsoleProgressState) {
|
||||
$sync.ConsoleProgressState = [hashtable]::Synchronized(@{
|
||||
LastText = ""
|
||||
LastWrite = [datetime]::MinValue
|
||||
LineLength = 0
|
||||
LineOpen = $false
|
||||
})
|
||||
}
|
||||
$state = $sync.ConsoleProgressState
|
||||
|
||||
$text = if ([string]::IsNullOrWhiteSpace($Status)) { $state.LastText } else { $Status }
|
||||
if ([string]::IsNullOrWhiteSpace($text)) {
|
||||
return
|
||||
}
|
||||
|
||||
$redirected = [Console]::IsOutputRedirected
|
||||
$throttleMs = if ($redirected) { 1000 } else { 150 }
|
||||
|
||||
$now = Get-Date
|
||||
$sinceLast = ($now - $state.LastWrite).TotalMilliseconds
|
||||
|
||||
# Redirected output cannot be rewritten in place, so every update is its own line and the
|
||||
# throttle holds even when the text changed. A job reporting per package would otherwise
|
||||
# scroll a screenful.
|
||||
if ($redirected) {
|
||||
if ($sinceLast -lt $throttleMs) { return }
|
||||
} elseif ($text -eq $state.LastText -and $sinceLast -lt $throttleMs) {
|
||||
return
|
||||
}
|
||||
|
||||
$state.LastText = $text
|
||||
$state.LastWrite = $now
|
||||
|
||||
$prefix = if ($Percent -ge 0) { "[{0,3}%] " -f $Percent } else { "[ =] " }
|
||||
$line = "$prefix$text"
|
||||
|
||||
if ($redirected) {
|
||||
Write-Host $line -ForegroundColor DarkCyan
|
||||
return
|
||||
}
|
||||
|
||||
# Pad to the previous length so a shorter line does not leave the tail of the longer one
|
||||
$padding = [Math]::Max(0, $state.LineLength - $line.Length)
|
||||
Write-Host ("`r$line" + (" " * $padding)) -NoNewline -ForegroundColor DarkCyan
|
||||
$state.LineLength = $line.Length
|
||||
$state.LineOpen = $true
|
||||
}
|
||||
|
||||
function Complete-WinUtilConsoleProgress {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Ends the progress line so the next thing printed starts on its own
|
||||
|
||||
.DESCRIPTION
|
||||
The line is rewritten in place and therefore left without a newline. Anything else
|
||||
reaching the console has to close it first, or it lands on top of the progress.
|
||||
#>
|
||||
|
||||
$state = $sync.ConsoleProgressState
|
||||
if ($null -eq $state -or -not $state.LineOpen) {
|
||||
return
|
||||
}
|
||||
|
||||
Write-Host ""
|
||||
$state.LineOpen = $false
|
||||
$state.LineLength = 0
|
||||
$state.LastText = ""
|
||||
}
|
||||
@@ -0,0 +1,132 @@
|
||||
function Copy-WinUtilEnvironmentReportExportFile {
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$SourcePath,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
[string]$DestinationPath,
|
||||
|
||||
[switch]$Overwrite
|
||||
)
|
||||
|
||||
[System.IO.File]::Copy($SourcePath, $DestinationPath, $Overwrite)
|
||||
}
|
||||
|
||||
function Remove-WinUtilEnvironmentReportExportFile {
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Path
|
||||
)
|
||||
|
||||
[System.IO.File]::Delete($Path)
|
||||
}
|
||||
|
||||
function Write-WinUtilEnvironmentReportExport {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Publishes an environment report and its optional log companion as one export.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$JsonPath,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Json,
|
||||
|
||||
[Parameter(Mandatory)]
|
||||
[string]$LogsPath,
|
||||
|
||||
[AllowNull()]
|
||||
[string]$Logs,
|
||||
|
||||
[switch]$IncludeLogs
|
||||
)
|
||||
|
||||
$suffix = [guid]::NewGuid().ToString("N")
|
||||
$jsonTempPath = "$JsonPath.$suffix.tmp"
|
||||
$logsTempPath = "$LogsPath.$suffix.tmp"
|
||||
$jsonBackupPath = "$JsonPath.$suffix.bak"
|
||||
$logsBackupPath = "$LogsPath.$suffix.bak"
|
||||
$hadJson = [System.IO.File]::Exists($JsonPath)
|
||||
$hadLogs = $IncludeLogs -and [System.IO.File]::Exists($LogsPath)
|
||||
$encoding = [System.Text.UTF8Encoding]::new($false)
|
||||
$retainJsonBackup = $false
|
||||
$retainLogsBackup = $false
|
||||
|
||||
try {
|
||||
# Prepare both outputs before replacing either destination. If collection or a temporary
|
||||
# write fails, the previous export remains untouched.
|
||||
[System.IO.File]::WriteAllText($jsonTempPath, $Json, $encoding)
|
||||
if ($IncludeLogs) {
|
||||
[System.IO.File]::WriteAllText($logsTempPath, $Logs, $encoding)
|
||||
}
|
||||
|
||||
if ($hadJson) {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $JsonPath -DestinationPath $jsonBackupPath
|
||||
}
|
||||
if ($hadLogs) {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $LogsPath -DestinationPath $logsBackupPath
|
||||
}
|
||||
|
||||
try {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $jsonTempPath -DestinationPath $JsonPath -Overwrite
|
||||
if ($IncludeLogs) {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $logsTempPath -DestinationPath $LogsPath -Overwrite
|
||||
}
|
||||
} catch {
|
||||
$publishError = $_
|
||||
$rollbackErrors = [System.Collections.Generic.List[string]]::new()
|
||||
|
||||
# A two-file replacement is not atomic on Windows. Restore both destinations if the
|
||||
# second publish fails so an old companion cannot appear to belong to new JSON.
|
||||
try {
|
||||
if ($hadJson) {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $jsonBackupPath -DestinationPath $JsonPath -Overwrite
|
||||
} elseif ([System.IO.File]::Exists($JsonPath)) {
|
||||
Remove-WinUtilEnvironmentReportExportFile -Path $JsonPath
|
||||
}
|
||||
} catch {
|
||||
$retainJsonBackup = $hadJson
|
||||
$recoveryNote = if ($hadJson) { " Recovery copy retained at $jsonBackupPath." } else { "" }
|
||||
$rollbackErrors.Add("JSON restore failed: $($_.Exception.Message).$recoveryNote")
|
||||
}
|
||||
|
||||
if ($IncludeLogs) {
|
||||
try {
|
||||
if ($hadLogs) {
|
||||
Copy-WinUtilEnvironmentReportExportFile -SourcePath $logsBackupPath -DestinationPath $LogsPath -Overwrite
|
||||
} elseif ([System.IO.File]::Exists($LogsPath)) {
|
||||
Remove-WinUtilEnvironmentReportExportFile -Path $LogsPath
|
||||
}
|
||||
} catch {
|
||||
$retainLogsBackup = $hadLogs
|
||||
$recoveryNote = if ($hadLogs) { " Recovery copy retained at $logsBackupPath." } else { "" }
|
||||
$rollbackErrors.Add("logs restore failed: $($_.Exception.Message).$recoveryNote")
|
||||
}
|
||||
}
|
||||
|
||||
if ($rollbackErrors.Count -gt 0) {
|
||||
$message = "Environment export publish failed: $($publishError.Exception.Message) Rollback also failed: $($rollbackErrors -join ' ')"
|
||||
throw [System.IO.IOException]::new($message, $publishError.Exception)
|
||||
}
|
||||
|
||||
throw $publishError
|
||||
}
|
||||
} finally {
|
||||
$cleanupTargets = @(
|
||||
[pscustomobject]@{ Path = $jsonTempPath; Retain = $false },
|
||||
[pscustomobject]@{ Path = $logsTempPath; Retain = $false },
|
||||
[pscustomobject]@{ Path = $jsonBackupPath; Retain = $retainJsonBackup },
|
||||
[pscustomobject]@{ Path = $logsBackupPath; Retain = $retainLogsBackup }
|
||||
)
|
||||
foreach ($cleanupTarget in $cleanupTargets) {
|
||||
if (-not $cleanupTarget.Retain -and [System.IO.File]::Exists($cleanupTarget.Path)) {
|
||||
try {
|
||||
Remove-WinUtilEnvironmentReportExportFile -Path $cleanupTarget.Path
|
||||
} catch {
|
||||
Write-Warning "Could not remove temporary environment-export file $($cleanupTarget.Path): $($_.Exception.Message)"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,58 @@
|
||||
function Write-WinUtilErrorRecord {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Logs a failure with enough context to act on it
|
||||
|
||||
.DESCRIPTION
|
||||
A bare "You cannot call a method on a null-valued expression." says nothing about
|
||||
where it came from. This records the message together with the exception type, the
|
||||
command and line that raised it, and the script stack, under a component name.
|
||||
|
||||
.PARAMETER ErrorRecord
|
||||
The error to report, normally $_ from a catch block.
|
||||
|
||||
.PARAMETER Component
|
||||
Which part of WinUtil was running, for example Install or UI.
|
||||
|
||||
.PARAMETER Context
|
||||
What was being attempted, for example the button name or the package.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
$ErrorRecord,
|
||||
|
||||
[string]$Component = "WinUtil",
|
||||
|
||||
[string]$Context,
|
||||
|
||||
# The underlying failure was already logged and counted; retain this call's context and
|
||||
# stack as diagnostic detail without adding a second headline error.
|
||||
[switch]$DetailOnly
|
||||
)
|
||||
|
||||
$headline = if ($Context) { "$Context : $($ErrorRecord.Exception.Message)" } else { $ErrorRecord.Exception.Message }
|
||||
Write-WinUtilLog -Level "ERROR" -Component $Component -Message $headline -Detail:$DetailOnly
|
||||
|
||||
$invocation = $ErrorRecord.InvocationInfo
|
||||
if ($invocation) {
|
||||
$where = "$($invocation.ScriptName):$($invocation.ScriptLineNumber)"
|
||||
if ([string]::IsNullOrWhiteSpace($invocation.ScriptName)) {
|
||||
$where = "line $($invocation.ScriptLineNumber)"
|
||||
}
|
||||
Write-WinUtilLog -Level "ERROR" -Detail -Component $Component -Message " at $where in $($invocation.MyCommand): $($invocation.Line.Trim())"
|
||||
}
|
||||
|
||||
Write-WinUtilLog -Level "ERROR" -Detail -Component $Component -Message " type $($ErrorRecord.Exception.GetType().FullName), category $($ErrorRecord.CategoryInfo.Category)"
|
||||
|
||||
if ($ErrorRecord.ScriptStackTrace) {
|
||||
foreach ($frame in ($ErrorRecord.ScriptStackTrace -split "`r?`n")) {
|
||||
if (-not [string]::IsNullOrWhiteSpace($frame)) {
|
||||
Write-WinUtilLog -Level "ERROR" -Detail -Component $Component -Message " $($frame.Trim())"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (-not $DetailOnly) {
|
||||
Write-Host "$Component : $headline" -ForegroundColor Red
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,54 @@
|
||||
function Write-WinUtilJobBanner {
|
||||
<#
|
||||
.SYNOPSIS
|
||||
Writes the boxed start, finish or failure line a job prints to the console
|
||||
|
||||
.DESCRIPTION
|
||||
One place decides what a running operation looks like in the terminal, so every
|
||||
workflow announces itself the same way instead of hand-drawing its own box. Called
|
||||
by Start-WinUtilJob, not by job bodies.
|
||||
|
||||
.PARAMETER Message
|
||||
The line to box, for example "Installing apps".
|
||||
|
||||
.PARAMETER Level
|
||||
INFO for a normal banner, ERROR to colour it as a failure.
|
||||
#>
|
||||
param(
|
||||
[Parameter(Mandatory)]
|
||||
[string]$Message,
|
||||
|
||||
[ValidateSet("INFO", "ERROR")]
|
||||
[string]$Level = "INFO"
|
||||
)
|
||||
|
||||
# A progress line is rewritten in place and left open, so the box would be drawn on top of it
|
||||
Complete-WinUtilConsoleProgress
|
||||
|
||||
# Wrapped, because a failure listing several packages would otherwise draw a box wider
|
||||
# than the console
|
||||
$width = 76
|
||||
$words = $Message -split '\s+'
|
||||
$lines = [System.Collections.Generic.List[string]]::new()
|
||||
$current = ""
|
||||
foreach ($word in $words) {
|
||||
if ($current.Length -gt 0 -and ($current.Length + 1 + $word.Length) -gt $width) {
|
||||
$lines.Add($current)
|
||||
$current = $word
|
||||
} else {
|
||||
$current = if ($current.Length -eq 0) { $word } else { "$current $word" }
|
||||
}
|
||||
}
|
||||
if ($current.Length -gt 0) { $lines.Add($current) }
|
||||
|
||||
$longest = ($lines | Measure-Object -Property Length -Maximum).Maximum
|
||||
$border = "=" * ($longest + 6)
|
||||
$colour = if ($Level -eq "ERROR") { "Red" } else { "Cyan" }
|
||||
|
||||
Write-Host ""
|
||||
Write-Host $border -ForegroundColor $colour
|
||||
foreach ($line in $lines) {
|
||||
Write-Host ("-- {0}$(' ' * ($longest - $line.Length)) --" -f $line) -ForegroundColor $colour
|
||||
}
|
||||
Write-Host $border -ForegroundColor $colour
|
||||
}
|
||||
@@ -4,6 +4,11 @@ function Write-WinUtilLog {
|
||||
.SYNOPSIS
|
||||
Writes a timestamped WinUtil log entry to the active session log.
|
||||
|
||||
.DESCRIPTION
|
||||
Called from the interface thread and from every job body. When Start-Transcript owns the
|
||||
active session log, entries go through the host so the transcript records them without a
|
||||
competing file write. Standalone callers use a named mutex to serialize direct appends.
|
||||
|
||||
.PARAMETER Message
|
||||
The message to write.
|
||||
|
||||
@@ -21,9 +26,20 @@ function Write-WinUtilLog {
|
||||
[ValidateSet("INFO", "WARN", "ERROR", "DEBUG")]
|
||||
[string]$Level = "INFO",
|
||||
|
||||
[string]$Component = "WinUtil"
|
||||
[string]$Component = "WinUtil",
|
||||
|
||||
# Continuation of an error already counted, such as a stack frame
|
||||
[switch]$Detail
|
||||
)
|
||||
|
||||
if ($Level -eq "ERROR" -and -not $Detail -and $null -ne $sync.LoggedErrors) {
|
||||
$null = $sync.LoggedErrors.Add("[$Component] $Message")
|
||||
}
|
||||
|
||||
if ($Level -eq "ERROR" -and -not $Detail -and $global:WinUtilIsJobWorker) {
|
||||
$global:WinUtilJobErrorCount++
|
||||
}
|
||||
|
||||
try {
|
||||
$logPath = $null
|
||||
$transcriptPath = $null
|
||||
@@ -70,10 +86,29 @@ function Write-WinUtilLog {
|
||||
return
|
||||
}
|
||||
|
||||
$mutex = [System.Threading.Mutex]::new($false, "WinUtilSessionLog")
|
||||
$held = $false
|
||||
try {
|
||||
try {
|
||||
$held = $mutex.WaitOne(2000)
|
||||
} catch [System.Threading.AbandonedMutexException] {
|
||||
# A thread died holding the mutex; ownership transfers to us either way
|
||||
$held = $true
|
||||
}
|
||||
|
||||
if (-not $held) {
|
||||
# Writing anyway is what interleaves lines, and the wait only times out when
|
||||
# contention is at its worst
|
||||
Write-Host $line
|
||||
return
|
||||
}
|
||||
|
||||
Add-Content -Path $logPath -Value $line -Encoding UTF8 -ErrorAction Stop
|
||||
} catch [System.IO.IOException] {
|
||||
Write-Host $line
|
||||
} finally {
|
||||
if ($held) { $mutex.ReleaseMutex() }
|
||||
$mutex.Dispose()
|
||||
}
|
||||
} catch {
|
||||
Write-Warning "Unable to write WinUtil log entry: $($_.Exception.Message)"
|
||||
|
||||
Reference in New Issue
Block a user