strapi/packages/core/database/lib/index.js

'use strict';

const knex = require('knex');

const { getDialect } = require('./dialects');
const createSchemaProvider = require('./schema');
const createMetadata = require('./metadata');
const { createEntityManager } = require('./entity-manager');

// TODO: move back into strapi
const { transformContentTypes } = require('./utils/content-types');
// const Configuration = require('./configuration');
// const { resolveConnector } = require('./connector');

class Database {
  constructor(config) {
    this.metadata = createMetadata(config.models);

    // TODO: validate meta
    // this.metadata.validate();

    // this.connector = resolveConnector(this.config);

    this.connection = knex(config.connection);
    this.dialect = getDialect(this);

    this.schema = createSchemaProvider(this);

    // TODO: migrations -> allow running them through cli before startup

    this.entityManager = createEntityManager(this);
  }

  async initialize() {
    await this.dialect.initialize();
  }

  query(uid) {
    if (!this.metadata.has(uid)) {
      throw new Error(`Model ${uid} not found`);
    }

    return this.entityManager.getRepository(uid);
  }

  async destroy() {
    await this.connection.destroy();
  }
}

Database.transformContentTypes = transformContentTypes;
Database.init = async config => {
  const db = new Database(config);

  await db.initialize();

  return db;
};

module.exports = {
  Database,
};