Space-Banane
97 lines
2.8 KiB
TypeScript
97 lines
2.8 KiB
TypeScript
import chalk from "chalk";
|
||
import { getApiClient } from "../../api.js";
|
||
|
||
export const updateFunctionDefinition = {
|
||
name: "add",
|
||
description:
|
||
"Adds or updates environment variables for a serverless function.",
|
||
options: [
|
||
{ name: "--id <id>", description: "Function ID", required: true },
|
||
{
|
||
name: "--name <name>",
|
||
description: "Environment variable name",
|
||
required: true,
|
||
},
|
||
{
|
||
name: "--value <value>",
|
||
description: "Environment variable value",
|
||
required: true,
|
||
},
|
||
],
|
||
action: async (options: any) => {
|
||
if (!options.id || !options.name || !options.value) {
|
||
console.error(
|
||
`${chalk.red("✗")} Error: Missing required options (--id, --name, --value)`,
|
||
);
|
||
return;
|
||
}
|
||
|
||
const client = await getApiClient();
|
||
let existingEnv: { name: string; value: string }[] = [];
|
||
|
||
try {
|
||
const response = await client.get(`/api/function/${options.id}`);
|
||
const functionData = response.data.data || response.data;
|
||
existingEnv = functionData.env || [];
|
||
|
||
if (typeof existingEnv === 'string') {
|
||
try {
|
||
existingEnv = JSON.parse(existingEnv);
|
||
} catch (e: any) {
|
||
existingEnv = [];
|
||
}
|
||
}
|
||
if (!Array.isArray(existingEnv)) {
|
||
existingEnv = [];
|
||
}
|
||
} catch (error: any) {
|
||
console.error(
|
||
`${chalk.red("✗")} Error fetching function data: ${error.message}`,
|
||
);
|
||
return;
|
||
}
|
||
|
||
const newEnv = [...existingEnv];
|
||
const existingIndex = newEnv.findIndex((env) => env.name === options.name);
|
||
const isUpdate = existingIndex !== -1;
|
||
|
||
if (isUpdate) {
|
||
newEnv[existingIndex].value = options.value;
|
||
} else {
|
||
newEnv.push({ name: options.name, value: options.value });
|
||
}
|
||
|
||
const totalCount = newEnv.length;
|
||
|
||
try {
|
||
const response = await client.patch(`/api/function/${options.id}`, {
|
||
environment: newEnv,
|
||
});
|
||
|
||
if (response.status === 200) {
|
||
const actionText = isUpdate ? "updated" : "added";
|
||
console.log(
|
||
`${chalk.green("✓")} Environment variable ${chalk.cyan(options.name)} ${actionText} successfully for function ${chalk.cyan(options.id)}.`,
|
||
);
|
||
console.log(
|
||
`${chalk.blue("ℹ")} Total environment variables: ${totalCount}`,
|
||
);
|
||
} else {
|
||
console.error(
|
||
`${chalk.red("✗")} Unexpected response from server: ${response.status}`,
|
||
);
|
||
}
|
||
} catch (error: any) {
|
||
if (error.response?.status === 404) {
|
||
console.error(
|
||
`${chalk.red("✗")} Function with ID ${chalk.yellow(options.id)} not found.`,
|
||
);
|
||
} else {
|
||
console.error(
|
||
`${chalk.red("✗")} Failed to update: ${chalk.yellow(error.response?.data?.message || error.message)}`,
|
||
);
|
||
}
|
||
}
|
||
},
|
||
};
|