kysely-codegen
Version:
`kysely-codegen` generates Kysely type definitions from your database. That's it.
59 lines • 1.95 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.SymbolCollection = void 0;
const case_converter_1 = require("../utils/case-converter");
class SymbolCollection {
constructor(options) {
this.symbolNames = {};
this.symbols = {};
this.identifierStyle = options?.identifierStyle ?? 'kysely-pascal-case';
const entries = options?.entries?.sort(([a], [b]) => a.localeCompare(b)) ?? [];
for (const [id, symbol] of entries) {
this.set(id, symbol);
}
}
entries() {
return Object.entries(this.symbols)
.sort(([a], [b]) => a.localeCompare(b))
.map(([id, symbol]) => ({
id,
name: this.symbolNames[id],
symbol: symbol,
}));
}
get(id) {
return this.symbols[id];
}
getName(id) {
return this.symbolNames[id];
}
has(id) {
return this.symbols[id] !== undefined;
}
set(id, symbol) {
let symbolName = this.symbolNames[id];
if (symbolName) {
return symbolName;
}
const symbolNames = new Set(Object.values(this.symbolNames));
const caseConverter = this.identifierStyle === 'screaming-snake-case'
? case_converter_1.toScreamingSnakeCase
: case_converter_1.toKyselyPascalCase;
symbolName = caseConverter(id.replaceAll(/[^\w$]/g, '_'));
if (symbolNames.has(symbolName)) {
let suffix = 2;
while (symbolNames.has(`${symbolName}${suffix}`)) {
suffix++;
}
symbolName += suffix;
}
if (/^\d/.test(symbolName)) {
symbolName = `_${symbolName}`;
}
this.symbols[id] = symbol;
this.symbolNames[id] = symbolName;
return symbolName;
}
}
exports.SymbolCollection = SymbolCollection;
//# sourceMappingURL=symbol-collection.js.map