Skip to content

Commit

Permalink
Remove parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
criamico committed May 16, 2022
1 parent 555a1df commit 3a374ca
Showing 1 changed file with 7 additions and 19 deletions.
26 changes: 7 additions & 19 deletions x-pack/plugins/fleet/server/routes/agent/upgrade_handler.ts
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ export const postAgentUpgradeHandler: RequestHandler<
const { version, source_uri: sourceUri, force } = request.body;
const kibanaVersion = appContextService.getKibanaVersion();
try {
checkKibanaVersion(version, kibanaVersion, true);
checkKibanaVersion(version, kibanaVersion);
checkSourceUriAllowed(sourceUri);
} catch (err) {
return response.customError({
Expand Down Expand Up @@ -99,7 +99,7 @@ export const postBulkAgentsUpgradeHandler: RequestHandler<
} = request.body;
const kibanaVersion = appContextService.getKibanaVersion();
try {
checkKibanaVersion(version, kibanaVersion, false);
checkKibanaVersion(version, kibanaVersion);
checkSourceUriAllowed(sourceUri);
await checkFleetServerVersion(version, agents, soClient, esClient);
} catch (err) {
Expand Down Expand Up @@ -135,30 +135,18 @@ export const postBulkAgentsUpgradeHandler: RequestHandler<
}
};

export const checkKibanaVersion = (
version: string,
kibanaVersion: string,
shouldBeSame: boolean
) => {
export const checkKibanaVersion = (version: string, kibanaVersion: string) => {
// get version number only in case "-SNAPSHOT" is in it
const kibanaVersionNumber = semverCoerce(kibanaVersion)?.version;
if (!kibanaVersionNumber) throw new Error(`kibanaVersion ${kibanaVersionNumber} is not valid`);
const versionToUpgradeNumber = semverCoerce(version)?.version;
if (!versionToUpgradeNumber)
throw new Error(`version to upgrade ${versionToUpgradeNumber} is not valid`);

if (shouldBeSame) {
// only allow upgrading to the same version as the installed kibana version
if (kibanaVersionNumber !== versionToUpgradeNumber)
throw new Error(
`cannot upgrade agent to ${versionToUpgradeNumber} because it is different than the installed kibana version ${kibanaVersionNumber}`
);
} else {
if (semverGt(version, kibanaVersion))
throw new Error(
`cannot upgrade agent to ${versionToUpgradeNumber} because it is higher than the installed kibana version ${kibanaVersionNumber}`
);
}
if (semverGt(version, kibanaVersion))
throw new Error(
`cannot upgrade agent to ${versionToUpgradeNumber} because it is higher than the installed kibana version ${kibanaVersionNumber}`
);
};

const checkSourceUriAllowed = (sourceUri?: string) => {
Expand Down

0 comments on commit 3a374ca

Please sign in to comment.