dont import cloud in core commands module

This commit is contained in:
bymyself
2025-12-04 12:14:00 -08:00
parent 9d889198f8
commit afaceb5dd8
2 changed files with 19 additions and 8 deletions

View File

@@ -1,3 +1,5 @@
import type { ComputedRef } from 'vue'
import { useCurrentUser } from '@/composables/auth/useCurrentUser'
import { useFirebaseAuthActions } from '@/composables/auth/useFirebaseAuthActions'
import { useSelectedLiteGraphItems } from '@/composables/canvas/useSelectedLiteGraphItems'
@@ -46,6 +48,7 @@ import { useQueueSettingsStore, useQueueStore } from '@/stores/queueStore'
import { useSubgraphNavigationStore } from '@/stores/subgraphNavigationStore'
import { useSubgraphStore } from '@/stores/subgraphStore'
import { useBottomPanelStore } from '@/stores/workspace/bottomPanelStore'
import { isCloud } from '@/platform/distribution/types'
import { useColorPaletteStore } from '@/stores/workspace/colorPaletteStore'
import { useRightSidePanelStore } from '@/stores/workspace/rightSidePanelStore'
import { useSearchBoxStore } from '@/stores/workspace/searchBoxStore'
@@ -63,7 +66,10 @@ import { ManagerTab } from '@/workbench/extensions/manager/types/comfyManagerTyp
import { useWorkflowTemplateSelectorDialog } from './useWorkflowTemplateSelectorDialog'
const { isSubscribedOrIsNotCloud, showSubscriptionDialog } = useSubscription()
const defaultSubscriptionState: Pick<ComputedRef<boolean>, 'value'> = {
value: true
}
const noop = () => {}
const moveSelectedNodesVersionAdded = '1.22.2'
@@ -85,6 +91,11 @@ export function useCoreCommands(): ComfyCommand[] {
useSelectedLiteGraphItems()
const getTracker = () => workflowStore.activeWorkflow?.changeTracker
const subscription = isCloud ? useSubscription() : null
const subscriptionState =
subscription?.isSubscribedOrIsNotCloud ?? defaultSubscriptionState
const subscriptionDialog = subscription?.showSubscriptionDialog ?? noop
const moveSelectedNodes = (
positionUpdater: (pos: Point, gridSize: number) => Point
) => {
@@ -475,8 +486,8 @@ export function useCoreCommands(): ComfyCommand[] {
trigger_source?: ExecutionTriggerSource
}) => {
useTelemetry()?.trackRunButton(metadata)
if (!isSubscribedOrIsNotCloud.value) {
showSubscriptionDialog()
if (!subscriptionState.value) {
subscriptionDialog()
return
}
@@ -498,8 +509,8 @@ export function useCoreCommands(): ComfyCommand[] {
trigger_source?: ExecutionTriggerSource
}) => {
useTelemetry()?.trackRunButton(metadata)
if (!isSubscribedOrIsNotCloud.value) {
showSubscriptionDialog()
if (!subscriptionState.value) {
subscriptionDialog()
return
}
@@ -520,8 +531,8 @@ export function useCoreCommands(): ComfyCommand[] {
trigger_source?: ExecutionTriggerSource
}) => {
useTelemetry()?.trackRunButton(metadata)
if (!isSubscribedOrIsNotCloud.value) {
showSubscriptionDialog()
if (!subscriptionState.value) {
subscriptionDialog()
return
}

View File

@@ -100,7 +100,7 @@ vi.mock('@/composables/auth/useFirebaseAuthActions', () => ({
vi.mock('@/platform/cloud/subscription/composables/useSubscription', () => ({
useSubscription: vi.fn(() => ({
isSubscribedOrIsNotCloud: vi.fn().mockReturnValue(true),
isSubscribedOrIsNotCloud: { value: true },
showSubscriptionDialog: vi.fn()
}))
}))