supermemory/apps/web/lib/billing-utils.ts

const COMPANY_BRAIN_PRODUCT_ID = "company_brain"

// Add-on resolved by product presence, not tier.
// better-auth returns org.metadata as a JSON string, so accept string or object.
export function hasCompanyBrain(
	metadataRaw: Record<string, unknown> | string | null | undefined,
): boolean {
	if (!metadataRaw) return false
	let metadata: Record<string, unknown>
	if (typeof metadataRaw === "string") {
		try {
			metadata = JSON.parse(metadataRaw) as Record<string, unknown>
		} catch {
			return false
		}
	} else {
		metadata = metadataRaw
	}
	const overrides = metadata.featureOverrides as
		| Record<string, { allow?: boolean }>
		| undefined
	const override = overrides?.[COMPANY_BRAIN_PRODUCT_ID]
	if (override) return Boolean(override.allow)
	const activeProducts = Array.isArray(metadata.activeProducts)
		? (metadata.activeProducts as string[])
		: []
	return activeProducts.includes(COMPANY_BRAIN_PRODUCT_ID)
}

// Explicit concierge override for company_brain, or undefined when none is set.
export function getCompanyBrainOverride(
	metadataRaw: Record<string, unknown> | string | null | undefined,
): boolean | undefined {
	if (!metadataRaw) return undefined
	let metadata: Record<string, unknown>
	if (typeof metadataRaw === "string") {
		try {
			metadata = JSON.parse(metadataRaw) as Record<string, unknown>
		} catch {
			return undefined
		}
	} else {
		metadata = metadataRaw
	}
	const overrides = metadata.featureOverrides as
		| Record<string, { allow?: boolean }>
		| undefined
	const override = overrides?.[COMPANY_BRAIN_PRODUCT_ID]
	return override ? Boolean(override.allow) : undefined
}

// Origin of the org. Consumer (app.supermemory) orgs get company_brain attached,
// but the add-on lands async — signupSource is set at creation, so it's the
// reliable "this org uses brain spaces" signal in the UI.
export function getSignupSource(
	metadataRaw: Record<string, unknown> | string | null | undefined,
): string | null {
	if (!metadataRaw) return null
	let metadata: Record<string, unknown>
	if (typeof metadataRaw === "string") {
		try {
			metadata = JSON.parse(metadataRaw) as Record<string, unknown>
		} catch {
			return null
		}
	} else {
		metadata = metadataRaw
	}
	return typeof metadata.signupSource === "string"
		? (metadata.signupSource as string)
		: null
}

// Brain mode chosen during onboarding ("personal" | "team"). Set synchronously
// at org creation, so it's the reliable pre-webhook signal for company brain.
export function getBrainMode(
	metadataRaw: Record<string, unknown> | string | null | undefined,
): string | null {
	if (!metadataRaw) return null
	let metadata: Record<string, unknown>
	if (typeof metadataRaw === "string") {
		try {
			metadata = JSON.parse(metadataRaw) as Record<string, unknown>
		} catch {
			return null
		}
	} else {
		metadata = metadataRaw
	}
	return typeof metadata.brainMode === "string"
		? (metadata.brainMode as string)
		: null
}

/**
 * Format a number with K/M suffix for display
 * @example formatUsageNumber(1500000) => "1.5M"
 * @example formatUsageNumber(50000) => "50K"
 */
export function formatUsageNumber(value: number): string {
	if (value >= 1_000_000) {
		const millions = value / 1_000_000
		return millions % 1 === 0 ? `${millions}M` : `${millions.toFixed(1)}M`
	}
	if (value >= 1_000) {
		const thousands = value / 1_000
		return thousands % 1 === 0 ? `${thousands}K` : `${thousands.toFixed(1)}K`
	}
	return value.toString()
}

/**
 * Calculate usage percentage, clamped between 0 and 100
 */
export function calculateUsagePercent(used: number, limit: number): number {
	if (limit <= 0) return 0
	return Math.min(Math.max((used / limit) * 100, 0), 100)
}

/**
 * Get the number of days remaining until a reset date
 * @param resetAt - Unix timestamp in milliseconds (from Autumn billing)
 * @returns number of days, or null if no date provided
 */
export function getDaysRemaining(
	resetAt: number | null | undefined,
): number | null {
	if (!resetAt) return null
	const end = new Date(resetAt)
	const now = new Date()
	const diffMs = end.getTime() - now.getTime()
	const diffDays = Math.ceil(diffMs / (1000 * 60 * 60 * 24))
	return Math.max(0, diffDays)
}