emotions-mcp-server
Version:
MCP Server for accessing PostgreSQL emotions database
295 lines (294 loc) • 10.6 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.EmotionsDbService = void 0;
const pg_1 = require("pg");
class EmotionsDbService {
constructor(connectionString) {
this.tableName = 'emotions';
this.pool = new pg_1.Pool({
connectionString,
});
}
async closeConnection() {
await this.pool.end();
}
async tableExists() {
const client = await this.pool.connect();
try {
const result = await client.query(`
SELECT EXISTS (
SELECT FROM information_schema.tables
WHERE table_name = $1
)
`, [this.tableName]);
return result.rows[0].exists;
}
finally {
client.release();
}
}
async createTable() {
const client = await this.pool.connect();
try {
await client.query(`
CREATE TABLE IF NOT EXISTS ${this.tableName} (
id SERIAL PRIMARY KEY,
user_context TEXT NOT NULL,
nummer INTEGER NOT NULL,
emotion TEXT NOT NULL,
datum DATE,
alter INTEGER,
quellenart TEXT CHECK (quellenart IN ('Eigene Emotion', 'Übernommene Emotion', 'Geerbte Emotion')),
quelle TEXT,
koerperteil TEXT,
auswirkungen TEXT,
bemerkungen TEXT,
UNIQUE(user_context, nummer)
)
`);
}
finally {
client.release();
}
}
async insertEmotion(userContext, emotion) {
// Validate required fields
if (!emotion.emotion) {
throw new Error('Das Feld "emotion" ist erforderlich');
}
// Validate quelle field when necessary
if ((emotion.quellenart === 'Übernommene Emotion' || emotion.quellenart === 'Geerbte Emotion') && !emotion.quelle) {
throw new Error('Das Feld "quelle" ist erforderlich für übernommene oder geerbte Emotionen');
}
const client = await this.pool.connect();
try {
// Start a transaction
await client.query('BEGIN');
// Always generate the next nummer value for this user context
const nextNummerResult = await client.query(`SELECT COALESCE(MAX(nummer), 0) + 1 AS next_nummer FROM ${this.tableName} WHERE user_context = $1`, [userContext]);
const nummer = nextNummerResult.rows[0].next_nummer;
const result = await client.query(`INSERT INTO ${this.tableName} (
user_context, nummer, emotion, datum, alter, quellenart, quelle, koerperteil, auswirkungen, bemerkungen
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
RETURNING id, user_context AS "userContext", nummer, emotion, datum, alter, quellenart, quelle, koerperteil, auswirkungen, bemerkungen`, [
userContext,
nummer,
emotion.emotion,
emotion.datum || null,
emotion.alter || null,
emotion.quellenart || null,
emotion.quelle || null,
emotion.koerperteil || null,
emotion.auswirkungen || null,
emotion.bemerkungen || null
]);
// Commit the transaction
await client.query('COMMIT');
return result.rows[0];
}
catch (error) {
// Rollback in case of error
await client.query('ROLLBACK');
throw error;
}
finally {
client.release();
}
}
async deleteEmotion(userContext, emotionNumber) {
const client = await this.pool.connect();
try {
const result = await client.query(`DELETE FROM ${this.tableName} WHERE nummer = $1 AND user_context = $2 RETURNING id`, [emotionNumber, userContext]);
return (result.rowCount ?? 0) > 0;
}
finally {
client.release();
}
}
async getEmotionByNumber(userContext, emotionNumber) {
const client = await this.pool.connect();
try {
const result = await client.query(`SELECT
id,
user_context AS "userContext",
nummer,
emotion,
datum,
alter,
quellenart,
quelle,
koerperteil,
auswirkungen,
bemerkungen
FROM ${this.tableName}
WHERE nummer = $1 AND user_context = $2`, [emotionNumber, userContext]);
return result.rows.length > 0 ? result.rows[0] : null;
}
finally {
client.release();
}
}
buildFilterQuery(filter) {
const conditions = [];
const params = [];
let paramIndex = 1;
// Add user_context as first parameter
params.push(''); // Will be set in getEmotions method
conditions.push(`user_context = $${paramIndex++}`);
if (filter.nummer !== undefined) {
conditions.push(`nummer = $${paramIndex++}`);
params.push(filter.nummer);
}
if (filter.emotion) {
conditions.push(`emotion ILIKE $${paramIndex++}`);
params.push(`%${filter.emotion}%`);
}
if (filter.datumVon) {
conditions.push(`datum >= $${paramIndex++}`);
params.push(filter.datumVon);
}
if (filter.datumBis) {
conditions.push(`datum <= $${paramIndex++}`);
params.push(filter.datumBis);
}
if (filter.alterVon !== undefined) {
conditions.push(`alter >= $${paramIndex++}`);
params.push(filter.alterVon);
}
if (filter.alterBis !== undefined) {
conditions.push(`alter <= $${paramIndex++}`);
params.push(filter.alterBis);
}
if (filter.quellenart) {
conditions.push(`quellenart = $${paramIndex++}`);
params.push(filter.quellenart);
}
if (filter.quelle) {
conditions.push(`quelle ILIKE $${paramIndex++}`);
params.push(`%${filter.quelle}%`);
}
if (filter.koerperteil) {
conditions.push(`koerperteil ILIKE $${paramIndex++}`);
params.push(`%${filter.koerperteil}%`);
}
if (filter.auswirkungen) {
conditions.push(`auswirkungen ILIKE $${paramIndex++}`);
params.push(`%${filter.auswirkungen}%`);
}
if (filter.bemerkungen) {
conditions.push(`bemerkungen ILIKE $${paramIndex++}`);
params.push(`%${filter.bemerkungen}%`);
}
const query = conditions.length > 0
? `WHERE ${conditions.join(' AND ')}`
: '';
return { query, params };
}
async getEmotions(userContext, filter = {}) {
const { query, params } = this.buildFilterQuery(filter);
params[0] = userContext; // Set the first parameter to userContext
const client = await this.pool.connect();
try {
const result = await client.query(`SELECT
id,
user_context AS "userContext",
nummer,
emotion,
datum,
alter,
quellenart,
quelle,
koerperteil,
auswirkungen,
bemerkungen
FROM ${this.tableName}
${query}
ORDER BY nummer ASC`, params);
return result.rows;
}
finally {
client.release();
}
}
async updateEmotion(userContext, emotionNumber, emotion) {
// Validate required fields
if (!emotion.emotion) {
throw new Error('Das Feld "emotion" ist erforderlich');
}
// Validate quelle field when necessary
if ((emotion.quellenart === 'Übernommene Emotion' || emotion.quellenart === 'Geerbte Emotion') && !emotion.quelle) {
throw new Error('Das Feld "quelle" ist erforderlich für übernommene oder geerbte Emotionen');
}
const client = await this.pool.connect();
try {
// Start a transaction
await client.query('BEGIN');
// Check if the emotion exists for this user
const checkResult = await client.query(`SELECT id, nummer FROM ${this.tableName} WHERE nummer = $1 AND user_context = $2`, [emotionNumber, userContext]);
if (checkResult.rows.length === 0) {
throw new Error(`Emotion mit der Nummer ${emotionNumber} wurde nicht gefunden`);
}
// Use the existing nummer - prevent nummer from being changed
const existingNummer = checkResult.rows[0].nummer;
const result = await client.query(`UPDATE ${this.tableName} SET
emotion = $1,
datum = $2,
alter = $3,
quellenart = $4,
quelle = $5,
koerperteil = $6,
auswirkungen = $7,
bemerkungen = $8
WHERE nummer = $9 AND user_context = $10
RETURNING id, user_context AS "userContext", nummer, emotion, datum, alter, quellenart, quelle, koerperteil, auswirkungen, bemerkungen`, [
emotion.emotion,
emotion.datum || null,
emotion.alter || null,
emotion.quellenart || null,
emotion.quelle || null,
emotion.koerperteil || null,
emotion.auswirkungen || null,
emotion.bemerkungen || null,
emotionNumber,
userContext
]);
// Commit the transaction
await client.query('COMMIT');
return result.rows.length > 0 ? result.rows[0] : null;
}
catch (error) {
// Rollback in case of error
await client.query('ROLLBACK');
throw error;
}
finally {
client.release();
}
}
async getRecentEmotions(userContext, limit) {
const client = await this.pool.connect();
try {
const result = await client.query(`SELECT
id,
user_context AS "userContext",
nummer,
emotion,
datum,
alter,
quellenart,
quelle,
koerperteil,
auswirkungen,
bemerkungen
FROM ${this.tableName}
WHERE user_context = $1
ORDER BY nummer DESC
LIMIT $2`, [userContext, limit]);
return result.rows;
}
finally {
client.release();
}
}
}
exports.EmotionsDbService = EmotionsDbService;