image-edit-tools 1.0.0

package/tests/unit/remove-bg.test.ts

@@ -0,0 +1,42 @@
+import { describe, it, expect, beforeAll, vi } from 'vitest'
+import { readFileSync } from 'fs'
+import { join, dirname } from 'path'
+import { fileURLToPath } from 'url'
+import { removeBg } from '../../src/ops/remove-bg.js'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const fixture = (name: string) => readFileSync(join(__dirname, '../fixtures', name))
+
+// Since tests run in CI/offline and we shouldn't download huge models, 
+// we test that the function gracefully hits the MODEL_NOT_FOUND error 
+// if AutoModel fails, or passes if the network enables it.
+// We explicitly mock '@xenova/transformers' to simulate network failure or success.
+
+vi.mock('@xenova/transformers', async (importOriginal) => {
+  return {
+    ...(await importOriginal<typeof import('@xenova/transformers')>()),
+    AutoModel: {
+      from_pretrained: vi.fn().mockRejectedValue(new Error('Mock network failure'))
+    },
+    AutoProcessor: {
+      from_pretrained: vi.fn()
+    }
+  }
+})
+
+describe('removeBg', () => {
+  let sampleJpeg: Buffer
+
+  beforeAll(() => {
+    sampleJpeg = fixture('sample.jpg') 
+  })
+
+  it('handles unavailable model offline correctly', async () => {
+    const result = await removeBg(sampleJpeg, {})
+    expect(result.ok).toBe(false)
+    if (!result.ok) {
+      expect(result.code).toBe('MODEL_NOT_FOUND')
+    }
+  })
+})

package/tests/unit/resize.test.ts

@@ -0,0 +1,70 @@
+import { describe, it, expect, beforeAll } from 'vitest'
+import { readFileSync } from 'fs'
+import { join, dirname } from 'path'
+import { fileURLToPath } from 'url'
+import { resize } from '../../src/ops/resize.js'
+import sharp from 'sharp'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const fixture = (name: string) => readFileSync(join(__dirname, '../fixtures', name))
+
+describe('resize', () => {
+  let sampleJpeg: Buffer
+
+  beforeAll(() => {
+    sampleJpeg = fixture('sample.jpg') // 400x300
+  })
+
+  it('resizes using px dimensions', async () => {
+    const result = await resize(sampleJpeg, { width: 100, height: 100 })
+    expect(result.ok).toBe(true)
+    if (!result.ok) return
+    const meta = await sharp(result.data).metadata()
+    expect(meta.width).toBe(100)
+    expect(meta.height).toBe(100)
+  })
+
+  it('resizes using scale', async () => {
+    const result = await resize(sampleJpeg, { scale: 0.5 })
+    expect(result.ok).toBe(true)
+    if (!result.ok) return
+    const meta = await sharp(result.data).metadata()
+    expect(meta.width).toBe(200)
+    expect(meta.height).toBe(150)
+  })
+
+  it('preserves aspect ratio for single axis', async () => {
+    const result = await resize(sampleJpeg, { width: 200 })
+    expect(result.ok).toBe(true)
+    if (!result.ok) return
+    const meta = await sharp(result.data).metadata()
+    expect(meta.width).toBe(200)
+    expect(meta.height).toBe(150) // 400:300 is 4:3 -> 200:150
+  })
+
+  it('applies fit parameter correctly', async () => {
+    const result = await resize(sampleJpeg, { width: 100, height: 200, fit: 'contain' })
+    expect(result.ok).toBe(true)
+    if (!result.ok) return
+    const meta = await sharp(result.data).metadata()
+    expect(meta.width).toBe(100)
+    expect(meta.height).toBe(200)
+  })
+
+  it('respects withoutEnlargement', async () => {
+    const result = await resize(sampleJpeg, { width: 1000, height: 1000, withoutEnlargement: true })
+    expect(result.ok).toBe(true)
+    if (!result.ok) return
+    const meta = await sharp(result.data).metadata()
+    expect(meta.width).toBe(400) // Original width
+    expect(meta.height).toBe(300) // Original height
+  })
+
+  it('returns error on zero dimensions', async () => {
+    const result = await resize(sampleJpeg, { width: 0 })
+    expect(result.ok).toBe(false)
+    if (result.ok) return
+    expect(result.code).toBe('INVALID_INPUT')
+  })
+})

package/tests/unit/watermark.test.ts

@@ -0,0 +1,54 @@
+import { describe, it, expect, beforeAll } from 'vitest'
+import { readFileSync } from 'fs'
+import { join, dirname } from 'path'
+import { fileURLToPath } from 'url'
+import { watermark } from '../../src/ops/watermark.js'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const fixture = (name: string) => readFileSync(join(__dirname, '../fixtures', name))
+
+describe('watermark', () => {
+  let sampleJpeg: Buffer
+  let logoPng: Buffer
+
+  beforeAll(() => {
+    sampleJpeg = fixture('sample.jpg') // 400x300
+    logoPng = fixture('logo.png')      // 100x100
+  })
+
+  it('adds text watermark at bottom-right', async () => {
+    const result = await watermark(sampleJpeg, {
+      type: 'text', text: 'Confidential', position: 'bottom-right'
+    })
+    expect(result.ok).toBe(true)
+  })
+
+  it('adds text watermark tiled', async () => {
+    const result = await watermark(sampleJpeg, {
+      type: 'text', text: 'Draft', position: 'tile', tileSpacing: 100
+    })
+    expect(result.ok).toBe(true)
+  })
+
+  it('adds image watermark at top-left', async () => {
+    const result = await watermark(sampleJpeg, {
+      type: 'image', image: logoPng, position: 'top-left'
+    })
+    expect(result.ok).toBe(true)
+  })
+
+  it('adds image watermark tiled with opacity and scale', async () => {
+    const result = await watermark(sampleJpeg, {
+      type: 'image', image: logoPng, position: 'tile', opacity: 0.3, scale: 0.5, tileSpacing: 20
+    })
+    expect(result.ok).toBe(true)
+  })
+
+  it('returns error if missing valid type', async () => {
+    const result = await watermark(sampleJpeg, {} as any)
+    expect(result.ok).toBe(false)
+    if (result.ok) return
+    expect(result.code).toBe('INVALID_INPUT')
+  })
+})

package/tsconfig.json

@@ -0,0 +1,15 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "outDir": "dist",
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "strict": true,
+    "esModuleInterop": true
+  },
+  "include": ["src"],
+  "exclude": ["node_modules", "dist", "tests"]
+}

package/vitest.config.ts

@@ -0,0 +1,27 @@
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  test: {
+    globals: true,
+    environment: 'node',
+    coverage: {
+      provider: 'v8',
+      reporter: ['text', 'json', 'html'],
+      thresholds: {
+        lines: 70,
+        functions: 70,
+        branches: 50,
+      },
+      exclude: [
+        'scripts/**',
+        'src/index.ts',
+        'src/mcp/index.ts',
+        'src/mcp/server.ts',
+        'src/mcp/tools.ts', // mostly wiring, E2E tested
+        'src/ops/detect-subject.ts', // unimplemented/stubbed
+        'src/ops/remove-bg.ts' // heavy AI model, requires optional mocking
+      ]
+    },
+    testTimeout: 30000,
+  },
+})