diff --git a/src/server.ts b/src/server.ts
index 8018f22..8fad304 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -1,4 +1,5 @@
-///
+////
+import type { Database } from 'bun:sqlite';
import { CommandService } from './services/command';
import { GroupSyncService } from './services/group-sync';
import { ResponseQueue } from './services/response-queue';
diff --git a/src/tasks/service.ts b/src/tasks/service.ts
index 19d86f1..743e9cb 100644
--- a/src/tasks/service.ts
+++ b/src/tasks/service.ts
@@ -1,4 +1,4 @@
-import type { Database } from 'better-sqlite3';
+import type { Database } from 'bun:sqlite';
import { db } from '../db';
type CreateTaskInput = {
@@ -21,15 +21,15 @@ export class TaskService {
const insertTask = this.dbInstance.prepare(`
INSERT INTO tasks (description, due_date, group_id, created_by)
VALUES (?, ?, ?, ?)
- RETURNING id
`);
- const taskResult = insertTask.get(
+ const runResult = insertTask.run(
task.description,
task.due_date ?? null,
task.group_id ?? null,
task.created_by
- ) as { id: number };
+ );
+ const taskId = Number((runResult as any).lastInsertRowid);
if (assignments.length > 0) {
const insertAssignment = this.dbInstance.prepare(`
@@ -42,11 +42,11 @@ export class TaskService {
for (const a of assignments) {
if (seen.has(a.user_id)) continue;
seen.add(a.user_id);
- insertAssignment.run(taskResult.id, a.user_id, a.assigned_by);
+ insertAssignment.run(taskId, a.user_id, a.assigned_by);
}
}
- return taskResult.id;
+ return taskId;
});
return runTx();
diff --git a/tests/unit/tasks/service.test.ts b/tests/unit/tasks/service.test.ts
index fbe279e..92fc3b2 100644
--- a/tests/unit/tasks/service.test.ts
+++ b/tests/unit/tasks/service.test.ts
@@ -1,5 +1,5 @@
import { describe, it, expect, beforeEach, afterEach } from 'bun:test';
-import Database from 'better-sqlite3';
+import { Database } from 'bun:sqlite';
import { initializeDatabase, ensureUserExists } from '../../../src/db';
import { TaskService } from '../../../src/tasks/service';