wrangler で cron 実行の Cloudflare Workers を作る

name = "delete-old-deploy"
main = "src/index.ts"
compatibility_date = "2023-08-07"
workers_dev = false

[vars]
PROJECT_NAMES=["xxxx", "yyyy"]

[triggers]
crons = ["8 1 3 * *"]

/**
 * @link https://developers.cloudflare.com/api/operations/pages-deployment-get-deployments
 */
interface Deployment {
	id: string;
	created_on: string;
}
export interface Deployments {
	result: Deployment[];
	success: boolean;
	errors: {
		code: number;
		message: string;
	}[];
}

const API_ROOT = 'https://api.cloudflare.com/client/v4/accounts';
const EXPIRATION_DAYS = 30;

export async function fetchDeployments({ accountId, projectName, token }: { accountId: string; projectName: string; token: string }) {
	const endPoint = `${API_ROOT}/${accountId}/pages/projects/${projectName}/deployments`;
	const response = await fetch(endPoint, {
		headers: {
			'Content-Type': 'application/json;charset=UTF-8',
			Authorization: `Bearer ${token}`,
		},
	});
	const deployments: Deployments = await response.json();

	if (!deployments.success) {
		throw new Error(`error: ${deployments.errors[0].message}`);
	}
	return deployments.result;
}

export async function deleteDeployment({
	deployment,
	accountId,
	projectName,
	token,
}: {
	deployment: Deployment;
	accountId: string;
	projectName: string;
	token: string;
}) {
	if ((Date.now() - new Date(deployment.created_on).getTime()) / 86400000 <= EXPIRATION_DAYS) {
		return;
	}

	await fetch(`${API_ROOT}/${accountId}/pages/projects/${projectName}/deployments/${deployment.id}`, {
		method: 'DELETE',
		headers: {
			'Content-Type': 'application/json;charset=UTF-8',
			Authorization: `Bearer ${token}`,
		},
	});
}

import { deleteDeployment, fetchDeployments } from './pages';

export interface Env {
	ACCOUNT_ID: string;
	PROJECT_NAMES: string[];
	API_TOKEN: string;
}

export default {
	async scheduled(_: Request, env: Env): Promise<void> {
		await Promise.all(
			env.PROJECT_NAMES.map(async (projectName) => {
				const deployments = await fetchDeployments({
					accountId: env.ACCOUNT_ID,
					projectName: projectName,
					token: env.API_TOKEN,
				});
				await Promise.all(
					deployments
						.flat()
						.map((deployment) =>
							deleteDeployment({ deployment, accountId: env.ACCOUNT_ID, projectName: projectName, token: env.API_TOKEN }),
						),
				);
			}),
		);
	},
};

curl "http://localhost:8787/__scheduled?cron=*+*+*+*+*"

import { rest } from 'msw';
import { setupServer } from 'msw/node';
import type { Deployments } from '../../src/pages';

export const deployments: Deployments = {
	result: [
		{
			id: 'dummy-1',
			created_on: '2021-04-09T00:00:00.000000Z',
		},
		{
			id: 'dummy-2',
			created_on: '2021-04-10T00:00:00.000000Z',
		},
		{
			id: 'dummy-3',
			created_on: '2021-04-11T00:00:00.000000Z',
		},
	],
	success: true,
	errors: [],
};

// This configures a Service Worker with the given request handlers.
export const server = setupServer(
	rest.get('https://api.cloudflare.com/client/v4/accounts/*/pages/projects/*/deployments', (_, res, ctx) => {
		return res(ctx.status(200), ctx.json(deployments));
	}),
);

import { afterAll, afterEach, beforeAll } from 'vitest';
import { server } from './mocks/server';

beforeAll(() => server.listen({ onUnhandledRequest: 'error' }));
afterAll(() => server.close());
afterEach(() => server.resetHandlers());

import { defineConfig } from 'vite';

export default defineConfig({
	test: {
		setupFiles: ['./test/setup.ts'],
	},
});

import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
import { fetchDeployments, deleteDeployment } from '../src/pages';
import { rest } from 'msw';
import { deployments, server } from './mocks/server';

describe('pages', () => {
	describe('fetchDeployments', () => {
		it('deployments を取得できる', async () => {
			const result = await fetchDeployments({
				accountId: 'test',
				projectName: 'test',
				token: 'test',
			});
			expect(result.length).toBe(3);
			expect(result[0].id).toBe('dummy-1');
			expect(result[1].id).toBe('dummy-2');
			expect(result[2].id).toBe('dummy-3');
		});
	});
	describe('deleteDeployment', () => {
		const mockFn = vi.fn();

		beforeEach(() => {
			vi.useFakeTimers();
			server.use(
				rest.delete('https://api.cloudflare.com/client/v4/accounts/*/pages/projects/*/deployments/*', (_, res, ctx) => {
					mockFn();
					return res(ctx.status(200));
				}),
			);
		});

		afterEach(() => {
			vi.useRealTimers();
		});

		it('30日より前のデプロイを削除できる', async () => {
			vi.setSystemTime(new Date(2021, 4, 11, 0, 0, 0));
			await Promise.all(
				deployments.result.map((deployment) => deleteDeployment({ deployment, accountId: 'test', projectName: 'test', token: 'test' })),
			);
			expect(mockFn).toHaveBeenCalledTimes(2);
		});
	});
});

npm run test

npm wrangler login

npm wrangler secret put ACCOUNT_ID
npm wrangler secret put API_TOKEN

npm run deploy