@aws-amplify/graphql-schema-generator
Version:
Amplify GraphQL schema generator
90 lines • 3.5 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
Object.defineProperty(exports, "__esModule", { value: true });
exports.DataSourceAdapter = void 0;
const os = __importStar(require("os"));
const schema_representation_1 = require("../schema-representation");
class DataSourceAdapter {
constructor() {
this.useVPC = false;
this.vpcSchemaInspectorLambda = undefined;
this.vpcLambdaRegion = undefined;
}
getModels() {
const tableNames = this.getTablesList();
const models = [];
for (const table of tableNames) {
models.push(this.describeTable(table));
}
return models;
}
describeTable(tableName) {
const fields = this.getFields(tableName);
const primaryKey = this.getPrimaryKey(tableName);
const indexes = this.getIndexes(tableName);
const model = new schema_representation_1.Model(tableName);
fields.forEach((field) => model.addField(field));
primaryKey && model.setPrimaryKey(primaryKey.getFields());
indexes.forEach((index) => model.addIndex(index.name, index.getFields()));
return model;
}
useVpc(vpcSchemaInspectorLambda, region) {
this.useVPC = true;
this.vpcSchemaInspectorLambda = vpcSchemaInspectorLambda;
this.vpcLambdaRegion = region;
}
queryToCSV(queryResult) {
if (queryResult.length === 0) {
return '';
}
const headers = Object.keys(queryResult[0]);
const headerIndices = Object.fromEntries(headers.map((key, index) => [index, key]));
const rows = queryResult.map((row) => [...Array(headers.length).keys()]
.map((index) => {
const value = row[headerIndices[index]];
if (typeof value === 'string' && value.includes(',')) {
return `"${value}"`;
}
if (value === null) {
return 'NULL';
}
return value;
})
.join(','));
return headers.join(',') + os.EOL + rows.join(os.EOL);
}
}
exports.DataSourceAdapter = DataSourceAdapter;
//# sourceMappingURL=datasource-adapter.js.map