fix workflow code (#140)

* fix workflow code

* fix nested code

* fixes

Author: ctate

lib/codegen-templates/database-query.ts

@@ -1,21 +1,30 @@
 /**
  * Code template for Database Query action step
  * This is a string template used for code generation - keep as string export
+ *
+ * Requires: pnpm add postgres
+ * Environment: DATABASE_URL
  */
 export default `export async function databaseQueryStep(input: {
   query: string;
 }) {
   "use step";
   
-  // Database Query - You need to set up your database connection
-  // Install: pnpm add postgres (or your preferred database library)
-  // Set DATABASE_URL in your environment variables
+  const databaseUrl = process.env.DATABASE_URL;
+  if (!databaseUrl) {
+    return { success: false, error: "DATABASE_URL environment variable is not set" };
+  }
   
-  // Example using postgres library:
-  // import postgres from 'postgres';
-  // const sql = postgres(process.env.DATABASE_URL!);
-  // const result = await sql.unsafe(input.query);
-  // await sql.end();
+  const postgres = await import("postgres");
+  const sql = postgres.default(databaseUrl, { max: 1 });
   
-  throw new Error('Database Query not implemented - see comments in generated code');
+  try {
+    const result = await sql.unsafe(input.query);
+    await sql.end();
+    return { success: true, rows: result, count: result.length };
+  } catch (error) {
+    await sql.end();
+    const message = error instanceof Error ? error.message : String(error);
+    return { success: false, error: \`Database query failed: \${message}\` };
+  }
 }`;