221 lines
8.8 KiB
Go
221 lines
8.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
func main() {
|
|
dbURL := os.Getenv("DATABASE_URL")
|
|
if dbURL == "" {
|
|
dbURL = "postgres://multica:multica@localhost:5432/multica?sslmode=disable"
|
|
}
|
|
|
|
ctx := context.Background()
|
|
pool, err := pgxpool.New(ctx, dbURL)
|
|
if err != nil {
|
|
log.Fatalf("Unable to connect to database: %v", err)
|
|
}
|
|
defer pool.Close()
|
|
|
|
// Create seed user
|
|
var userID string
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO "user" (name, email, avatar_url)
|
|
VALUES ('Jiayuan Zhang', 'jiayuan@multica.ai', NULL)
|
|
ON CONFLICT (email) DO UPDATE SET name = EXCLUDED.name
|
|
RETURNING id
|
|
`).Scan(&userID)
|
|
if err != nil {
|
|
log.Fatalf("Failed to create user: %v", err)
|
|
}
|
|
fmt.Printf("User created: %s\n", userID)
|
|
|
|
// Create seed workspace
|
|
var workspaceID string
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO workspace (name, slug, description)
|
|
VALUES ('Multica', 'multica', 'AI-native task management')
|
|
ON CONFLICT (slug) DO UPDATE SET name = EXCLUDED.name
|
|
RETURNING id
|
|
`).Scan(&workspaceID)
|
|
if err != nil {
|
|
log.Fatalf("Failed to create workspace: %v", err)
|
|
}
|
|
fmt.Printf("Workspace created: %s\n", workspaceID)
|
|
|
|
// Add user as owner
|
|
_, err = pool.Exec(ctx, `
|
|
INSERT INTO member (workspace_id, user_id, role)
|
|
VALUES ($1, $2, 'owner')
|
|
ON CONFLICT (workspace_id, user_id) DO NOTHING
|
|
`, workspaceID, userID)
|
|
if err != nil {
|
|
log.Fatalf("Failed to create member: %v", err)
|
|
}
|
|
fmt.Println("Member created")
|
|
|
|
// Create some agents
|
|
agents := []struct {
|
|
name string
|
|
description string
|
|
runtimeMode string
|
|
status string
|
|
skills string
|
|
tools string
|
|
triggers string
|
|
}{
|
|
{
|
|
"Deep Research Agent",
|
|
"Performs deep research on topics using web search and analysis",
|
|
"local", "idle",
|
|
"# Deep Research Agent\n\nYou are a research agent that performs thorough analysis on assigned topics.\n\n## Workflow\n1. Break down the research question into sub-questions\n2. Use web search to gather information from multiple sources\n3. Cross-reference and validate findings\n4. Synthesize a comprehensive report\n5. Post the report as a comment on the issue\n\n## Output Format\nAlways produce a structured report with:\n- Executive Summary\n- Key Findings\n- Sources\n- Recommendations",
|
|
`[{"id":"tool-1","name":"Google Search","description":"Search the web for information","auth_type":"api_key","connected":true,"config":{}},{"id":"tool-2","name":"Web Scraper","description":"Extract content from web pages","auth_type":"none","connected":true,"config":{}}]`,
|
|
`[{"id":"trigger-1","type":"on_assign","enabled":true,"config":{}}]`,
|
|
},
|
|
{
|
|
"Code Review Bot",
|
|
"Reviews pull requests and provides feedback on code quality",
|
|
"cloud", "idle",
|
|
"# Code Review Bot\n\nYou review code changes and provide constructive feedback.\n\n## Review Criteria\n- Code correctness and logic\n- Performance implications\n- Security vulnerabilities\n- Code style and readability\n- Test coverage\n\n## Process\n1. Read the issue description for context\n2. Analyze code changes\n3. Post review comments on specific lines\n4. Provide an overall summary",
|
|
`[{"id":"tool-3","name":"GitHub","description":"Access GitHub repositories and PRs","auth_type":"oauth","connected":true,"config":{}}]`,
|
|
`[{"id":"trigger-2","type":"on_assign","enabled":true,"config":{}}]`,
|
|
},
|
|
{
|
|
"Daily Standup Bot",
|
|
"Generates daily standup summaries from recent activity",
|
|
"cloud", "working",
|
|
"# Daily Standup Bot\n\nGenerate a daily standup summary based on workspace activity.\n\n## Tasks\n1. Collect all issue status changes from the last 24 hours\n2. Summarize what each team member worked on\n3. Identify blocked items\n4. Post the summary to the team channel",
|
|
`[{"id":"tool-4","name":"Slack","description":"Send messages to Slack channels","auth_type":"oauth","connected":true,"config":{"channel":"#standup"}}]`,
|
|
`[{"id":"trigger-3","type":"scheduled","enabled":true,"config":{"cron":"0 9 * * 1-5","timezone":"Asia/Shanghai"}}]`,
|
|
},
|
|
{
|
|
"Local Dev Agent",
|
|
"A local development agent running on your machine",
|
|
"local", "offline",
|
|
"",
|
|
`[]`,
|
|
`[{"id":"trigger-4","type":"on_assign","enabled":true,"config":{}}]`,
|
|
},
|
|
}
|
|
|
|
for _, a := range agents {
|
|
var agentID string
|
|
// Check if agent already exists
|
|
err = pool.QueryRow(ctx, `
|
|
SELECT id FROM agent WHERE workspace_id = $1 AND name = $2
|
|
`, workspaceID, a.name).Scan(&agentID)
|
|
if err == nil {
|
|
fmt.Printf("Agent exists: %s (%s)\n", a.name, agentID)
|
|
continue
|
|
}
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO agent_runtime (workspace_id, daemon_id, name, runtime_mode, provider, status, device_info, metadata, last_seen_at)
|
|
VALUES (
|
|
$1,
|
|
NULL,
|
|
$2,
|
|
$3,
|
|
$4,
|
|
$5,
|
|
$6,
|
|
'{"seed":true}'::jsonb,
|
|
CASE WHEN $5 = 'online' THEN now() ELSE NULL END
|
|
)
|
|
RETURNING id
|
|
`,
|
|
workspaceID,
|
|
a.name+" Runtime",
|
|
a.runtimeMode,
|
|
map[string]string{"cloud": "multica_agent", "local": "seed_local"}[a.runtimeMode],
|
|
map[bool]string{true: "offline", false: "online"}[a.status == "offline"],
|
|
map[string]string{"cloud": "Seeded cloud runtime", "local": "Seeded local runtime"}[a.runtimeMode],
|
|
).Scan(&agentID)
|
|
if err != nil {
|
|
log.Printf("Failed to create runtime for agent %s: %v", a.name, err)
|
|
continue
|
|
}
|
|
runtimeID := agentID
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO agent (workspace_id, name, description, runtime_mode, runtime_id, status, owner_id, skills, tools, triggers)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9::jsonb, $10::jsonb)
|
|
RETURNING id
|
|
`, workspaceID, a.name, a.description, a.runtimeMode, runtimeID, a.status, userID, a.skills, a.tools, a.triggers).Scan(&agentID)
|
|
if err != nil {
|
|
log.Printf("Failed to create agent %s: %v", a.name, err)
|
|
continue
|
|
}
|
|
fmt.Printf("Agent created: %s (%s)\n", a.name, agentID)
|
|
}
|
|
|
|
// Create seed issues
|
|
issues := []struct {
|
|
title string
|
|
description string
|
|
status string
|
|
priority string
|
|
position float64
|
|
}{
|
|
{"Add multi-workspace support", "Users should be able to create and switch between multiple workspaces.", "backlog", "medium", 1},
|
|
{"Agent long-term memory persistence", "Agents need persistent memory across sessions for better context.", "backlog", "low", 2},
|
|
{"Design the agent config UI", "Create a configuration interface for agent settings and capabilities.", "todo", "high", 3},
|
|
{"Implement issue list API endpoint", "Build the REST API for listing, filtering, and paginating issues.", "in_progress", "urgent", 4},
|
|
{"Implement OAuth login flow", "Set up OAuth 2.0 with Google for user authentication.", "in_progress", "high", 5},
|
|
{"Add WebSocket reconnection logic", "Handle disconnections gracefully with exponential backoff.", "in_review", "medium", 6},
|
|
{"Set up CI/CD pipeline", "Configure GitHub Actions for automated testing and deployment.", "done", "high", 7},
|
|
{"Design database schema", "Create the initial PostgreSQL schema for all entities.", "done", "urgent", 8},
|
|
{"Implement real-time notifications", "Push notifications to users via WebSocket when issues change.", "todo", "medium", 9},
|
|
{"Agent task queue management", "Build the task dispatching and queue management system for agents.", "todo", "high", 10},
|
|
}
|
|
|
|
for _, iss := range issues {
|
|
var issueID string
|
|
// Check if issue already exists
|
|
err = pool.QueryRow(ctx, `
|
|
SELECT id FROM issue WHERE workspace_id = $1 AND title = $2
|
|
`, workspaceID, iss.title).Scan(&issueID)
|
|
if err == nil {
|
|
fmt.Printf("Issue exists: %s (%s)\n", iss.title, issueID)
|
|
continue
|
|
}
|
|
err = pool.QueryRow(ctx, `
|
|
INSERT INTO issue (workspace_id, title, description, status, priority, creator_type, creator_id, position)
|
|
VALUES ($1, $2, $3, $4, $5, 'member', $6, $7)
|
|
RETURNING id
|
|
`, workspaceID, iss.title, iss.description, iss.status, iss.priority, userID, iss.position).Scan(&issueID)
|
|
if err != nil {
|
|
log.Printf("Failed to create issue %s: %v", iss.title, err)
|
|
continue
|
|
}
|
|
fmt.Printf("Issue created: %s (%s)\n", iss.title, issueID)
|
|
}
|
|
|
|
// Create seed comment (only if not already present)
|
|
var commentExists bool
|
|
_ = pool.QueryRow(ctx, `
|
|
SELECT EXISTS(
|
|
SELECT 1 FROM comment c
|
|
JOIN issue i ON c.issue_id = i.id
|
|
WHERE i.workspace_id = $1 AND i.title = 'Implement issue list API endpoint'
|
|
AND c.content = 'This is a high priority item for Q2.'
|
|
)
|
|
`, workspaceID).Scan(&commentExists)
|
|
if !commentExists {
|
|
_, err = pool.Exec(ctx, `
|
|
INSERT INTO comment (issue_id, author_type, author_id, content, type)
|
|
SELECT i.id, 'member', $2, 'This is a high priority item for Q2.', 'comment'
|
|
FROM issue i WHERE i.workspace_id = $1 AND i.title = 'Implement issue list API endpoint'
|
|
`, workspaceID, userID)
|
|
if err != nil {
|
|
log.Printf("Failed to create comment: %v", err)
|
|
}
|
|
}
|
|
|
|
fmt.Println("\nSeed data created successfully!")
|
|
fmt.Printf("\nUser ID: %s\n", userID)
|
|
fmt.Printf("Workspace ID: %s\n", workspaceID)
|
|
}
|