All files / web/src/composables/node useNodeHelpers.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type { NodeType } from '@/types/generated/NodeType'
import type { Node as VueFlowNode } from '@vue-flow/core'
import type { Node as BackendNode } from '@/types/generated/Node'
 
// Union type to handle different node formats
export type AnyNode = VueFlowNode | BackendNode | { type?: string; node_type?: NodeType }
 
// Trigger node types
const TRIGGER_TYPES: NodeType[] = [
  'WebhookTrigger',
  'ScheduleTrigger',
  'ManualTrigger',
]
 
export function useNodeHelpers() {
  /**
   * Check if a node is a trigger type
   */
  const isNodeATrigger = (node: AnyNode): boolean => {
    // Handle different node formats
    const nodeType = (node as any)?.type || (node as any)?.node_type
    return nodeType ? TRIGGER_TYPES.includes(nodeType as NodeType) : false
  }
 
  /**
   * Get node category
   */
  const getNodeCategory = (nodeType: NodeType): 'trigger' | 'action' => {
    return TRIGGER_TYPES.includes(nodeType) ? 'trigger' : 'action'
  }
 
  /**
   * Node type constants for template usage
   */
  const NODE_TYPES = {
    // Trigger nodes
    WEBHOOK_TRIGGER: 'WebhookTrigger' as NodeType,
    SCHEDULE_TRIGGER: 'ScheduleTrigger' as NodeType,
    MANUAL_TRIGGER: 'ManualTrigger' as NodeType,
    
    // Action nodes
    AGENT: 'Agent' as NodeType,
    HTTP_REQUEST: 'HttpRequest' as NodeType,
    PRINT: 'Print' as NodeType,
    DATA_TRANSFORM: 'DataTransform' as NodeType,
  } as const
 
  return {
    isNodeATrigger,
    getNodeCategory,
    NODE_TYPES,
    TRIGGER_TYPES,
  }
}
 
// Export standalone function for backwards compatibility
export const isNodeATrigger = (node: AnyNode): boolean => {
  const { isNodeATrigger: checkTrigger } = useNodeHelpers()
  return checkTrigger(node)
}
 
// Export NODE_TYPES for backwards compatibility
export const NODE_TYPES = {
  WEBHOOK_TRIGGER: 'WebhookTrigger' as NodeType,
  SCHEDULE_TRIGGER: 'ScheduleTrigger' as NodeType,
  MANUAL_TRIGGER: 'ManualTrigger' as NodeType,
  AGENT: 'Agent' as NodeType,
  HTTP_REQUEST: 'HttpRequest' as NodeType,
  PYTHON: 'Python' as NodeType,
  PRINT: 'Print' as NodeType,
  DATA_TRANSFORM: 'DataTransform' as NodeType,
} as const