hristudio/scripts/seed-dev.ts

import bcrypt from "bcryptjs";
import { drizzle } from "drizzle-orm/postgres-js";
import { sql } from "drizzle-orm";
import postgres from "postgres";
import * as schema from "../src/server/db/schema";

// Database connection
const connectionString = process.env.DATABASE_URL!;
const connection = postgres(connectionString);
const db = drizzle(connection, { schema });

// --- NAO6 Plugin Definitions (Synced from seed-nao6-plugin.ts) ---
const NAO_PLUGIN_DEF = {
  name: "NAO6 Robot (Enhanced ROS2 Integration)",
  version: "2.0.0",
  description: "Comprehensive NAO6 robot integration for HRIStudio experiments via ROS2.",
  actions: [
    { id: "nao_speak", name: "Speak Text", category: "speech", parametersSchema: { type: "object", properties: { text: { type: "string" }, volume: { type: "number", default: 0.7 } }, required: ["text"] } },
    { id: "nao_gesture", name: "Perform Gesture", category: "interaction", parametersSchema: { type: "object", properties: { gesture: { type: "string", enum: ["wave", "bow", "point"] }, speed: { type: "number", default: 0.8 } } } },
    { id: "nao_look_at", name: "Look At", category: "movement", parametersSchema: { type: "object", properties: { target: { type: "string", enum: ["participant", "screen", "away"] }, duration: { type: "number", default: 2.0 } } } },
    { id: "nao_nod", name: "Nod Head", category: "interaction", parametersSchema: { type: "object", properties: { speed: { type: "number", default: 1.0 } } } },
    { id: "nao_shake_head", name: "Shake Head", category: "interaction", parametersSchema: { type: "object", properties: { speed: { type: "number", default: 1.0 } } } },
    { id: "nao_bow", name: "Bow", category: "interaction", parametersSchema: { type: "object", properties: {} } },
    { id: "nao_open_hand", name: "Present (Open Hand)", category: "interaction", parametersSchema: { type: "object", properties: { hand: { type: "string", enum: ["left", "right", "both"], default: "right" } } } }
  ]
};

