image-edit-tools 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.gitattributes +2 -0
- package/README.md +41 -0
- package/dist/index.d.ts +24 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +24 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/index.d.ts +3 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +3 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/server.d.ts +3 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +15 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools.d.ts +4 -0
- package/dist/mcp/tools.d.ts.map +1 -0
- package/dist/mcp/tools.js +285 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/ops/add-text.d.ts +5 -0
- package/dist/ops/add-text.d.ts.map +1 -0
- package/dist/ops/add-text.js +129 -0
- package/dist/ops/add-text.js.map +1 -0
- package/dist/ops/adjust.d.ts +3 -0
- package/dist/ops/adjust.d.ts.map +1 -0
- package/dist/ops/adjust.js +71 -0
- package/dist/ops/adjust.js.map +1 -0
- package/dist/ops/batch.d.ts +3 -0
- package/dist/ops/batch.d.ts.map +1 -0
- package/dist/ops/batch.js +35 -0
- package/dist/ops/batch.js.map +1 -0
- package/dist/ops/blur-region.d.ts +5 -0
- package/dist/ops/blur-region.d.ts.map +1 -0
- package/dist/ops/blur-region.js +54 -0
- package/dist/ops/blur-region.js.map +1 -0
- package/dist/ops/composite.d.ts +5 -0
- package/dist/ops/composite.d.ts.map +1 -0
- package/dist/ops/composite.js +53 -0
- package/dist/ops/composite.js.map +1 -0
- package/dist/ops/convert.d.ts +3 -0
- package/dist/ops/convert.d.ts.map +1 -0
- package/dist/ops/convert.js +45 -0
- package/dist/ops/convert.js.map +1 -0
- package/dist/ops/crop.d.ts +3 -0
- package/dist/ops/crop.d.ts.map +1 -0
- package/dist/ops/crop.js +105 -0
- package/dist/ops/crop.js.map +1 -0
- package/dist/ops/detect-faces.d.ts +3 -0
- package/dist/ops/detect-faces.d.ts.map +1 -0
- package/dist/ops/detect-faces.js +41 -0
- package/dist/ops/detect-faces.js.map +1 -0
- package/dist/ops/detect-subject.d.ts +3 -0
- package/dist/ops/detect-subject.d.ts.map +1 -0
- package/dist/ops/detect-subject.js +78 -0
- package/dist/ops/detect-subject.js.map +1 -0
- package/dist/ops/extract-text.d.ts +5 -0
- package/dist/ops/extract-text.d.ts.map +1 -0
- package/dist/ops/extract-text.js +21 -0
- package/dist/ops/extract-text.js.map +1 -0
- package/dist/ops/filter.d.ts +3 -0
- package/dist/ops/filter.d.ts.map +1 -0
- package/dist/ops/filter.js +53 -0
- package/dist/ops/filter.js.map +1 -0
- package/dist/ops/get-dominant-colors.d.ts +3 -0
- package/dist/ops/get-dominant-colors.d.ts.map +1 -0
- package/dist/ops/get-dominant-colors.js +48 -0
- package/dist/ops/get-dominant-colors.js.map +1 -0
- package/dist/ops/get-metadata.d.ts +3 -0
- package/dist/ops/get-metadata.d.ts.map +1 -0
- package/dist/ops/get-metadata.js +30 -0
- package/dist/ops/get-metadata.js.map +1 -0
- package/dist/ops/optimize.d.ts +3 -0
- package/dist/ops/optimize.d.ts.map +1 -0
- package/dist/ops/optimize.js +78 -0
- package/dist/ops/optimize.js.map +1 -0
- package/dist/ops/overlay.d.ts +3 -0
- package/dist/ops/overlay.d.ts.map +1 -0
- package/dist/ops/overlay.js +52 -0
- package/dist/ops/overlay.js.map +1 -0
- package/dist/ops/pad.d.ts +3 -0
- package/dist/ops/pad.d.ts.map +1 -0
- package/dist/ops/pad.js +62 -0
- package/dist/ops/pad.js.map +1 -0
- package/dist/ops/pipeline.d.ts +5 -0
- package/dist/ops/pipeline.d.ts.map +1 -0
- package/dist/ops/pipeline.js +81 -0
- package/dist/ops/pipeline.js.map +1 -0
- package/dist/ops/remove-bg.d.ts +3 -0
- package/dist/ops/remove-bg.d.ts.map +1 -0
- package/dist/ops/remove-bg.js +79 -0
- package/dist/ops/remove-bg.js.map +1 -0
- package/dist/ops/resize.d.ts +3 -0
- package/dist/ops/resize.d.ts.map +1 -0
- package/dist/ops/resize.js +54 -0
- package/dist/ops/resize.js.map +1 -0
- package/dist/ops/watermark.d.ts +3 -0
- package/dist/ops/watermark.d.ts.map +1 -0
- package/dist/ops/watermark.js +142 -0
- package/dist/ops/watermark.js.map +1 -0
- package/dist/types.d.ts +233 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +12 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/load-image.d.ts +9 -0
- package/dist/utils/load-image.d.ts.map +1 -0
- package/dist/utils/load-image.js +22 -0
- package/dist/utils/load-image.js.map +1 -0
- package/dist/utils/result.d.ts +4 -0
- package/dist/utils/result.d.ts.map +1 -0
- package/dist/utils/result.js +3 -0
- package/dist/utils/result.js.map +1 -0
- package/dist/utils/validate.d.ts +16 -0
- package/dist/utils/validate.d.ts.map +1 -0
- package/dist/utils/validate.js +20 -0
- package/dist/utils/validate.js.map +1 -0
- package/docs/AGENTS.md +18 -0
- package/docs/MCP.md +106 -0
- package/package.json +52 -0
- package/scripts/generate-fixtures.js +33 -0
- package/src/index.ts +24 -0
- package/src/mcp/index.ts +2 -0
- package/src/mcp/server.ts +21 -0
- package/src/mcp/tools.ts +276 -0
- package/src/ops/add-text.ts +139 -0
- package/src/ops/adjust.ts +68 -0
- package/src/ops/batch.ts +41 -0
- package/src/ops/blur-region.ts +58 -0
- package/src/ops/composite.ts +56 -0
- package/src/ops/convert.ts +46 -0
- package/src/ops/crop.ts +101 -0
- package/src/ops/detect-faces.ts +41 -0
- package/src/ops/detect-subject.ts +80 -0
- package/src/ops/extract-text.ts +19 -0
- package/src/ops/filter.ts +51 -0
- package/src/ops/get-dominant-colors.ts +41 -0
- package/src/ops/get-metadata.ts +28 -0
- package/src/ops/optimize.ts +77 -0
- package/src/ops/overlay.ts +51 -0
- package/src/ops/pad.ts +63 -0
- package/src/ops/pipeline.ts +61 -0
- package/src/ops/remove-bg.ts +82 -0
- package/src/ops/resize.ts +54 -0
- package/src/ops/watermark.ts +141 -0
- package/src/types/color-thief-node.d.ts +4 -0
- package/src/types.ts +267 -0
- package/src/utils/load-image.ts +21 -0
- package/src/utils/result.ts +4 -0
- package/src/utils/validate.ts +21 -0
- package/tests/fixtures/logo.png +0 -0
- package/tests/fixtures/sample.jpg +0 -0
- package/tests/fixtures/sample.png +0 -0
- package/tests/fixtures/sample.webp +0 -0
- package/tests/integration/error-handling.test.ts +22 -0
- package/tests/integration/load-image.test.ts +45 -0
- package/tests/unit/add-text.test.ts +56 -0
- package/tests/unit/adjust.test.ts +81 -0
- package/tests/unit/batch.test.ts +38 -0
- package/tests/unit/blur-region.test.ts +52 -0
- package/tests/unit/composite.test.ts +58 -0
- package/tests/unit/convert.test.ts +55 -0
- package/tests/unit/crop.test.ts +100 -0
- package/tests/unit/detect-faces.test.ts +32 -0
- package/tests/unit/detect-subject.test.ts +37 -0
- package/tests/unit/extract-text.test.ts +34 -0
- package/tests/unit/filter.test.ts +39 -0
- package/tests/unit/get-dominant-colors.test.ts +25 -0
- package/tests/unit/get-metadata.test.ts +36 -0
- package/tests/unit/mcp.test.ts +104 -0
- package/tests/unit/optimize.test.ts +47 -0
- package/tests/unit/overlay.test.ts +39 -0
- package/tests/unit/pad.test.ts +56 -0
- package/tests/unit/pipeline.test.ts +48 -0
- package/tests/unit/remove-bg.test.ts +42 -0
- package/tests/unit/resize.test.ts +70 -0
- package/tests/unit/watermark.test.ts +54 -0
- package/tsconfig.json +15 -0
- package/vitest.config.ts +27 -0
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { ErrorCode } from '../types.js';
|
|
2
|
+
import { loadImage } from '../utils/load-image.js';
|
|
3
|
+
import { err, ok } from '../utils/result.js';
|
|
4
|
+
import Tesseract from 'tesseract.js';
|
|
5
|
+
export async function extractText(input, options = {}) {
|
|
6
|
+
try {
|
|
7
|
+
const buffer = await loadImage(input);
|
|
8
|
+
const lang = options.lang || 'eng';
|
|
9
|
+
const result = await Tesseract.recognize(buffer, lang);
|
|
10
|
+
return ok(result.data.text);
|
|
11
|
+
}
|
|
12
|
+
catch (e) {
|
|
13
|
+
const msg = e.message || '';
|
|
14
|
+
if (msg.includes('HTTP'))
|
|
15
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
16
|
+
if (msg.includes('ENOENT'))
|
|
17
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
18
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
//# sourceMappingURL=extract-text.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extract-text.js","sourceRoot":"","sources":["../../src/ops/extract-text.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,SAAS,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,SAAS,MAAM,cAAc,CAAC;AAErC,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAiB,EAAE,UAA6B,EAAE;IAClF,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,KAAK,CAAC;QAEnC,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACvD,OAAO,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"filter.d.ts","sourceRoot":"","sources":["../../src/ops/filter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAKhF,wBAAsB,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,WAAW,CAAC,CA6C5F"}
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { ErrorCode } from '../types.js';
|
|
2
|
+
import { loadImage } from '../utils/load-image.js';
|
|
3
|
+
import { err, ok } from '../utils/result.js';
|
|
4
|
+
import sharp from 'sharp';
|
|
5
|
+
export async function filter(input, options) {
|
|
6
|
+
try {
|
|
7
|
+
const buffer = await loadImage(input);
|
|
8
|
+
let pipeline = sharp(buffer);
|
|
9
|
+
switch (options.preset) {
|
|
10
|
+
case 'grayscale':
|
|
11
|
+
pipeline = pipeline.grayscale();
|
|
12
|
+
break;
|
|
13
|
+
case 'sepia':
|
|
14
|
+
pipeline = pipeline.tint({ r: 112, g: 66, b: 20 }).modulate({ saturation: 0.5 });
|
|
15
|
+
break;
|
|
16
|
+
case 'invert':
|
|
17
|
+
pipeline = pipeline.negate();
|
|
18
|
+
break;
|
|
19
|
+
case 'vintage':
|
|
20
|
+
// vintage: grayscale -> sepia -> slight contrast reduction -> add film grain (skip grain as composite noise is complex, but we can do a slight tint instead)
|
|
21
|
+
pipeline = pipeline
|
|
22
|
+
.grayscale()
|
|
23
|
+
.tint({ r: 112, g: 66, b: 20 })
|
|
24
|
+
.modulate({ saturation: 0.8, brightness: 1.1 })
|
|
25
|
+
.linear(0.9, -(128 * 0.9) + 128);
|
|
26
|
+
break;
|
|
27
|
+
case 'unsharp':
|
|
28
|
+
pipeline = pipeline.sharpen({ sigma: 1.5, m1: 0.5, m2: 0.1 });
|
|
29
|
+
break;
|
|
30
|
+
case 'blur':
|
|
31
|
+
if (options.radius === undefined || options.radius <= 0) {
|
|
32
|
+
return err('Blur radius must be a positive number', ErrorCode.INVALID_INPUT);
|
|
33
|
+
}
|
|
34
|
+
pipeline = pipeline.blur(options.radius);
|
|
35
|
+
break;
|
|
36
|
+
default:
|
|
37
|
+
return err('Invalid filter preset', ErrorCode.INVALID_INPUT);
|
|
38
|
+
}
|
|
39
|
+
const output = await pipeline.toBuffer();
|
|
40
|
+
return ok(output);
|
|
41
|
+
}
|
|
42
|
+
catch (e) {
|
|
43
|
+
const msg = e.message || '';
|
|
44
|
+
if (msg.includes('HTTP'))
|
|
45
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
46
|
+
if (msg.includes('ENOENT'))
|
|
47
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
48
|
+
if (msg.includes('unsupported image format'))
|
|
49
|
+
return err('Corrupt or unsupported input', ErrorCode.INVALID_INPUT);
|
|
50
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
//# sourceMappingURL=filter.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"filter.js","sourceRoot":"","sources":["../../src/ops/filter.ts"],"names":[],"mappings":"AAAA,OAAO,EAA0C,SAAS,EAAE,MAAM,aAAa,CAAC;AAChF,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,MAAM,CAAC,KAAK,UAAU,MAAM,CAAC,KAAiB,EAAE,OAAsB;IACpE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAE7B,QAAQ,OAAO,CAAC,MAAM,EAAE,CAAC;YACvB,KAAK,WAAW;gBACd,QAAQ,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAChC,MAAM;YACR,KAAK,OAAO;gBACV,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;gBACjF,MAAM;YACR,KAAK,QAAQ;gBACX,QAAQ,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAC7B,MAAM;YACR,KAAK,SAAS;gBACZ,6JAA6J;gBAC7J,QAAQ,GAAG,QAAQ;qBAChB,SAAS,EAAE;qBACX,IAAI,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;qBAC9B,QAAQ,CAAC,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,CAAC;qBAC9C,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;gBACnC,MAAM;YACR,KAAK,SAAS;gBACZ,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;gBAC9D,MAAM;YACR,KAAK,MAAM;gBACT,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;oBACxD,OAAO,GAAG,CAAC,uCAAuC,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;gBAC/E,CAAC;gBACD,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACzC,MAAM;YACR;gBACE,OAAO,GAAG,CAAC,uBAAuB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,CAAC;QACzC,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,IAAI,GAAG,CAAC,QAAQ,CAAC,0BAA0B,CAAC;YAAE,OAAO,GAAG,CAAC,8BAA8B,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClH,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-dominant-colors.d.ts","sourceRoot":"","sources":["../../src/ops/get-dominant-colors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,EAAa,MAAM,aAAa,CAAC;AAQ5D,wBAAsB,iBAAiB,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,GAAE,MAAU,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAgCvG"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import { ErrorCode } from '../types.js';
|
|
2
|
+
import { loadImage } from '../utils/load-image.js';
|
|
3
|
+
import { err, ok } from '../utils/result.js';
|
|
4
|
+
import { getPaletteFromURL } from 'color-thief-node';
|
|
5
|
+
import { writeFileSync, unlinkSync } from 'fs';
|
|
6
|
+
import { join } from 'path';
|
|
7
|
+
import { tmpdir } from 'os';
|
|
8
|
+
export async function getDominantColors(input, count = 5) {
|
|
9
|
+
let tmpFile = '';
|
|
10
|
+
try {
|
|
11
|
+
const buffer = await loadImage(input);
|
|
12
|
+
tmpFile = join(tmpdir(), `ct-${Date.now()}-${Math.random()}.jpg`);
|
|
13
|
+
writeFileSync(tmpFile, buffer);
|
|
14
|
+
const palette = await getPaletteFromURL(tmpFile, count);
|
|
15
|
+
if (tmpFile) {
|
|
16
|
+
try {
|
|
17
|
+
unlinkSync(tmpFile);
|
|
18
|
+
}
|
|
19
|
+
catch (e) { }
|
|
20
|
+
}
|
|
21
|
+
const hexColors = palette.map((rgb) => {
|
|
22
|
+
return '#' + rgb.map((x) => x.toString(16).padStart(2, '0')).join('');
|
|
23
|
+
});
|
|
24
|
+
while (hexColors.length > count)
|
|
25
|
+
hexColors.pop();
|
|
26
|
+
while (hexColors.length > 0 && hexColors.length < count)
|
|
27
|
+
hexColors.push(hexColors[hexColors.length - 1]);
|
|
28
|
+
// If image has only 1 color and getPaletteFromURL completely fails to return array,
|
|
29
|
+
if (hexColors.length === 0)
|
|
30
|
+
hexColors.push('#000000');
|
|
31
|
+
return ok(hexColors);
|
|
32
|
+
}
|
|
33
|
+
catch (e) {
|
|
34
|
+
if (tmpFile) {
|
|
35
|
+
try {
|
|
36
|
+
unlinkSync(tmpFile);
|
|
37
|
+
}
|
|
38
|
+
catch (e) { }
|
|
39
|
+
}
|
|
40
|
+
const msg = e.message || '';
|
|
41
|
+
if (msg.includes('HTTP'))
|
|
42
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
43
|
+
if (msg.includes('ENOENT'))
|
|
44
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
45
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
//# sourceMappingURL=get-dominant-colors.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-dominant-colors.js","sourceRoot":"","sources":["../../src/ops/get-dominant-colors.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,SAAS,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAE5B,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,KAAiB,EAAE,QAAgB,CAAC;IAC1E,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,MAAM,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClE,aAAa,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAE/B,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACxD,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,CAAC;gBAAC,UAAU,CAAC,OAAO,CAAC,CAAA;YAAC,CAAC;YAAC,OAAM,CAAC,EAAC,CAAC,CAAA,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,GAAa,EAAE,EAAE;YAC9C,OAAO,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,OAAO,SAAS,CAAC,MAAM,GAAG,KAAK;YAAE,SAAS,CAAC,GAAG,EAAE,CAAC;QACjD,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,KAAK;YAAE,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;QAEzG,oFAAoF;QACpF,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEtD,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,OAAO,EAAE,CAAC;YACV,IAAI,CAAC;gBAAC,UAAU,CAAC,OAAO,CAAC,CAAA;YAAC,CAAC;YAAC,OAAM,CAAC,EAAC,CAAC,CAAA,CAAC;QAC1C,CAAC;QACD,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-metadata.d.ts","sourceRoot":"","sources":["../../src/ops/get-metadata.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,EAAa,MAAM,aAAa,CAAC;AAI3E,wBAAsB,WAAW,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAsBnF"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import sharp from 'sharp';
|
|
2
|
+
import { ErrorCode } from '../types.js';
|
|
3
|
+
import { loadImage } from '../utils/load-image.js';
|
|
4
|
+
import { err, ok } from '../utils/result.js';
|
|
5
|
+
export async function getMetadata(input) {
|
|
6
|
+
try {
|
|
7
|
+
const buffer = await loadImage(input);
|
|
8
|
+
const meta = await sharp(buffer).metadata();
|
|
9
|
+
return ok({
|
|
10
|
+
width: meta.width || 0,
|
|
11
|
+
height: meta.height || 0,
|
|
12
|
+
format: meta.format || 'unknown',
|
|
13
|
+
fileSize: meta.size || buffer.length,
|
|
14
|
+
colorSpace: meta.space,
|
|
15
|
+
hasAlpha: meta.hasAlpha || false,
|
|
16
|
+
channels: meta.channels || 3,
|
|
17
|
+
density: meta.density,
|
|
18
|
+
exif: meta.exif ? { bufferLength: meta.exif.length } : undefined
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
catch (e) {
|
|
22
|
+
const msg = e.message || '';
|
|
23
|
+
if (msg.includes('HTTP'))
|
|
24
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
25
|
+
if (msg.includes('ENOENT'))
|
|
26
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
27
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=get-metadata.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"get-metadata.js","sourceRoot":"","sources":["../../src/ops/get-metadata.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAqC,SAAS,EAAE,MAAM,aAAa,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAE7C,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAiB;IACjD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAE5C,OAAO,EAAE,CAAC;YACR,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,CAAC;YACtB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,SAAS;YAChC,QAAQ,EAAE,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,MAAM;YACpC,UAAU,EAAE,IAAI,CAAC,KAAK;YACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,KAAK;YAChC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,CAAC;YAC5B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;SACjE,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"optimize.d.ts","sourceRoot":"","sources":["../../src/ops/optimize.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAKlF,wBAAsB,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,WAAW,CAAC,CAsEhG"}
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
import sharp from 'sharp';
|
|
2
|
+
import { ErrorCode } from '../types.js';
|
|
3
|
+
import { loadImage } from '../utils/load-image.js';
|
|
4
|
+
import { getImageMetadata } from '../utils/validate.js';
|
|
5
|
+
import { err, ok } from '../utils/result.js';
|
|
6
|
+
export async function optimize(input, options) {
|
|
7
|
+
try {
|
|
8
|
+
const buffer = await loadImage(input);
|
|
9
|
+
const meta = await getImageMetadata(buffer);
|
|
10
|
+
let pipeline = sharp(buffer);
|
|
11
|
+
let format = meta.format;
|
|
12
|
+
const autoFormat = options.autoFormat ?? true;
|
|
13
|
+
if (autoFormat) {
|
|
14
|
+
format = meta.hasAlpha ? 'webp' : 'jpeg';
|
|
15
|
+
}
|
|
16
|
+
if (options.maxDimension) {
|
|
17
|
+
if (options.maxDimension <= 0)
|
|
18
|
+
return err('maxDimension must be positive', ErrorCode.INVALID_INPUT);
|
|
19
|
+
if (meta.width > options.maxDimension || meta.height > options.maxDimension) {
|
|
20
|
+
if (meta.width > meta.height) {
|
|
21
|
+
pipeline = pipeline.resize({ width: options.maxDimension, withoutEnlargement: true });
|
|
22
|
+
}
|
|
23
|
+
else {
|
|
24
|
+
pipeline = pipeline.resize({ height: options.maxDimension, withoutEnlargement: true });
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
}
|
|
28
|
+
const applyFormat = (p, f, q) => {
|
|
29
|
+
if (f === 'webp')
|
|
30
|
+
return p.webp({ quality: q });
|
|
31
|
+
if (f === 'png')
|
|
32
|
+
return p.png({ quality: q, compressionLevel: 9 });
|
|
33
|
+
if (f === 'avif')
|
|
34
|
+
return p.avif({ quality: q });
|
|
35
|
+
return p.jpeg({ quality: q });
|
|
36
|
+
};
|
|
37
|
+
if (options.maxSizeKB) {
|
|
38
|
+
if (options.maxSizeKB <= 0)
|
|
39
|
+
return err('maxSizeKB must be positive', ErrorCode.INVALID_INPUT);
|
|
40
|
+
const targetBytes = options.maxSizeKB * 1024;
|
|
41
|
+
let minQ = 10, maxQ = 95;
|
|
42
|
+
let bestBuffer = null;
|
|
43
|
+
let bestDiff = Infinity;
|
|
44
|
+
for (let i = 0; i < 7; i++) {
|
|
45
|
+
const midQ = Math.floor((minQ + maxQ) / 2);
|
|
46
|
+
const testBuf = await applyFormat(pipeline.clone(), format, midQ).toBuffer();
|
|
47
|
+
const diff = targetBytes - testBuf.length;
|
|
48
|
+
if (testBuf.length <= targetBytes && diff < bestDiff && diff >= 0) {
|
|
49
|
+
bestBuffer = testBuf;
|
|
50
|
+
bestDiff = diff;
|
|
51
|
+
}
|
|
52
|
+
if (testBuf.length > targetBytes) {
|
|
53
|
+
maxQ = midQ - 1;
|
|
54
|
+
}
|
|
55
|
+
else {
|
|
56
|
+
minQ = midQ + 1;
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
if (bestBuffer)
|
|
60
|
+
return ok(bestBuffer);
|
|
61
|
+
const fallback = await applyFormat(pipeline.clone(), format, 10).toBuffer();
|
|
62
|
+
return ok(fallback);
|
|
63
|
+
}
|
|
64
|
+
else {
|
|
65
|
+
const output = await applyFormat(pipeline, format, 80).toBuffer();
|
|
66
|
+
return ok(output);
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
catch (e) {
|
|
70
|
+
const msg = e.message || '';
|
|
71
|
+
if (msg.includes('HTTP'))
|
|
72
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
73
|
+
if (msg.includes('ENOENT'))
|
|
74
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
75
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
//# sourceMappingURL=optimize.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"optimize.js","sourceRoot":"","sources":["../../src/ops/optimize.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAA4C,SAAS,EAAE,MAAM,aAAa,CAAC;AAClF,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AACxD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAE7C,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,KAAiB,EAAE,OAAwB;IACxE,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAE5C,IAAI,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAE7B,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QACzB,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,IAAI,CAAC;QAC9C,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC;QAC3C,CAAC;QAED,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACzB,IAAI,OAAO,CAAC,YAAY,IAAI,CAAC;gBAAE,OAAO,GAAG,CAAC,+BAA+B,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;YACpG,IAAI,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,YAAY,IAAI,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;gBAC5E,IAAI,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;oBAC7B,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,YAAY,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;gBACxF,CAAC;qBAAM,CAAC;oBACN,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,YAAY,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzF,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,CAAC,CAAc,EAAE,CAAS,EAAE,CAAS,EAAE,EAAE;YAC3D,IAAI,CAAC,KAAK,MAAM;gBAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChD,IAAI,CAAC,KAAK,KAAK;gBAAE,OAAO,CAAC,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,CAAC;YACnE,IAAI,CAAC,KAAK,MAAM;gBAAE,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;YAChD,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC;QAEF,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACtB,IAAI,OAAO,CAAC,SAAS,IAAI,CAAC;gBAAE,OAAO,GAAG,CAAC,4BAA4B,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;YAC9F,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;YAE7C,IAAI,IAAI,GAAG,EAAE,EAAE,IAAI,GAAG,EAAE,CAAC;YACzB,IAAI,UAAU,GAAkB,IAAI,CAAC;YACrC,IAAI,QAAQ,GAAG,QAAQ,CAAC;YAExB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC3C,MAAM,OAAO,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;gBAC7E,MAAM,IAAI,GAAG,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;gBAE1C,IAAI,OAAO,CAAC,MAAM,IAAI,WAAW,IAAI,IAAI,GAAG,QAAQ,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;oBAClE,UAAU,GAAG,OAAO,CAAC;oBACrB,QAAQ,GAAG,IAAI,CAAC;gBAClB,CAAC;gBAED,IAAI,OAAO,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;oBACjC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC;gBAClB,CAAC;qBAAM,CAAC;oBACN,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC;YAED,IAAI,UAAU;gBAAE,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC;YAEtC,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;YAC5E,OAAO,EAAE,CAAC,QAAQ,CAAC,CAAC;QACtB,CAAC;aAAM,CAAC;YACN,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;YAClE,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"overlay.d.ts","sourceRoot":"","sources":["../../src/ops/overlay.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAIjF,wBAAsB,OAAO,CAAC,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,OAAO,GAAE,cAAmB,GAAG,OAAO,CAAC,WAAW,CAAC,CA6C7H"}
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
import sharp from 'sharp';
|
|
2
|
+
import { ErrorCode } from '../types.js';
|
|
3
|
+
import { loadImage } from '../utils/load-image.js';
|
|
4
|
+
import { err, ok } from '../utils/result.js';
|
|
5
|
+
export async function overlay(input, overlayImage, options = {}) {
|
|
6
|
+
try {
|
|
7
|
+
const buffer = await loadImage(input);
|
|
8
|
+
let layerBuf = await loadImage(overlayImage);
|
|
9
|
+
if (options.opacity !== undefined && options.opacity >= 0 && options.opacity < 1) {
|
|
10
|
+
layerBuf = await sharp(layerBuf)
|
|
11
|
+
.ensureAlpha()
|
|
12
|
+
.composite([
|
|
13
|
+
{
|
|
14
|
+
input: Buffer.from([255, 255, 255, Math.round(options.opacity * 255)]),
|
|
15
|
+
raw: { width: 1, height: 1, channels: 4 },
|
|
16
|
+
tile: true,
|
|
17
|
+
blend: 'dest-in'
|
|
18
|
+
}
|
|
19
|
+
])
|
|
20
|
+
.toBuffer();
|
|
21
|
+
}
|
|
22
|
+
let gravity = options.gravity ? options.gravity.toLowerCase() : undefined;
|
|
23
|
+
const compositeOpts = {
|
|
24
|
+
input: layerBuf,
|
|
25
|
+
blend: options.blend || 'over'
|
|
26
|
+
};
|
|
27
|
+
if (gravity) {
|
|
28
|
+
compositeOpts.gravity = gravity;
|
|
29
|
+
}
|
|
30
|
+
else {
|
|
31
|
+
if (options.offsetX !== undefined)
|
|
32
|
+
compositeOpts.left = Math.floor(options.offsetX);
|
|
33
|
+
if (options.offsetY !== undefined)
|
|
34
|
+
compositeOpts.top = Math.floor(options.offsetY);
|
|
35
|
+
}
|
|
36
|
+
const output = await sharp(buffer)
|
|
37
|
+
.composite([compositeOpts])
|
|
38
|
+
.toBuffer();
|
|
39
|
+
return ok(output);
|
|
40
|
+
}
|
|
41
|
+
catch (e) {
|
|
42
|
+
const msg = e.message || '';
|
|
43
|
+
if (msg.includes('HTTP'))
|
|
44
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
45
|
+
if (msg.includes('ENOENT'))
|
|
46
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
47
|
+
if (msg.includes('unsupported image format'))
|
|
48
|
+
return err('Corrupt or unsupported input', ErrorCode.INVALID_INPUT);
|
|
49
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
//# sourceMappingURL=overlay.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"overlay.js","sourceRoot":"","sources":["../../src/ops/overlay.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAA2C,SAAS,EAAE,MAAM,aAAa,CAAC;AACjF,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAE7C,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,KAAiB,EAAE,YAAwB,EAAE,UAA0B,EAAE;IACrG,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,QAAQ,GAAG,MAAM,SAAS,CAAC,YAAY,CAAC,CAAC;QAE7C,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,OAAO,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;YACjF,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC;iBAC7B,WAAW,EAAE;iBACb,SAAS,CAAC;gBACT;oBACE,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC;oBACtE,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;oBACzC,IAAI,EAAE,IAAI;oBACV,KAAK,EAAE,SAAS;iBACjB;aACF,CAAC;iBACD,QAAQ,EAAE,CAAC;QAChB,CAAC;QAED,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAE1E,MAAM,aAAa,GAAyB;YAC1C,KAAK,EAAE,QAAQ;YACf,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,MAAM;SAC/B,CAAC;QAEF,IAAI,OAAO,EAAE,CAAC;YACZ,aAAa,CAAC,OAAO,GAAG,OAAO,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS;gBAAE,aAAa,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YACpF,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS;gBAAE,aAAa,CAAC,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC;aAC/B,SAAS,CAAC,CAAC,aAAa,CAAC,CAAC;aAC1B,QAAQ,EAAE,CAAC;QAEd,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,IAAI,GAAG,CAAC,QAAQ,CAAC,0BAA0B,CAAC;YAAE,OAAO,GAAG,CAAC,8BAA8B,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClH,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pad.d.ts","sourceRoot":"","sources":["../../src/ops/pad.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAK7E,wBAAsB,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC,CAwDtF"}
|
package/dist/ops/pad.js
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import sharp from 'sharp';
|
|
2
|
+
import { ErrorCode } from '../types.js';
|
|
3
|
+
import { loadImage } from '../utils/load-image.js';
|
|
4
|
+
import { err, ok } from '../utils/result.js';
|
|
5
|
+
import { getImageMetadata } from '../utils/validate.js';
|
|
6
|
+
export async function pad(input, options) {
|
|
7
|
+
try {
|
|
8
|
+
const buffer = await loadImage(input);
|
|
9
|
+
let top = 0, right = 0, bottom = 0, left = 0;
|
|
10
|
+
if (options.size !== undefined) {
|
|
11
|
+
if (options.size <= 0)
|
|
12
|
+
return err('Size must be positive', ErrorCode.INVALID_INPUT);
|
|
13
|
+
const meta = await getImageMetadata(buffer);
|
|
14
|
+
const targetSize = options.size;
|
|
15
|
+
if (targetSize < meta.width || targetSize < meta.height) {
|
|
16
|
+
return err('Target size must be larger than image dimensions', ErrorCode.INVALID_INPUT);
|
|
17
|
+
}
|
|
18
|
+
const missingTotalW = targetSize - meta.width;
|
|
19
|
+
const missingTotalH = targetSize - meta.height;
|
|
20
|
+
left = Math.floor(missingTotalW / 2);
|
|
21
|
+
right = missingTotalW - left;
|
|
22
|
+
top = Math.floor(missingTotalH / 2);
|
|
23
|
+
bottom = missingTotalH - top;
|
|
24
|
+
}
|
|
25
|
+
else {
|
|
26
|
+
top = options.top || 0;
|
|
27
|
+
right = options.right || 0;
|
|
28
|
+
bottom = options.bottom || 0;
|
|
29
|
+
left = options.left || 0;
|
|
30
|
+
}
|
|
31
|
+
if (top < 0 || right < 0 || bottom < 0 || left < 0) {
|
|
32
|
+
return err('Padding values cannot be negative', ErrorCode.INVALID_INPUT);
|
|
33
|
+
}
|
|
34
|
+
let background = options.color || '#ffffff';
|
|
35
|
+
let pipeline = sharp(buffer);
|
|
36
|
+
if (options.color?.toLowerCase() === 'transparent') {
|
|
37
|
+
background = { r: 0, g: 0, b: 0, alpha: 0 };
|
|
38
|
+
pipeline = pipeline.png();
|
|
39
|
+
}
|
|
40
|
+
const output = await pipeline
|
|
41
|
+
.extend({
|
|
42
|
+
top,
|
|
43
|
+
bottom,
|
|
44
|
+
left,
|
|
45
|
+
right,
|
|
46
|
+
background
|
|
47
|
+
})
|
|
48
|
+
.toBuffer();
|
|
49
|
+
return ok(output);
|
|
50
|
+
}
|
|
51
|
+
catch (e) {
|
|
52
|
+
const msg = e.message || '';
|
|
53
|
+
if (msg.includes('HTTP'))
|
|
54
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
55
|
+
if (msg.includes('ENOENT'))
|
|
56
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
57
|
+
if (msg.includes('unsupported image format'))
|
|
58
|
+
return err('Corrupt or unsupported input', ErrorCode.INVALID_INPUT);
|
|
59
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
60
|
+
}
|
|
61
|
+
}
|
|
62
|
+
//# sourceMappingURL=pad.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pad.js","sourceRoot":"","sources":["../../src/ops/pad.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAuC,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7E,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,oBAAoB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAExD,MAAM,CAAC,KAAK,UAAU,GAAG,CAAC,KAAiB,EAAE,OAAmB;IAC9D,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,GAAG,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,MAAM,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC;QAE7C,IAAI,OAAO,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC/B,IAAI,OAAO,CAAC,IAAI,IAAI,CAAC;gBAAE,OAAO,GAAG,CAAC,uBAAuB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;YACpF,MAAM,IAAI,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC;YAEhC,IAAI,UAAU,GAAG,IAAI,CAAC,KAAK,IAAI,UAAU,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBACvD,OAAO,GAAG,CAAC,kDAAkD,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;YAC3F,CAAC;YAED,MAAM,aAAa,GAAG,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC;YAC9C,MAAM,aAAa,GAAG,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;YAC/C,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YACrC,KAAK,GAAG,aAAa,GAAG,IAAI,CAAC;YAC7B,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YACpC,MAAM,GAAG,aAAa,GAAG,GAAG,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;YACvB,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC;YAC3B,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,CAAC,CAAC;YAC7B,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,IAAI,MAAM,GAAG,CAAC,IAAI,IAAI,GAAG,CAAC,EAAE,CAAC;YACnD,OAAO,GAAG,CAAC,mCAAmC,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,UAAU,GAAwB,OAAO,CAAC,KAAK,IAAI,SAAS,CAAC;QACjE,IAAI,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QAC7B,IAAI,OAAO,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,aAAa,EAAE,CAAC;YACnD,UAAU,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YAC5C,QAAQ,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC;QAC5B,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ;aAC1B,MAAM,CAAC;YACN,GAAG;YACH,MAAM;YACN,IAAI;YACJ,KAAK;YACL,UAAU;SACX,CAAC;aACD,QAAQ,EAAE,CAAC;QAEd,OAAO,EAAE,CAAC,MAAM,CAAC,CAAC;IACpB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,CAAC;QAClE,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClF,IAAI,GAAG,CAAC,QAAQ,CAAC,0BAA0B,CAAC;YAAE,OAAO,GAAG,CAAC,8BAA8B,EAAE,SAAS,CAAC,aAAa,CAAC,CAAC;QAClH,OAAO,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../../src/ops/pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,iBAAiB,EAAa,MAAM,aAAa,CAAC;AAcpF,wBAAsB,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,WAAW,GAAG;IAAE,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC,CA8C3H"}
|
|
@@ -0,0 +1,81 @@
|
|
|
1
|
+
import { ErrorCode } from '../types.js';
|
|
2
|
+
import { crop } from './crop.js';
|
|
3
|
+
import { resize } from './resize.js';
|
|
4
|
+
import { pad } from './pad.js';
|
|
5
|
+
import { adjust } from './adjust.js';
|
|
6
|
+
import { filter } from './filter.js';
|
|
7
|
+
import { blurRegion } from './blur-region.js';
|
|
8
|
+
import { addText } from './add-text.js';
|
|
9
|
+
import { composite } from './composite.js';
|
|
10
|
+
import { watermark } from './watermark.js';
|
|
11
|
+
import { convert } from './convert.js';
|
|
12
|
+
import { optimize } from './optimize.js';
|
|
13
|
+
import { removeBg } from './remove-bg.js';
|
|
14
|
+
export async function pipeline(input, operations) {
|
|
15
|
+
let currentImage = input;
|
|
16
|
+
for (let i = 0; i < operations.length; i++) {
|
|
17
|
+
const op = operations[i];
|
|
18
|
+
let result;
|
|
19
|
+
try {
|
|
20
|
+
switch (op.op) {
|
|
21
|
+
case 'crop':
|
|
22
|
+
result = await crop(currentImage, op);
|
|
23
|
+
break;
|
|
24
|
+
case 'resize':
|
|
25
|
+
result = await resize(currentImage, op);
|
|
26
|
+
break;
|
|
27
|
+
case 'pad':
|
|
28
|
+
result = await pad(currentImage, op);
|
|
29
|
+
break;
|
|
30
|
+
case 'adjust':
|
|
31
|
+
result = await adjust(currentImage, op);
|
|
32
|
+
break;
|
|
33
|
+
case 'filter':
|
|
34
|
+
result = await filter(currentImage, op);
|
|
35
|
+
break;
|
|
36
|
+
case 'blurRegion':
|
|
37
|
+
result = await blurRegion(currentImage, { regions: op.regions });
|
|
38
|
+
break;
|
|
39
|
+
case 'addText':
|
|
40
|
+
result = await addText(currentImage, { layers: op.layers });
|
|
41
|
+
break;
|
|
42
|
+
case 'composite':
|
|
43
|
+
result = await composite(currentImage, { layers: op.layers });
|
|
44
|
+
break;
|
|
45
|
+
case 'watermark':
|
|
46
|
+
result = await watermark(currentImage, op);
|
|
47
|
+
break;
|
|
48
|
+
case 'convert':
|
|
49
|
+
result = await convert(currentImage, op);
|
|
50
|
+
break;
|
|
51
|
+
case 'optimize':
|
|
52
|
+
result = await optimize(currentImage, op);
|
|
53
|
+
break;
|
|
54
|
+
case 'removeBg':
|
|
55
|
+
result = await removeBg(currentImage, op);
|
|
56
|
+
break;
|
|
57
|
+
default:
|
|
58
|
+
return { ok: false, error: `Unknown operation`, code: ErrorCode.INVALID_INPUT, step: i };
|
|
59
|
+
}
|
|
60
|
+
if (!result.ok) {
|
|
61
|
+
return { ...result, step: i };
|
|
62
|
+
}
|
|
63
|
+
currentImage = result.data;
|
|
64
|
+
}
|
|
65
|
+
catch (e) {
|
|
66
|
+
return { ok: false, error: e.message, code: ErrorCode.PROCESSING_FAILED, step: i };
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
if (operations.length === 0) {
|
|
70
|
+
const { loadImage } = await import('../utils/load-image.js');
|
|
71
|
+
try {
|
|
72
|
+
const buf = await loadImage(input);
|
|
73
|
+
return { ok: true, data: buf };
|
|
74
|
+
}
|
|
75
|
+
catch (e) {
|
|
76
|
+
return { ok: false, error: 'Failed to load initial image', code: ErrorCode.INVALID_INPUT, step: 0 };
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
return { ok: true, data: currentImage };
|
|
80
|
+
}
|
|
81
|
+
//# sourceMappingURL=pipeline.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"pipeline.js","sourceRoot":"","sources":["../../src/ops/pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,EAA8C,SAAS,EAAE,MAAM,aAAa,CAAC;AACpF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,MAAM,CAAC,KAAK,UAAU,QAAQ,CAAC,KAAiB,EAAE,UAA+B;IAC/E,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3C,MAAM,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;QACzB,IAAI,MAAmB,CAAC;QAExB,IAAI,CAAC;YACH,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBACd,KAAK,MAAM;oBAAE,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAC1D,KAAK,QAAQ;oBAAE,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAC9D,KAAK,KAAK;oBAAE,MAAM,GAAG,MAAM,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBACxD,KAAK,QAAQ;oBAAE,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAC9D,KAAK,QAAQ;oBAAE,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAC9D,KAAK,YAAY;oBAAE,MAAM,GAAG,MAAM,UAAU,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAC3F,KAAK,SAAS;oBAAE,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;oBAAC,MAAM;gBACnF,KAAK,WAAW;oBAAE,MAAM,GAAG,MAAM,SAAS,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;oBAAC,MAAM;gBACvF,KAAK,WAAW;oBAAE,MAAM,GAAG,MAAM,SAAS,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBACpE,KAAK,SAAS;oBAAE,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAChE,KAAK,UAAU;oBAAE,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAClE,KAAK,UAAU;oBAAE,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;oBAAC,MAAM;gBAClE;oBACE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,mBAAmB,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC7F,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;gBACf,OAAO,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAChC,CAAC;YAED,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC;QAC7B,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YACf,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC,iBAAiB,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACtF,CAAC;IACH,CAAC;IAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,wBAAwB,CAAC,CAAC;QAC7D,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;YACnC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QACjC,CAAC;QAAC,OAAM,CAAK,EAAE,CAAC;YACd,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,8BAA8B,EAAE,IAAI,EAAE,SAAS,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACtG,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,YAAsB,EAAE,CAAC;AACpD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remove-bg.d.ts","sourceRoot":"","sources":["../../src/ops/remove-bg.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAa,MAAM,aAAa,CAAC;AAMlF,wBAAsB,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,OAAO,GAAE,eAAoB,GAAG,OAAO,CAAC,WAAW,CAAC,CA0ErG"}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
import sharp from 'sharp';
|
|
2
|
+
import { ErrorCode } from '../types.js';
|
|
3
|
+
import { loadImage } from '../utils/load-image.js';
|
|
4
|
+
import { err, ok } from '../utils/result.js';
|
|
5
|
+
import { getImageMetadata } from '../utils/validate.js';
|
|
6
|
+
import { AutoModel, AutoProcessor, RawImage } from '@xenova/transformers';
|
|
7
|
+
export async function removeBg(input, options = {}) {
|
|
8
|
+
try {
|
|
9
|
+
const buffer = await loadImage(input);
|
|
10
|
+
const meta = await getImageMetadata(buffer);
|
|
11
|
+
let model, processor;
|
|
12
|
+
try {
|
|
13
|
+
model = await AutoModel.from_pretrained('briaai/RMBG-1.4', {
|
|
14
|
+
config: { model_type: 'custom' },
|
|
15
|
+
quantized: true // Use quantized for speed and avoiding OOM
|
|
16
|
+
});
|
|
17
|
+
processor = await AutoProcessor.from_pretrained('briaai/RMBG-1.4', {
|
|
18
|
+
config: {
|
|
19
|
+
do_normalize: true,
|
|
20
|
+
do_pad: false,
|
|
21
|
+
do_rescale: true,
|
|
22
|
+
do_resize: true,
|
|
23
|
+
image_mean: [0.5, 0.5, 0.5],
|
|
24
|
+
image_std: [1, 1, 1],
|
|
25
|
+
resample: 2,
|
|
26
|
+
size: { width: 1024, height: 1024 },
|
|
27
|
+
}
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
catch (e) {
|
|
31
|
+
return err('Model unavailable. Run: npx image-edit-tools download-models', ErrorCode.MODEL_NOT_FOUND);
|
|
32
|
+
}
|
|
33
|
+
// Process image
|
|
34
|
+
const rawImage = await RawImage.fromURL(URL.createObjectURL(new Blob([buffer]))); // Alternative for node:
|
|
35
|
+
// RawImage from buffer is easier using sharp. We need uint8 array of RGB or RGBA.
|
|
36
|
+
const rawRgb = await sharp(buffer).ensureAlpha().raw().toBuffer();
|
|
37
|
+
const img = new RawImage(new Uint8ClampedArray(rawRgb), meta.width, meta.height, 4);
|
|
38
|
+
const { pixel_values } = await processor(img);
|
|
39
|
+
const { output } = await model({ input: pixel_values });
|
|
40
|
+
// Output is a mask tensor. Resize mask back to original dimensions using sharp.
|
|
41
|
+
// The model outputs size 1024x1024.
|
|
42
|
+
const maskData = new Uint8Array(output.data.length);
|
|
43
|
+
for (let i = 0; i < output.data.length; ++i) {
|
|
44
|
+
maskData[i] = Math.max(0, Math.min(255, Math.round(output.data[i] * 255))); // Assume output is 0.0 to 1.0 floats
|
|
45
|
+
}
|
|
46
|
+
const maskBuffer = await sharp(Buffer.from(maskData), { raw: { width: 1024, height: 1024, channels: 1 } })
|
|
47
|
+
.resize(meta.width, meta.height, { fit: 'fill' }) // match original size
|
|
48
|
+
.toBuffer();
|
|
49
|
+
let outputBuf = await sharp(buffer)
|
|
50
|
+
.ensureAlpha()
|
|
51
|
+
.joinChannel(maskBuffer) // join mask as alpha channel
|
|
52
|
+
.png()
|
|
53
|
+
.toBuffer();
|
|
54
|
+
// Replace color or image
|
|
55
|
+
if (options.replaceColor) {
|
|
56
|
+
outputBuf = await sharp(outputBuf)
|
|
57
|
+
.flatten({ background: options.replaceColor })
|
|
58
|
+
.toBuffer();
|
|
59
|
+
}
|
|
60
|
+
else if (options.replaceImage) {
|
|
61
|
+
const bgBuf = await loadImage(options.replaceImage);
|
|
62
|
+
// Resize background to match image if needed, or simply composite subject over background
|
|
63
|
+
const sizedBgBuf = await sharp(bgBuf).resize(meta.width, meta.height, { fit: 'cover' }).toBuffer();
|
|
64
|
+
outputBuf = await sharp(sizedBgBuf)
|
|
65
|
+
.composite([{ input: outputBuf, blend: 'over' }])
|
|
66
|
+
.toBuffer();
|
|
67
|
+
}
|
|
68
|
+
return ok(outputBuf);
|
|
69
|
+
}
|
|
70
|
+
catch (e) {
|
|
71
|
+
const msg = e.message || '';
|
|
72
|
+
if (msg.includes('HTTP'))
|
|
73
|
+
return err(msg, ErrorCode.FETCH_FAILED);
|
|
74
|
+
if (msg.includes('ENOENT'))
|
|
75
|
+
return err('File not found', ErrorCode.INVALID_INPUT);
|
|
76
|
+
return err(msg, ErrorCode.PROCESSING_FAILED);
|
|
77
|
+
}
|
|
78
|
+
}
|
|
79
|
+
//# sourceMappingURL=remove-bg.js.map
|