fix(installer): retain stale modules during updates

This commit is contained in:
Dicky Moore 2026-05-18 13:09:52 +01:00
parent 1541840ec6
commit 3cdee7913c
4 changed files with 139 additions and 111 deletions

View File

@ -76,25 +76,23 @@ class Installer {
const results = [];
const addResult = (step, status, detail = '', meta = {}) => results.push({ step, status, detail, ...meta });
// Capture previously installed skill IDs before they get overwritten
const previousSkillIds = new Set();
const prevCsvPath = path.join(paths.bmadDir, '_config', 'skill-manifest.csv');
if (await fs.pathExists(prevCsvPath)) {
try {
const csvParse = require('csv-parse/sync');
const content = await fs.readFile(prevCsvPath, 'utf8');
const records = csvParse.parse(content, { columns: true, skip_empty_lines: true });
for (const r of records) {
if (r.canonicalId) previousSkillIds.add(r.canonicalId);
}
} catch (error) {
await prompts.log.warn(`Failed to parse skill-manifest.csv: ${error.message}`);
}
}
// Capture previously installed skill rows before they get overwritten
const preservedModules = originalConfig._preserveModules || [];
const previousSkillManifestRows = await this._readSkillManifestRows(paths.bmadDir);
const previousSkillIds = this._getPreviousSkillIdsForCleanup(previousSkillManifestRows, preservedModules);
const allModules = config.modules || [];
await this._installAndConfigure(config, originalConfig, paths, allModules, allModules, addResult, officialModules);
await this._installAndConfigure(
config,
originalConfig,
paths,
allModules,
allModules,
addResult,
officialModules,
previousSkillManifestRows,
);
await this._setupIdes(config, allModules, paths, addResult, previousSkillIds);
@ -213,7 +211,16 @@ class Installer {
/**
* Install modules, create directories, generate configs and manifests.
*/
async _installAndConfigure(config, originalConfig, paths, officialModuleIds, allModules, addResult, officialModules) {
async _installAndConfigure(
config,
originalConfig,
paths,
officialModuleIds,
allModules,
addResult,
officialModules,
previousSkillManifestRows = [],
) {
const isQuickUpdate = config.isQuickUpdate();
const moduleConfigs = officialModules.moduleConfigs;
@ -292,25 +299,29 @@ class Installer {
message('Generating manifests...');
const manifestGen = new ManifestGenerator();
const preservedModules = originalConfig._preserveModules || [];
const allModulesForManifest = config.isQuickUpdate()
? originalConfig._existingModules || allModules || []
: originalConfig._preserveModules
? [...allModules, ...originalConfig._preserveModules]
: preservedModules.length > 0
? [...allModules, ...preservedModules]
: allModules || [];
let modulesForCsvPreserve;
if (config.isQuickUpdate()) {
modulesForCsvPreserve = originalConfig._existingModules || allModules || [];
} else {
modulesForCsvPreserve = originalConfig._preserveModules ? [...allModules, ...originalConfig._preserveModules] : allModules;
modulesForCsvPreserve = preservedModules.length > 0 ? [...allModules, ...preservedModules] : allModules;
}
await this._trackPreservedModuleFiles(paths.bmadDir, preservedModules);
await manifestGen.generateManifests(paths.bmadDir, allModulesForManifest, [...this.installedFiles], {
ides: config.ides || [],
preservedModules: modulesForCsvPreserve,
moduleConfigs,
});
await this._appendPreservedSkillManifestRows(paths.bmadDir, previousSkillManifestRows, preservedModules);
// Apply post-install --set TOML patches. Runs after writeCentralConfig
// (inside generateManifests above) so the patch operates on the
@ -412,6 +423,62 @@ class Installer {
}
}
async _readSkillManifestRows(bmadDir) {
const csvPath = path.join(bmadDir, '_config', 'skill-manifest.csv');
if (!(await fs.pathExists(csvPath))) return [];
try {
const csvParse = require('csv-parse/sync');
const content = await fs.readFile(csvPath, 'utf8');
return csvParse.parse(content, { columns: true, skip_empty_lines: true });
} catch (error) {
await prompts.log.warn(`Failed to parse skill-manifest.csv: ${error.message}`);
return [];
}
}
_getPreviousSkillIdsForCleanup(previousRows, preservedModules = []) {
const preservedModuleSet = new Set(preservedModules || []);
const ids = new Set();
for (const row of previousRows || []) {
if (row.canonicalId && !preservedModuleSet.has(row.module)) {
ids.add(row.canonicalId);
}
}
return ids;
}
async _appendPreservedSkillManifestRows(bmadDir, previousRows, preservedModules = []) {
if (!previousRows || previousRows.length === 0 || preservedModules.length === 0) return;
const preservedModuleSet = new Set(preservedModules);
const rowsToPreserve = previousRows.filter((row) => row.canonicalId && row.module && preservedModuleSet.has(row.module));
if (rowsToPreserve.length === 0) return;
const csvPath = path.join(bmadDir, '_config', 'skill-manifest.csv');
if (!(await fs.pathExists(csvPath))) return;
const currentRows = await this._readSkillManifestRows(bmadDir);
const activeIds = new Set(currentRows.map((row) => row.canonicalId).filter(Boolean));
const appendedRows = [];
for (const row of rowsToPreserve) {
if (activeIds.has(row.canonicalId)) continue;
activeIds.add(row.canonicalId);
appendedRows.push(
[row.canonicalId, row.name || row.canonicalId, row.description || '', row.module, row.path || '']
.map((field) => this.escapeCSVField(field))
.join(','),
);
}
if (appendedRows.length === 0) return;
const currentContent = await fs.readFile(csvPath, 'utf8');
const prefix = currentContent.endsWith('\n') ? currentContent : `${currentContent}\n`;
await fs.writeFile(csvPath, prefix + appendedRows.join('\n') + '\n', 'utf8');
}
/**
* Restore custom and modified files that were backed up before the update.
* No-op for fresh installs (updateState is null).
@ -598,6 +665,15 @@ class Installer {
}
}
async _trackPreservedModuleFiles(bmadDir, preservedModules = []) {
for (const moduleName of preservedModules) {
const modulePath = path.join(bmadDir, moduleName);
if (await fs.pathExists(modulePath)) {
await this._trackFilesRecursive(modulePath);
}
}
}
/**
* Install official (non-custom) modules.
* @param {Object} config - Installation configuration
@ -1270,11 +1346,6 @@ class Installer {
}
}
for (const moduleName of skippedModules) {
if (quickModules.collectedConfig[moduleName] || !quickModules.existingConfig?.[moduleName]) continue;
quickModules.collectedConfig[moduleName] = { ...quickModules.existingConfig[moduleName] };
}
if (!promptedForNewFields) {
await prompts.log.success('All configuration is up to date, no new options to configure');
}

View File

@ -435,14 +435,12 @@ class ManifestGenerator {
// this means user-scoped keys (e.g. user_name) could mis-file into the
// team config, so the operator should notice.
const scopeByModuleKey = {};
const unresolvedModuleNames = new Set();
// Maps installer moduleName (may be full display name) → module code field
// from module.yaml, so TOML sections use [modules.<code>] not [modules.<name>].
const codeByModuleName = {};
for (const moduleName of this.updatedModules) {
const moduleYamlPath = await resolveInstalledModuleYaml(moduleName);
if (!moduleYamlPath) {
unresolvedModuleNames.add(moduleName);
console.warn(
`[warn] writeCentralConfig: could not locate module.yaml for '${moduleName}'. ` +
`Answers from this module will default to team scope — user-scoped keys may mis-file into config.toml.`,
@ -460,7 +458,6 @@ class ManifestGenerator {
}
}
} catch (error) {
unresolvedModuleNames.add(moduleName);
console.warn(
`[warn] writeCentralConfig: could not parse module.yaml for '${moduleName}' (${error.message}). ` +
`Answers from this module will default to team scope — user-scoped keys may mis-file into config.toml.`,
@ -547,35 +544,9 @@ class ManifestGenerator {
userLines.push('');
}
const updatedModuleSet = new Set(this.updatedModules);
const preservedModuleBlocks = { team: [], user: [] };
const preservedModuleCodes = new Set();
if (updatedModuleSet.size > 0) {
const collectPreservedModuleBlocks = async (filePath, target) => {
if (!(await fs.pathExists(filePath))) return;
try {
const prev = await fs.readFile(filePath, 'utf8');
for (const block of extractModuleBlocks(prev)) {
if (!updatedModuleSet.has(block.code)) continue;
if (!unresolvedModuleNames.has(block.code) && moduleConfigs[block.code] && Object.keys(moduleConfigs[block.code]).length > 0)
continue;
preservedModuleCodes.add(block.code);
target.push(block.body);
}
} catch (error) {
console.warn(
`[warn] writeCentralConfig: could not read prior ${path.basename(filePath)} to preserve module config: ${error.message}`,
);
}
};
await collectPreservedModuleBlocks(teamPath, preservedModuleBlocks.team);
await collectPreservedModuleBlocks(userPath, preservedModuleBlocks.user);
}
// [modules.<code>] — split per module
for (const moduleName of this.updatedModules) {
if (moduleName === 'core') continue;
if (unresolvedModuleNames.has(moduleName) && preservedModuleCodes.has(moduleName)) continue;
const cfg = moduleConfigs[moduleName];
if (!cfg || Object.keys(cfg).length === 0) continue;
// Use the module's code field from module.yaml as the TOML key so the
@ -603,13 +574,6 @@ class ManifestGenerator {
}
}
for (const body of preservedModuleBlocks.team) {
teamLines.push(body, '');
}
for (const body of preservedModuleBlocks.user) {
userLines.push(body, '');
}
// [agents.<code>] — always team (agent roster is organizational).
// Freshly collected agents come from module.yaml this run. If a module
// was preserved (e.g. during quickUpdate when its source isn't available),
@ -892,26 +856,4 @@ function extractAgentBlocks(tomlContent) {
return blocks;
}
function extractModuleBlocks(tomlContent) {
const lines = tomlContent.split(/\r?\n/);
const blocks = [];
for (let i = 0; i < lines.length; i++) {
const startMatch = lines[i].match(/^\[modules\.([^\]]+)\]\s*$/);
if (!startMatch) continue;
const code = startMatch[1];
const bodyLines = [lines[i]];
i++;
while (i < lines.length && !/^\[[^\]]+\]\s*$/.test(lines[i])) {
bodyLines.push(lines[i]);
i++;
}
i--;
while (bodyLines.length > 0 && bodyLines.at(-1).trim() === '') {
bodyLines.pop();
}
blocks.push({ code, body: bodyLines.join('\n') });
}
return blocks;
}
module.exports = { ManifestGenerator };

View File

@ -501,7 +501,7 @@ class ConfigDrivenIdeSetup {
// Build removal set: previously installed skills + removals.txt entries
let removalSet;
if (options.previousSkillIds && options.previousSkillIds.size > 0) {
if (options.previousSkillIds) {
// Install/update flow: use pre-captured skill IDs (before manifest was overwritten)
removalSet = new Set(options.previousSkillIds);
if (resolvedBmadDir) {
@ -547,7 +547,7 @@ class ConfigDrivenIdeSetup {
// previousSkillIds — full uninstall or per-IDE removal via
// cleanupByList), don't spare anything; the IDE itself is going away,
// so its pointers should go with it.
const isInstallFlow = options.previousSkillIds && options.previousSkillIds.size > 0;
const isInstallFlow = !!options.previousSkillIds;
const activeSkillIds = isInstallFlow ? await this._readActiveSkillIds(resolvedBmadDir) : new Set();
const extension = this.installerConfig.commands_extension || '.md';
await this.cleanupCommandPointers(

View File

@ -7,7 +7,6 @@ const { CLIUtils } = require('./cli-utils');
const { ExternalModuleManager } = require('./modules/external-manager');
const { resolveModuleVersion } = require('./modules/version-resolver');
const { Manifest } = require('./core/manifest');
const { resolveInstalledModuleYaml } = require('./project-root');
const {
parseChannelOptions,
buildPlan,
@ -111,21 +110,42 @@ async function getModuleVersion(moduleCode, { repoUrl = null, registryDefault =
* UI utilities for the installer
*/
class UI {
async _findUnavailableInstalledModules(installedModuleIds = new Set(), bmadDir) {
const unavailable = [];
for (const moduleId of installedModuleIds) {
if (moduleId === 'core' || moduleId === 'bmm') continue;
const moduleYamlPath = await resolveInstalledModuleYaml(moduleId);
if (moduleYamlPath) continue;
async _retainUnavailableInstalledModules(selectedModules, installedModuleIds, bmadDir, options = {}) {
const { OfficialModules } = require('./modules/official-modules');
const officialCodes = new Set(['core']);
const { CustomModuleManager } = require('./modules/custom-module-manager');
const customMgr = new CustomModuleManager();
const localSource = await customMgr.findModuleSourceByCode(moduleId, { bmadDir });
if (localSource) continue;
unavailable.push(moduleId);
const builtInModules = (await new OfficialModules().listAvailable()).modules || [];
for (const mod of builtInModules) {
officialCodes.add(mod.id);
}
return unavailable;
const externalManager = new ExternalModuleManager();
const registryModules = await externalManager.listAvailable();
for (const mod of registryModules) {
officialCodes.add(mod.code);
}
const { CustomModuleManager } = require('./modules/custom-module-manager');
const customMgr = new CustomModuleManager();
const selectedSet = new Set(selectedModules);
const preserveModules = [];
for (const moduleId of installedModuleIds) {
if (moduleId === 'core') continue;
if (!selectedSet.has(moduleId) && !options.preserveUnselected) continue;
if (officialCodes.has(moduleId)) continue;
const customSource = await customMgr.findModuleSourceByCode(moduleId, { bmadDir });
if (!customSource) {
preserveModules.push(moduleId);
}
}
const preservedSet = new Set(preserveModules);
return {
selectedModules: selectedModules.filter((moduleId) => !preservedSet.has(moduleId)),
preserveModules,
};
}
/**
@ -291,12 +311,15 @@ class UI {
selectedModules.unshift('core');
}
const preservedModules = await this._findUnavailableInstalledModules(installedModuleIds, bmadDir);
const retainedModuleResult = await this._retainUnavailableInstalledModules(selectedModules, installedModuleIds, bmadDir, {
preserveUnselected: options.yes && !options.modules,
});
selectedModules = retainedModuleResult.selectedModules;
const preservedModules = retainedModuleResult.preserveModules;
if (preservedModules.length > 0) {
const preservedSet = new Set(preservedModules);
selectedModules = selectedModules.filter((moduleName) => !preservedSet.has(moduleName));
await prompts.log.warn(
`Retaining ${preservedModules.length} installed module(s) with no source available: ${preservedModules.join(', ')}`,
`Retaining ${preservedModules.length} installed module(s) with no available source: ${preservedModules.join(', ')}`,
);
}
@ -319,14 +342,6 @@ class UI {
...options,
channelOptions,
});
if (preservedModules.length > 0) {
const preservedConfigLoader = new (require('./modules/official-modules').OfficialModules)({ channelOptions });
await preservedConfigLoader.loadExistingConfig(confirmedDirectory);
for (const moduleName of preservedModules) {
if (moduleConfigs[moduleName] || !preservedConfigLoader.existingConfig?.[moduleName]) continue;
moduleConfigs[moduleName] = { ...preservedConfigLoader.existingConfig[moduleName] };
}
}
// Warn about --pin/--next flags that refer to modules the user didn't
// select, or that target bundled modules (core/bmm) where channel