asyavee/medusa-store
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat(migrations): CLI generate command (#8103)

Browse Source
This commit is contained in:
Adrien de Peretti
2024-07-12 13:12:49 +02:00
committed by GitHub
parent 4c2e9a3239
commit 104b00d4e9
14 changed files with 341 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
packages/core/modules-sdk/src/loaders/utils/load-internal.ts
View File

@@ -2,6 +2,7 @@ import {
Constructor,
IModuleService,
InternalModuleDeclaration,
LoaderOptions,
Logger,
MedusaContainer,
ModuleExports,
@@ -32,6 +33,11 @@ type ModuleResource = {
normalizedPath: string
}
type MigrationFunction = (
options: LoaderOptions<any>,
moduleDeclaration?: InternalModuleDeclaration
) => Promise<void>
export async function loadInternalModule(
container: MedusaContainer,
resolution: ModuleResolution,
@@ -171,7 +177,11 @@ export async function loadInternalModule(
export async function loadModuleMigrations(
resolution: ModuleResolution,
moduleExports?: ModuleExports
): Promise<[Function | undefined, Function | undefined]> {
): Promise<{
runMigrations?: MigrationFunction
revertMigration?: MigrationFunction
generateMigration?: MigrationFunction
}> {
let loadedModule: ModuleExports
try {
loadedModule =
@@ -179,6 +189,7 @@ export async function loadModuleMigrations(
let runMigrations = loadedModule.runMigrations
let revertMigration = loadedModule.revertMigration
let generateMigration = loadedModule.generateMigration
if (!runMigrations || !revertMigration) {
const moduleResources = await loadResources(
@@ -189,6 +200,7 @@ export async function loadModuleMigrations(
const migrationScriptOptions = {
moduleName: resolution.definition.key,
models: moduleResources.models,
pathToMigrations: join(moduleResources.normalizedPath, "migrations"),
}
@@ -199,11 +211,15 @@ export async function loadModuleMigrations(
revertMigration ??= ModulesSdkUtils.buildRevertMigrationScript(
migrationScriptOptions
)
generateMigration ??= ModulesSdkUtils.buildGenerateMigrationScript(
migrationScriptOptions
)
}
return [runMigrations, revertMigration]
return { runMigrations, revertMigration, generateMigration }
} catch {
return [undefined, undefined]
return {}
}
}

70
packages/core/modules-sdk/src/medusa-app.ts
View File

@@ -31,7 +31,11 @@ import {
import { asValue } from "awilix"
import type { Knex } from "knex"
import { MODULE_PACKAGE_NAMES } from "./definitions"
import { MedusaModule, RegisterModuleJoinerConfig } from "./medusa-module"
import {
MedusaModule,
MigrationOptions,
RegisterModuleJoinerConfig,
} from "./medusa-module"
import { RemoteLink } from "./remote-link"
import { RemoteQuery } from "./remote-query"
import { MODULE_RESOURCE_TYPE, MODULE_SCOPE } from "./types"
@@ -51,6 +55,7 @@ declare module "@medusajs/types" {
export type RunMigrationFn = () => Promise<void>
export type RevertMigrationFn = (moduleNames: string[]) => Promise<void>
export type GenerateMigrations = (moduleNames: string[]) => Promise<void>
export type MedusaModuleConfig = {
[key: string | Modules]:
@@ -225,6 +230,7 @@ export type MedusaAppOutput = {
notFound?: Record<string, Record<string, string>>
runMigrations: RunMigrationFn
revertMigrations: RevertMigrationFn
generateMigrations: GenerateMigrations
onApplicationShutdown: () => Promise<void>
onApplicationPrepareShutdown: () => Promise<void>
sharedContainer?: MedusaContainer
@@ -354,6 +360,9 @@ async function MedusaApp_({
revertMigrations: async () => {
throw new Error("Revert migrations not allowed in loaderOnly mode")
},
generateMigrations: async () => {
throw new Error("Generate migrations not allowed in loaderOnly mode")
},
}
}
@@ -406,10 +415,10 @@ async function MedusaApp_({
const applyMigration = async ({
modulesNames,
revert = false,
action = "run",
}: {
modulesNames: string[]
revert?: boolean
action?: "run" | "revert" | "generate"
}) => {
const moduleResolutions = modulesNames.map((moduleName) => {
return {
@@ -423,7 +432,6 @@ async function MedusaApp_({
.map(({ moduleName }) => moduleName)
if (missingModules.length) {
const action = revert ? "revert" : "run"
const error = new MedusaError(
MedusaError.Types.UNKNOWN_MODULES,
`Cannot ${action} migrations for unknown module(s) ${missingModules.join(
@@ -443,22 +451,20 @@ async function MedusaApp_({
}
}
if (revert) {
await MedusaModule.migrateDown(
moduleResolution.definition.key,
moduleResolution.resolutionPath as string,
sharedContainer,
moduleResolution.options,
moduleResolution.moduleExports
)
const migrationOptions: MigrationOptions = {
moduleKey: moduleResolution.definition.key,
modulePath: moduleResolution.resolutionPath as string,
container: sharedContainer,
options: moduleResolution.options,
moduleExports: moduleResolution.moduleExports,
}
if (action === "revert") {
await MedusaModule.migrateDown(migrationOptions)
} else if (action === "run") {
await MedusaModule.migrateUp(migrationOptions)
} else {
await MedusaModule.migrateUp(
moduleResolution.definition.key,
moduleResolution.resolutionPath as string,
sharedContainer,
moduleResolution.options,
moduleResolution.moduleExports
)
await MedusaModule.migrateGenerate(migrationOptions)
}
}
}
@@ -493,7 +499,7 @@ async function MedusaApp_({
): Promise<void> => {
await applyMigration({
modulesNames,
revert: true,
action: "revert",
})
const options: Partial<ModuleServiceInitializeOptions> =
@@ -516,6 +522,15 @@ async function MedusaApp_({
)
}
const generateMigrations: GenerateMigrations = async (
modulesNames
): Promise<void> => {
await applyMigration({
modulesNames,
action: "generate",
})
}
return {
onApplicationShutdown,
onApplicationPrepareShutdown,
@@ -526,6 +541,7 @@ async function MedusaApp_({
notFound,
runMigrations,
revertMigrations,
generateMigrations,
sharedContainer: sharedContainer_,
}
}
@@ -566,3 +582,17 @@ export async function MedusaAppMigrateDown(
await revertMigrations(moduleNames).finally(MedusaModule.clearInstances)
}
export async function MedusaAppMigrateGenerate(
moduleNames: string[],
options: MedusaAppOptions = {}
): Promise<void> {
const migrationOnly = true
const { generateMigrations } = await MedusaApp_({
...options,
migrationOnly,
})
await generateMigrations(moduleNames).finally(MedusaModule.clearInstances)
}

91
packages/core/modules-sdk/src/medusa-module.ts
View File

@@ -52,6 +52,14 @@ type ModuleAlias = {
main?: boolean
}
export type MigrationOptions = {
moduleKey: string
modulePath: string
container?: MedusaContainer
options?: Record<string, any>
moduleExports?: ModuleExports
}
export type ModuleBootstrapOptions = {
moduleKey: string
defaultPath: string
@@ -536,13 +544,13 @@ class MedusaModule {
return services
}
public static async migrateUp(
moduleKey: string,
modulePath: string,
container?: MedusaContainer,
options?: Record<string, any>,
moduleExports?: ModuleExports
): Promise<void> {
public static async migrateGenerate({
options,
container,
moduleExports,
moduleKey,
modulePath,
}: MigrationOptions): Promise<void> {
const moduleResolutions = registerMedusaModule(moduleKey, {
scope: MODULE_SCOPE.INTERNAL,
resources: MODULE_RESOURCE_TYPE.ISOLATED,
@@ -555,28 +563,31 @@ class MedusaModule {
allowUnregistered: true,
}) ?? logger
container ??= createMedusaContainer()
for (const mod in moduleResolutions) {
const [migrateUp] = await loadModuleMigrations(
const { generateMigration } = await loadModuleMigrations(
moduleResolutions[mod],
moduleExports
)
if (typeof migrateUp === "function") {
await migrateUp({
if (typeof generateMigration === "function") {
await generateMigration({
options,
container: container!,
logger: logger_,
})
}
}
}
public static async migrateDown(
moduleKey: string,
modulePath: string,
container?: MedusaContainer,
options?: Record<string, any>,
moduleExports?: ModuleExports
): Promise<void> {
public static async migrateUp({
options,
container,
moduleExports,
moduleKey,
modulePath,
}: MigrationOptions): Promise<void> {
const moduleResolutions = registerMedusaModule(moduleKey, {
scope: MODULE_SCOPE.INTERNAL,
resources: MODULE_RESOURCE_TYPE.ISOLATED,
@@ -589,15 +600,55 @@ class MedusaModule {
allowUnregistered: true,
}) ?? logger
container ??= createMedusaContainer()
for (const mod in moduleResolutions) {
const [, migrateDown] = await loadModuleMigrations(
const { runMigrations } = await loadModuleMigrations(
moduleResolutions[mod],
moduleExports
)
if (typeof migrateDown === "function") {
await migrateDown({
if (typeof runMigrations === "function") {
await runMigrations({
options,
container: container!,
logger: logger_,
})
}
}
}
public static async migrateDown({
options,
container,
moduleExports,
moduleKey,
modulePath,
}: MigrationOptions): Promise<void> {
const moduleResolutions = registerMedusaModule(moduleKey, {
scope: MODULE_SCOPE.INTERNAL,
resources: MODULE_RESOURCE_TYPE.ISOLATED,
resolve: modulePath,
options,
})
const logger_ =
container?.resolve(ContainerRegistrationKeys.LOGGER, {
allowUnregistered: true,
}) ?? logger
container ??= createMedusaContainer()
for (const mod in moduleResolutions) {
const { revertMigration } = await loadModuleMigrations(
moduleResolutions[mod],
moduleExports
)
if (typeof revertMigration === "function") {
await revertMigration({
options,
container: container!,
logger: logger_,
})
}