async function main() {
  console.log("🌱 Starting realistic seed script...");

  try {
    // 1. Clean existing data (Full Wipe)
    console.log("🧹 Cleaning existing data...");
    await db.delete(schema.mediaCaptures).where(sql`1=1`);
    await db.delete(schema.trialEvents).where(sql`1=1`);
    await db.delete(schema.trials).where(sql`1=1`);
    await db.delete(schema.actions).where(sql`1=1`);
    await db.delete(schema.steps).where(sql`1=1`);
    await db.delete(schema.experiments).where(sql`1=1`);
    await db.delete(schema.participants).where(sql`1=1`);
    await db.delete(schema.studyPlugins).where(sql`1=1`);
    await db.delete(schema.studyMembers).where(sql`1=1`);
    await db.delete(schema.studies).where(sql`1=1`);
    await db.delete(schema.plugins).where(sql`1=1`);
    await db.delete(schema.pluginRepositories).where(sql`1=1`);
    await db.delete(schema.userSystemRoles).where(sql`1=1`);
    await db.delete(schema.users).where(sql`1=1`);
    await db.delete(schema.robots).where(sql`1=1`);

    // 2. Create Users
    console.log("👥 Creating users...");
    const hashedPassword = await bcrypt.hash("password123", 12);

    const [adminUser] = await db.insert(schema.users).values({
      name: "Sean O'Connor",
      email: "sean@soconnor.dev",
      password: hashedPassword,
      emailVerified: new Date(),
      image: "https://api.dicebear.com/7.x/avataaars/svg?seed=Sean",
    }).returning();

    const [researcherUser] = await db.insert(schema.users).values({
      name: "Dr. Felipe Perrone",
      email: "felipe.perrone@bucknell.edu",
      password: hashedPassword,
      emailVerified: new Date(),
      image: "https://api.dicebear.com/7.x/avataaars/svg?seed=Felipe",
    }).returning();

    if (!adminUser) throw new Error("Failed to create admin user");

    await db.insert(schema.userSystemRoles).values({ userId: adminUser.id, role: "administrator" });

    // 3. Create Robots & Plugins
    console.log("🤖 Creating robots and plugins...");
    const [naoRobot] = await db.insert(schema.robots).values({
      name: "NAO6",
      manufacturer: "SoftBank Robotics",
      model: "NAO V6",
      description: "Humanoid robot for social interaction studies.",
      capabilities: ["speech", "vision", "bipedal_walking", "gestures"],
      communicationProtocol: "ros2",
    }).returning();

    const [naoRepo] = await db.insert(schema.pluginRepositories).values({
      name: "HRIStudio Official Plugins",
      url: "https://github.com/hristudio/plugins",
      description: "Official verified plugins",
      trustLevel: "official",
      isEnabled: true,
      isOfficial: true,
      createdBy: adminUser.id,
    }).returning();

    const [naoPlugin] = await db.insert(schema.plugins).values({
      robotId: naoRobot!.id,
      name: NAO_PLUGIN_DEF.name,
      version: NAO_PLUGIN_DEF.version,
      description: NAO_PLUGIN_DEF.description,
      author: "HRIStudio Team",
      trustLevel: "official",
      status: "active",
      repositoryUrl: naoRepo!.url,
      actionDefinitions: NAO_PLUGIN_DEF.actions,
      configurationSchema: { type: "object", properties: { robotIp: { type: "string", default: "192.168.1.100" } } },
      metadata: { category: "robot_control" }
    }).returning();

    // 4. Create Study & Experiment - Comparative WoZ Study
    console.log("📚 Creating 'Comparative WoZ Study'...");
    const [study] = await db.insert(schema.studies).values({
      name: "Comparative WoZ Study",
      description: "Comparison of HRIStudio vs Choregraphe for The Interactive Storyteller scenario.",
      institution: "Bucknell University",
      irbProtocol: "2024-HRI-COMP",
      status: "active",
      createdBy: adminUser.id,
    }).returning();

    await db.insert(schema.studyMembers).values([
      { studyId: study!.id, userId: adminUser.id, role: "owner" },
      { studyId: study!.id, userId: researcherUser!.id, role: "researcher" }
    ]);

    await db.insert(schema.studyPlugins).values({
      studyId: study!.id,
      pluginId: naoPlugin!.id,
      configuration: { robotIp: "10.0.0.42" },
      installedBy: adminUser.id
    });

    const [experiment] = await db.insert(schema.experiments).values({
      studyId: study!.id,
      name: "The Interactive Storyteller",
      description: "A storytelling scenario where the robot tells a story and asks questions to the participant.",
      version: 1,
      status: "ready",
      robotId: naoRobot!.id,
      createdBy: adminUser.id,
    }).returning();

    // 5. Create Steps & Actions (The Interactive Storyteller Protocol)
    console.log("🎬 Creating experiment steps (Interactive Storyteller)...");

    // --- Step 1: The Hook ---
    const [step1] = await db.insert(schema.steps).values({
      experimentId: experiment!.id,
      name: "The Hook",
      description: "Initial greeting and engagement",
      type: "robot",
      orderIndex: 0,
      required: true,
      durationEstimate: 30
    }).returning();

    await db.insert(schema.actions).values([
      {
        stepId: step1!.id,
        name: "Greet Participant",
        type: "nao6.nao_speak",
        orderIndex: 0,
        parameters: { text: "Hello there! I have a wonderful story to share with you today.", volume: 0.8 },
        pluginId: naoPlugin!.id,
        category: "speech",
        retryable: true
      },
      {
        stepId: step1!.id,
        name: "Wave Greeting",
        type: "nao6.nao_gesture",
        orderIndex: 1,
        parameters: { gesture: "wave" },
        pluginId: naoPlugin!.id,
        category: "interaction",
        retryable: true
      }
    ]);

    // --- Step 2: The Narrative (Part 1) ---
    const [step2] = await db.insert(schema.steps).values({
      experimentId: experiment!.id,
      name: "The Narrative - Part 1",
      description: "Robot tells the first part of the story",
      type: "robot",
      orderIndex: 1,
      required: true,
      durationEstimate: 60
    }).returning();

    await db.insert(schema.actions).values([
      {
        stepId: step2!.id,
        name: "Tell Story Part 1",
        type: "nao6.nao_speak",
        orderIndex: 0,
        parameters: { text: "Once upon a time, in a land far away, there lived a curious robot named Alpha.", volume: 0.8 },
        pluginId: naoPlugin!.id,
        category: "speech"
      },
      {
        stepId: step2!.id,
        name: "Present Gesture",
        type: "nao6.nao_open_hand",
        orderIndex: 1,
        parameters: { hand: "right" },
        pluginId: naoPlugin!.id,
        category: "interaction"
      }
    ]);

    // --- Step 3: Comprehension Check (Wizard Decision) ---
    // Note: In a real visual designer, this would be a Branch/Conditional.
    // Here we model it as a Wizard Step where they explicitly choose the next robot action.
    const [step3] = await db.insert(schema.steps).values({
      experimentId: experiment!.id,
      name: "Comprehension Check",
      description: "Wizard verifies participant understanding",
      type: "wizard",
      orderIndex: 2,
      required: true,
      durationEstimate: 45
    }).returning();

    await db.insert(schema.actions).values([
      {
        stepId: step3!.id,
        name: "Ask Question",
        type: "nao6.nao_speak",
        orderIndex: 0,
        parameters: { text: "What was the robot's name?", volume: 0.8 },
        pluginId: naoPlugin!.id,
        category: "speech"
      },
      {
        stepId: step3!.id,
        name: "Wait for Wizard Input",
        type: "core.wait_for_response",
        orderIndex: 1,
        parameters: { prompt: "Did participant answer 'Alpha'?", options: ["Yes", "No"] },
        sourceKind: "core",
        category: "wizard"
      }
    ]);

    // --- Step 4: Feedback (Positive/Negative branches implied) ---
    // For linear seed, we just add the Positive feedback step
    const [step4] = await db.insert(schema.steps).values({
      experimentId: experiment!.id,
      name: "Positive Feedback",
      description: "Correct answer response",
      type: "robot",
      orderIndex: 3,
      required: true,
      durationEstimate: 15
    }).returning();

    await db.insert(schema.actions).values([
      {
        stepId: step4!.id,
        name: "Nod Affirmation",
        type: "nao6.nao_nod",
        orderIndex: 0,
        parameters: { speed: 1.0 },
        pluginId: naoPlugin!.id,
        category: "interaction"
      },
      {
        stepId: step4!.id,
        name: "Say Correct",
        type: "nao6.nao_speak",
        orderIndex: 1,
        parameters: { text: "That is correct! Well done.", volume: 0.8 },
        pluginId: naoPlugin!.id,
        category: "speech"
      }
    ]);

    // --- Step 5: Conclusion ---
    const [step5] = await db.insert(schema.steps).values({
      experimentId: experiment!.id,
      name: "Conclusion",
      description: "Wrap up the story",
      type: "robot",
      orderIndex: 4,
      required: true,
      durationEstimate: 30
    }).returning();

    await db.insert(schema.actions).values([
      {
        stepId: step5!.id,
        name: "Finish Story",
        type: "nao6.nao_speak",
        orderIndex: 0,
        parameters: { text: "Alpha explored the world and learned many things. The end.", volume: 0.8 },
        pluginId: naoPlugin!.id,
        category: "speech"
      },
      {
        stepId: step5!.id,
        name: "Bow Goodbye",
        type: "nao6.nao_bow",
        orderIndex: 1,
        parameters: {},
        pluginId: naoPlugin!.id,
        category: "interaction"
      }
    ]);

    // 6. Participants (N=20 for study)
    console.log("👤 Creating 20 participants for N=20 study...");
    const participants = [];
    for (let i = 1; i <= 20; i++) {
      participants.push({
        studyId: study!.id,
        participantCode: `P${100 + i}`,
        name: `Participant ${100 + i}`,
        consentGiven: true,
        consentGivenAt: new Date(),
        notes: i % 2 === 0 ? "Condition: HRIStudio" : "Condition: Choregraphe"
      });
    }
    const insertedParticipants = await db.insert(schema.participants).values(participants).returning();

    console.log("\n✅ Database seeded successfully!");
    console.log(`Summary:`);
    console.log(`- 1 Admin User (sean@soconnor.dev)`);
    console.log(`- Study: 'Comparative WoZ Study'`);
    console.log(`- Experiment: 'The Interactive Storyteller' (${5} steps created)`);
    console.log(`- ${insertedParticipants.length} Participants`);

  } catch (error) {
    console.error("❌ Seeding failed:", error);
    process.exit(1);
  } finally {
    await connection.end();
  }
}

main();