418 lines
12 KiB
TypeScript
418 lines
12 KiB
TypeScript
'use client'
|
|
|
|
import { useCallback, useState, useEffect } from 'react'
|
|
import {
|
|
ReactFlow,
|
|
Controls,
|
|
Background,
|
|
BackgroundVariant,
|
|
useNodesState,
|
|
useEdgesState,
|
|
addEdge,
|
|
reconnectEdge,
|
|
Connection,
|
|
MarkerType,
|
|
Panel,
|
|
} from '@xyflow/react'
|
|
import '@xyflow/react/dist/style.css'
|
|
|
|
import FunnelNode from './nodes/FunnelNode'
|
|
import OutcomeNode from './nodes/OutcomeNode'
|
|
import type { FlowNode, FlowEdge, FunnelNodeData, OutcomeNodeData } from '@/lib/types'
|
|
|
|
const nodeTypes = {
|
|
funnel: FunnelNode,
|
|
outcome: OutcomeNode,
|
|
}
|
|
|
|
// Colors for allocations
|
|
const SPENDING_COLORS = ['#3b82f6', '#8b5cf6', '#ec4899', '#06b6d4', '#10b981', '#6366f1']
|
|
const OVERFLOW_COLORS = ['#f59e0b', '#ef4444', '#f97316', '#eab308', '#dc2626', '#ea580c']
|
|
|
|
const initialNodes: FlowNode[] = [
|
|
// Main Treasury Funnel (receives inflows from external sources)
|
|
{
|
|
id: 'treasury',
|
|
type: 'funnel',
|
|
position: { x: 350, y: 0 },
|
|
data: {
|
|
label: 'Treasury',
|
|
currentValue: 85000,
|
|
minThreshold: 20000,
|
|
maxThreshold: 70000,
|
|
maxCapacity: 100000,
|
|
inflowRate: 1000,
|
|
// OUTFLOWS go SIDEWAYS to other funnels
|
|
overflowAllocations: [
|
|
{ targetId: 'public-goods', percentage: 40, color: OVERFLOW_COLORS[0] },
|
|
{ targetId: 'research', percentage: 35, color: OVERFLOW_COLORS[1] },
|
|
{ targetId: 'emergency', percentage: 25, color: OVERFLOW_COLORS[2] },
|
|
],
|
|
// SPENDING goes DOWN to outcomes
|
|
spendingAllocations: [
|
|
{ targetId: 'treasury-ops', percentage: 100, color: SPENDING_COLORS[0] },
|
|
],
|
|
} as FunnelNodeData,
|
|
},
|
|
// Sub-funnels (receive INFLOWS from Treasury outflows)
|
|
{
|
|
id: 'public-goods',
|
|
type: 'funnel',
|
|
position: { x: 50, y: 250 },
|
|
data: {
|
|
label: 'Public Goods',
|
|
currentValue: 45000,
|
|
minThreshold: 15000,
|
|
maxThreshold: 50000,
|
|
maxCapacity: 70000,
|
|
inflowRate: 400,
|
|
overflowAllocations: [],
|
|
spendingAllocations: [
|
|
{ targetId: 'pg-infra', percentage: 50, color: SPENDING_COLORS[0] },
|
|
{ targetId: 'pg-education', percentage: 30, color: SPENDING_COLORS[1] },
|
|
{ targetId: 'pg-tooling', percentage: 20, color: SPENDING_COLORS[2] },
|
|
],
|
|
} as FunnelNodeData,
|
|
},
|
|
{
|
|
id: 'research',
|
|
type: 'funnel',
|
|
position: { x: 350, y: 250 },
|
|
data: {
|
|
label: 'Research',
|
|
currentValue: 28000,
|
|
minThreshold: 20000,
|
|
maxThreshold: 45000,
|
|
maxCapacity: 60000,
|
|
inflowRate: 350,
|
|
overflowAllocations: [],
|
|
spendingAllocations: [
|
|
{ targetId: 'research-grants', percentage: 70, color: SPENDING_COLORS[0] },
|
|
{ targetId: 'research-papers', percentage: 30, color: SPENDING_COLORS[1] },
|
|
],
|
|
} as FunnelNodeData,
|
|
},
|
|
{
|
|
id: 'emergency',
|
|
type: 'funnel',
|
|
position: { x: 650, y: 250 },
|
|
data: {
|
|
label: 'Emergency',
|
|
currentValue: 12000,
|
|
minThreshold: 25000,
|
|
maxThreshold: 60000,
|
|
maxCapacity: 80000,
|
|
inflowRate: 250,
|
|
overflowAllocations: [],
|
|
spendingAllocations: [
|
|
{ targetId: 'emergency-response', percentage: 100, color: SPENDING_COLORS[0] },
|
|
],
|
|
} as FunnelNodeData,
|
|
},
|
|
// Outcome nodes (receive SPENDING from funnels via BOTTOM)
|
|
{
|
|
id: 'treasury-ops',
|
|
type: 'outcome',
|
|
position: { x: 350, y: 500 },
|
|
data: {
|
|
label: 'Treasury Ops',
|
|
description: 'Day-to-day treasury management',
|
|
fundingReceived: 15000,
|
|
fundingTarget: 25000,
|
|
status: 'in-progress',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'pg-infra',
|
|
type: 'outcome',
|
|
position: { x: -20, y: 500 },
|
|
data: {
|
|
label: 'Infrastructure',
|
|
description: 'Core infrastructure development',
|
|
fundingReceived: 22000,
|
|
fundingTarget: 30000,
|
|
status: 'in-progress',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'pg-education',
|
|
type: 'outcome',
|
|
position: { x: 50, y: 620 },
|
|
data: {
|
|
label: 'Education',
|
|
description: 'Developer education programs',
|
|
fundingReceived: 12000,
|
|
fundingTarget: 20000,
|
|
status: 'in-progress',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'pg-tooling',
|
|
type: 'outcome',
|
|
position: { x: 180, y: 500 },
|
|
data: {
|
|
label: 'Dev Tooling',
|
|
description: 'Open-source developer tools',
|
|
fundingReceived: 5000,
|
|
fundingTarget: 15000,
|
|
status: 'not-started',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'research-grants',
|
|
type: 'outcome',
|
|
position: { x: 300, y: 500 },
|
|
data: {
|
|
label: 'Grants',
|
|
description: 'Academic research grants',
|
|
fundingReceived: 18000,
|
|
fundingTarget: 25000,
|
|
status: 'in-progress',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'research-papers',
|
|
type: 'outcome',
|
|
position: { x: 420, y: 500 },
|
|
data: {
|
|
label: 'Papers',
|
|
description: 'Peer-reviewed publications',
|
|
fundingReceived: 8000,
|
|
fundingTarget: 10000,
|
|
status: 'in-progress',
|
|
} as OutcomeNodeData,
|
|
},
|
|
{
|
|
id: 'emergency-response',
|
|
type: 'outcome',
|
|
position: { x: 650, y: 500 },
|
|
data: {
|
|
label: 'Response Fund',
|
|
description: 'Rapid response for critical issues',
|
|
fundingReceived: 5000,
|
|
fundingTarget: 50000,
|
|
status: 'not-started',
|
|
} as OutcomeNodeData,
|
|
},
|
|
]
|
|
|
|
// Generate edges from node allocations
|
|
function generateEdges(nodes: FlowNode[]): FlowEdge[] {
|
|
const edges: FlowEdge[] = []
|
|
|
|
// Track which side to use for each target
|
|
const targetSides: Record<string, 'left' | 'right'> = {}
|
|
|
|
nodes.forEach((node) => {
|
|
if (node.type !== 'funnel') return
|
|
const data = node.data as FunnelNodeData
|
|
const sourceX = node.position.x
|
|
|
|
// OUTFLOW edges - go SIDEWAYS from source to target's TOP (inflow)
|
|
data.overflowAllocations?.forEach((alloc, idx) => {
|
|
const strokeWidth = 2 + (alloc.percentage / 100) * 6
|
|
const targetNode = nodes.find(n => n.id === alloc.targetId)
|
|
if (!targetNode) return
|
|
|
|
const targetX = targetNode.position.x
|
|
const goingRight = targetX > sourceX
|
|
const sourceHandle = goingRight ? 'outflow-right' : 'outflow-left'
|
|
|
|
edges.push({
|
|
id: `outflow-${node.id}-${alloc.targetId}`,
|
|
source: node.id,
|
|
target: alloc.targetId,
|
|
sourceHandle: sourceHandle,
|
|
targetHandle: undefined, // Goes to top (default target)
|
|
animated: true,
|
|
style: {
|
|
stroke: alloc.color,
|
|
strokeWidth,
|
|
opacity: 0.8,
|
|
},
|
|
markerEnd: {
|
|
type: MarkerType.ArrowClosed,
|
|
color: alloc.color,
|
|
width: 12,
|
|
height: 12,
|
|
},
|
|
data: {
|
|
allocation: alloc.percentage,
|
|
color: alloc.color,
|
|
edgeType: 'overflow' as const,
|
|
},
|
|
type: 'smoothstep',
|
|
})
|
|
})
|
|
|
|
// SPENDING edges - go DOWN from BOTTOM to outcomes
|
|
data.spendingAllocations?.forEach((alloc) => {
|
|
const strokeWidth = 2 + (alloc.percentage / 100) * 6
|
|
|
|
edges.push({
|
|
id: `spending-${node.id}-${alloc.targetId}`,
|
|
source: node.id,
|
|
target: alloc.targetId,
|
|
sourceHandle: 'spending-out', // Explicit bottom handle
|
|
animated: true,
|
|
style: {
|
|
stroke: alloc.color,
|
|
strokeWidth,
|
|
opacity: 0.9,
|
|
},
|
|
markerEnd: {
|
|
type: MarkerType.ArrowClosed,
|
|
color: alloc.color,
|
|
width: 12,
|
|
height: 12,
|
|
},
|
|
data: {
|
|
allocation: alloc.percentage,
|
|
color: alloc.color,
|
|
edgeType: 'spending' as const,
|
|
},
|
|
})
|
|
})
|
|
})
|
|
|
|
return edges
|
|
}
|
|
|
|
export default function FlowCanvas() {
|
|
const [nodes, setNodes, onNodesChange] = useNodesState(initialNodes)
|
|
const [edges, setEdges, onEdgesChange] = useEdgesState(generateEdges(initialNodes))
|
|
const [isSimulating, setIsSimulating] = useState(true)
|
|
|
|
const onConnect = useCallback(
|
|
(params: Connection) =>
|
|
setEdges((eds) =>
|
|
addEdge(
|
|
{
|
|
...params,
|
|
animated: true,
|
|
style: { stroke: '#64748b', strokeWidth: 3 },
|
|
markerEnd: { type: MarkerType.ArrowClosed, color: '#64748b' },
|
|
},
|
|
eds
|
|
)
|
|
),
|
|
[setEdges]
|
|
)
|
|
|
|
// Allow edges to be reconnected by dragging
|
|
const onReconnect = useCallback(
|
|
(oldEdge: FlowEdge, newConnection: Connection) =>
|
|
setEdges((eds) => reconnectEdge(oldEdge, newConnection, eds) as FlowEdge[]),
|
|
[setEdges]
|
|
)
|
|
|
|
// Simulation effect
|
|
useEffect(() => {
|
|
if (!isSimulating) return
|
|
|
|
const interval = setInterval(() => {
|
|
setNodes((nds) =>
|
|
nds.map((node) => {
|
|
if (node.type === 'funnel') {
|
|
const data = node.data as FunnelNodeData
|
|
const change = (Math.random() - 0.45) * 300
|
|
return {
|
|
...node,
|
|
data: {
|
|
...data,
|
|
currentValue: Math.max(0, Math.min(data.maxCapacity * 1.1, data.currentValue + change)),
|
|
},
|
|
}
|
|
} else if (node.type === 'outcome') {
|
|
const data = node.data as OutcomeNodeData
|
|
const change = Math.random() * 80
|
|
const newReceived = Math.min(data.fundingTarget * 1.05, data.fundingReceived + change)
|
|
return {
|
|
...node,
|
|
data: {
|
|
...data,
|
|
fundingReceived: newReceived,
|
|
status: newReceived >= data.fundingTarget ? 'completed' :
|
|
data.status === 'not-started' && newReceived > 0 ? 'in-progress' : data.status,
|
|
},
|
|
}
|
|
}
|
|
return node
|
|
})
|
|
)
|
|
}, 500)
|
|
|
|
return () => clearInterval(interval)
|
|
}, [isSimulating, setNodes])
|
|
|
|
// Regenerate edges when nodes change
|
|
useEffect(() => {
|
|
setEdges(generateEdges(nodes))
|
|
}, [nodes, setEdges])
|
|
|
|
return (
|
|
<div className="w-full h-full">
|
|
<ReactFlow
|
|
nodes={nodes}
|
|
edges={edges}
|
|
onNodesChange={onNodesChange}
|
|
onEdgesChange={onEdgesChange}
|
|
onConnect={onConnect}
|
|
onReconnect={onReconnect}
|
|
nodeTypes={nodeTypes}
|
|
edgesReconnectable={true}
|
|
fitView
|
|
fitViewOptions={{ padding: 0.15 }}
|
|
className="bg-slate-50"
|
|
>
|
|
<Background variant={BackgroundVariant.Dots} gap={20} size={1} color="#e2e8f0" />
|
|
<Controls className="bg-white border border-slate-200 rounded-lg shadow-sm" />
|
|
|
|
{/* Title Panel */}
|
|
<Panel position="top-left" className="bg-white rounded-lg shadow-lg border border-slate-200 p-4 m-4">
|
|
<h1 className="text-lg font-bold text-slate-800">Threshold-Based Flow Funding</h1>
|
|
<p className="text-xs text-slate-500 mt-1">
|
|
<span className="text-emerald-600">↓ Inflows</span> (top) •
|
|
<span className="text-amber-600 ml-1">→ Outflows</span> (sides) •
|
|
<span className="text-blue-600 ml-1">↓ Outcomes</span> (bottom)
|
|
</p>
|
|
<p className="text-[10px] text-slate-400 mt-1">Double-click funnels to edit allocations</p>
|
|
</Panel>
|
|
|
|
{/* Simulation Toggle */}
|
|
<Panel position="top-right" className="m-4">
|
|
<button
|
|
onClick={() => setIsSimulating(!isSimulating)}
|
|
className={`px-4 py-2 rounded-lg font-medium shadow-sm transition-all text-sm ${
|
|
isSimulating
|
|
? 'bg-emerald-500 text-white hover:bg-emerald-600'
|
|
: 'bg-slate-200 text-slate-700 hover:bg-slate-300'
|
|
}`}
|
|
>
|
|
{isSimulating ? '⏸ Pause' : '▶ Start'}
|
|
</button>
|
|
</Panel>
|
|
|
|
{/* Legend */}
|
|
<Panel position="bottom-left" className="bg-white rounded-lg shadow-lg border border-slate-200 p-3 m-4">
|
|
<div className="text-[10px] font-medium text-slate-500 uppercase tracking-wide mb-2">Flow Types</div>
|
|
<div className="space-y-1.5 text-xs">
|
|
<div className="flex items-center gap-2">
|
|
<div className="w-3 h-3 rounded-full bg-emerald-500" />
|
|
<span className="text-slate-600">Inflows (top)</span>
|
|
</div>
|
|
<div className="flex items-center gap-2">
|
|
<div className="w-3 h-3 rounded-full bg-amber-500" />
|
|
<span className="text-slate-600">Outflows (sides)</span>
|
|
</div>
|
|
<div className="flex items-center gap-2">
|
|
<div className="w-3 h-3 rounded-full bg-blue-500" />
|
|
<span className="text-slate-600">Outcomes (bottom)</span>
|
|
</div>
|
|
</div>
|
|
</Panel>
|
|
</ReactFlow>
|
|
</div>
|
|
)
|
|
}
|