Page MenuHomeSealhub
Files F9582636

No OneTemporary
Actions

View Options

diff --git a/src/index.ts b/src/index.ts
index 329020d..9b0a02c 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,211 +1,206 @@
import Router from "@koa/router";
import sharp from "sharp";
import crypto from "crypto";
import { access, mkdir, copyFile, readFile, stat } from "fs/promises";
import { extname, basename } from "path";
import { Middleware } from "koa";
type correctExtension = "jpeg" | "png" | "avif" | "webp";
function isCorrectExtension(type: unknown): type is correctExtension {
const extensions = ["avif", "webp", "jpeg", "png"];
return extensions.includes(type as string);
}
const MONTH = 60 * 60 * 24 * 30;
export default class KoaResponsiveImageRouter extends Router {
router: Router;
hashToResolutions: Record<string, number[]> = {};
hashToLossless: Record<string, boolean> = {};
constructor(public static_path: string, public tmp_dir: string) {
super();
this.router = new Router();
this.router.get("/:hash/:filename", async (ctx) => {
const { hash, filename } = ctx.params;
const resolution = parseInt(filename.split(".")[0]);
const destination = `${this.getHashedPath(hash)}`;
const type = extname(filename).split(".").pop();
if (
this.hashToResolutions[hash].find(
(el: number) => el === resolution
) &&
type !== undefined &&
isCorrectExtension(type)
) {
ctx.set("Cache-Control", `public, max-age=${MONTH}`);
ctx.set("etag", `W/"${hash}:${filename}"`);
ctx.status = 200; //otherwise the `.fresh` check won't work, see https://koajs.com/
if (ctx.fresh) {
ctx.status = 304;
return;
}
try {
await access(destination);
ctx.body = await this.getImage({ hash, resolution, type });
ctx.type = `image/${type}`;
- ctx.set(
- "Cache-Control",
- `public, max-age=${MONTH}, immutable`
- );
- ctx.set("etag", `${hash}:${filename}`);
} catch (error) {
ctx.response.status = 404;
}
} else {
ctx.response.status = 404;
}
});
}
async image({
resolutions,
sizes_attr,
path,
lossless = false,
}: {
resolutions: number[];
sizes_attr: string;
path: string;
lossless?: boolean;
}): Promise<string> {
const hash = await this.getHash(path, resolutions);
this.hashToResolutions[hash] = resolutions;
this.hashToLossless[hash] = lossless;
await this.generateDirectory(path, hash);
await this.copySourceFile(path, hash);
const destination = `${this.static_path}/${hash}`;
const extensions = [
"webp",
"png",
...(lossless ? [] : ["jpg", "avif"]),
];
let html = "<picture>";
for (let j = 0; j < extensions.length; j++) {
html += '\n<source\nsrcset="\n';
for (let i = 0; i < resolutions.length; i++) {
html += `${destination}/${resolutions[i]}.${extensions[j]} ${resolutions[i]}w`;
if (i !== resolutions.length - 1) {
html += ",";
} else {
html += `\n"`;
}
html += `\n`;
}
html += `src="${destination}/${
resolutions[Math.round(resolutions.length / 2)]
}.${extensions[j]}"\n`;
html += `sizes="${sizes_attr}"\ntype="image/${extensions[j]}"\n/>\n`;
}
html += `<img loading="lazy" src="${destination}/${
resolutions[Math.round(resolutions.length / 2)]
}.jpeg" /></picture>`;
return html;
}
getRoutes(): Middleware {
return this.router.routes();
}
private async getHash(original_file_path: string, resolutions: number[]) {
return crypto
.createHash("md5")
.update(
`
${basename(original_file_path)}${(
await stat(original_file_path)
).atime.getTime()}${JSON.stringify(resolutions)}`
)
.digest("hex");
}
private getHashedPath(hash: string) {
return `${this.tmp_dir}/${hash}`;
}
private async generateDirectory(original_file_path: string, hash: string) {
const destination = `${this.tmp_dir}/${hash}`;
try {
await access(destination);
} catch {
try {
await mkdir(destination, { recursive: true });
} catch (error) {
console.log("directory exist");
}
}
}
private async copySourceFile(original_file_path: string, hash: string) {
const source = original_file_path;
const destination = `${this.getHashedPath(hash)}/original-file`;
await copyFile(source, destination);
}
private async getImage({
hash,
resolution,
type,
}: {
hash: string;
resolution: number;
type: correctExtension;
}): Promise<Buffer> {
try {
return Buffer.from(
((await readFile(
`${this.getHashedPath(hash)}/${resolution}.${type}`
)) as unknown) as string,
"base64" as BufferEncoding
);
} catch {
const buffer = await this.generateImage({ hash, resolution, type });
this.saveImage({ hash, resolution, type, buffer });
return buffer;
}
}
private saveImage({
hash,
resolution,
type,
buffer,
}: {
hash: string;
resolution: number;
type: string;
buffer: Buffer;
}) {
void sharp(buffer).toFile(
`${this.getHashedPath(hash)}/${resolution}.${type}`
);
}
private async generateImage({
hash,
resolution,
type,
}: {
hash: string;
resolution: number;
type: correctExtension;
}) {
const lossless = this.hashToLossless[hash];
return await sharp(`${this.getHashedPath(hash)}/original-file`)
.resize(resolution)
.toFormat(type, lossless ? { lossless: true } : {})
.toBuffer();
}
}

File Metadata

Mime Type
text/x-diff
Expires
Sat, Oct 11, 07:21 (11 h, 45 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
983965
Default Alt Text
(5 KB)

Event Timeline