86 lines
2.1 KiB
JavaScript
86 lines
2.1 KiB
JavaScript
const path = require("path")
|
|
|
|
const { createConnection } = require("typeorm")
|
|
|
|
const DB_HOST = process.env.DB_HOST
|
|
const DB_USERNAME = process.env.DB_USERNAME
|
|
const DB_PASSWORD = process.env.DB_PASSWORD
|
|
const DB_NAME = process.env.DB_NAME
|
|
const DB_URL = `postgres://${DB_USERNAME}:${DB_PASSWORD}@${DB_HOST}/${DB_NAME}`
|
|
|
|
require("./dev-require")
|
|
|
|
async function createDB() {
|
|
const connection = await createConnection({
|
|
type: "postgres",
|
|
url: `postgres://${DB_USERNAME}:${DB_PASSWORD}@${DB_HOST}`,
|
|
})
|
|
|
|
await connection.query(`DROP DATABASE IF EXISTS "${DB_NAME}";`)
|
|
await connection.query(`CREATE DATABASE "${DB_NAME}";`)
|
|
await connection.close()
|
|
}
|
|
|
|
module.exports = {
|
|
initDb: async function () {
|
|
const cwd = path.resolve(path.join(__dirname, "../.."))
|
|
|
|
const configPath = path.resolve(
|
|
path.join(__dirname, "../api/medusa-config.js")
|
|
)
|
|
const { featureFlags } = require(configPath)
|
|
|
|
const basePath = path.join(cwd, "packages/medusa/src")
|
|
|
|
const featureFlagsLoader = require(path.join(
|
|
basePath,
|
|
`loaders`,
|
|
`feature-flags`
|
|
)).default
|
|
|
|
const featureFlagsRouter = featureFlagsLoader({ featureFlags })
|
|
|
|
const modelsLoader = require(path.join(
|
|
basePath,
|
|
`loaders`,
|
|
`models`
|
|
)).default
|
|
|
|
const entities = modelsLoader({}, { register: false })
|
|
|
|
// get migraitons with enabled featureflags
|
|
const migrationDir = path.resolve(
|
|
path.join(basePath, `migrations`, `*.{j,t}s`)
|
|
)
|
|
|
|
const { getEnabledMigrations } = require(path.join(
|
|
basePath,
|
|
`commands`,
|
|
`utils`,
|
|
`get-migrations`
|
|
))
|
|
|
|
const enabledMigrations = await getEnabledMigrations(
|
|
[migrationDir],
|
|
(flag) => featureFlagsRouter.isFeatureEnabled(flag)
|
|
)
|
|
|
|
const enabledEntities = entities.filter(
|
|
(e) => typeof e.isFeatureEnabled === "undefined" || e.isFeatureEnabled()
|
|
)
|
|
|
|
await createDB()
|
|
const dbConnection = await createConnection({
|
|
type: "postgres",
|
|
url: DB_URL,
|
|
entities: enabledEntities,
|
|
migrations: enabledMigrations,
|
|
//logging: true,
|
|
})
|
|
|
|
await dbConnection.runMigrations()
|
|
|
|
return dbConnection
|
|
},
|
|
}
|