@positronic/cloudflare 0.0.3 → 0.0.5

package/src/api.ts

@@ -1,579 +0,0 @@
-import { Hono, type Context } from 'hono';
-import { v4 as uuidv4 } from 'uuid';
-import { AwsClient } from 'aws4fetch';
-import { parseCronExpression } from 'cron-schedule';
-import type { BrainRunnerDO } from './brain-runner-do.js';
-import { getManifest } from './brain-runner-do.js';
-import type { MonitorDO } from './monitor-do.js';
-import type { ScheduleDO } from './schedule-do.js';
-import type { R2Bucket, R2Object } from '@cloudflare/workers-types';
-import { type ResourceEntry, RESOURCE_TYPES } from '@positronic/core';
-
-type Bindings = {
-  BRAIN_RUNNER_DO: DurableObjectNamespace<BrainRunnerDO>;
-  MONITOR_DO: DurableObjectNamespace<MonitorDO>;
-  SCHEDULE_DO: DurableObjectNamespace<ScheduleDO>;
-  RESOURCES_BUCKET: R2Bucket;
-  NODE_ENV?: string;
-  R2_ACCESS_KEY_ID?: string;
-  R2_SECRET_ACCESS_KEY?: string;
-  R2_ACCOUNT_ID?: string;
-  R2_BUCKET_NAME?: string;
-};
-
-type CreateBrainRunRequest = {
-  brainName: string;
-};
-
-type CreateBrainRunResponse = {
-  brainRunId: string;
-};
-
-// Override ResourceEntry to make path optional for resources that aren't in version control
-type R2Resource = Omit<ResourceEntry, 'path'> & {
-  path?: string;
-  size: number;
-  lastModified: string;
-  local: boolean;
-};
-
-const app = new Hono<{ Bindings: Bindings }>();
-
-app.post('/brains/runs', async (context: Context) => {
-  const { brainName } = await context.req.json<CreateBrainRunRequest>();
-
-  if (!brainName) {
-    return context.json({ error: 'Missing brainName in request body' }, 400);
-  }
-
-  // Validate that the brain exists before starting it
-  const manifest = getManifest();
-  if (!manifest) {
-    return context.json({ error: 'Manifest not initialized' }, 500);
-  }
-
-  const brain = await manifest.import(brainName);
-  if (!brain) {
-    return context.json({ error: `Brain '${brainName}' not found` }, 404);
-  }
-
-  const brainRunId = uuidv4();
-  const namespace = context.env.BRAIN_RUNNER_DO;
-  const doId = namespace.idFromName(brainRunId);
-  const stub = namespace.get(doId);
-  await stub.start(brainName, brainRunId);
-  const response: CreateBrainRunResponse = {
-    brainRunId,
-  };
-  return context.json(response, 201);
-});
-
-app.post('/brains/runs/rerun', async (context: Context) => {
-  const { brainName, runId, startsAt, stopsAfter } = await context.req.json();
-
-  if (!brainName) {
-    return context.json({ error: 'Missing brainName in request body' }, 400);
-  }
-
-  // Validate that the brain exists
-  const manifest = getManifest();
-  if (!manifest) {
-    return context.json({ error: 'Manifest not initialized' }, 500);
-  }
-
-  const brain = await manifest.import(brainName);
-  if (!brain) {
-    return context.json({ error: `Brain '${brainName}' not found` }, 404);
-  }
-
-  // If runId is provided, validate it exists
-  if (runId) {
-    const monitorId = context.env.MONITOR_DO.idFromName('singleton');
-    const monitorStub = context.env.MONITOR_DO.get(monitorId);
-    const existingRun = await monitorStub.getLastEvent(runId);
-    
-    if (!existingRun) {
-      return context.json({ error: `Brain run '${runId}' not found` }, 404);
-    }
-  }
-
-  // Create a new brain run with rerun parameters
-  const newBrainRunId = uuidv4();
-  const namespace = context.env.BRAIN_RUNNER_DO;
-  const doId = namespace.idFromName(newBrainRunId);
-  const stub = namespace.get(doId);
-  
-  // Start the brain with rerun options
-  const rerunOptions = {
-    ...(runId && { originalRunId: runId }),
-    ...(startsAt !== undefined && { startsAt }),
-    ...(stopsAfter !== undefined && { stopsAfter }),
-  };
-  
-  await stub.start(brainName, newBrainRunId, rerunOptions);
-  
-  const response: CreateBrainRunResponse = {
-    brainRunId: newBrainRunId,
-  };
-  return context.json(response, 201);
-});
-
-app.get('/brains/runs/:runId/watch', async (context: Context) => {
-  const runId = context.req.param('runId');
-  const namespace = context.env.BRAIN_RUNNER_DO;
-  const doId = namespace.idFromName(runId);
-  const stub = namespace.get(doId);
-  const response = await stub.fetch(new Request(`http://do/watch`));
-  return response;
-});
-
-app.get('/brains/:brainName/history', async (context: Context) => {
-  const brainName = context.req.param('brainName');
-  const limit = Number(context.req.query('limit') || '10');
-
-  // Get the monitor singleton instance
-  const monitorId = context.env.MONITOR_DO.idFromName('singleton');
-  const monitorStub = context.env.MONITOR_DO.get(monitorId);
-
-  const runs = await monitorStub.history(brainName, limit);
-  return context.json({ runs });
-});
-
-app.get('/brains/:brainName/active-runs', async (context: Context) => {
-  const brainName = context.req.param('brainName');
-
-  // Get the monitor singleton instance
-  const monitorId = context.env.MONITOR_DO.idFromName('singleton');
-  const monitorStub = context.env.MONITOR_DO.get(monitorId);
-
-  const runs = await monitorStub.activeRuns(brainName);
-  return context.json({ runs });
-});
-
-app.get('/brains/watch', async (context: Context) => {
-  const monitorId = context.env.MONITOR_DO.idFromName('singleton');
-  const monitorStub = context.env.MONITOR_DO.get(monitorId);
-  const response = await monitorStub.fetch(new Request(`http://do/watch`));
-  return response;
-});
-
-app.get('/brains', async (context: Context) => {
-  const manifest = getManifest();
-  
-  if (!manifest) {
-    return context.json({ error: 'Manifest not initialized' }, 500);
-  }
-
-  const brainNames = manifest.list();
-  const brains = await Promise.all(
-    brainNames.map(async (name) => {
-      const brain = await manifest.import(name);
-      if (!brain) {
-        return null;
-      }
-      
-      return {
-        name,
-        title: brain.title,
-        description: `${brain.title} brain`, // Default description since property is private
-      };
-    })
-  );
-
-  // Filter out any null entries
-  const validBrains = brains.filter(brain => brain !== null);
-
-  return context.json({
-    brains: validBrains,
-    count: validBrains.length,
-  });
-});
-
-// Schedule endpoints
-
-// Create a new schedule
-app.post('/brains/schedules', async (context: Context) => {
-  try {
-    const body = await context.req.json();
-    const { brainName, cronExpression } = body;
-
-    if (!brainName) {
-      return context.json({ error: 'Missing required field "brainName"' }, 400);
-    }
-    if (!cronExpression) {
-      return context.json(
-        { error: 'Missing required field "cronExpression"' },
-        400
-      );
-    }
-
-    // Validate cron expression before calling DO
-    try {
-      parseCronExpression(cronExpression);
-    } catch {
-      return context.json(
-        { error: `Invalid cron expression: ${cronExpression}` },
-        400
-      );
-    }
-
-    // Get the schedule singleton instance
-    const scheduleId = context.env.SCHEDULE_DO.idFromName('singleton');
-    const scheduleStub = context.env.SCHEDULE_DO.get(scheduleId);
-
-    const schedule = await scheduleStub.createSchedule(
-      brainName,
-      cronExpression
-    );
-    return context.json(schedule, 201);
-  } catch (error) {
-    const errorMessage =
-      error instanceof Error ? error.message : 'Failed to create schedule';
-    return context.json({ error: errorMessage }, 400);
-  }
-});
-
-// List all schedules
-app.get('/brains/schedules', async (context: Context) => {
-  const scheduleId = context.env.SCHEDULE_DO.idFromName('singleton');
-  const scheduleStub = context.env.SCHEDULE_DO.get(scheduleId);
-
-  const result = await scheduleStub.listSchedules();
-  return context.json(result);
-});
-
-// Get scheduled run history - MUST be before :scheduleId route
-app.get('/brains/schedules/runs', async (context: Context) => {
-  const scheduleIdParam = context.req.query('scheduleId');
-  const limit = Number(context.req.query('limit') || '100');
-
-  const scheduleDoId = context.env.SCHEDULE_DO.idFromName('singleton');
-  const scheduleStub = context.env.SCHEDULE_DO.get(scheduleDoId);
-
-  const result = await scheduleStub.getAllRuns(scheduleIdParam, limit);
-  return context.json(result);
-});
-
-// Delete a schedule
-app.delete('/brains/schedules/:scheduleId', async (context: Context) => {
-  const scheduleIdParam = context.req.param('scheduleId');
-
-  const scheduleDoId = context.env.SCHEDULE_DO.idFromName('singleton');
-  const scheduleStub = context.env.SCHEDULE_DO.get(scheduleDoId);
-
-  const deleted = await scheduleStub.deleteSchedule(scheduleIdParam);
-
-  if (!deleted) {
-    return context.json({ error: 'Schedule not found' }, 404);
-  }
-
-  return new Response(null, { status: 204 });
-});
-
-app.get('/brains/:brainName', async (context: Context) => {
-  const brainName = context.req.param('brainName');
-  const manifest = getManifest();
-  
-  if (!manifest) {
-    return context.json({ error: 'Manifest not initialized' }, 500);
-  }
-
-  const brain = await manifest.import(brainName);
-  if (!brain) {
-    return context.json({ error: `Brain '${brainName}' not found` }, 404);
-  }
-
-  // Get the brain structure
-  const structure = brain.structure;
-
-  return context.json({
-    name: brainName,
-    title: structure.title,
-    description: structure.description || `${structure.title} brain`,
-    steps: structure.steps,
-  });
-});
-
-app.get('/resources', async (context: Context) => {
-  const bucket = context.env.RESOURCES_BUCKET;
-
-  try {
-    // List all objects in the bucket
-    // R2 returns up to 1000 objects by default
-    const listed = await bucket.list();
-
-    const resources = await Promise.all(
-      listed.objects.map(async (object: R2Object) => {
-        // Get the object to access its custom metadata
-        const r2Object = await bucket.head(object.key);
-
-        if (!r2Object) {
-          throw new Error(`Resource "${object.key}" not found`);
-        }
-
-        if (!r2Object.customMetadata?.type) {
-          throw new Error(
-            `Resource "${object.key}" is missing required metadata field "type"`
-          );
-        }
-
-        const resource: R2Resource = {
-          type: r2Object.customMetadata.type as (typeof RESOURCE_TYPES)[number],
-          ...(r2Object.customMetadata.path && {
-            path: r2Object.customMetadata.path,
-          }),
-          key: object.key,
-          size: object.size,
-          lastModified: object.uploaded.toISOString(),
-          local: r2Object.customMetadata.local === 'true', // R2 metadata is always strings
-        };
-
-        return resource;
-      })
-    );
-
-    return context.json({
-      resources,
-      truncated: listed.truncated,
-      count: resources.length,
-    });
-  } catch (error) {
-    const errorMessage =
-      error instanceof Error ? error.message : 'Unknown error occurred';
-    return context.json({ error: errorMessage }, 500);
-  }
-});
-
-app.get('/status', async (context: Context) => {
-  return context.json({ ready: true });
-});
-
-app.post('/resources', async (context: Context) => {
-  const bucket = context.env.RESOURCES_BUCKET;
-
-  const formData = await context.req.formData();
-
-  const file = formData.get('file') as File | null;
-  const type = formData.get('type') as string | null;
-  const path = formData.get('path') as string | null;
-  const key = formData.get('key') as string | null;
-  const local = formData.get('local') as string | null;
-
-  if (!file) {
-    return context.json({ error: 'Missing required field "file"' }, 400);
-  }
-
-  if (!type) {
-    return context.json({ error: 'Missing required field "type"' }, 400);
-  }
-
-  if (!RESOURCE_TYPES.includes(type as any)) {
-    return context.json(
-      { error: `Field "type" must be one of: ${RESOURCE_TYPES.join(', ')}` },
-      400
-    );
-  }
-
-  // Either key or path must be provided
-  if (!key && !path) {
-    return context.json(
-      { error: 'Either "key" or "path" must be provided' },
-      400
-    );
-  }
-
-  // Use key if provided, otherwise use path
-  const objectKey = key || path!;
-
-  try {
-    // Upload to R2 with custom metadata
-    const arrayBuffer = await file.arrayBuffer();
-    const uploadedObject = await bucket.put(objectKey, arrayBuffer, {
-      customMetadata: {
-        type,
-        ...(path && { path }),
-        local: local === 'true' ? 'true' : 'false', // R2 metadata must be strings
-      },
-    });
-
-    const resource: R2Resource = {
-      type: type as (typeof RESOURCE_TYPES)[number],
-      ...(path && { path }),
-      key: objectKey,
-      size: uploadedObject.size,
-      lastModified: uploadedObject.uploaded.toISOString(),
-      local: local === 'true',
-    };
-
-    return context.json(resource, 201);
-  } catch (error) {
-    const errorMessage =
-      error instanceof Error ? error.message : 'Failed to upload resource';
-    return context.json({ error: errorMessage }, 500);
-  }
-});
-
-// Delete a single resource by key
-app.delete('/resources/:key', async (context: Context) => {
-  const bucket = context.env.RESOURCES_BUCKET;
-  const key = context.req.param('key');
-
-  // URL decode the key since it might contain slashes
-  const decodedKey = decodeURIComponent(key);
-
-  try {
-    // Delete the resource - R2 delete is idempotent, so it's safe to delete non-existent resources
-    await bucket.delete(decodedKey);
-    return new Response(null, { status: 204 });
-  } catch (error) {
-    console.error(`Failed to delete resource "${decodedKey}":`, error);
-    return context.json(
-      {
-        error: `Failed to delete resource: ${
-          error instanceof Error ? error.message : 'Unknown error'
-        }`,
-      },
-      500
-    );
-  }
-});
-
-// Delete all resources (bulk delete) - only available in development mode
-app.delete('/resources', async (context: Context) => {
-  // Check if we're in development mode
-  const isDevelopment = context.env.NODE_ENV === 'development';
-
-  if (!isDevelopment) {
-    return context.json(
-      { error: 'Bulk delete is only available in development mode' },
-      403
-    );
-  }
-
-  const bucket = context.env.RESOURCES_BUCKET;
-
-  // List all objects
-  const listed = await bucket.list();
-  let deletedCount = 0;
-
-  // Delete each object
-  for (const object of listed.objects) {
-    await bucket.delete(object.key);
-    deletedCount++;
-  }
-
-  // Handle pagination if there are more than 1000 objects
-  let cursor = listed.cursor;
-  while (listed.truncated && cursor) {
-    const nextBatch = await bucket.list({ cursor });
-    for (const object of nextBatch.objects) {
-      await bucket.delete(object.key);
-      deletedCount++;
-    }
-    cursor = nextBatch.cursor;
-  }
-
-  return context.json({ deletedCount });
-});
-
-// Generate presigned URL for large file uploads
-app.post('/resources/presigned-link', async (context: Context) => {
-  try {
-    const body = await context.req.json();
-    const { key, type, size } = body;
-
-    // Validate required fields
-    if (!key) {
-      return context.json({ error: 'Missing required field "key"' }, 400);
-    }
-    if (!type) {
-      return context.json({ error: 'Missing required field "type"' }, 400);
-    }
-    if (size === undefined || size === null) {
-      return context.json({ error: 'Missing required field "size"' }, 400);
-    }
-
-    // Validate type
-    if (type !== 'text' && type !== 'binary') {
-      return context.json(
-        { error: 'type must be either "text" or "binary"' },
-        400
-      );
-    }
-
-    // Check if R2 credentials are configured
-    const {
-      R2_ACCESS_KEY_ID,
-      R2_SECRET_ACCESS_KEY,
-      R2_ACCOUNT_ID,
-      R2_BUCKET_NAME,
-    } = context.env;
-
-    if (
-      !R2_ACCESS_KEY_ID ||
-      !R2_SECRET_ACCESS_KEY ||
-      !R2_ACCOUNT_ID ||
-      !R2_BUCKET_NAME
-    ) {
-      return context.json(
-        {
-          error:
-            'R2 credentials not configured. Please set R2_ACCESS_KEY_ID, R2_SECRET_ACCESS_KEY, R2_ACCOUNT_ID, and R2_BUCKET_NAME environment variables.',
-        },
-        400
-      );
-    }
-
-    // Create AWS client with R2 credentials
-    const client = new AwsClient({
-      accessKeyId: R2_ACCESS_KEY_ID,
-      secretAccessKey: R2_SECRET_ACCESS_KEY,
-    });
-
-    // Construct the R2 URL
-    const url = new URL(
-      `https://${R2_BUCKET_NAME}.${R2_ACCOUNT_ID}.r2.cloudflarestorage.com/${key}`
-    );
-
-    // Set expiration to 1 hour (3600 seconds)
-    const expiresIn = 3600;
-    url.searchParams.set('X-Amz-Expires', expiresIn.toString());
-
-    // Create a request to sign
-    const requestToSign = new Request(url, {
-      method: 'PUT',
-      headers: {
-        'x-amz-meta-type': type,
-        'x-amz-meta-local': 'false', // Manual uploads are not local
-      },
-    });
-
-    // Sign the request
-    const signedRequest = await client.sign(requestToSign, {
-      aws: {
-        signQuery: true,
-        service: 's3',
-      },
-    });
-
-    // Return the presigned URL
-    return context.json({
-      url: signedRequest.url,
-      method: 'PUT',
-      expiresIn,
-    });
-  } catch (error) {
-    console.error('Error generating presigned URL:', error);
-    return context.json(
-      {
-        error: `Failed to generate presigned URL: ${
-          error instanceof Error ? error.message : 'Unknown error'
-        }`,
-      },
-      500
-    );
-  }
-});
-
-
-export default app;