import { artifact } from '@ai-sdk-tools/artifacts'
import { z } from 'zod'
import { tool } from 'ai'

// Define artifact schema
const Dashboard = artifact('dashboard', z.object({
  title: z.string(),
  metrics: z.object({
    revenue: z.number(),
    users: z.number(),
    conversion: z.number()
  }),
  status: z.enum(['loading', 'streaming', 'complete', 'error']),
  progress: z.number().min(0).max(1).default(0)
}))

// AI tool that streams dashboard data
export const analyzeDashboardTool = tool({
  description: 'Analyze business metrics and create live dashboard',
  parameters: z.object({
    company: z.string()
  }),
  execute: async ({ company }) => {
    const dashboard = Dashboard.stream({
      title: `${company} Analytics Dashboard`,
      status: 'loading',
      progress: 0,
      metrics: { revenue: 0, users: 0, conversion: 0 }
    })

    await dashboard.update({
      status: 'streaming',
      progress: 0.5,
      metrics: { revenue: 2500000, users: 12500, conversion: 3.2 }
    })

    await dashboard.complete({
      status: 'complete',
      progress: 1.0,
      metrics: { revenue: 3000000, users: 15200, conversion: 3.8 }
    })

    return 'Analysis complete'
  }
})

import { useArtifact } from '@ai-sdk-tools/artifacts'
import { useState } from 'react'

function DashboardComponent() {
  const { 
    data, 
    status, 
    progress, 
    error, 
    isActive 
  } = useArtifact(Dashboard, {
    onUpdate: (newData) => {
      console.log('Data updated:', newData)
    },
    onComplete: (finalData) => {
      console.log('Analysis complete!', finalData)
    },
    onError: (error) => {
      console.error('Analysis failed:', error)
    },
    onProgress: (progress) => {
      console.log('Progress:', Math.round(progress * 100) + '%')
    },
    onStatusChange: (newStatus, prevStatus) => {
      console.log('Status:', prevStatus, '→', newStatus)
    }
  })

  if (!data && !isActive) {
    return <div>No data available</div>
  }

  return ...
}