fix(auto-replace): use regex replace to avoid infinite loops

This commit is contained in:
Rhys Arkins 2020-03-01 20:52:43 +01:00
parent 435096d97e
commit 6f46153e09
2 changed files with 27 additions and 22 deletions

View file

@ -2,6 +2,7 @@ import { logger } from '../../logger';
import { get } from '../../manager';
import { WORKER_FILE_UPDATE_FAILED } from '../../constants/error-messages';
import { matchAt, replaceAt } from '../../util/string';
import { regEx } from '../../util/regex';
export async function confirmIfDepUpdated(
upgrade,
@ -59,6 +60,10 @@ export async function checkBranchDepsMatchBaseDeps(
}
}
function escapeRegExp(input: string): string {
return input.replace(/[.*+\-?^${}()|[\]\\]/g, '\\$&'); // $& means the whole matched string
}
export async function doAutoReplace(
upgrade,
existingContent: string,
@ -77,27 +82,33 @@ export async function doAutoReplace(
return existingContent;
}
const {
depName,
currentValue,
newValue,
currentDigest,
newDigest,
autoReplaceData,
autoReplaceData = {},
} = upgrade;
const replaceString = autoReplaceData.replaceString || currentValue;
logger.trace({ depName, replaceString }, 'autoReplace replaceString');
let searchIndex = existingContent.indexOf(replaceString);
if (searchIndex === -1) {
logger.error(
{ depName },
'Cannot find replaceString in current file content'
);
throw new Error(WORKER_FILE_UPDATE_FAILED);
}
try {
let newString = replaceString;
do {
newString = newString.replace(currentValue, newValue);
} while (newString.includes(currentValue));
if (currentDigest) {
do {
newString = newString.replace(currentDigest, newDigest);
} while (newString.includes(currentDigest));
}
let searchIndex = existingContent.indexOf(replaceString);
if (searchIndex === -1) {
logger.error('Cannot find replaceString in current file content');
throw new Error(WORKER_FILE_UPDATE_FAILED);
let newString = replaceString.replace(
regEx(escapeRegExp(currentValue), 'g'),
newValue
);
if (currentDigest && newDigest) {
newString = newString.replace(
regEx(escapeRegExp(currentDigest), 'g'),
newDigest
);
}
logger.debug(`Starting search at index ${searchIndex}`);
// Iterate through the rest of the file
@ -117,9 +128,9 @@ export async function doAutoReplace(
}
}
}
} catch (err) {
} catch (err) /* istanbul ignore next */ {
logger.debug({ err }, 'doAutoReplace error');
}
logger.error('Could not autoReplace');
// istanbul ignore next
throw new Error(WORKER_FILE_UPDATE_FAILED);
}

View file

@ -21,11 +21,6 @@ describe('workers/branch/auto-replace', () => {
};
parentBranch = undefined;
});
it('throws on error', async () => {
await expect(
doAutoReplace(upgrade, 'existing content', parentBranch)
).rejects.toThrow();
});
it('rebases if the deps list has changed', async () => {
upgrade.baseDeps = extractPackageFile(sampleHtml).deps;
parentBranch = 'some existing branch';
@ -54,7 +49,6 @@ describe('workers/branch/auto-replace', () => {
upgrade.newValue = '7.1.1';
upgrade.autoReplaceData = {
depIndex: 0,
replaceString: script,
};
const res = await doAutoReplace(upgrade, src, parentBranch);
expect(res).toMatchSnapshot();