@lobehub/chat/packages/file-loaders/src/loaders/doc/index.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import debug from 'debug';
import WordExtractor from 'word-extractor';

import type { DocumentPage, FileLoaderInterface } from '../../types';

const log = debug('file-loaders:doc');

/**
 * Loads legacy Word documents (.doc) using word-extractor.
 * Extracts plain text content and basic metadata from DOC files.
 */
export class DocLoader implements FileLoaderInterface {
  async loadPages(filePath: string): Promise<DocumentPage[]> {
    log('Loading DOC file:', filePath);
    try {
      const extractor = new WordExtractor();
      const extracted: any = await extractor.extract(filePath);

      // Prefer getBody() if available; fallback to common fields
      const pageContent: string =
        extracted && typeof extracted.getBody === 'function'
          ? extracted.getBody()
          : ((extracted?.text as string) ?? '');

      const lines = pageContent.split('\n');
      const lineCount = lines.length;
      const charCount = pageContent.length;

      const page: DocumentPage = {
        charCount,
        lineCount,
        metadata: { pageNumber: 1 },
        pageContent,
      };

      log('DOC loading completed');
      return [page];
    } catch (e) {
      const error = e as Error;
      log('Error encountered while loading DOC file');
      console.error(`Error loading DOC file ${filePath}: ${error.message}`);

      const errorPage: DocumentPage = {
        charCount: 0,
        lineCount: 0,
        metadata: { error: `Failed to load DOC file: ${error.message}` },
        pageContent: '',
      };
      return [errorPage];
    }
  }

  async aggregateContent(pages: DocumentPage[]): Promise<string> {
    log('Aggregating content from', pages.length, 'DOC pages');
    return pages.map((p) => p.pageContent).join('\n\n');
  }
}