mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-09 17:40:09 +00:00
don't use compound name, use isSubscriptionRequirementMet
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
<template>
|
||||
<component
|
||||
:is="currentButton"
|
||||
:key="isSubscribedOrIsNotCloud ? 'queue' : 'subscribe'"
|
||||
:key="isSubscriptionRequirementMet ? 'queue' : 'subscribe'"
|
||||
/>
|
||||
</template>
|
||||
<script setup lang="ts">
|
||||
@@ -11,9 +11,9 @@ import ComfyQueueButton from '@/components/actionbar/ComfyRunButton/ComfyQueueBu
|
||||
import SubscribeToRunButton from '@/platform/cloud/subscription/components/SubscribeToRun.vue'
|
||||
import { useSubscription } from '@/platform/cloud/subscription/composables/useSubscription'
|
||||
|
||||
const { isSubscribedOrIsNotCloud } = useSubscription()
|
||||
const { isSubscriptionRequirementMet } = useSubscription()
|
||||
|
||||
const currentButton = computed(() =>
|
||||
isSubscribedOrIsNotCloud.value ? ComfyQueueButton : SubscribeToRunButton
|
||||
isSubscriptionRequirementMet.value ? ComfyQueueButton : SubscribeToRunButton
|
||||
)
|
||||
</script>
|
||||
|
||||
@@ -15,7 +15,7 @@
|
||||
<UserCredit text-class="text-3xl font-bold" />
|
||||
<Skeleton v-if="loading" width="2rem" height="2rem" />
|
||||
<Button
|
||||
v-else-if="isSubscribedOrIsNotCloud"
|
||||
v-else-if="isSubscriptionRequirementMet"
|
||||
:label="$t('credits.purchaseCredits')"
|
||||
:loading="loading"
|
||||
@click="handlePurchaseCreditsClick"
|
||||
@@ -146,9 +146,9 @@ const authActions = useFirebaseAuthActions()
|
||||
const commandStore = useCommandStore()
|
||||
const telemetry = useTelemetry()
|
||||
const subscription = isCloud ? useSubscription() : null
|
||||
const isSubscribedOrIsNotCloud = computed(() => {
|
||||
const isSubscriptionRequirementMet = computed(() => {
|
||||
if (!isCloud) return true
|
||||
return subscription?.isSubscribedOrIsNotCloud.value ?? false
|
||||
return subscription?.isSubscriptionRequirementMet.value ?? false
|
||||
})
|
||||
const loading = computed(() => authStore.loading)
|
||||
const balanceLoading = computed(() => authStore.isFetchingBalance)
|
||||
|
||||
@@ -82,7 +82,7 @@ vi.mock('@/stores/firebaseAuthStore', () => ({
|
||||
const mockFetchStatus = vi.fn().mockResolvedValue(undefined)
|
||||
vi.mock('@/platform/cloud/subscription/composables/useSubscription', () => ({
|
||||
useSubscription: vi.fn(() => ({
|
||||
isSubscribedOrIsNotCloud: { value: true },
|
||||
isSubscriptionRequirementMet: { value: true },
|
||||
fetchStatus: mockFetchStatus
|
||||
}))
|
||||
}))
|
||||
|
||||
@@ -24,7 +24,7 @@
|
||||
</div>
|
||||
|
||||
<div
|
||||
v-if="isSubscribedOrIsNotCloud"
|
||||
v-if="isSubscriptionRequirementMet"
|
||||
class="flex items-center justify-between"
|
||||
>
|
||||
<div class="flex flex-col gap-1">
|
||||
@@ -71,7 +71,7 @@
|
||||
/>
|
||||
|
||||
<Button
|
||||
v-if="isSubscribedOrIsNotCloud"
|
||||
v-if="isSubscriptionRequirementMet"
|
||||
class="justify-start"
|
||||
:label="$t(planSettingsLabel)"
|
||||
icon="pi pi-receipt"
|
||||
@@ -98,7 +98,7 @@
|
||||
<script setup lang="ts">
|
||||
import Button from 'primevue/button'
|
||||
import Divider from 'primevue/divider'
|
||||
import { onMounted } from 'vue'
|
||||
import { computed, onMounted } from 'vue'
|
||||
|
||||
import UserAvatar from '@/components/common/UserAvatar.vue'
|
||||
import UserCredit from '@/components/common/UserCredit.vue'
|
||||
@@ -125,7 +125,10 @@ const { userDisplayName, userEmail, userPhotoUrl, handleSignOut } =
|
||||
useCurrentUser()
|
||||
const authActions = useFirebaseAuthActions()
|
||||
const dialogService = useDialogService()
|
||||
const { isSubscribedOrIsNotCloud, fetchStatus } = useSubscription()
|
||||
const subscription = isCloud ? useSubscription() : null
|
||||
const isSubscriptionRequirementMet =
|
||||
subscription?.isSubscriptionRequirementMet ?? computed(() => true)
|
||||
const fetchStatus = subscription?.fetchStatus ?? (async () => {})
|
||||
|
||||
const handleOpenUserSettings = () => {
|
||||
dialogService.showSettingsDialog('user')
|
||||
|
||||
@@ -6,7 +6,6 @@ import { useErrorHandling } from '@/composables/useErrorHandling'
|
||||
import type { ErrorRecoveryStrategy } from '@/composables/useErrorHandling'
|
||||
import { t } from '@/i18n'
|
||||
import { isCloud } from '@/platform/distribution/types'
|
||||
import { useSubscription } from '@/platform/cloud/subscription/composables/useSubscription'
|
||||
import { useTelemetry } from '@/platform/telemetry'
|
||||
import { useToastStore } from '@/platform/updates/common/toastStore'
|
||||
import { useDialogService } from '@/services/dialogService'
|
||||
@@ -83,8 +82,11 @@ export const useFirebaseAuthActions = () => {
|
||||
|
||||
const purchaseCredits = wrapWithErrorHandlingAsync(async (amount: number) => {
|
||||
if (isCloud) {
|
||||
const { isSubscribedOrIsNotCloud } = useSubscription()
|
||||
if (!isSubscribedOrIsNotCloud.value) return
|
||||
const { useSubscription } = await import(
|
||||
'@/platform/cloud/subscription/composables/useSubscription'
|
||||
)
|
||||
const { isSubscriptionRequirementMet } = useSubscription()
|
||||
if (!isSubscriptionRequirementMet.value) return
|
||||
}
|
||||
|
||||
const response = await authStore.initiateCreditPurchase({
|
||||
|
||||
@@ -91,7 +91,7 @@ export function useCoreCommands(): ComfyCommand[] {
|
||||
|
||||
const subscription = isCloud ? useSubscription() : null
|
||||
const subscriptionState =
|
||||
subscription?.isSubscribedOrIsNotCloud ?? defaultSubscriptionState
|
||||
subscription?.isSubscriptionRequirementMet ?? defaultSubscriptionState
|
||||
const subscriptionDialog = subscription?.showSubscriptionDialog ?? noop
|
||||
|
||||
const moveSelectedNodes = (
|
||||
|
||||
@@ -51,7 +51,8 @@ const emit = defineEmits<{
|
||||
subscribed: []
|
||||
}>()
|
||||
|
||||
const { subscribe, isSubscribedOrIsNotCloud, fetchStatus } = useSubscription()
|
||||
const { subscribe, isSubscriptionRequirementMet, fetchStatus } =
|
||||
useSubscription()
|
||||
const telemetry = useTelemetry()
|
||||
|
||||
const isLoading = ref(false)
|
||||
@@ -76,7 +77,7 @@ const startPollingSubscriptionStatus = () => {
|
||||
|
||||
await fetchStatus()
|
||||
|
||||
if (isSubscribedOrIsNotCloud.value) {
|
||||
if (isSubscriptionRequirementMet.value) {
|
||||
stopPolling()
|
||||
telemetry?.trackMonthlySubscriptionSucceeded()
|
||||
emit('subscribed')
|
||||
|
||||
@@ -4,7 +4,7 @@
|
||||
<div class="flex items-baseline gap-2">
|
||||
<span class="text-2xl font-inter font-semibold leading-tight">
|
||||
{{
|
||||
isSubscribedOrIsNotCloud
|
||||
isSubscriptionRequirementMet
|
||||
? $t('subscription.title')
|
||||
: $t('subscription.titleUnsubscribed')
|
||||
}}
|
||||
@@ -27,7 +27,7 @@
|
||||
}}</span>
|
||||
</div>
|
||||
<div
|
||||
v-if="isSubscribedOrIsNotCloud"
|
||||
v-if="isSubscriptionRequirementMet"
|
||||
class="text-sm text-text-secondary"
|
||||
>
|
||||
<template v-if="isCancelled">
|
||||
@@ -47,7 +47,7 @@
|
||||
</div>
|
||||
</div>
|
||||
<Button
|
||||
v-if="isSubscribedOrIsNotCloud"
|
||||
v-if="isSubscriptionRequirementMet"
|
||||
:label="$t('subscription.manageSubscription')"
|
||||
severity="secondary"
|
||||
class="text-xs bg-interface-menu-component-surface-selected"
|
||||
@@ -196,7 +196,7 @@
|
||||
{{ $t('subscription.viewUsageHistory') }}
|
||||
</a>
|
||||
<Button
|
||||
v-if="isSubscribedOrIsNotCloud"
|
||||
v-if="isSubscriptionRequirementMet"
|
||||
:label="$t('subscription.addCredits')"
|
||||
severity="secondary"
|
||||
class="p-2 min-h-8 bg-interface-menu-component-surface-selected"
|
||||
@@ -320,7 +320,7 @@ import { cn } from '@/utils/tailwindUtil'
|
||||
const { buildDocsUrl } = useExternalLink()
|
||||
|
||||
const {
|
||||
isSubscribedOrIsNotCloud,
|
||||
isSubscriptionRequirementMet,
|
||||
isCancelled,
|
||||
formattedRenewalDate,
|
||||
formattedEndDate,
|
||||
|
||||
@@ -31,7 +31,7 @@ function useSubscriptionInternal() {
|
||||
const subscriptionStatus = ref<CloudSubscriptionStatusResponse | null>(null)
|
||||
const telemetry = useTelemetry()
|
||||
|
||||
const isSubscribedOrIsNotCloud = computed(() => {
|
||||
const isSubscriptionRequirementMet = computed(() => {
|
||||
if (!isCloud || !window.__CONFIG__?.subscription_required) return true
|
||||
|
||||
return subscriptionStatus.value?.is_active ?? false
|
||||
@@ -111,7 +111,7 @@ function useSubscriptionInternal() {
|
||||
const { startCancellationWatcher, stopCancellationWatcher } =
|
||||
useSubscriptionCancellationWatcher({
|
||||
fetchStatus,
|
||||
isSubscribedOrIsNotCloud,
|
||||
isSubscriptionRequirementMet,
|
||||
subscriptionStatus,
|
||||
telemetry,
|
||||
shouldWatchCancellation
|
||||
@@ -125,7 +125,7 @@ function useSubscriptionInternal() {
|
||||
const requireActiveSubscription = async (): Promise<void> => {
|
||||
await fetchSubscriptionStatus()
|
||||
|
||||
if (!isSubscribedOrIsNotCloud.value) {
|
||||
if (!isSubscriptionRequirementMet.value) {
|
||||
showSubscriptionDialog()
|
||||
}
|
||||
}
|
||||
@@ -223,7 +223,7 @@ function useSubscriptionInternal() {
|
||||
|
||||
return {
|
||||
// State
|
||||
isSubscribedOrIsNotCloud,
|
||||
isSubscriptionRequirementMet,
|
||||
isCancelled,
|
||||
formattedRenewalDate,
|
||||
formattedEndDate,
|
||||
|
||||
@@ -36,8 +36,8 @@ export function useSubscriptionActions() {
|
||||
void handleRefresh()
|
||||
})
|
||||
|
||||
const handleAddApiCredits = () => {
|
||||
dialogService.showTopUpCreditsDialog()
|
||||
const handleAddApiCredits = async () => {
|
||||
await dialogService.showTopUpCreditsDialog()
|
||||
}
|
||||
|
||||
const handleMessageSupport = async () => {
|
||||
|
||||
@@ -12,7 +12,7 @@ const CANCELLATION_BACKOFF_MULTIPLIER = 3 // 5s, 15s, 45s, 135s intervals
|
||||
|
||||
type CancellationWatcherOptions = {
|
||||
fetchStatus: () => Promise<CloudSubscriptionStatusResponse | null | void>
|
||||
isSubscribedOrIsNotCloud: ComputedRef<boolean>
|
||||
isSubscriptionRequirementMet: ComputedRef<boolean>
|
||||
subscriptionStatus: Ref<CloudSubscriptionStatusResponse | null>
|
||||
telemetry: Pick<TelemetryProvider, 'trackMonthlySubscriptionCancelled'> | null
|
||||
shouldWatchCancellation: () => boolean
|
||||
@@ -20,7 +20,7 @@ type CancellationWatcherOptions = {
|
||||
|
||||
export function useSubscriptionCancellationWatcher({
|
||||
fetchStatus,
|
||||
isSubscribedOrIsNotCloud,
|
||||
isSubscriptionRequirementMet,
|
||||
subscriptionStatus,
|
||||
telemetry,
|
||||
shouldWatchCancellation
|
||||
@@ -73,7 +73,7 @@ export function useSubscriptionCancellationWatcher({
|
||||
try {
|
||||
await fetchStatus()
|
||||
|
||||
if (!isSubscribedOrIsNotCloud.value) {
|
||||
if (!isSubscriptionRequirementMet.value) {
|
||||
if (!cancellationTracked.value) {
|
||||
cancellationTracked.value = true
|
||||
try {
|
||||
|
||||
@@ -681,8 +681,8 @@ export class ComfyApp {
|
||||
isInsufficientCredits: true
|
||||
})
|
||||
} else {
|
||||
const { isSubscribedOrIsNotCloud } = useSubscription()
|
||||
if (isSubscribedOrIsNotCloud.value) {
|
||||
const { isSubscriptionRequirementMet } = useSubscription()
|
||||
if (isSubscriptionRequirementMet.value) {
|
||||
useDialogService().showTopUpCreditsDialog({
|
||||
isInsufficientCredits: true
|
||||
})
|
||||
|
||||
@@ -17,7 +17,6 @@ import SettingDialogHeader from '@/components/dialog/header/SettingDialogHeader.
|
||||
import { t } from '@/i18n'
|
||||
import { useTelemetry } from '@/platform/telemetry'
|
||||
import { isCloud } from '@/platform/distribution/types'
|
||||
import { useSubscription } from '@/platform/cloud/subscription/composables/useSubscription'
|
||||
import SettingDialogContent from '@/platform/settings/components/SettingDialogContent.vue'
|
||||
import type { ExecutionErrorWsMessage } from '@/schemas/apiSchema'
|
||||
import { useDialogStore } from '@/stores/dialogStore'
|
||||
@@ -377,12 +376,15 @@ export const useDialogService = () => {
|
||||
})
|
||||
}
|
||||
|
||||
function showTopUpCreditsDialog(options?: {
|
||||
async function showTopUpCreditsDialog(options?: {
|
||||
isInsufficientCredits?: boolean
|
||||
}) {
|
||||
if (isCloud) {
|
||||
const { isSubscribedOrIsNotCloud } = useSubscription()
|
||||
if (!isSubscribedOrIsNotCloud.value) return
|
||||
const { useSubscription } = await import(
|
||||
'@/platform/cloud/subscription/composables/useSubscription'
|
||||
)
|
||||
const { isSubscriptionRequirementMet } = useSubscription()
|
||||
if (!isSubscriptionRequirementMet.value) return
|
||||
}
|
||||
|
||||
return dialogStore.showDialog({
|
||||
|
||||
Reference in New Issue
Block a user