simstudioai--sim/apps/sim/tools/hubspot/create_company.ts

import { createLogger } from '@sim/logger'
import type {
  HubSpotCreateCompanyParams,
  HubSpotCreateCompanyResponse,
} from '@/tools/hubspot/types'
import { COMPANY_OBJECT_OUTPUT } from '@/tools/hubspot/types'
import type { ToolConfig } from '@/tools/types'

const logger = createLogger('HubSpotCreateCompany')

export const hubspotCreateCompanyTool: ToolConfig<
  HubSpotCreateCompanyParams,
  HubSpotCreateCompanyResponse
> = {
  id: 'hubspot_create_company',
  name: 'Create Company in HubSpot',
  description: 'Create a new company in HubSpot',
  version: '1.0.0',

  oauth: {
    required: true,
    provider: 'hubspot',
  },

  params: {
    accessToken: {
      type: 'string',
      required: true,
      visibility: 'hidden',
      description: 'The access token for the HubSpot API',
    },
    properties: {
      type: 'object',
      required: true,
      visibility: 'user-or-llm',
      description:
        'Company properties as JSON object (e.g., {"name": "Acme Inc", "domain": "acme.com", "industry": "Technology"})',
    },
    associations: {
      type: 'array',
      required: false,
      visibility: 'user-or-llm',
      description:
        'Array of associations to create with the company as JSON (each with "to.id" and "types" containing "associationCategory" and "associationTypeId")',
    },
  },

  request: {
    url: () => 'https://api.hubapi.com/crm/v3/objects/companies',
    method: 'POST',
    headers: (params) => {
      if (!params.accessToken) {
        throw new Error('Access token is required')
      }

      return {
        Authorization: `Bearer ${params.accessToken}`,
        'Content-Type': 'application/json',
      }
    },
    body: (params) => {
      let properties = params.properties
      if (typeof properties === 'string') {
        try {
          properties = JSON.parse(properties)
        } catch (e) {
          throw new Error('Invalid JSON format for properties. Please provide a valid JSON object.')
        }
      }

      const body: any = {
        properties,
      }

      let associations = params.associations
      if (typeof associations === 'string') {
        try {
          associations = JSON.parse(associations)
        } catch (e) {
          throw new Error(
            'Invalid JSON format for associations. Please provide a valid JSON array.'
          )
        }
      }
      if (Array.isArray(associations) && associations.length > 0) {
        body.associations = associations
      }

      return body
    },
  },

  transformResponse: async (response: Response) => {
    const data = await response.json()

    if (!response.ok) {
      logger.error('HubSpot API request failed', { data, status: response.status })
      throw new Error(data.message || 'Failed to create company in HubSpot')
    }

    return {
      success: true,
      output: {
        company: data,
        companyId: data.id,
        success: true,
      },
    }
  },

  outputs: {
    company: COMPANY_OBJECT_OUTPUT,
    companyId: { type: 'string', description: 'The created company ID' },
    success: { type: 'boolean', description: 'Operation success status' },
  },
}