Add rate-limit to vercel sync fns

This commit is contained in:
carlosmonastyrski
2025-04-09 12:36:43 -03:00
parent d01cb282f9
commit ed5c18b5ac
@@ -11,7 +11,14 @@ function isVercelDefaultEnvType(value: string): value is DefaultVercelEnvType {
return Object.values(VercelEnvironmentType).map(String).includes(value); return Object.values(VercelEnvironmentType).map(String).includes(value);
} }
const getVercelSecrets = async (secretSync: TVercelSyncWithCredentials) => { const MAX_RETRIES = 5;
const sleep = async () =>
new Promise((resolve) => {
setTimeout(resolve, 1000);
});
const getVercelSecrets = async (secretSync: TVercelSyncWithCredentials, attempt = 0): Promise<VercelApiSecret[]> => {
const { const {
destinationConfig, destinationConfig,
connection: { connection: {
@@ -23,7 +30,7 @@ const getVercelSecrets = async (secretSync: TVercelSyncWithCredentials) => {
decrypt: "true", decrypt: "true",
...(destinationConfig.branch ? { gitBranch: destinationConfig.branch } : {}) ...(destinationConfig.branch ? { gitBranch: destinationConfig.branch } : {})
}; };
try {
const { data } = await request.get<{ envs: VercelApiSecret[] }>( const { data } = await request.get<{ envs: VercelApiSecret[] }>(
`${IntegrationUrls.VERCEL_API_URL}/v9/projects/${destinationConfig.app}/env?teamId=${destinationConfig.teamId}`, `${IntegrationUrls.VERCEL_API_URL}/v9/projects/${destinationConfig.app}/env?teamId=${destinationConfig.teamId}`,
{ {
@@ -78,9 +85,20 @@ const getVercelSecrets = async (secretSync: TVercelSyncWithCredentials) => {
); );
return secretsWithValues; return secretsWithValues;
} catch (error) {
if ((error as { code: string }).code === "rate_limited" && attempt < MAX_RETRIES) {
await sleep();
return await getVercelSecrets(secretSync, attempt + 1);
}
throw error;
}
}; };
const deleteSecret = async (secretSync: TVercelSyncWithCredentials, vercelSecret: VercelApiSecret) => { const deleteSecret = async (
secretSync: TVercelSyncWithCredentials,
vercelSecret: VercelApiSecret,
attempt = 0
): Promise<void> => {
const { const {
destinationConfig, destinationConfig,
connection: { connection: {
@@ -99,6 +117,10 @@ const deleteSecret = async (secretSync: TVercelSyncWithCredentials, vercelSecret
} }
); );
} catch (error) { } catch (error) {
if ((error as { code: string }).code === "rate_limited" && attempt < MAX_RETRIES) {
await sleep();
return await deleteSecret(secretSync, vercelSecret, attempt + 1);
}
throw new SecretSyncError({ throw new SecretSyncError({
error, error,
secretKey: vercelSecret.key secretKey: vercelSecret.key
@@ -106,7 +128,12 @@ const deleteSecret = async (secretSync: TVercelSyncWithCredentials, vercelSecret
} }
}; };
const createSecret = async (secretSync: TVercelSyncWithCredentials, secretMap: TSecretMap, key: string) => { const createSecret = async (
secretSync: TVercelSyncWithCredentials,
secretMap: TSecretMap,
key: string,
attempt = 0
): Promise<void> => {
try { try {
const { const {
destinationConfig, destinationConfig,
@@ -135,6 +162,10 @@ const createSecret = async (secretSync: TVercelSyncWithCredentials, secretMap: T
} }
); );
} catch (error) { } catch (error) {
if ((error as { code: string }).code === "rate_limited" && attempt < MAX_RETRIES) {
await sleep();
return await createSecret(secretSync, secretMap, key, attempt + 1);
}
throw new SecretSyncError({ throw new SecretSyncError({
error, error,
secretKey: key secretKey: key
@@ -145,8 +176,9 @@ const createSecret = async (secretSync: TVercelSyncWithCredentials, secretMap: T
const updateSecret = async ( const updateSecret = async (
secretSync: TVercelSyncWithCredentials, secretSync: TVercelSyncWithCredentials,
secretMap: TSecretMap, secretMap: TSecretMap,
vercelSecret: VercelApiSecret vercelSecret: VercelApiSecret,
) => { attempt = 0
): Promise<void> => {
try { try {
const { const {
destinationConfig, destinationConfig,
@@ -187,6 +219,10 @@ const updateSecret = async (
} }
); );
} catch (error) { } catch (error) {
if ((error as { code: string }).code === "rate_limited" && attempt < MAX_RETRIES) {
await sleep();
return await updateSecret(secretSync, secretMap, vercelSecret, attempt + 1);
}
throw new SecretSyncError({ throw new SecretSyncError({
error, error,
secretKey: vercelSecret.key secretKey: vercelSecret.key