Infrastructure/astro
0
Fork 0
mirror of https://github.com/withastro/astro.git synced 2025-03-31 23:31:30 -05:00
Code Issues Activity

feat: good enough return type on seed :)

Browse source
This commit is contained in:
bholmesdev 2024-02-05 17:28:04 -05:00
parent 1ef45b1a09
commit 37848a6bd2
2 changed files with 21 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
packages/db/src/core/queries.ts
View file

@ -48,12 +48,17 @@ export async function setupDbTables({
if (data) {
try {
await data({
async seed({ table }, values) {
seed: async ({ table }, values) => {
const result = Array.isArray(values)
? // TODO: fix values typing once we can infer fields type correctly
await db.insert(table).values(values).returning()
: await db.insert(table).values(values).returning().get();
return result;
? db.insert(table).values(values).returning()
: db
.insert(table)
.values(values as any)
.returning()
.get();
// Drizzle types don't *quite* line up, and it's tough to debug why.
// we're casting and calling this close enough :)
return result as any;
},
db,
mode,

14
packages/db/src/core/types.ts
View file

@ -1,4 +1,5 @@
import type { SQLiteInsertValue } from 'drizzle-orm/sqlite-core';
import type { InferSelectModel } from 'drizzle-orm';
import type { SqliteDB, Table } from '../runtime/index.js';
import { z } from 'zod';
import { getTableName } from 'drizzle-orm';
@ -158,10 +159,17 @@ export type WritableDBCollection = z.infer<typeof writableCollectionSchema>;
export type DBDataContext = {
db: SqliteDB;
seed<TFields extends FieldsConfig>(
seed: <
TFields extends FieldsConfig,
TData extends MaybeArray<SQLiteInsertValue<Table<string, TFields>>>,
>(
collection: ResolvedCollectionConfig<TFields>,
data: MaybeArray<SQLiteInsertValue<Table<string, TFields>>>
): Promise<any> /** TODO: type output */;
data: TData
) => Promise<
TData extends Array<SQLiteInsertValue<Table<string, TFields>>>
? InferSelectModel<Table<string, TFields>>[]
: InferSelectModel<Table<string, TFields>>
>;
mode: 'dev' | 'build';
};