Initial public upload from Projekte/learn-harness-engineering (part 3)

bde76b9 verified about 1 month ago

2.57 kB

import * as fs from 'fs';
import * as path from 'path';
import { v4 as uuidv4 } from 'uuid';
import { Document } from '../shared/types';
import { PersistenceService } from './persistence-service';

const DOCUMENTS_META = 'documents-meta.json';

export class DocumentService {
  private persistence: PersistenceService;

  constructor(persistence: PersistenceService) {
    this.persistence = persistence;
  }

  /** List all imported documents. */
  listDocuments(): Document[] {
    const docs = this.persistence.readJson<Document[]>(DOCUMENTS_META);
    return docs ?? [];
  }

  /** Import a file from the given path. */
  importDocument(filePath: string): Document {
    if (!fs.existsSync(filePath)) {
      throw new Error(`File not found: ${filePath}`);
    }

    const filename = path.basename(filePath);
    const content = fs.readFileSync(filePath, 'utf-8');
    const stats = fs.statSync(filePath);

    const doc: Document = {
      id: uuidv4(),
      title: filename.replace(/\.[^.]+$/, ''),
      filename,
      importedAt: new Date().toISOString(),
      size: stats.size,
      status: 'imported',
    };

    // Copy file to data directory
    this.persistence.copyFileToDocuments(filePath, filename);

    // Store content for indexing
    this.persistence.writeText(`content/${doc.id}.txt`, content);

    // Update metadata
    const docs = this.listDocuments();
    docs.push(doc);
    this.persistence.writeJson(DOCUMENTS_META, docs);

    return doc;
  }

  /** Get a single document by ID. */
  getDocument(id: string): Document | null {
    const docs = this.listDocuments();
    return docs.find(d => d.id === id) ?? null;
  }

  /** Get the text content of a document. */
  getDocumentContent(id: string): string | null {
    return this.persistence.readText(`content/${id}.txt`);
  }

  /** Update a document's metadata. */
  updateDocument(id: string, updates: Partial<Document>): Document | null {
    const docs = this.listDocuments();
    const index = docs.findIndex(d => d.id === id);
    if (index === -1) return null;

    docs[index] = { ...docs[index], ...updates };
    this.persistence.writeJson(DOCUMENTS_META, docs);
    return docs[index];
  }

  /** Delete a document by ID. */
  deleteDocument(id: string): boolean {
    const docs = this.listDocuments();
    const doc = docs.find(d => d.id === id);
    if (!doc) return false;

    this.persistence.deleteFromDocuments(doc.filename);

    const updated = docs.filter(d => d.id !== id);
    this.persistence.writeJson(DOCUMENTS_META, updated);
    return true;
  }
}

	import * as fs from 'fs';
	import * as path from 'path';
	import { v4 as uuidv4 } from 'uuid';
	import { Document } from '../shared/types';
	import { PersistenceService } from './persistence-service';

	const DOCUMENTS_META = 'documents-meta.json';

	export class DocumentService {
	private persistence: PersistenceService;

	constructor(persistence: PersistenceService) {
	this.persistence = persistence;
	}

	/** List all imported documents. */
	listDocuments(): Document[] {
	const docs = this.persistence.readJson<Document[]>(DOCUMENTS_META);
	return docs ?? [];
	}

	/** Import a file from the given path. */
	importDocument(filePath: string): Document {
	if (!fs.existsSync(filePath)) {
	throw new Error(`File not found: ${filePath}`);
	}

	const filename = path.basename(filePath);
	const content = fs.readFileSync(filePath, 'utf-8');
	const stats = fs.statSync(filePath);

	const doc: Document = {
	id: uuidv4(),
	title: filename.replace(/\.[^.]+$/, ''),
	filename,
	importedAt: new Date().toISOString(),
	size: stats.size,
	status: 'imported',
	};

	// Copy file to data directory
	this.persistence.copyFileToDocuments(filePath, filename);

	// Store content for indexing
	this.persistence.writeText(`content/${doc.id}.txt`, content);

	// Update metadata
	const docs = this.listDocuments();
	docs.push(doc);
	this.persistence.writeJson(DOCUMENTS_META, docs);

	return doc;
	}

	/** Get a single document by ID. */
	getDocument(id: string): Document \| null {
	const docs = this.listDocuments();
	return docs.find(d => d.id === id) ?? null;
	}

	/** Get the text content of a document. */
	getDocumentContent(id: string): string \| null {
	return this.persistence.readText(`content/${id}.txt`);
	}

	/** Update a document's metadata. */
	updateDocument(id: string, updates: Partial<Document>): Document \| null {
	const docs = this.listDocuments();
	const index = docs.findIndex(d => d.id === id);
	if (index === -1) return null;

	docs[index] = { ...docs[index], ...updates };
	this.persistence.writeJson(DOCUMENTS_META, docs);
	return docs[index];
	}

	/** Delete a document by ID. */
	deleteDocument(id: string): boolean {
	const docs = this.listDocuments();
	const doc = docs.find(d => d.id === id);
	if (!doc) return false;

	this.persistence.deleteFromDocuments(doc.filename);

	const updated = docs.filter(d => d.id !== id);
	this.persistence.writeJson(DOCUMENTS_META, updated);
	return true;
	}
	}