From bb9b7341de4f9f87bcae4cae530fc9c7379a7c7c Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Tue, 2 Jun 2026 04:23:46 -0700
Subject: [PATCH] WSL fallback: create Desktop + Start Menu shortcuts that
 launch WSL Studio

The fallback returns before install.ps1's native shortcut code, so it created no
shortcuts. Add a WSL launcher (launch-studio-wsl.ps1) plus Desktop and Start Menu
.lnk shortcuts that start `unsloth studio` inside WSL and open http://localhost:8888
in the browser once the backend is healthy. Best-effort + wrapped in try/catch so a
shortcut failure never aborts the install.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
---
 install.ps1 | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/install.ps1 b/install.ps1
index 79bcbe0e39..d9063e832c 100644
--- a/install.ps1
+++ b/install.ps1
@@ -1558,6 +1558,38 @@ shell.Run cmd, 0, False
             } catch {
                 substep "(shim creation failed; launch manually):  wsl -d $distro -u root -- bash -lic 'unsloth studio -p 8888'" "Yellow"
             }
+            # Desktop + Start Menu shortcuts: launch the WSL Studio and open the browser when ready.
+            try {
+                $appDir = Join-Path $env:LOCALAPPDATA "Unsloth"
+                New-Item -ItemType Directory -Force -Path $appDir *> $null
+                $launcher = Join-Path $appDir "launch-studio-wsl.ps1"
+                $L = @(
+                    '$ErrorActionPreference = "SilentlyContinue"',
+                    ('$distro = "' + $distro + '"'),
+                    'Start-Job { for ($i=0; $i -lt 120; $i++) { try { if ((Invoke-WebRequest "http://localhost:8888/api/health" -UseBasicParsing -TimeoutSec 2).StatusCode -eq 200) { Start-Process "http://localhost:8888"; break } } catch {}; Start-Sleep 1 } } | Out-Null',
+                    'Write-Host "Starting Unsloth Studio in WSL ($distro); browser opens at http://localhost:8888 when ready (Ctrl+C to stop)..."',
+                    'wsl.exe -d $distro -u root -- bash -lic "unsloth studio -p 8888"'
+                )
+                Set-Content -LiteralPath $launcher -Value $L -Encoding UTF8
+                $icon = Join-Path $appDir "unsloth.ico"
+                try { if (-not (Test-Path -LiteralPath $icon)) { Invoke-WebRequest "https://raw.githubusercontent.com/unslothai/unsloth/main/studio/frontend/public/unsloth.ico" -OutFile $icon -UseBasicParsing -TimeoutSec 15 *> $null } } catch {}
+                $wsh = New-Object -ComObject WScript.Shell
+                $lnks = @()
+                $dd = [Environment]::GetFolderPath("Desktop"); if ($dd -and $dd.Trim()) { $lnks += (Join-Path $dd "Unsloth Studio.lnk") }
+                if ($env:APPDATA -and $env:APPDATA.Trim()) { $smd = Join-Path $env:APPDATA "Microsoft\Windows\Start Menu\Programs"; New-Item -ItemType Directory -Force -Path $smd *> $null; $lnks += (Join-Path $smd "Unsloth Studio.lnk") }
+                foreach ($lnk in $lnks) {
+                    $sc = $wsh.CreateShortcut($lnk)
+                    $sc.TargetPath = (Join-Path $env:SystemRoot "System32\WindowsPowerShell\v1.0\powershell.exe")
+                    $sc.Arguments = "-NoExit -NoProfile -ExecutionPolicy Bypass -File `"$launcher`""
+                    $sc.WorkingDirectory = $appDir
+                    if (Test-Path -LiteralPath $icon) { $sc.IconLocation = $icon }
+                    $sc.Description = "Unsloth Studio (GPU via WSL)"
+                    $sc.Save()
+                }
+                step "shortcuts" "created Desktop + Start Menu shortcuts (launch WSL Studio + open browser)" "Green"
+            } catch {
+                substep "(could not create shortcuts: $($_.Exception.Message))" "Yellow"
+            }
         } else {
             step "wsl" "WSL Studio install did not finish cleanly (torch.cuda not detected; inner exit $wslRc) -- see log above." "Yellow"
             substep "retry, or launch manually:  wsl -d $distro -u root -- bash -lic 'unsloth studio -p 8888'" "Cyan"