@balena/pinejs 15.1.0-build-web-resource-4-c82f246bf4abb69c47ca357f99014eb18f1434a8-1 → 15.1.0-build-upsert-6f9a458cc52c017ddaf48ceb7e4597522cdf4dfc-1

package/src/server-glue/webresource-handlers/NoopHandler.ts

@@ -1,20 +0,0 @@
-import {
-	IncomingFile,
-	UploadResponse,
-	WebResourceHandler,
-} from '../webresource-handler';
-
-export class NoopHandler implements WebResourceHandler {
-	public async handleFile(resource: IncomingFile): Promise<UploadResponse> {
-		// handleFile must consume the file stream
-		resource.stream.resume();
-		return {
-			filename: 'noop',
-			size: 0,
-		};
-	}
-
-	public async removeFile(_fileReference: string): Promise<void> {
-		return;
-	}
-}

package/src/server-glue/webresource-handlers/S3Handler.ts

@@ -1,99 +0,0 @@
-import { optionalVar, requiredVar, intVar } from '@balena/env-parsing';
-import {
-	IncomingFile,
-	UploadResponse,
-	WebResourceHandler,
-} from '../webresource-handler';
-import {
-	S3Client,
-	S3ClientConfig,
-	DeleteObjectCommand,
-	PutObjectCommandInput,
-} from '@aws-sdk/client-s3';
-import { Upload } from '@aws-sdk/lib-storage';
-
-import { randomUUID } from 'crypto';
-
-export class S3Handler implements WebResourceHandler {
-	private readonly config: S3ClientConfig;
-	private readonly bucket: string;
-	private readonly endpoint: string;
-	private readonly maxFileSize = intVar(
-		'PINEJS_WEBRESOURCE_MAXFILESIZE',
-		52428800,
-	);
-	private client: S3Client;
-
-	constructor() {
-		this.endpoint = requiredVar('S3_ENDPOINT');
-		this.config = {
-			region: optionalVar('S3_REGION', 'us-east-1'),
-			credentials: {
-				accessKeyId: requiredVar('S3_ACCESS_KEY'),
-				secretAccessKey: requiredVar('S3_SECRET_KEY'),
-			},
-			endpoint: this.endpoint,
-			forcePathStyle: true,
-		};
-
-		this.bucket = optionalVar(
-			'S3_STORAGE_ADAPTER_BUCKET',
-			'balena-pine-web-resources',
-		);
-		this.client = new S3Client(this.config);
-	}
-
-	public async handleFile(resource: IncomingFile): Promise<UploadResponse> {
-		let size = 0;
-		const key = `${resource.fieldname}_${randomUUID()}_${
-			resource.originalname
-		}`;
-		const params: PutObjectCommandInput = {
-			Bucket: this.bucket,
-			ACL: 'public-read',
-			StorageClass: 'STANDARD',
-			Key: key,
-			Body: resource.stream,
-			ContentType: resource.mimetype,
-		};
-		const upload = new Upload({ client: this.client, params });
-
-		upload.on('httpUploadProgress', async (ev) => {
-			size = ev.total ? ev.total : ev.loaded!;
-			if (size > this.maxFileSize) {
-				await upload.abort();
-			}
-		});
-
-		try {
-			await upload.done();
-		} catch (err: any) {
-			resource.stream.resume();
-			if (size > this.maxFileSize) {
-				throw new Error('File size exceeded');
-			}
-			throw err;
-		}
-
-		const filename = this.getS3URL(key);
-		return { size, filename };
-	}
-
-	public async removeFile(fileReference: string): Promise<void> {
-		const fileReferences = fileReference.split('/');
-		const fileKey = fileReferences[fileReferences.length - 1];
-
-		const command = new DeleteObjectCommand({
-			Bucket: this.bucket,
-			Key: fileKey,
-		});
-
-		await this.client.send(command);
-	}
-
-	private getS3URL(key: string) {
-		return this.endpoint.includes(this.bucket)
-			? `${this.endpoint}/${key}`
-			: `${this.endpoint}/${this.bucket}/${key}`;
-	}
-}