Merge pull request #50 from zmanian/tall-bag

Add Claude Code version check and update button

Author: mcintyre94

Wisp/ViewModels/SpriteOverviewViewModel.swift

@@ -8,6 +8,16 @@ enum SpritesCLIAuth {
     case unknown, checking, authenticated, notAuthenticated
 }
 
+enum ClaudeCodeVersionStatus: Equatable {
+    case unknown
+    case checking
+    case upToDate(version: String)
+    case updateAvailable(current: String, latest: String)
+    case updating
+    case updateFailed(error: String)
+    case failed
+}
+
 @Observable
 @MainActor
 final class SpriteOverviewViewModel {
@@ -19,6 +29,7 @@ final class SpriteOverviewViewModel {
     var isAuthenticatingGitHub = false
     var spritesCLIAuthStatus: SpritesCLIAuth = .unknown
     var isAuthenticatingSprites = false
+    var claudeCodeVersionStatus: ClaudeCodeVersionStatus = .unknown
     var errorMessage: String?
     var isUploading = false
     var uploadResult: SpritesAPIClient.FileUploadResponse?
@@ -195,6 +206,74 @@ final class SpriteOverviewViewModel {
         }
     }
 
+    func checkClaudeCodeVersion(apiClient: SpritesAPIClient) async {
+        claudeCodeVersionStatus = .checking
+
+        async let execResult = apiClient.runExec(
+            spriteName: sprite.name,
+            command: "claude --version 2>/dev/null || echo CLAUDE_NOT_FOUND"
+        )
+        async let latestVersion = fetchNpmLatestVersion()
+
+        let (output, success) = await execResult
+        let npmVersion = await latestVersion
+
+        if !success || output.contains("CLAUDE_NOT_FOUND") || output.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty {
+            claudeCodeVersionStatus = .failed
+            return
+        }
+
+        let installed = output.trimmingCharacters(in: .whitespacesAndNewlines)
+
+        if let npm = npmVersion, let installedSemver = extractSemver(from: installed) {
+            if installedSemver == npm {
+                claudeCodeVersionStatus = .upToDate(version: installed)
+            } else {
+                claudeCodeVersionStatus = .updateAvailable(current: installed, latest: npm)
+            }
+        } else {
+            claudeCodeVersionStatus = .upToDate(version: installed)
+        }
+    }
+
+    func updateClaudeCode(apiClient: SpritesAPIClient) async {
+        claudeCodeVersionStatus = .updating
+        let (output, success) = await apiClient.runExec(
+            spriteName: sprite.name,
+            command: "claude update 2>&1 && claude --version",
+            timeout: 120
+        )
+        if success {
+            let lines = output.trimmingCharacters(in: .whitespacesAndNewlines).split(separator: "\n")
+            if let lastLine = lines.last {
+                let newInstalled = String(lastLine).trimmingCharacters(in: .whitespacesAndNewlines)
+                let npmVersion = await fetchNpmLatestVersion()
+                if let npm = npmVersion, let semver = extractSemver(from: newInstalled), semver == npm {
+                    claudeCodeVersionStatus = .upToDate(version: newInstalled)
+                } else {
+                    claudeCodeVersionStatus = .upToDate(version: newInstalled)
+                }
+            } else {
+                claudeCodeVersionStatus = .updateFailed(error: "Update succeeded but could not read version")
+            }
+        } else {
+            claudeCodeVersionStatus = .updateFailed(error: "Update failed")
+        }
+    }
+
+    private func fetchNpmLatestVersion() async -> String? {
+        guard let url = URL(string: "https://registry.npmjs.org/@anthropic-ai/claude-code/latest") else { return nil }
+        guard let (data, _) = try? await URLSession.shared.data(from: url) else { return nil }
+        struct NpmResponse: Decodable { let version: String }
+        return try? JSONDecoder().decode(NpmResponse.self, from: data).version
+    }
+
+    private func extractSemver(from string: String) -> String? {
+        let pattern = #"\d+\.\d+\.\d+"#
+        return string.range(of: pattern, options: .regularExpression)
+            .map { String(string[$0]) }
+    }
+
     func authenticateSprites(apiClient: SpritesAPIClient) async {
         guard let token = apiClient.spritesToken else { return }
         isAuthenticatingSprites = true

Wisp/Views/SpriteDetail/Overview/SpriteOverviewView.swift

@@ -145,6 +145,106 @@ struct SpriteOverviewView: View {
                 }
             }
 
+            Section("Claude Code") {
+                switch viewModel.claudeCodeVersionStatus {
+                case .unknown, .checking:
+                    HStack(spacing: 8) {
+                        Text("Version")
+                        Spacer()
+                        ProgressView()
+                        Text("Checking...")
+                            .foregroundStyle(.secondary)
+                    }
+                case .upToDate(let version):
+                    HStack {
+                        Text("Version")
+                        Spacer()
+                        Image(systemName: "checkmark.circle.fill")
+                            .foregroundStyle(.green)
+                        Text(version)
+                            .foregroundStyle(.secondary)
+                    }
+                    .contextMenu {
+                        Button {
+                            UIPasteboard.general.string = version
+                        } label: {
+                            Label("Copy Version", systemImage: "doc.on.doc")
+                        }
+                    }
+                case .updateAvailable(let current, _):
+                    HStack {
+                        Text("Version")
+                        Spacer()
+                        Image(systemName: "arrow.triangle.2.circlepath")
+                            .foregroundStyle(.orange)
+                        Text(current)
+                            .foregroundStyle(.secondary)
+                    }
+                    .contextMenu {
+                        Button {
+                            UIPasteboard.general.string = current
+                        } label: {
+                            Label("Copy Version", systemImage: "doc.on.doc")
+                        }
+                    }
+                case .updating:
+                    HStack(spacing: 8) {
+                        Text("Version")
+                        Spacer()
+                        ProgressView()
+                        Text("Updating...")
+                            .foregroundStyle(.secondary)
+                    }
+                case .updateFailed(let error):
+                    HStack {
+                        Text("Version")
+                        Spacer()
+                        Text(error)
+                            .foregroundStyle(.red)
+                            .font(.caption)
+                    }
+                case .failed:
+                    HStack {
+                        Text("Version")
+                        Spacer()
+                        Text("Not installed")
+                            .foregroundStyle(.secondary)
+                    }
+                }
+
+                Button {
+                    Task { await viewModel.updateClaudeCode(apiClient: apiClient) }
+                } label: {
+                    HStack {
+                        if case .updateAvailable(_, let latest) = viewModel.claudeCodeVersionStatus {
+                            Text("Update to \(latest)")
+                                .foregroundStyle(.primary)
+                        } else {
+                            Text("Update Claude Code")
+                                .foregroundStyle(.primary)
+                        }
+                        Spacer()
+                        if case .updating = viewModel.claudeCodeVersionStatus {
+                            ProgressView()
+                        } else if case .checking = viewModel.claudeCodeVersionStatus {
+                            EmptyView()
+                        } else if case .upToDate = viewModel.claudeCodeVersionStatus {
+                            Text("Latest version")
+                                .foregroundStyle(.secondary)
+                        } else {
+                            Image(systemName: "arrow.triangle.2.circlepath")
+                                .foregroundStyle(Color.accentColor)
+                        }
+                    }
+                }
+                .disabled({
+                    if case .updating = viewModel.claudeCodeVersionStatus { return true }
+                    if case .checking = viewModel.claudeCodeVersionStatus { return true }
+                    if case .upToDate = viewModel.claudeCodeVersionStatus { return true }
+                    return false
+                }())
+            }
+
             Section("Sprites CLI") {
                 switch viewModel.spritesCLIAuthStatus {
                 case .unknown, .checking:
@@ -238,8 +338,10 @@ struct SpriteOverviewView: View {
         .task {
             loadWorkingDirectory()
             await viewModel.refresh(apiClient: apiClient)
-            await viewModel.checkSpritesAuth(apiClient: apiClient)
-            await viewModel.checkGitHubAuth(apiClient: apiClient)
+            async let claude: Void = viewModel.checkClaudeCodeVersion(apiClient: apiClient)
+            async let sprites: Void = viewModel.checkSpritesAuth(apiClient: apiClient)
+            async let github: Void = viewModel.checkGitHubAuth(apiClient: apiClient)
+            _ = await (claude, sprites, github)
         }
         .task {
             await viewModel.pollStatus(apiClient: apiClient)