chore: import upstream snapshot with attribution

apps/sim/app/api/custom-blocks/[id]/usages/route.test.ts

@@ -0,0 +1,91 @@
+/**
+ * @vitest-environment node
+ */
+import { createMockRequest } from '@sim/testing'
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const { mockGetSession, mockIsFeatureEnabled, mockHasWorkspaceAdminAccess, mockOperations } =
+  vi.hoisted(() => ({
+    mockGetSession: vi.fn(),
+    mockIsFeatureEnabled: vi.fn(),
+    mockHasWorkspaceAdminAccess: vi.fn(),
+    mockOperations: {
+      getCustomBlockManageContext: vi.fn(),
+      getCustomBlockUsageCounts: vi.fn(),
+    },
+  }))
+
+vi.mock('@/lib/auth', () => ({
+  getSession: mockGetSession,
+}))
+
+vi.mock('@/lib/core/config/feature-flags', () => ({
+  isFeatureEnabled: mockIsFeatureEnabled,
+}))
+
+vi.mock('@/lib/workspaces/permissions/utils', () => ({
+  hasWorkspaceAdminAccess: mockHasWorkspaceAdminAccess,
+}))
+
+vi.mock('@/lib/workflows/custom-blocks/operations', () => mockOperations)
+
+import { GET } from '@/app/api/custom-blocks/[id]/usages/route'
+
+const MANAGE_CONTEXT = {
+  organizationId: 'org-1',
+  sourceWorkspaceId: 'ws-1',
+  type: 'custom_block_abc123',
+  name: 'Invoice Parser',
+}
+
+const USAGE_COUNTS = { usageCount: 3, deployedUsageCount: 2 }
+
+function callRoute(id = 'cb-1') {
+  return GET(createMockRequest('GET'), { params: Promise.resolve({ id }) })
+}
+
+describe('GET /api/custom-blocks/[id]/usages', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    mockGetSession.mockResolvedValue({ user: { id: 'user-1' } })
+    mockIsFeatureEnabled.mockResolvedValue(true)
+    mockHasWorkspaceAdminAccess.mockResolvedValue(true)
+    mockOperations.getCustomBlockManageContext.mockResolvedValue(MANAGE_CONTEXT)
+    mockOperations.getCustomBlockUsageCounts.mockResolvedValue(USAGE_COUNTS)
+  })
+
+  it('returns 401 without a session', async () => {
+    mockGetSession.mockResolvedValue(null)
+    const response = await callRoute()
+    expect(response.status).toBe(401)
+  })
+
+  it('returns 404 for an unknown block', async () => {
+    mockOperations.getCustomBlockManageContext.mockResolvedValue(null)
+    const response = await callRoute()
+    expect(response.status).toBe(404)
+  })
+
+  it('returns 403 when the feature flag is off', async () => {
+    mockIsFeatureEnabled.mockResolvedValue(false)
+    const response = await callRoute()
+    expect(response.status).toBe(403)
+  })
+
+  it('returns 403 for a non-admin of the source workspace', async () => {
+    mockHasWorkspaceAdminAccess.mockResolvedValue(false)
+    const response = await callRoute()
+    expect(response.status).toBe(403)
+    expect(mockOperations.getCustomBlockUsageCounts).not.toHaveBeenCalled()
+  })
+
+  it('returns the org-scoped usage counts for the block type', async () => {
+    const response = await callRoute()
+    expect(response.status).toBe(200)
+    expect(await response.json()).toEqual(USAGE_COUNTS)
+    expect(mockOperations.getCustomBlockUsageCounts).toHaveBeenCalledWith(
+      'org-1',
+      'custom_block_abc123'
+    )
+  })
+})

apps/sim/app/api/custom-blocks/[id]/usages/route.ts

@@ -0,0 +1,26 @@
+import type { NextRequest } from 'next/server'
+import { NextResponse } from 'next/server'
+import { getCustomBlockUsageCountsContract } from '@/lib/api/contracts/custom-blocks'
+import { parseRequest } from '@/lib/api/server'
+import { getSession } from '@/lib/auth'
+import { withRouteHandler } from '@/lib/core/utils/with-route-handler'
+import { getCustomBlockUsageCounts } from '@/lib/workflows/custom-blocks/operations'
+import { authorizeManage } from '@/app/api/custom-blocks/[id]/authorize-manage'
+
+type RouteContext = { params: Promise<{ id: string }> }
+
+export const GET = withRouteHandler(async (request: NextRequest, context: RouteContext) => {
+  const session = await getSession()
+  if (!session?.user?.id) {
+    return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
+  }
+
+  const parsed = await parseRequest(getCustomBlockUsageCountsContract, request, context)
+  if (!parsed.success) return parsed.response
+
+  const authz = await authorizeManage(session.user.id, parsed.data.params.id)
+  if (authz.error) return authz.error
+
+  const counts = await getCustomBlockUsageCounts(authz.ctx.organizationId, authz.ctx.type)
+  return NextResponse.json(counts)
+})