From 0df8423e6dda0ca1651ac207ecd29f8818f79362 Mon Sep 17 00:00:00 2001
From: Alexander Cerutti <cerutti.alexander@gmail.com>
Date: Sun, 16 Jan 2022 00:15:01 +0100
Subject: [PATCH] Fixed serverless model reading .pass extensions issues

---
 examples/serverless/src/shared.ts | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/examples/serverless/src/shared.ts b/examples/serverless/src/shared.ts
index a70df2a..36dc87e 100644
--- a/examples/serverless/src/shared.ts
+++ b/examples/serverless/src/shared.ts
@@ -27,7 +27,16 @@ export async function getModel(
 ): Promise<string | { [key: string]: Buffer }> {
 	if (process.env.IS_OFFLINE === "true") {
 		console.log("model offline retrieving");
-		return path.resolve(__dirname, "../../../", `models/${modelName}`);
+
+		const standardModelName = modelName.endsWith(".pass")
+			? modelName
+			: `${modelName}.pass`;
+
+		return path.resolve(
+			__dirname,
+			"../../../",
+			`models/${standardModelName}`,
+		);
 	}
 
 	const s3 = await getS3Instance();
@@ -127,7 +136,7 @@ export async function getSpecificFileInModel(
 	const model = await getModel(modelName);
 
 	if (typeof model === "string") {
-		return fs.readFile(path.resolve(`${model}.pass`, fileName));
+		return fs.readFile(path.resolve(model, fileName));
 	}
 
 	return model[fileName];