201 lines
5.4 KiB
JavaScript
201 lines
5.4 KiB
JavaScript
/// <reference types="./sqlite.d.ts" />
|
||
import { sql } from '../raw-builder/sql.js';
|
||
import { getJsonObjectArgs } from '../util/json-object-args.js';
|
||
/**
|
||
* A SQLite helper for aggregating a subquery into a JSON array.
|
||
*
|
||
* NOTE: This helper only works correctly if you've installed the `ParseJSONResultsPlugin`.
|
||
* Otherwise the nested selections will be returned as JSON strings.
|
||
*
|
||
* The plugin can be installed like this:
|
||
*
|
||
* ```ts
|
||
* import * as Sqlite from 'better-sqlite3'
|
||
* import { Kysely, ParseJSONResultsPlugin, SqliteDialect } from 'kysely'
|
||
* import type { Database } from 'type-editor' // imaginary module
|
||
*
|
||
* const db = new Kysely<Database>({
|
||
* dialect: new SqliteDialect({
|
||
* database: new Sqlite(':memory:')
|
||
* }),
|
||
* plugins: [new ParseJSONResultsPlugin()]
|
||
* })
|
||
* ```
|
||
*
|
||
* ### Examples
|
||
*
|
||
* ```ts
|
||
* import { jsonArrayFrom } from 'kysely/helpers/sqlite'
|
||
*
|
||
* const result = await db
|
||
* .selectFrom('person')
|
||
* .select((eb) => [
|
||
* 'id',
|
||
* jsonArrayFrom(
|
||
* eb.selectFrom('pet')
|
||
* .select(['pet.id as pet_id', 'pet.name'])
|
||
* .whereRef('pet.owner_id', '=', 'person.id')
|
||
* .orderBy('pet.name')
|
||
* ).as('pets')
|
||
* ])
|
||
* .execute()
|
||
*
|
||
* result[0]?.id
|
||
* result[0]?.pets[0].pet_id
|
||
* result[0]?.pets[0].name
|
||
* ```
|
||
*
|
||
* The generated SQL (SQLite):
|
||
*
|
||
* ```sql
|
||
* select "id", (
|
||
* select coalesce(json_group_array(json_object(
|
||
* 'pet_id', "agg"."pet_id",
|
||
* 'name', "agg"."name"
|
||
* )), '[]') from (
|
||
* select "pet"."id" as "pet_id", "pet"."name"
|
||
* from "pet"
|
||
* where "pet"."owner_id" = "person"."id"
|
||
* order by "pet"."name"
|
||
* ) as "agg"
|
||
* ) as "pets"
|
||
* from "person"
|
||
* ```
|
||
*/
|
||
export function jsonArrayFrom(expr) {
|
||
return sql `(select coalesce(json_group_array(json_object(${sql.join(getSqliteJsonObjectArgs(expr.toOperationNode(), 'agg'))})), '[]') from ${expr} as agg)`;
|
||
}
|
||
/**
|
||
* A SQLite helper for turning a subquery into a JSON object.
|
||
*
|
||
* The subquery must only return one row.
|
||
*
|
||
* NOTE: This helper only works correctly if you've installed the `ParseJSONResultsPlugin`.
|
||
* Otherwise the nested selections will be returned as JSON strings.
|
||
*
|
||
* The plugin can be installed like this:
|
||
*
|
||
* ```ts
|
||
* import * as Sqlite from 'better-sqlite3'
|
||
* import { Kysely, ParseJSONResultsPlugin, SqliteDialect } from 'kysely'
|
||
* import type { Database } from 'type-editor' // imaginary module
|
||
*
|
||
* const db = new Kysely<Database>({
|
||
* dialect: new SqliteDialect({
|
||
* database: new Sqlite(':memory:')
|
||
* }),
|
||
* plugins: [new ParseJSONResultsPlugin()]
|
||
* })
|
||
* ```
|
||
*
|
||
* ### Examples
|
||
*
|
||
* ```ts
|
||
* import { jsonObjectFrom } from 'kysely/helpers/sqlite'
|
||
*
|
||
* const result = await db
|
||
* .selectFrom('person')
|
||
* .select((eb) => [
|
||
* 'id',
|
||
* jsonObjectFrom(
|
||
* eb.selectFrom('pet')
|
||
* .select(['pet.id as pet_id', 'pet.name'])
|
||
* .whereRef('pet.owner_id', '=', 'person.id')
|
||
* .where('pet.is_favorite', '=', true)
|
||
* ).as('favorite_pet')
|
||
* ])
|
||
* .execute()
|
||
*
|
||
* result[0]?.id
|
||
* result[0]?.favorite_pet?.pet_id
|
||
* result[0]?.favorite_pet?.name
|
||
* ```
|
||
*
|
||
* The generated SQL (SQLite):
|
||
*
|
||
* ```sql
|
||
* select "id", (
|
||
* select json_object(
|
||
* 'pet_id', "obj"."pet_id",
|
||
* 'name', "obj"."name"
|
||
* ) from (
|
||
* select "pet"."id" as "pet_id", "pet"."name"
|
||
* from "pet"
|
||
* where "pet"."owner_id" = "person"."id"
|
||
* and "pet"."is_favorite" = ?
|
||
* ) as obj
|
||
* ) as "favorite_pet"
|
||
* from "person";
|
||
* ```
|
||
*/
|
||
export function jsonObjectFrom(expr) {
|
||
return sql `(select json_object(${sql.join(getSqliteJsonObjectArgs(expr.toOperationNode(), 'obj'))}) from ${expr} as obj)`;
|
||
}
|
||
/**
|
||
* The SQLite `json_object` function.
|
||
*
|
||
* NOTE: This helper only works correctly if you've installed the `ParseJSONResultsPlugin`.
|
||
* Otherwise the nested selections will be returned as JSON strings.
|
||
*
|
||
* The plugin can be installed like this:
|
||
*
|
||
* ```ts
|
||
* import * as Sqlite from 'better-sqlite3'
|
||
* import { Kysely, ParseJSONResultsPlugin, SqliteDialect } from 'kysely'
|
||
* import type { Database } from 'type-editor' // imaginary module
|
||
*
|
||
* const db = new Kysely<Database>({
|
||
* dialect: new SqliteDialect({
|
||
* database: new Sqlite(':memory:')
|
||
* }),
|
||
* plugins: [new ParseJSONResultsPlugin()]
|
||
* })
|
||
* ```
|
||
*
|
||
* ### Examples
|
||
*
|
||
* ```ts
|
||
* import { sql } from 'kysely'
|
||
* import { jsonBuildObject } from 'kysely/helpers/sqlite'
|
||
*
|
||
* const result = await db
|
||
* .selectFrom('person')
|
||
* .select((eb) => [
|
||
* 'id',
|
||
* jsonBuildObject({
|
||
* first: eb.ref('first_name'),
|
||
* last: eb.ref('last_name'),
|
||
* full: sql<string>`first_name || ' ' || last_name`
|
||
* }).as('name')
|
||
* ])
|
||
* .execute()
|
||
*
|
||
* result[0]?.id
|
||
* result[0]?.name.first
|
||
* result[0]?.name.last
|
||
* result[0]?.name.full
|
||
* ```
|
||
*
|
||
* The generated SQL (SQLite):
|
||
*
|
||
* ```sql
|
||
* select "id", json_object(
|
||
* 'first', first_name,
|
||
* 'last', last_name,
|
||
* 'full', "first_name" || ' ' || "last_name"
|
||
* ) as "name"
|
||
* from "person"
|
||
* ```
|
||
*/
|
||
export function jsonBuildObject(obj) {
|
||
return sql `json_object(${sql.join(Object.keys(obj).flatMap((k) => [sql.lit(k), obj[k]]))})`;
|
||
}
|
||
function getSqliteJsonObjectArgs(node, table) {
|
||
try {
|
||
return getJsonObjectArgs(node, table);
|
||
}
|
||
catch {
|
||
throw new Error('SQLite jsonArrayFrom and jsonObjectFrom functions can only handle explicit selections due to limitations of the json_object function. selectAll() is not allowed in the subquery.');
|
||
}
|
||
}
|