touhou-tagger/dist/cli/dumper.js

从 THBWiki 自动填写东方 Project CD 曲目信息.

import { log } from '../core/debug.js';
import { simplifyMetadataInfo } from '../core/metadata/normalize/normalize.js';
import { CliCommandBase } from './command-base.js';
import { getMetadataConfig } from './options.js';
const handleBufferStringify = (key, value) => {
    const isBuffer = typeof value === 'object' && value !== null && value.type === 'Buffer';
    if (isBuffer) {
        return `<Buffer length=${value.data?.length ?? 0}>`;
    }
    return value;
};
const dumpCover = async (metadatas) => {
    const { writeFileSync } = await import('fs');
    const { resolve } = await import('path');
    const metadata = metadatas.find(m => m.coverImage);
    if (!metadata) {
        return;
    }
    const { default: imageType } = await import('image-type');
    const type = imageType(metadata.coverImage);
    if (!type) {
        return;
    }
    const coverFilename = resolve(process.cwd(), `cover.${type.ext}`);
    log('cover file', coverFilename);
    writeFileSync(coverFilename, metadata.coverImage);
};
export class CliDumper extends CliCommandBase {
    constructor() {
        super();
    }
    async run() {
        await this.loadAlbumOptions();
        const { glob } = await import('glob');
        const { extname, resolve } = await import('path');
        const { writeFileSync, readFileSync } = await import('fs');
        const { readerMappings } = await import('../core/reader/reader-mappings.js');
        const globTypes = Object.keys(readerMappings)
            .map(readerType => readerType.replace(/^\./, ''))
            .join('|');
        const files = (await glob(`./**/*.@(${globTypes})`, { posix: true, cwd: this.workingDir })).sort();
        log({ globTypes });
        log(files);
        if (files.length === 0) {
            console.log('没有找到能够提取的音乐文件');
            return;
        }
        const results = await Promise.all(files.map(async (file) => {
            const type = extname(file);
            const reader = readerMappings[type];
            reader.config = getMetadataConfig(this.options);
            const buffer = readFileSync(resolve(this.workingDir, file));
            const rawTag = await reader.readRaw(buffer);
            const metadata = await reader.read(rawTag);
            return {
                rawTag,
                metadata,
            };
        }));
        const metadatas = results.map(it => it.metadata);
        const rawTags = results.map(it => it.rawTag);
        await simplifyMetadataInfo({
            metadatas,
        });
        writeFileSync(resolve(this.workingDir, 'metadata.json'), JSON.stringify(metadatas.map(({ coverImage, ...restParts }) => {
            return restParts;
        }), undefined, 2));
        if (this.options.debug) {
            writeFileSync(resolve(this.workingDir, 'metadata.debug.json'), JSON.stringify(rawTags, handleBufferStringify, 2));
        }
        if (this.options.cover) {
            await dumpCover(metadatas);
        }
    }
}