managarten/services/matrix-project-doc-bot/src/generation/generation.service.ts

import { Injectable, Inject, Logger } from '@nestjs/common';
import { ConfigService } from '@nestjs/config';
import OpenAI from 'openai';
import { eq, desc } from 'drizzle-orm';
import { DATABASE_CONNECTION } from '../database/database.module';
import { generations, projectItems, projects } from '../database/schema';
import { BLOG_STYLES } from '../config/configuration';
import type { PostgresJsDatabase } from 'drizzle-orm/postgres-js';
import type * as schema from '../database/schema';

type Database = PostgresJsDatabase<typeof schema>;

@Injectable()
export class GenerationService {
	private readonly logger = new Logger(GenerationService.name);
	private readonly openai: OpenAI;
	private readonly model: string;

	constructor(
		@Inject(DATABASE_CONNECTION) private db: Database,
		private configService: ConfigService
	) {
		this.openai = new OpenAI({
			apiKey: this.configService.get<string>('openai.apiKey'),
		});
		this.model = this.configService.get<string>('openai.model') || 'gpt-4o-mini';
	}

	async generateBlogpost(projectId: string, style: keyof typeof BLOG_STYLES): Promise<string> {
		const apiKey = this.configService.get<string>('openai.apiKey');
		if (!apiKey) {
			throw new Error('OpenAI API key not configured');
		}

		// Get project info
		const [project] = await this.db.select().from(projects).where(eq(projects.id, projectId));
		if (!project) {
			throw new Error('Project not found');
		}

		// Get all project items
		const items = await this.db
			.select()
			.from(projectItems)
			.where(eq(projectItems.projectId, projectId))
			.orderBy(projectItems.createdAt);

		if (items.length === 0) {
			throw new Error('Keine Inhalte im Projekt. Füge zuerst Fotos, Sprachnotizen oder Text hinzu.');
		}

		// Build content summary
		const contentSummary = items
			.map((item, index) => {
				const timestamp = item.createdAt.toLocaleString('de-DE');
				switch (item.type) {
					case 'photo':
						return `[Foto ${index + 1}] ${timestamp}${item.content ? `: ${item.content}` : ''}`;
					case 'voice':
						return `[Sprachnotiz ${index + 1}] ${timestamp}: "${item.content || 'Keine Transkription'}"`;
					case 'text':
						return `[Notiz ${index + 1}] ${timestamp}: "${item.content}"`;
					default:
						return '';
				}
			})
			.filter(Boolean)
			.join('\n\n');

		const styleConfig = BLOG_STYLES[style];

		const systemPrompt = `Du bist ein erfahrener Blogger und Content-Creator. ${styleConfig.prompt}

Projektname: "${project.name}"
Erstellt am: ${project.createdAt.toLocaleDateString('de-DE')}

Die folgenden Inhalte wurden während des Projekts gesammelt:`;

		const response = await this.openai.chat.completions.create({
			model: this.model,
			messages: [
				{ role: 'system', content: systemPrompt },
				{ role: 'user', content: contentSummary },
			],
			temperature: 0.7,
			max_tokens: 2000,
		});

		const content = response.choices[0]?.message?.content || '';

		// Save generation
		await this.db.insert(generations).values({
			projectId,
			style,
			content,
		});

		this.logger.log(`Generated ${style} blogpost for project ${projectId}`);
		return content;
	}

	async getLatestGeneration(projectId: string) {
		const [generation] = await this.db
			.select()
			.from(generations)
			.where(eq(generations.projectId, projectId))
			.orderBy(desc(generations.createdAt))
			.limit(1);

		return generation;
	}
}