Send to account tab if required

Author: mrubens

src/core/webview/ClineProvider.ts

@@ -529,11 +529,6 @@ export class ClineProvider
 			>
 		> = {},
 	) {
-		// Check MDM compliance before proceeding
-		if (!this.checkMdmCompliance()) {
-			return // Block task creation if not compliant
-		}
-
 		const {
 			apiConfiguration,
 			organizationAllowList,
@@ -1252,6 +1247,11 @@ export class ClineProvider
 
 		// Update VSCode context for experiments
 		await this.updateVSCodeContext()
+
+		// Check MDM compliance and send user to account tab if not compliant
+		if (!this.checkMdmCompliance()) {
+			await this.postMessageToWebview({ type: "action", action: "accountButtonClicked" })
+		}
 	}
 
 	/**
@@ -1466,6 +1466,7 @@ export class ClineProvider
 				codebaseIndexEmbedderBaseUrl: "",
 				codebaseIndexEmbedderModelId: "",
 			},
+			mdmCompliant: this.checkMdmCompliance(),
 		}
 	}
 
@@ -1721,11 +1722,6 @@ export class ClineProvider
 		const compliance = this.mdmService.isCompliant()
 
 		if (!compliance.compliant) {
-			vscode.window.showErrorMessage(compliance.reason, "Sign In").then((selection) => {
-				if (selection === "Sign In") {
-					this.postMessageToWebview({ type: "action", action: "accountButtonClicked" })
-				}
-			})
 			return false
 		}
 

webview-ui/src/App.tsx

@@ -42,6 +42,7 @@ const App = () => {
 		experiments,
 		cloudUserInfo,
 		cloudIsAuthenticated,
+		mdmCompliant,
 	} = useExtensionState()
 
 	// Create a persistent state manager
@@ -63,15 +64,23 @@ const App = () => {
 	const settingsRef = useRef<SettingsViewRef>(null)
 	const chatViewRef = useRef<ChatViewRef>(null)
 
-	const switchTab = useCallback((newTab: Tab) => {
-		setCurrentSection(undefined)
+	const switchTab = useCallback(
+		(newTab: Tab) => {
+			// Check MDM compliance before allowing tab switching
+			if (mdmCompliant === false && newTab !== "account") {
+				return
+			}
 
-		if (settingsRef.current?.checkUnsaveChanges) {
-			settingsRef.current.checkUnsaveChanges(() => setTab(newTab))
-		} else {
-			setTab(newTab)
-		}
-	}, [])
+			setCurrentSection(undefined)
+
+			if (settingsRef.current?.checkUnsaveChanges) {
+				settingsRef.current.checkUnsaveChanges(() => setTab(newTab))
+			} else {
+				setTab(newTab)
+			}
+		},
+		[mdmCompliant],
+	)
 
 	const [currentSection, setCurrentSection] = useState<string | undefined>(undefined)
 

webview-ui/src/context/ExtensionStateContext.tsx

@@ -37,6 +37,7 @@ export interface ExtensionStateContextType extends ExtensionState {
 	cloudIsAuthenticated: boolean
 	sharingEnabled: boolean
 	maxConcurrentFileReads?: number
+	mdmCompliant?: boolean
 	condensingApiConfigId?: string
 	setCondensingApiConfigId: (value: string) => void
 	customCondensingPrompt?: string