diff options
author | Madeline <46743919+MaddyUnderStars@users.noreply.github.com> | 2022-08-22 22:16:11 +1000 |
---|---|---|
committer | Madeline <46743919+MaddyUnderStars@users.noreply.github.com> | 2022-08-22 22:16:11 +1000 |
commit | 8720744091e71efe0b2e5f88956e1e0a2e9ef702 (patch) | |
tree | f00521f59f3c2ece7e790f2d0b8469c051586a92 /src/cdn/util/S3Storage.ts | |
parent | Actually respect with_mutual_guilds (diff) | |
parent | Merge remote-tracking branch 'Puyodead1/patch/prettier-config' into staging (diff) | |
download | server-8720744091e71efe0b2e5f88956e1e0a2e9ef702.tar.xz |
Merge remote-tracking branch 'upstream/staging' into fix/sendMemberInProfile
Diffstat (limited to 'src/cdn/util/S3Storage.ts')
-rw-r--r-- | src/cdn/util/S3Storage.ts | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/src/cdn/util/S3Storage.ts b/src/cdn/util/S3Storage.ts new file mode 100644 index 00000000..c4066817 --- /dev/null +++ b/src/cdn/util/S3Storage.ts @@ -0,0 +1,60 @@ +import { S3 } from "@aws-sdk/client-s3"; +import { Readable } from "stream"; +import { Storage } from "./Storage"; + +const readableToBuffer = (readable: Readable): Promise<Buffer> => + new Promise((resolve, reject) => { + const chunks: Buffer[] = []; + readable.on("data", (chunk) => chunks.push(chunk)); + readable.on("error", reject); + readable.on("end", () => resolve(Buffer.concat(chunks))); + }); + +export class S3Storage implements Storage { + public constructor( + private client: S3, + private bucket: string, + private basePath?: string + ) {} + + /** + * Always return a string, to ensure consistency. + */ + get bucketBasePath() { + return this.basePath ?? ""; + } + + async set(path: string, data: Buffer): Promise<void> { + await this.client.putObject({ + Bucket: this.bucket, + Key: `${this.bucketBasePath}${path}`, + Body: data, + }); + } + + async get(path: string): Promise<Buffer | null> { + try { + const s3Object = await this.client.getObject({ + Bucket: this.bucket, + Key: `${this.bucketBasePath ?? ""}${path}`, + }); + + if (!s3Object.Body) return null; + + const body = s3Object.Body; + + return await readableToBuffer(<Readable>body); + } catch (err) { + console.error(`[CDN] Unable to get S3 object at path ${path}.`); + console.error(err); + return null; + } + } + + async delete(path: string): Promise<void> { + await this.client.deleteObject({ + Bucket: this.bucket, + Key: `${this.bucketBasePath}${path}`, + }); + } +} |