import 'package:trainhub_flutter/data/database/app_database.dart';
import 'package:trainhub_flutter/domain/entities/chat_session.dart';
import 'package:trainhub_flutter/domain/entities/chat_message.dart';

class ChatMapper {
  ChatMapper._();

  static ChatSessionEntity sessionToEntity(ChatSession row) {
    return ChatSessionEntity(
      id: row.id,
      title: row.title,
      createdAt: row.createdAt,
      updatedAt: row.updatedAt,
    );
  }

  static ChatMessageEntity messageToEntity(ChatMessage row) {
    return ChatMessageEntity(
      id: row.id,
      sessionId: row.sessionId,
      role: row.role,
      content: row.content,
      createdAt: row.createdAt,
    );
  }
}