@chriswiegman/hugo-tools 0.1.2 → 0.2.1

package/src/pick-image.js

@@ -1,80 +1,141 @@
 #!/usr/bin/env node
 'use strict';
 
-// Lists images in the configured images directory for the current year/month
-// and copies the selected path to the clipboard.
+// Moves an image into assets/images/<year>/<month>/ and inserts a reference
+// into a Hugo content file — appending to the frontmatter images: array when
+// present, or appending a markdown image tag to the body otherwise.
 
 const fs = require('fs');
 const path = require('path');
-const readline = require('readline');
 const { execSync, execFileSync } = require('child_process');
 const config = require('./config');
 
-config.requireHugoSite();
-
 const IMAGE_EXTS = new Set(['.jpg', '.jpeg', '.png', '.webp', '.gif']);
 
-const now = new Date();
-const year = now.getFullYear().toString();
-const month = String(now.getMonth() + 1).padStart(2, '0');
-const imageDir = path.join(process.cwd(), config.imagesDir, year, month);
-const imagePath = `/${config.imagesDir}/${year}/${month}`;
+function urlFromImagesDir(imagesDir, year, month, filename) {
+	const urlBase = imagesDir.replace(/^assets\//, '');
 
-if (!fs.existsSync(imageDir)) {
-  console.error(`No image directory found at ${imageDir}`);
-  process.exit(1);
+	return `/${urlBase}/${year}/${month}/${filename}`;
 }
 
-const files = fs.readdirSync(imageDir)
-  .filter(f => IMAGE_EXTS.has(path.extname(f).toLowerCase()))
-  .sort();
-
-if (files.length === 0) {
-  console.error(`No images found in ${imageDir}`);
-  process.exit(1);
+function appendMarkdown(content, urlPath) {
+	return `${content.trimEnd()}\n\n![](${urlPath})\n`;
 }
 
-console.log(`\nImages in ${imagePath}/:\n`);
-files.forEach((f, i) => console.log(`  ${i + 1}) ${f}`));
-console.log('');
+function insertImage(content, urlPath) {
+	const fmEnd = content.indexOf('\n---', 4);
+
+	if (!content.startsWith('---\n') || fmEnd === -1) {
+		return appendMarkdown(content, urlPath);
+	}
+
+	const frontmatter = content.slice(4, fmEnd);
+
+	if (!/^images:/m.test(frontmatter)) {
+		return appendMarkdown(content, urlPath);
+	}
+
+	return content.replace(
+		/^(images:)((?:\n {2}-[^\n]*)*)/m,
+		(_, key, block) => {
+			const existing = block.split('\n').filter((line) => /^ {2}- \S/.test(line));
+
+			return [key, ...existing, `  - ${urlPath}`].join('\n');
+		},
+	);
+}
 
 function copyToClipboard(text) {
-  const platform = process.platform;
-  if (platform === 'darwin') {
-    execSync(`echo ${JSON.stringify(text)} | pbcopy`);
-    return;
-  }
-  if (platform === 'win32') {
-    execFileSync('clip', { input: text });
-    return;
-  }
-  // Linux: try common clipboard tools in order
-  const tools = [
-    ['xclip', ['-selection', 'clipboard']],
-    ['xsel', ['--clipboard', '--input']],
-    ['wl-copy', []],
-  ];
-  for (const [cmd, args] of tools) {
-    try {
-      execFileSync(cmd, args, { input: text });
-      return;
-    } catch {
-      // try next
-    }
-  }
-  console.log('(clipboard unavailable — copy the path above manually)');
+	const platform = process.platform;
+
+	if (platform === 'darwin') {
+		execSync(`echo ${JSON.stringify(text)} | pbcopy`);
+		return;
+	}
+
+	if (platform === 'win32') {
+		execFileSync('clip', { input: text });
+		return;
+	}
+
+	const tools = [
+		['xclip', ['-selection', 'clipboard']],
+		['xsel', ['--clipboard', '--input']],
+		['wl-copy', []],
+	];
+
+	for (const [cmd, args] of tools) {
+		try {
+			execFileSync(cmd, args, { input: text });
+			return;
+		} catch {
+			// try next
+		}
+	}
+	console.log('(clipboard unavailable — copy the path above manually)');
+}
+
+if (require.main === module) {
+	config.requireHugoSite();
+
+	const args = process.argv.slice(2);
+
+	if (args.length === 0) {
+		console.error('Usage: pick-image <image-file> [content-file]');
+		process.exit(1);
+	}
+
+	const imageFile = path.resolve(args[0]);
+	const contentFile = args[1] ? path.resolve(args[1]) : null;
+
+	if (!fs.existsSync(imageFile)) {
+		console.error(`Image not found: ${imageFile}`);
+		process.exit(1);
+	}
+
+	const ext = path.extname(imageFile).toLowerCase();
+
+	if (!IMAGE_EXTS.has(ext)) {
+		console.error(`Unsupported image format: ${ext}`);
+		process.exit(1);
+	}
+
+	const now = new Date();
+	const year = now.getFullYear().toString();
+	const month = String(now.getMonth() + 1).padStart(2, '0');
+	const filename = path.basename(imageFile);
+	const destDir = path.join(process.cwd(), config.imagesDir, year, month);
+	const destFile = path.join(destDir, filename);
+	const urlPath = urlFromImagesDir(config.imagesDir, year, month, filename);
+
+	fs.mkdirSync(destDir, { recursive: true });
+
+	if (path.resolve(imageFile) !== path.resolve(destFile)) {
+		if (fs.existsSync(destFile)) {
+			console.error(`File already exists at destination: ${destFile}`);
+			process.exit(1);
+		}
+
+		fs.renameSync(imageFile, destFile);
+		console.log(`Moved to: ${destFile}`);
+	}
+
+	if (!contentFile) {
+		copyToClipboard(urlPath);
+		console.log(`Path copied to clipboard: ${urlPath}`);
+		process.exit(0);
+	}
+
+	if (!fs.existsSync(contentFile)) {
+		console.error(`Content file not found: ${contentFile}`);
+		process.exit(1);
+	}
+
+	const original = fs.readFileSync(contentFile, 'utf8');
+	const updated = insertImage(original, urlPath);
+
+	fs.writeFileSync(contentFile, updated);
+	console.log(`Updated: ${path.relative(process.cwd(), contentFile)}`);
 }
 
-const rl = readline.createInterface({ input: process.stdin, output: process.stdout });
-rl.question(`Select image (1-${files.length}): `, answer => {
-  rl.close();
-  const choice = parseInt(answer, 10);
-  if (choice >= 1 && choice <= files.length) {
-    const result = `${imagePath}/${files[choice - 1]}`;
-    copyToClipboard(result);
-    console.log(`\nCopied to clipboard: ${result}`);
-  } else {
-    console.error('Invalid selection');
-    process.exit(1);
-  }
-});
+module.exports = { urlFromImagesDir, appendMarkdown, insertImage };

package/src/pick-image.test.mjs

@@ -0,0 +1,121 @@
+import assert from 'node:assert/strict';
+import { test } from 'node:test';
+import { createRequire } from 'node:module';
+
+const require = createRequire(import.meta.url);
+const { urlFromImagesDir, appendMarkdown, insertImage } = require('./pick-image.js');
+
+// ---------------------------------------------------------------------------
+// urlFromImagesDir
+// ---------------------------------------------------------------------------
+
+test('urlFromImagesDir: strips assets/ prefix', () => {
+	assert.equal(urlFromImagesDir('assets/images', '2026', '05', 'photo.jpg'), '/images/2026/05/photo.jpg');
+});
+
+test('urlFromImagesDir: preserves dir without assets/ prefix', () => {
+	assert.equal(urlFromImagesDir('images', '2026', '05', 'photo.jpg'), '/images/2026/05/photo.jpg');
+});
+
+test('urlFromImagesDir: preserves custom directory name', () => {
+	assert.equal(urlFromImagesDir('assets/media', '2026', '05', 'photo.jpg'), '/media/2026/05/photo.jpg');
+});
+
+// ---------------------------------------------------------------------------
+// appendMarkdown
+// ---------------------------------------------------------------------------
+
+test('appendMarkdown: appends image tag with blank line separator', () => {
+	const result = appendMarkdown('Body text.\n', '/images/2026/05/photo.jpg');
+
+	assert.equal(result, 'Body text.\n\n![](/images/2026/05/photo.jpg)\n');
+});
+
+test('appendMarkdown: trims trailing whitespace before appending', () => {
+	const result = appendMarkdown('Body text.\n\n\n', '/images/2026/05/photo.jpg');
+
+	assert.equal(result, 'Body text.\n\n![](/images/2026/05/photo.jpg)\n');
+});
+
+test('appendMarkdown: works when content has no trailing newline', () => {
+	const result = appendMarkdown('Body text.', '/images/2026/05/photo.jpg');
+
+	assert.equal(result, 'Body text.\n\n![](/images/2026/05/photo.jpg)\n');
+});
+
+// ---------------------------------------------------------------------------
+// insertImage — no frontmatter or no images: field → markdown fallback
+// ---------------------------------------------------------------------------
+
+test('insertImage: appends markdown when no frontmatter', () => {
+	const content = 'Just a plain body.\n';
+	const result = insertImage(content, '/images/2026/05/photo.jpg');
+
+	assert.ok(result.includes('![](/images/2026/05/photo.jpg)'));
+	assert.ok(result.includes('Just a plain body.'));
+});
+
+test('insertImage: appends markdown when frontmatter has no images: field', () => {
+	const content = '---\ntitle: "My Post"\n---\nBody text.\n';
+	const result = insertImage(content, '/images/2026/05/photo.jpg');
+
+	assert.ok(result.includes('![](/images/2026/05/photo.jpg)'));
+	assert.ok(result.includes('Body text.'));
+});
+
+// ---------------------------------------------------------------------------
+// insertImage — frontmatter images: array manipulation
+// ---------------------------------------------------------------------------
+
+test('insertImage: replaces placeholder (  -) with real path', () => {
+	const content = '---\ntitle: "My Post"\nimages:\n  -\n---\nBody.\n';
+	const result = insertImage(content, '/images/2026/05/photo.jpg');
+
+	assert.ok(result.includes('  - /images/2026/05/photo.jpg'));
+	assert.ok(!result.includes('\n  -\n'));
+	assert.ok(!result.includes('![]('));
+});
+
+test('insertImage: appends to existing single-item images array', () => {
+	const content = '---\ntitle: "My Post"\nimages:\n  - /images/2026/01/old.jpg\n---\nBody.\n';
+	const result = insertImage(content, '/images/2026/05/new.jpg');
+
+	assert.ok(result.includes('  - /images/2026/01/old.jpg'));
+	assert.ok(result.includes('  - /images/2026/05/new.jpg'));
+	assert.ok(!result.includes('![]('));
+});
+
+test('insertImage: appends to existing multi-item images array', () => {
+	const content = '---\nimages:\n  - /images/2026/01/a.jpg\n  - /images/2026/02/b.jpg\n---\n';
+	const result = insertImage(content, '/images/2026/05/c.jpg');
+	const lines = result.split('\n');
+	const imgLines = lines.filter((l) => l.startsWith('  - '));
+
+	assert.equal(imgLines.length, 3);
+	assert.equal(imgLines[2], '  - /images/2026/05/c.jpg');
+});
+
+test('insertImage: does not duplicate existing items', () => {
+	const content = '---\nimages:\n  - /images/2026/01/a.jpg\n---\n';
+	const result = insertImage(content, '/images/2026/05/b.jpg');
+	const count = (result.match(/ {2}- \/images/g) || []).length;
+
+	assert.equal(count, 2);
+});
+
+test('insertImage: leaves other frontmatter fields untouched', () => {
+	const content = '---\ntitle: "My Post"\nimages:\n  -\ncategories:\n  - Tech\ntags:\n  - go\n---\nBody.\n';
+	const result = insertImage(content, '/images/2026/05/photo.jpg');
+
+	assert.ok(result.includes('title: "My Post"'));
+	assert.ok(result.includes('categories:\n  - Tech'));
+	assert.ok(result.includes('tags:\n  - go'));
+	assert.ok(result.includes('Body.'));
+});
+
+test('insertImage: body content is preserved when updating frontmatter', () => {
+	const content = '---\ntitle: "My Post"\nimages:\n  -\n---\nThis is the body.\n';
+	const result = insertImage(content, '/images/2026/05/photo.jpg');
+
+	assert.ok(result.includes('This is the body.'));
+});