simstudioai--sim/apps/sim/lib/copilot/tools/server/blocks/get-trigger-blocks.ts

import { createLogger } from '@sim/logger'
import { z } from 'zod'
import type { BaseServerTool } from '@/lib/copilot/tools/server/base-tool'
import { getAllowedIntegrationsFromEnv } from '@/lib/core/config/env-flags'
import { getAllBlocks } from '@/blocks/registry'
import { getUserPermissionConfig } from '@/ee/access-control/utils/permission-check'

export const GetTriggerBlocksInput = z.object({})
export const GetTriggerBlocksResult = z.object({
  triggerBlockIds: z.array(z.string()),
})

export const getTriggerBlocksServerTool: BaseServerTool<
  ReturnType<typeof GetTriggerBlocksInput.parse>,
  ReturnType<typeof GetTriggerBlocksResult.parse>
> = {
  name: 'get_trigger_blocks',
  inputSchema: GetTriggerBlocksInput,
  outputSchema: GetTriggerBlocksResult,
  async execute(_args: unknown, context?: { userId: string; workspaceId?: string }) {
    const logger = createLogger('GetTriggerBlocksServerTool')
    logger.debug('Executing get_trigger_blocks')

    const permissionConfig =
      context?.userId && context?.workspaceId
        ? await getUserPermissionConfig(context.userId, context.workspaceId)
        : null
    const allowedIntegrations =
      permissionConfig?.allowedIntegrations ?? getAllowedIntegrationsFromEnv()

    const triggerBlockIds: string[] = []

    for (const blockConfig of getAllBlocks()) {
      const blockType = blockConfig.type
      if (blockConfig.hideFromToolbar) continue
      if (allowedIntegrations != null && !allowedIntegrations.includes(blockType.toLowerCase()))
        continue

      if (blockConfig.category === 'triggers') {
        triggerBlockIds.push(blockType)
      } else if ('triggerAllowed' in blockConfig && blockConfig.triggerAllowed === true) {
        triggerBlockIds.push(blockType)
      } else if (blockConfig.subBlocks?.some((subBlock) => subBlock.mode === 'trigger')) {
        triggerBlockIds.push(blockType)
      }
    }

    triggerBlockIds.sort()

    logger.debug(`Found ${triggerBlockIds.length} trigger blocks`)
    return GetTriggerBlocksResult.parse({ triggerBlockIds })
  },
}