Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

🐛 Fix isDbError()-guard does not work #12416

Open
wants to merge 10 commits into
base: main
Choose a base branch
from
5 changes: 5 additions & 0 deletions .changeset/breezy-radios-grab.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
---
'@astrojs/db': patch
---

fix isDbError()-guard for LibsqlError
bluwy marked this conversation as resolved.
Show resolved Hide resolved
8 changes: 3 additions & 5 deletions packages/db/src/core/cli/commands/execute/index.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import { existsSync } from 'node:fs';
import { LibsqlError } from '@libsql/client';
import type { AstroConfig } from 'astro';
import { green } from 'kleur/colors';
import type { Arguments } from 'yargs-parser';
Expand All @@ -16,6 +15,7 @@ import {
import { bundleFile, importBundledFile } from '../../../load-file.js';
import type { DBConfig } from '../../../types.js';
import { getManagedRemoteToken } from '../../../utils.js';
import { isDbError } from '../../../../runtime/utils.js';

export async function cmd({
astroConfig,
Expand Down Expand Up @@ -64,9 +64,7 @@ export async function cmd({
await mod.default();
console.info(`${green('✔')} File run successfully.`);
} catch (e) {
if (e instanceof LibsqlError) {
throw new Error(EXEC_ERROR(e.message));
}
throw e;
if (isDbError(e)) throw new Error(EXEC_ERROR(e.message));
else throw e;
}
}
5 changes: 2 additions & 3 deletions packages/db/src/core/cli/migration-queries.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
import { stripVTControlCharacters } from 'node:util';
import { LibsqlError } from '@libsql/client';
import deepDiff from 'deep-diff';
import { sql } from 'drizzle-orm';
import { SQLiteAsyncDialect } from 'drizzle-orm/sqlite-core';
Expand All @@ -8,7 +7,7 @@ import { customAlphabet } from 'nanoid';
import { hasPrimaryKey } from '../../runtime/index.js';
import { createRemoteDatabaseClient } from '../../runtime/index.js';
import { isSerializedSQL } from '../../runtime/types.js';
import { safeFetch } from '../../runtime/utils.js';
import { isDbError, safeFetch } from '../../runtime/utils.js';
import { MIGRATION_VERSION } from '../consts.js';
import { RENAME_COLUMN_ERROR, RENAME_TABLE_ERROR } from '../errors.js';
import {
Expand Down Expand Up @@ -454,7 +453,7 @@ async function getDbCurrentSnapshot(
} catch (error) {
// Don't handle errors that are not from libSQL
if (
error instanceof LibsqlError &&
isDbError(error) &&
// If the schema was never pushed to the database yet the table won't exist.
// Treat a missing snapshot table as an empty table.

Expand Down
5 changes: 2 additions & 3 deletions packages/db/src/core/integration/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@ import { mkdir, writeFile } from 'node:fs/promises';
import { dirname } from 'node:path';
import { fileURLToPath } from 'node:url';
import type { ManagedAppToken } from '@astrojs/studio';
import { LibsqlError } from '@libsql/client';
import type { AstroConfig, AstroIntegration } from 'astro';
import { blue, yellow } from 'kleur/colors';
import {
Expand All @@ -15,7 +14,7 @@ import {
mergeConfig,
} from 'vite';
import parseArgs from 'yargs-parser';
import { AstroDbError } from '../../runtime/utils.js';
import { AstroDbError, isDbError } from '../../runtime/utils.js';
import { CONFIG_FILE_NAMES, DB_PATH } from '../consts.js';
import { EXEC_DEFAULT_EXPORT_ERROR, EXEC_ERROR } from '../errors.js';
import { resolveDbConfig } from '../load-file.js';
Expand Down Expand Up @@ -209,7 +208,7 @@ async function executeSeedFile({
try {
await mod.default();
} catch (e) {
if (e instanceof LibsqlError) {
if (isDbError(e)) {
throw new AstroDbError(EXEC_ERROR(e.message));
}
throw e;
Expand Down
4 changes: 4 additions & 0 deletions packages/db/src/runtime/utils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,10 @@ export class DetailedLibsqlError extends LibsqlError {
}
}

export function isDbError(err: unknown): err is LibsqlError {
return err instanceof LibsqlError || (err instanceof Error && (err as any).libsqlError === true)
}

function slash(path: string) {
const isExtendedLengthPath = path.startsWith('\\\\?\\');

Expand Down
5 changes: 1 addition & 4 deletions packages/db/src/runtime/virtual.ts
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,6 @@ function createColumn<S extends string, T extends Record<string, unknown>>(type:
};
}

export function isDbError(err: unknown): err is LibsqlError {
return err instanceof LibsqlError;
}

export const column = {
number: <T extends NumberColumnOpts>(opts: T = {} as T) => {
return createColumn('number', opts) satisfies { type: 'number' };
Expand Down Expand Up @@ -90,3 +86,4 @@ export {
} from 'drizzle-orm';

export { alias } from 'drizzle-orm/sqlite-core';
export { isDbError } from './utils.js';
5 changes: 3 additions & 2 deletions packages/db/test/test-utils.js
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
import { createServer } from 'node:http';
import { LibsqlError, createClient } from '@libsql/client';
import { createClient } from '@libsql/client';
import { z } from 'zod';
import { cli } from '../dist/core/cli/index.js';
import { resolveDbConfig } from '../dist/core/load-file.js';
import { getCreateIndexQueries, getCreateTableQuery } from '../dist/core/queries.js';
import { isDbError } from '../dist/runtime/utils.js';

const singleQuerySchema = z.object({
sql: z.string(),
Expand Down Expand Up @@ -142,7 +143,7 @@ function createRemoteDbServer() {
JSON.stringify({
success: false,
error: {
code: e instanceof LibsqlError ? e.code : 'SQLITE_QUERY_FAILED',
code: isDbError(e) ? e.code : 'SQLITE_QUERY_FAILED',
abegehr marked this conversation as resolved.
Show resolved Hide resolved
details: e.message,
},
}),
Expand Down
Loading