better-auth
Version:
The most comprehensive authentication framework for TypeScript.
122 lines (120 loc) • 4.73 kB
JavaScript
import { getAuthTables } from "@better-auth/core/db";
import { APIError } from "better-call";
//#region src/db/schema.ts
const cache = /* @__PURE__ */ new WeakMap();
function parseOutputData(data, schema) {
const fields = schema.fields;
const parsedData = {};
for (const key in data) {
const field = fields[key];
if (!field) {
parsedData[key] = data[key];
continue;
}
if (field.returned === false && key !== "id") continue;
parsedData[key] = data[key];
}
return parsedData;
}
function getFields(options, table, mode) {
const cacheKey = `${table}:${mode}`;
if (!cache.has(options)) cache.set(options, /* @__PURE__ */ new Map());
const tableCache = cache.get(options);
if (tableCache.has(cacheKey)) return tableCache.get(cacheKey);
const coreSchema = mode === "output" ? getAuthTables(options)[table]?.fields ?? {} : {};
const additionalFields = table === "user" || table === "session" || table === "account" ? options[table]?.additionalFields : void 0;
let schema = {
...coreSchema,
...additionalFields ?? {}
};
for (const plugin of options.plugins || []) if (plugin.schema && plugin.schema[table]) schema = {
...schema,
...plugin.schema[table].fields
};
tableCache.set(cacheKey, schema);
return schema;
}
function parseUserOutput(options, user) {
return parseOutputData(user, { fields: getFields(options, "user", "output") });
}
function parseSessionOutput(options, session) {
return parseOutputData(session, { fields: getFields(options, "session", "output") });
}
function parseAccountOutput(options, account) {
const { accessToken: _accessToken, refreshToken: _refreshToken, idToken: _idToken, accessTokenExpiresAt: _accessTokenExpiresAt, refreshTokenExpiresAt: _refreshTokenExpiresAt, password: _password, ...rest } = parseOutputData(account, { fields: getFields(options, "account", "output") });
return rest;
}
function parseInputData(data, schema) {
const action = schema.action || "create";
const fields = schema.fields;
const parsedData = Object.assign(Object.create(null), null);
for (const key in fields) {
if (key in data) {
if (fields[key].input === false) {
if (fields[key].defaultValue !== void 0) {
if (action !== "update") {
parsedData[key] = fields[key].defaultValue;
continue;
}
}
if (data[key]) throw new APIError("BAD_REQUEST", { message: `${key} is not allowed to be set` });
continue;
}
if (fields[key].validator?.input && data[key] !== void 0) {
const result = fields[key].validator.input["~standard"].validate(data[key]);
if (result instanceof Promise) throw new APIError("INTERNAL_SERVER_ERROR", { message: "Async validation is not supported for additional fields" });
if ("issues" in result && result.issues) throw new APIError("BAD_REQUEST", { message: result.issues[0]?.message || "Validation Error" });
parsedData[key] = result.value;
continue;
}
if (fields[key].transform?.input && data[key] !== void 0) {
parsedData[key] = fields[key].transform?.input(data[key]);
continue;
}
parsedData[key] = data[key];
continue;
}
if (fields[key].defaultValue !== void 0 && action === "create") {
if (typeof fields[key].defaultValue === "function") {
parsedData[key] = fields[key].defaultValue();
continue;
}
parsedData[key] = fields[key].defaultValue;
continue;
}
if (fields[key].required && action === "create") throw new APIError("BAD_REQUEST", { message: `${key} is required` });
}
return parsedData;
}
function parseUserInput(options, user = {}, action) {
return parseInputData(user, {
fields: getFields(options, "user", "input"),
action
});
}
function parseAdditionalUserInput(options, user) {
const schema = getFields(options, "user", "input");
return parseInputData(user || {}, { fields: schema });
}
function parseAccountInput(options, account) {
return parseInputData(account, { fields: getFields(options, "account", "input") });
}
function parseSessionInput(options, session) {
return parseInputData(session, { fields: getFields(options, "session", "input") });
}
function mergeSchema(schema, newSchema) {
if (!newSchema) return schema;
for (const table in newSchema) {
const newModelName = newSchema[table]?.modelName;
if (newModelName) schema[table].modelName = newModelName;
for (const field in schema[table].fields) {
const newField = newSchema[table]?.fields?.[field];
if (!newField) continue;
schema[table].fields[field].fieldName = newField;
}
}
return schema;
}
//#endregion
export { mergeSchema, parseAccountInput, parseAccountOutput, parseAdditionalUserInput, parseInputData, parseSessionInput, parseSessionOutput, parseUserInput, parseUserOutput };
//# sourceMappingURL=schema.mjs.map