drizzle-orm
Version:
Drizzle ORM package for SQL databases
126 lines • 3.31 kB
JavaScript
import { entityKind } from "../entity.js";
import { SelectionProxyHandler } from "../selection-proxy.js";
import { getTableColumns } from "../utils.js";
import { QueryBuilder } from "./query-builders/query-builder.js";
import { mysqlTable } from "./table.js";
import { MySqlViewBase } from "./view-base.js";
import { MySqlViewConfig } from "./view-common.js";
class ViewBuilderCore {
constructor(name, schema) {
this.name = name;
this.schema = schema;
}
static [entityKind] = "MySqlViewBuilder";
config = {};
algorithm(algorithm) {
this.config.algorithm = algorithm;
return this;
}
sqlSecurity(sqlSecurity) {
this.config.sqlSecurity = sqlSecurity;
return this;
}
withCheckOption(withCheckOption) {
this.config.withCheckOption = withCheckOption ?? "cascaded";
return this;
}
}
class ViewBuilder extends ViewBuilderCore {
static [entityKind] = "MySqlViewBuilder";
as(qb) {
if (typeof qb === "function") {
qb = qb(new QueryBuilder());
}
const selectionProxy = new SelectionProxyHandler({
alias: this.name,
sqlBehavior: "error",
sqlAliasedBehavior: "alias",
replaceOriginalName: true
});
const aliasedSelection = new Proxy(qb.getSelectedFields(), selectionProxy);
return new Proxy(
new MySqlView({
mysqlConfig: this.config,
config: {
name: this.name,
schema: this.schema,
selectedFields: aliasedSelection,
query: qb.getSQL().inlineParams()
}
}),
selectionProxy
);
}
}
class ManualViewBuilder extends ViewBuilderCore {
static [entityKind] = "MySqlManualViewBuilder";
columns;
constructor(name, columns, schema) {
super(name, schema);
this.columns = getTableColumns(mysqlTable(name, columns));
}
existing() {
return new Proxy(
new MySqlView({
mysqlConfig: void 0,
config: {
name: this.name,
schema: this.schema,
selectedFields: this.columns,
query: void 0
}
}),
new SelectionProxyHandler({
alias: this.name,
sqlBehavior: "error",
sqlAliasedBehavior: "alias",
replaceOriginalName: true
})
);
}
as(query) {
return new Proxy(
new MySqlView({
mysqlConfig: this.config,
config: {
name: this.name,
schema: this.schema,
selectedFields: this.columns,
query: query.inlineParams()
}
}),
new SelectionProxyHandler({
alias: this.name,
sqlBehavior: "error",
sqlAliasedBehavior: "alias",
replaceOriginalName: true
})
);
}
}
class MySqlView extends MySqlViewBase {
static [entityKind] = "MySqlView";
[MySqlViewConfig];
constructor({ mysqlConfig, config }) {
super(config);
this[MySqlViewConfig] = mysqlConfig;
}
}
function mysqlViewWithSchema(name, selection, schema) {
if (selection) {
return new ManualViewBuilder(name, selection, schema);
}
return new ViewBuilder(name, schema);
}
function mysqlView(name, selection) {
return mysqlViewWithSchema(name, selection, void 0);
}
export {
ManualViewBuilder,
MySqlView,
ViewBuilder,
ViewBuilderCore,
mysqlView,
mysqlViewWithSchema
};
//# sourceMappingURL=view.js.map