Fix type errors

Author: Brayden

src/allowlist/index.ts

@@ -1,5 +1,5 @@
 import { StarbaseDBConfiguration } from "../handler";
-import { DataSource } from "../types";
+import { DataSource, QueryResult } from "../types";
 
 const parser = new (require("node-sql-parser").Parser)();
 
@@ -17,7 +17,7 @@ function normalizeSQL(sql: string) {
 async function loadAllowlist(dataSource: DataSource): Promise<string[]> {
   try {
     const statement = "SELECT sql_statement FROM tmp_allowlist_queries";
-    const result = await dataSource.rpc.executeQuery({ sql: statement });
+    const result = await dataSource.rpc.executeQuery({ sql: statement }) as QueryResult[]
     return result.map((row) => String(row.sql_statement));
   } catch (error) {
     console.error("Error loading allowlist:", error);

src/cache/index.ts

@@ -1,5 +1,5 @@
 import { StarbaseDBConfiguration } from "../handler";
-import { DataSource } from "../types";
+import { DataSource, QueryResult } from "../types";
 import sqlparser from "node-sql-parser";
 const parser = new sqlparser.Parser();
 
@@ -62,7 +62,7 @@ export async function beforeQueryCache(opts: {
   const result = await dataSource.rpc.executeQuery({
     sql: fetchCacheStatement,
     params: [sql],
-  });
+  }) as any[];
 
   if (result?.length) {
     const { timestamp, ttl, results } = result[0] as QueryResult;
@@ -115,7 +115,7 @@ export async function afterQueryCache(opts: {
     const exists = await dataSource.rpc.executeQuery({
       sql: "SELECT 1 FROM tmp_cache WHERE query = ? LIMIT 1",
       params: [sql],
-    });
+    }) as QueryResult[];
 
     const query = exists?.length
       ? {

src/rls/index.ts

@@ -1,5 +1,5 @@
 import { StarbaseDBConfiguration } from "../handler";
-import { DataSource } from "../types";
+import { DataSource, QueryResult } from "../types";
 
 const parser = new (require("node-sql-parser").Parser)();
 
@@ -53,7 +53,7 @@ async function loadPolicies(dataSource: DataSource): Promise<Policy[]> {
       'SELECT "actions", "schema", "table", "column", "value", "value_type", "operator" FROM tmp_rls_policies';
     const result = await dataSource.rpc.executeQuery({
       sql: statement,
-    });
+    }) as QueryResult[];
 
     if (!result || result.length === 0) {
       // Discussion point to be had here. For safety precautions I am ejecting

src/types.ts

@@ -57,23 +57,6 @@ export type DataSource = {
   cacheTTL?: number;
 };
 
-// export interface InternalConnection {
-//   durableObject: DatabaseStub;
-// }
-
-// export type DatabaseStub = DurableObjectStub & {
-//   fetch: (init?: RequestInit | Request) => Promise<Response>;
-//   executeQuery(
-//     sql: string,
-//     params: any[] | undefined,
-//     isRaw: boolean
-//   ): QueryResponse;
-//   executeTransaction(
-//     queries: { sql: string; params?: any[] }[],
-//     isRaw: boolean
-//   ): any[];
-// };
-
 export enum RegionLocationHint {
   AUTO = "auto",
   WNAM = "wnam", // Western North America