feat: change S3 node client to opendal. 😎

This commit is contained in:
Yufan Sheng 2024-06-21 02:21:43 +08:00
parent 2ddb050dd2
commit 11e9972df6
Signed by: syhily
GPG Key ID: 9D18A22A7DCD5A9B
5 changed files with 203 additions and 1804 deletions

View File

@ -91,6 +91,7 @@
"nofollow",
"nopd",
"noto",
"opendal",
"oppo",
"opposans",
"pandiyan",

View File

@ -1,4 +1,4 @@
FROM node:lts-alpine AS base
FROM node:lts AS base
WORKDIR /app
COPY package.json package-lock.json ./

1868
package-lock.json generated

File diff suppressed because it is too large Load Diff

View File

@ -56,7 +56,6 @@
},
"devDependencies": {
"@astrojs/check": "^0.7.0",
"@aws-sdk/client-s3": "^3.600.0",
"@biomejs/biome": "^1.8.2",
"@napi-rs/canvas": "^0.1.53",
"@types/lodash": "^4.17.5",
@ -67,6 +66,7 @@
"@types/unist": "^3.0.2",
"aplayer": "^1.10.1",
"bootstrap": "^5.3.3",
"opendal": "^0.46.1",
"prettier": "^3.3.2",
"prettier-plugin-astro": "^0.14.0",
"prettier-plugin-astro-organize-imports": "^0.4.8",

View File

@ -1,60 +1,78 @@
import { HeadObjectCommand, NotFound, PutObjectCommand, S3Client } from '@aws-sdk/client-s3';
import type { Logger } from '@smithy/types';
import type { AstroIntegration, AstroIntegrationLogger } from 'astro';
import { z } from 'astro/zod';
import fs from 'node:fs';
import path from 'node:path';
import { Operator } from 'opendal';
import { rimrafSync } from 'rimraf';
const S3Options = z.object({
const S3Options = z
.object({
// The directories that you want to upload to S3.
paths: z.array(z.string()).min(1),
region: z.string().min(1).default('us-east-1'),
// The S3 region, set it if you use AWS S3 service.
region: z.string().min(1).default('auto'),
// The endpoint, set it if you use 3rd-party S3 service.
endpoint: z.string().url().optional(),
// The name of the bucket.
bucket: z.string().min(1),
// The root directory you want to upload files.
root: z.string().default('/'),
// The access key id.
accessKey: z.string().min(1),
// The secret access key.
secretAccessKey: z.string().min(1),
});
// The extra options provided by opendal.
// All the methods in https://docs.rs/opendal/latest/opendal/services/struct.S3.html#implementations can be treated as an option.
extraOptions: z.record(z.string(), z.string()).default({}),
})
.strict()
.refine((opts) => (opts.region === 'auto' ? opts.endpoint !== undefined : true));
const parseOptions = (
opts: z.input<typeof S3Options>,
logger: AstroIntegrationLogger,
): { options: Record<string, string>; paths: string[] } => {
try {
const { paths, bucket, root, accessKey, secretAccessKey, region, endpoint, extraOptions } = S3Options.parse(opts);
// Create opendal operator.
// The common configurations are listed here https://docs.rs/opendal/latest/opendal/services/struct.S3.html#configuration
const options: Record<string, string> = {
...extraOptions,
root: root,
bucket: bucket,
region: region,
access_key_id: accessKey,
secret_access_key: secretAccessKey,
};
if (endpoint !== undefined) {
options.endpoint = endpoint;
}
return { options, paths };
} catch (err) {
if (err instanceof z.ZodError) {
logger.error(`Uploader options validation error, there are ${err.issues.length} errors:`);
for (const issue of err.issues) {
logger.error(issue.message);
}
}
throw err;
}
};
export const uploader = (opts: z.input<typeof S3Options>): AstroIntegration => ({
name: 'S3 Uploader',
hooks: {
'astro:build:done': async ({ dir, logger }: { dir: URL; logger: AstroIntegrationLogger }) => {
const { paths, bucket, accessKey, secretAccessKey, region, endpoint } = S3Options.parse(opts);
// Create S3 Client.
const clientLogger = (): Logger => {
const s3Logger = logger.fork('S3 Client');
return {
// biome-ignore lint/suspicious/noExplicitAny: It's define by external types.
debug: (...content: any[]): void => {
s3Logger.debug(content.join(' '));
},
// biome-ignore lint/suspicious/noExplicitAny: It's define by external types.
info: (...content: any[]): void => {
s3Logger.info(content.join(' '));
},
// biome-ignore lint/suspicious/noExplicitAny: It's define by external types.
warn: (...content: any[]): void => {
s3Logger.warn(content.join(' '));
},
// biome-ignore lint/suspicious/noExplicitAny: It's define by external types.
error: (...content: any[]): void => {
s3Logger.error(content.join(' '));
},
};
};
const client = new S3Client({
region: region,
endpoint: endpoint,
logger: clientLogger(),
credentials: { accessKeyId: accessKey, secretAccessKey: secretAccessKey },
useGlobalEndpoint: endpoint === '' || endpoint === '',
});
const { options, paths } = parseOptions(opts, logger);
const operator = new Operator('s3', options);
logger.info(`Start to upload static files in dir ${paths} to S3 compatible backend.`);
for (const current of paths) {
await uploadFile(client, logger, bucket, current, dir.pathname);
await uploadFile(operator, logger, current, dir.pathname);
rimrafSync(path.join(dir.pathname, current));
}
@ -68,32 +86,26 @@ const normalizePath = (current: string): string => {
return current.includes(path.win32.sep) ? current.split(path.win32.sep).join(path.posix.sep) : current;
};
const uploadFile = async (
client: S3Client,
logger: AstroIntegrationLogger,
bucket: string,
current: string,
root: string,
) => {
const uploadFile = async (operator: Operator, logger: AstroIntegrationLogger, current: string, root: string) => {
const filePath = path.join(root, current);
const isFile = !fs.statSync(filePath).isDirectory();
const uploadAction = async (key: string) => {
logger.info(`Start to upload file: ${key}`);
const body = fs.readFileSync(filePath);
await operator.write(key, body);
};
if (isFile) {
const key = normalizePath(current);
const headCmd = new HeadObjectCommand({ Bucket: bucket, Key: key });
try {
await client.send(headCmd);
const meta = await operator.stat(key);
if (meta.isFile()) {
logger.info(`${key} exists on backend, skip.`);
} catch (error) {
if (error instanceof NotFound) {
logger.info(`Start to upload file: ${key}`);
const body = fs.readFileSync(filePath);
const putCmd = new PutObjectCommand({ Bucket: bucket, Key: key, Body: body });
await client.send(putCmd);
} else {
throw error;
await uploadAction(key);
}
} catch (error) {
await uploadAction(key);
}
return;
}
@ -103,6 +115,6 @@ const uploadFile = async (
if (next.startsWith('.')) {
continue;
}
await uploadFile(client, logger, bucket, path.join(current, next), root);
await uploadFile(operator, logger, path.join(current, next), root);
}
};