mirror of
https://github.com/Azure/k8s-deploy.git
synced 2026-06-26 14:49:26 +08:00
Vidya reddy/prettier code (#203)
This commit is contained in:
Vidya
@@ -1,310 +1,313 @@
|
||||
import * as core from "@actions/core";
|
||||
import * as fs from "fs";
|
||||
import * as yaml from "js-yaml";
|
||||
import * as path from "path";
|
||||
import * as fileHelper from "./fileUtils";
|
||||
import { getTempDirectory } from "./fileUtils";
|
||||
import {
|
||||
InputObjectKindNotDefinedError,
|
||||
InputObjectMetadataNotDefinedError,
|
||||
isWorkloadEntity,
|
||||
KubernetesWorkload,
|
||||
NullInputObjectError,
|
||||
} from "../types/kubernetesTypes";
|
||||
import {
|
||||
getSpecSelectorLabels,
|
||||
setSpecSelectorLabels,
|
||||
} from "./manifestSpecLabelUtils";
|
||||
import {
|
||||
getImagePullSecrets,
|
||||
setImagePullSecrets,
|
||||
} from "./manifestPullSecretUtils";
|
||||
import { Resource } from "../types/kubectl";
|
||||
|
||||
export function updateManifestFiles(manifestFilePaths: string[]) {
|
||||
if (manifestFilePaths?.length === 0) {
|
||||
throw new Error("Manifest files not provided");
|
||||
}
|
||||
|
||||
// update container images
|
||||
const containers: string[] = core.getInput("images").split("\n");
|
||||
const manifestFiles = updateContainerImagesInManifestFiles(
|
||||
manifestFilePaths,
|
||||
containers
|
||||
);
|
||||
|
||||
// update pull secrets
|
||||
const imagePullSecrets: string[] = core
|
||||
.getInput("imagepullsecrets")
|
||||
.split("\n")
|
||||
.filter((secret) => secret.trim().length > 0);
|
||||
return updateImagePullSecretsInManifestFiles(manifestFiles, imagePullSecrets);
|
||||
}
|
||||
|
||||
export function UnsetClusterSpecificDetails(resource: any) {
|
||||
if (!resource) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Unset cluster specific details in the object
|
||||
if (!!resource) {
|
||||
const { metadata, status } = resource;
|
||||
|
||||
if (!!metadata) {
|
||||
resource.metadata = {
|
||||
annotations: metadata.annotations,
|
||||
labels: metadata.labels,
|
||||
name: metadata.name,
|
||||
};
|
||||
}
|
||||
|
||||
if (!!status) {
|
||||
resource.status = {};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function updateContainerImagesInManifestFiles(
|
||||
filePaths: string[],
|
||||
containers: string[]
|
||||
): string[] {
|
||||
if (filePaths?.length <= 0) return filePaths;
|
||||
|
||||
const newFilePaths = [];
|
||||
|
||||
// update container images
|
||||
filePaths.forEach((filePath: string) => {
|
||||
let contents = fs.readFileSync(filePath).toString();
|
||||
|
||||
containers.forEach((container: string) => {
|
||||
let [imageName] = container.split(":");
|
||||
if (imageName.indexOf("@") > 0) {
|
||||
imageName = imageName.split("@")[0];
|
||||
}
|
||||
|
||||
if (contents.indexOf(imageName) > 0)
|
||||
contents = substituteImageNameInSpecFile(
|
||||
contents,
|
||||
imageName,
|
||||
container
|
||||
);
|
||||
});
|
||||
|
||||
// write updated files
|
||||
const tempDirectory = getTempDirectory();
|
||||
const fileName = path.join(tempDirectory, path.basename(filePath));
|
||||
fs.writeFileSync(path.join(fileName), contents);
|
||||
newFilePaths.push(fileName);
|
||||
});
|
||||
|
||||
return newFilePaths;
|
||||
}
|
||||
|
||||
/*
|
||||
Example:
|
||||
|
||||
Input of
|
||||
currentString: `image: "example/example-image"`
|
||||
imageName: `example/example-image`
|
||||
imageNameWithNewTag: `example/example-image:identifiertag`
|
||||
|
||||
would return
|
||||
`image: "example/example-image:identifiertag"`
|
||||
*/
|
||||
export function substituteImageNameInSpecFile(
|
||||
spec: string,
|
||||
imageName: string,
|
||||
imageNameWithNewTag: string
|
||||
) {
|
||||
if (spec.indexOf(imageName) < 0) return spec;
|
||||
|
||||
return spec.split("\n").reduce((acc, line) => {
|
||||
const imageKeyword = line.match(/^ *-? *image:/);
|
||||
if (imageKeyword) {
|
||||
let [currentImageName] = line
|
||||
.substring(imageKeyword[0].length) // consume the line from keyword onwards
|
||||
.trim()
|
||||
.replace(/[',"]/g, "") // replace allowed quotes with nothing
|
||||
.split(":");
|
||||
|
||||
if (currentImageName?.indexOf(" ") > 0) {
|
||||
currentImageName = currentImageName.split(" ")[0]; // remove comments
|
||||
}
|
||||
|
||||
if (currentImageName === imageName) {
|
||||
return acc + `${imageKeyword[0]} ${imageNameWithNewTag}\n`;
|
||||
}
|
||||
}
|
||||
|
||||
return acc + line + "\n";
|
||||
}, "");
|
||||
}
|
||||
|
||||
export function getReplicaCount(inputObject: any): any {
|
||||
if (!inputObject) throw NullInputObjectError;
|
||||
|
||||
if (!inputObject.kind) {
|
||||
throw InputObjectKindNotDefinedError;
|
||||
}
|
||||
|
||||
const { kind } = inputObject;
|
||||
if (
|
||||
kind.toLowerCase() !== KubernetesWorkload.POD.toLowerCase() &&
|
||||
kind.toLowerCase() !== KubernetesWorkload.DAEMON_SET.toLowerCase()
|
||||
)
|
||||
return inputObject.spec.replicas;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
export function updateObjectLabels(
|
||||
inputObject: any,
|
||||
newLabels: Map<string, string>,
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError;
|
||||
|
||||
if (!inputObject.metadata) throw InputObjectMetadataNotDefinedError;
|
||||
|
||||
if (!newLabels) return;
|
||||
|
||||
if (override) {
|
||||
inputObject.metadata.labels = newLabels;
|
||||
} else {
|
||||
let existingLabels =
|
||||
inputObject.metadata.labels || new Map<string, string>();
|
||||
|
||||
Object.keys(newLabels).forEach(
|
||||
(key) => (existingLabels[key] = newLabels[key])
|
||||
);
|
||||
|
||||
inputObject.metadata.labels = existingLabels;
|
||||
}
|
||||
}
|
||||
|
||||
export function updateObjectAnnotations(
|
||||
inputObject: any,
|
||||
newAnnotations: Map<string, string>,
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError;
|
||||
|
||||
if (!inputObject.metadata) throw InputObjectMetadataNotDefinedError;
|
||||
|
||||
if (!newAnnotations) return;
|
||||
|
||||
if (override) {
|
||||
inputObject.metadata.annotations = newAnnotations;
|
||||
} else {
|
||||
const existingAnnotations =
|
||||
inputObject.metadata.annotations || new Map<string, string>();
|
||||
|
||||
Object.keys(newAnnotations).forEach(
|
||||
(key) => (existingAnnotations[key] = newAnnotations[key])
|
||||
);
|
||||
|
||||
inputObject.metadata.annotations = existingAnnotations;
|
||||
}
|
||||
}
|
||||
|
||||
export function updateImagePullSecrets(
|
||||
inputObject: any,
|
||||
newImagePullSecrets: string[],
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject?.spec || !newImagePullSecrets) return;
|
||||
|
||||
const newImagePullSecretsObjects = Array.from(newImagePullSecrets, (name) => {
|
||||
return { name };
|
||||
});
|
||||
let existingImagePullSecretObjects: any = getImagePullSecrets(inputObject);
|
||||
|
||||
if (override) {
|
||||
existingImagePullSecretObjects = newImagePullSecretsObjects;
|
||||
} else {
|
||||
existingImagePullSecretObjects = existingImagePullSecretObjects || [];
|
||||
|
||||
existingImagePullSecretObjects = existingImagePullSecretObjects.concat(
|
||||
newImagePullSecretsObjects
|
||||
);
|
||||
}
|
||||
|
||||
setImagePullSecrets(inputObject, existingImagePullSecretObjects);
|
||||
}
|
||||
|
||||
export function updateSelectorLabels(
|
||||
inputObject: any,
|
||||
newLabels: Map<string, string>,
|
||||
override: boolean
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError;
|
||||
|
||||
if (!inputObject.kind) throw InputObjectKindNotDefinedError;
|
||||
|
||||
if (!newLabels) return;
|
||||
|
||||
if (inputObject.kind.toLowerCase() === KubernetesWorkload.POD.toLowerCase())
|
||||
return;
|
||||
|
||||
let existingLabels = getSpecSelectorLabels(inputObject);
|
||||
if (override) {
|
||||
existingLabels = newLabels;
|
||||
} else {
|
||||
existingLabels = existingLabels || new Map<string, string>();
|
||||
Object.keys(newLabels).forEach(
|
||||
(key) => (existingLabels[key] = newLabels[key])
|
||||
);
|
||||
}
|
||||
|
||||
setSpecSelectorLabels(inputObject, existingLabels);
|
||||
}
|
||||
|
||||
export function getResources(
|
||||
filePaths: string[],
|
||||
filterResourceTypes: string[]
|
||||
): Resource[] {
|
||||
if (!filePaths) return [];
|
||||
|
||||
const resources: Resource[] = [];
|
||||
filePaths.forEach((filePath: string) => {
|
||||
const fileContents = fs.readFileSync(filePath).toString();
|
||||
yaml.safeLoadAll(fileContents, (inputObject) => {
|
||||
const inputObjectKind = inputObject?.kind || "";
|
||||
if (
|
||||
filterResourceTypes.filter(
|
||||
(type) => inputObjectKind.toLowerCase() === type.toLowerCase()
|
||||
).length > 0
|
||||
) {
|
||||
resources.push({
|
||||
type: inputObject.kind,
|
||||
name: inputObject.metadata.name,
|
||||
});
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
return resources;
|
||||
}
|
||||
|
||||
function updateImagePullSecretsInManifestFiles(
|
||||
filePaths: string[],
|
||||
imagePullSecrets: string[]
|
||||
): string[] {
|
||||
if (imagePullSecrets?.length <= 0) return filePaths;
|
||||
|
||||
const newObjectsList = [];
|
||||
filePaths.forEach((filePath: string) => {
|
||||
const fileContents = fs.readFileSync(filePath).toString();
|
||||
yaml.safeLoadAll(fileContents, (inputObject: any) => {
|
||||
if (inputObject?.kind) {
|
||||
const { kind } = inputObject;
|
||||
if (isWorkloadEntity(kind)) {
|
||||
updateImagePullSecrets(inputObject, imagePullSecrets);
|
||||
}
|
||||
newObjectsList.push(inputObject);
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
return fileHelper.writeObjectsToFile(newObjectsList);
|
||||
}
|
||||
import * as core from '@actions/core'
|
||||
import * as fs from 'fs'
|
||||
import * as yaml from 'js-yaml'
|
||||
import * as path from 'path'
|
||||
import * as fileHelper from './fileUtils'
|
||||
import {getTempDirectory} from './fileUtils'
|
||||
import {
|
||||
InputObjectKindNotDefinedError,
|
||||
InputObjectMetadataNotDefinedError,
|
||||
isWorkloadEntity,
|
||||
KubernetesWorkload,
|
||||
NullInputObjectError
|
||||
} from '../types/kubernetesTypes'
|
||||
import {
|
||||
getSpecSelectorLabels,
|
||||
setSpecSelectorLabels
|
||||
} from './manifestSpecLabelUtils'
|
||||
import {
|
||||
getImagePullSecrets,
|
||||
setImagePullSecrets
|
||||
} from './manifestPullSecretUtils'
|
||||
import {Resource} from '../types/kubectl'
|
||||
|
||||
export function updateManifestFiles(manifestFilePaths: string[]) {
|
||||
if (manifestFilePaths?.length === 0) {
|
||||
throw new Error('Manifest files not provided')
|
||||
}
|
||||
|
||||
// update container images
|
||||
const containers: string[] = core.getInput('images').split('\n')
|
||||
const manifestFiles = updateContainerImagesInManifestFiles(
|
||||
manifestFilePaths,
|
||||
containers
|
||||
)
|
||||
|
||||
// update pull secrets
|
||||
const imagePullSecrets: string[] = core
|
||||
.getInput('imagepullsecrets')
|
||||
.split('\n')
|
||||
.filter((secret) => secret.trim().length > 0)
|
||||
return updateImagePullSecretsInManifestFiles(manifestFiles, imagePullSecrets)
|
||||
}
|
||||
|
||||
export function UnsetClusterSpecificDetails(resource: any) {
|
||||
if (!resource) {
|
||||
return
|
||||
}
|
||||
|
||||
// Unset cluster specific details in the object
|
||||
if (!!resource) {
|
||||
const {metadata, status} = resource
|
||||
|
||||
if (!!metadata) {
|
||||
resource.metadata = {
|
||||
annotations: metadata.annotations,
|
||||
labels: metadata.labels,
|
||||
name: metadata.name
|
||||
}
|
||||
}
|
||||
|
||||
if (!!status) {
|
||||
resource.status = {}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
function updateContainerImagesInManifestFiles(
|
||||
filePaths: string[],
|
||||
containers: string[]
|
||||
): string[] {
|
||||
if (filePaths?.length <= 0) return filePaths
|
||||
|
||||
const newFilePaths = []
|
||||
|
||||
// update container images
|
||||
filePaths.forEach((filePath: string) => {
|
||||
let contents = fs.readFileSync(filePath).toString()
|
||||
|
||||
containers.forEach((container: string) => {
|
||||
let [imageName] = container.split(':')
|
||||
if (imageName.indexOf('@') > 0) {
|
||||
imageName = imageName.split('@')[0]
|
||||
}
|
||||
|
||||
if (contents.indexOf(imageName) > 0)
|
||||
contents = substituteImageNameInSpecFile(
|
||||
contents,
|
||||
imageName,
|
||||
container
|
||||
)
|
||||
})
|
||||
|
||||
// write updated files
|
||||
const tempDirectory = getTempDirectory()
|
||||
const fileName = path.join(tempDirectory, path.basename(filePath))
|
||||
fs.writeFileSync(path.join(fileName), contents)
|
||||
newFilePaths.push(fileName)
|
||||
})
|
||||
|
||||
return newFilePaths
|
||||
}
|
||||
|
||||
/*
|
||||
Example:
|
||||
|
||||
Input of
|
||||
currentString: `image: "example/example-image"`
|
||||
imageName: `example/example-image`
|
||||
imageNameWithNewTag: `example/example-image:identifiertag`
|
||||
|
||||
would return
|
||||
`image: "example/example-image:identifiertag"`
|
||||
*/
|
||||
export function substituteImageNameInSpecFile(
|
||||
spec: string,
|
||||
imageName: string,
|
||||
imageNameWithNewTag: string
|
||||
) {
|
||||
if (spec.indexOf(imageName) < 0) return spec
|
||||
|
||||
return spec.split('\n').reduce((acc, line) => {
|
||||
const imageKeyword = line.match(/^ *-? *image:/)
|
||||
if (imageKeyword) {
|
||||
let [currentImageName] = line
|
||||
.substring(imageKeyword[0].length) // consume the line from keyword onwards
|
||||
.trim()
|
||||
.replace(/[',"]/g, '') // replace allowed quotes with nothing
|
||||
.split(':')
|
||||
|
||||
if (currentImageName?.indexOf(' ') > 0) {
|
||||
currentImageName = currentImageName.split(' ')[0] // remove comments
|
||||
}
|
||||
|
||||
if (currentImageName === imageName) {
|
||||
return acc + `${imageKeyword[0]} ${imageNameWithNewTag}\n`
|
||||
}
|
||||
}
|
||||
|
||||
return acc + line + '\n'
|
||||
}, '')
|
||||
}
|
||||
|
||||
export function getReplicaCount(inputObject: any): any {
|
||||
if (!inputObject) throw NullInputObjectError
|
||||
|
||||
if (!inputObject.kind) {
|
||||
throw InputObjectKindNotDefinedError
|
||||
}
|
||||
|
||||
const {kind} = inputObject
|
||||
if (
|
||||
kind.toLowerCase() !== KubernetesWorkload.POD.toLowerCase() &&
|
||||
kind.toLowerCase() !== KubernetesWorkload.DAEMON_SET.toLowerCase()
|
||||
)
|
||||
return inputObject.spec.replicas
|
||||
|
||||
return 0
|
||||
}
|
||||
|
||||
export function updateObjectLabels(
|
||||
inputObject: any,
|
||||
newLabels: Map<string, string>,
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError
|
||||
|
||||
if (!inputObject.metadata) throw InputObjectMetadataNotDefinedError
|
||||
|
||||
if (!newLabels) return
|
||||
|
||||
if (override) {
|
||||
inputObject.metadata.labels = newLabels
|
||||
} else {
|
||||
let existingLabels =
|
||||
inputObject.metadata.labels || new Map<string, string>()
|
||||
|
||||
Object.keys(newLabels).forEach(
|
||||
(key) => (existingLabels[key] = newLabels[key])
|
||||
)
|
||||
|
||||
inputObject.metadata.labels = existingLabels
|
||||
}
|
||||
}
|
||||
|
||||
export function updateObjectAnnotations(
|
||||
inputObject: any,
|
||||
newAnnotations: Map<string, string>,
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError
|
||||
|
||||
if (!inputObject.metadata) throw InputObjectMetadataNotDefinedError
|
||||
|
||||
if (!newAnnotations) return
|
||||
|
||||
if (override) {
|
||||
inputObject.metadata.annotations = newAnnotations
|
||||
} else {
|
||||
const existingAnnotations =
|
||||
inputObject.metadata.annotations || new Map<string, string>()
|
||||
|
||||
Object.keys(newAnnotations).forEach(
|
||||
(key) => (existingAnnotations[key] = newAnnotations[key])
|
||||
)
|
||||
|
||||
inputObject.metadata.annotations = existingAnnotations
|
||||
}
|
||||
}
|
||||
|
||||
export function updateImagePullSecrets(
|
||||
inputObject: any,
|
||||
newImagePullSecrets: string[],
|
||||
override: boolean = false
|
||||
) {
|
||||
if (!inputObject?.spec || !newImagePullSecrets) return
|
||||
|
||||
const newImagePullSecretsObjects = Array.from(
|
||||
newImagePullSecrets,
|
||||
(name) => {
|
||||
return {name}
|
||||
}
|
||||
)
|
||||
let existingImagePullSecretObjects: any = getImagePullSecrets(inputObject)
|
||||
|
||||
if (override) {
|
||||
existingImagePullSecretObjects = newImagePullSecretsObjects
|
||||
} else {
|
||||
existingImagePullSecretObjects = existingImagePullSecretObjects || []
|
||||
|
||||
existingImagePullSecretObjects = existingImagePullSecretObjects.concat(
|
||||
newImagePullSecretsObjects
|
||||
)
|
||||
}
|
||||
|
||||
setImagePullSecrets(inputObject, existingImagePullSecretObjects)
|
||||
}
|
||||
|
||||
export function updateSelectorLabels(
|
||||
inputObject: any,
|
||||
newLabels: Map<string, string>,
|
||||
override: boolean
|
||||
) {
|
||||
if (!inputObject) throw NullInputObjectError
|
||||
|
||||
if (!inputObject.kind) throw InputObjectKindNotDefinedError
|
||||
|
||||
if (!newLabels) return
|
||||
|
||||
if (inputObject.kind.toLowerCase() === KubernetesWorkload.POD.toLowerCase())
|
||||
return
|
||||
|
||||
let existingLabels = getSpecSelectorLabels(inputObject)
|
||||
if (override) {
|
||||
existingLabels = newLabels
|
||||
} else {
|
||||
existingLabels = existingLabels || new Map<string, string>()
|
||||
Object.keys(newLabels).forEach(
|
||||
(key) => (existingLabels[key] = newLabels[key])
|
||||
)
|
||||
}
|
||||
|
||||
setSpecSelectorLabels(inputObject, existingLabels)
|
||||
}
|
||||
|
||||
export function getResources(
|
||||
filePaths: string[],
|
||||
filterResourceTypes: string[]
|
||||
): Resource[] {
|
||||
if (!filePaths) return []
|
||||
|
||||
const resources: Resource[] = []
|
||||
filePaths.forEach((filePath: string) => {
|
||||
const fileContents = fs.readFileSync(filePath).toString()
|
||||
yaml.safeLoadAll(fileContents, (inputObject) => {
|
||||
const inputObjectKind = inputObject?.kind || ''
|
||||
if (
|
||||
filterResourceTypes.filter(
|
||||
(type) => inputObjectKind.toLowerCase() === type.toLowerCase()
|
||||
).length > 0
|
||||
) {
|
||||
resources.push({
|
||||
type: inputObject.kind,
|
||||
name: inputObject.metadata.name
|
||||
})
|
||||
}
|
||||
})
|
||||
})
|
||||
|
||||
return resources
|
||||
}
|
||||
|
||||
function updateImagePullSecretsInManifestFiles(
|
||||
filePaths: string[],
|
||||
imagePullSecrets: string[]
|
||||
): string[] {
|
||||
if (imagePullSecrets?.length <= 0) return filePaths
|
||||
|
||||
const newObjectsList = []
|
||||
filePaths.forEach((filePath: string) => {
|
||||
const fileContents = fs.readFileSync(filePath).toString()
|
||||
yaml.safeLoadAll(fileContents, (inputObject: any) => {
|
||||
if (inputObject?.kind) {
|
||||
const {kind} = inputObject
|
||||
if (isWorkloadEntity(kind)) {
|
||||
updateImagePullSecrets(inputObject, imagePullSecrets)
|
||||
}
|
||||
newObjectsList.push(inputObject)
|
||||
}
|
||||
})
|
||||
})
|
||||
|
||||
return fileHelper.writeObjectsToFile(newObjectsList)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user