initial
This commit is contained in:
421
ts/api/handlers.mongodb.ts
Normal file
421
ts/api/handlers.mongodb.ts
Normal file
@@ -0,0 +1,421 @@
|
||||
import * as plugins from '../plugins.js';
|
||||
import type * as interfaces from '../interfaces/index.js';
|
||||
import type { TsView } from '../tsview.classes.tsview.js';
|
||||
|
||||
/**
|
||||
* Register MongoDB API handlers
|
||||
*/
|
||||
export async function registerMongoHandlers(
|
||||
typedrouter: plugins.typedrequest.TypedRouter,
|
||||
tsview: TsView
|
||||
): Promise<void> {
|
||||
// Helper to get the native MongoDB client
|
||||
const getMongoClient = async () => {
|
||||
const db = await tsview.getMongoDb();
|
||||
if (!db) {
|
||||
throw new Error('MongoDB not configured');
|
||||
}
|
||||
// Access the underlying MongoDB client through smartdata
|
||||
return (db as any).mongoDbClient;
|
||||
};
|
||||
|
||||
// Helper to create ObjectId filter
|
||||
const createIdFilter = (documentId: string) => {
|
||||
// Try to treat as ObjectId string - MongoDB driver will handle conversion
|
||||
try {
|
||||
// Import ObjectId from the mongodb package that smartdata uses
|
||||
const { ObjectId } = require('mongodb');
|
||||
if (ObjectId.isValid(documentId)) {
|
||||
return { _id: new ObjectId(documentId) };
|
||||
}
|
||||
} catch {
|
||||
// Fall through to string filter
|
||||
}
|
||||
return { _id: documentId };
|
||||
};
|
||||
|
||||
// List databases
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_ListDatabases>(
|
||||
'listDatabases',
|
||||
async () => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const adminDb = client.db().admin();
|
||||
const result = await adminDb.listDatabases();
|
||||
|
||||
const databases: interfaces.IMongoDatabase[] = result.databases.map((db: any) => ({
|
||||
name: db.name,
|
||||
sizeOnDisk: db.sizeOnDisk,
|
||||
empty: db.empty,
|
||||
}));
|
||||
|
||||
return { databases };
|
||||
} catch (err) {
|
||||
console.error('Error listing databases:', err);
|
||||
return { databases: [] };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// List collections
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_ListCollections>(
|
||||
'listCollections',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collectionsInfo = await db.listCollections().toArray();
|
||||
|
||||
const collections: interfaces.IMongoCollection[] = [];
|
||||
for (const coll of collectionsInfo) {
|
||||
const stats = await db.collection(coll.name).estimatedDocumentCount();
|
||||
collections.push({
|
||||
name: coll.name,
|
||||
count: stats,
|
||||
});
|
||||
}
|
||||
|
||||
return { collections };
|
||||
} catch (err) {
|
||||
console.error('Error listing collections:', err);
|
||||
return { collections: [] };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Create collection
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_CreateCollection>(
|
||||
'createCollection',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
await db.createCollection(reqData.collectionName);
|
||||
return { success: true };
|
||||
} catch (err) {
|
||||
console.error('Error creating collection:', err);
|
||||
return { success: false };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Find documents
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_FindDocuments>(
|
||||
'findDocuments',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const filter = reqData.filter || {};
|
||||
const projection = reqData.projection || {};
|
||||
const sort = reqData.sort || {};
|
||||
const skip = reqData.skip || 0;
|
||||
const limit = reqData.limit || 50;
|
||||
|
||||
const [documents, total] = await Promise.all([
|
||||
collection
|
||||
.find(filter)
|
||||
.project(projection)
|
||||
.sort(sort)
|
||||
.skip(skip)
|
||||
.limit(limit)
|
||||
.toArray(),
|
||||
collection.countDocuments(filter),
|
||||
]);
|
||||
|
||||
// Convert ObjectId to string for JSON serialization
|
||||
const serializedDocs = documents.map((doc: any) => {
|
||||
if (doc._id) {
|
||||
doc._id = doc._id.toString();
|
||||
}
|
||||
return doc;
|
||||
});
|
||||
|
||||
return { documents: serializedDocs, total };
|
||||
} catch (err) {
|
||||
console.error('Error finding documents:', err);
|
||||
return { documents: [], total: 0 };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Get single document
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_GetDocument>(
|
||||
'getDocument',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const filter = createIdFilter(reqData.documentId);
|
||||
const document = await collection.findOne(filter);
|
||||
|
||||
if (document && document._id) {
|
||||
document._id = document._id.toString();
|
||||
}
|
||||
|
||||
return { document };
|
||||
} catch (err) {
|
||||
console.error('Error getting document:', err);
|
||||
return { document: null };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Insert document
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_InsertDocument>(
|
||||
'insertDocument',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const result = await collection.insertOne(reqData.document);
|
||||
|
||||
return { insertedId: result.insertedId.toString() };
|
||||
} catch (err) {
|
||||
console.error('Error inserting document:', err);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Update document
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_UpdateDocument>(
|
||||
'updateDocument',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const filter = createIdFilter(reqData.documentId);
|
||||
|
||||
// Check if update has $ operators
|
||||
const hasOperators = Object.keys(reqData.update).some(k => k.startsWith('$'));
|
||||
const updateDoc = hasOperators ? reqData.update : { $set: reqData.update };
|
||||
|
||||
const result = await collection.updateOne(filter, updateDoc);
|
||||
|
||||
return {
|
||||
success: result.modifiedCount > 0 || result.matchedCount > 0,
|
||||
modifiedCount: result.modifiedCount,
|
||||
};
|
||||
} catch (err) {
|
||||
console.error('Error updating document:', err);
|
||||
return { success: false, modifiedCount: 0 };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Delete document
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_DeleteDocument>(
|
||||
'deleteDocument',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const filter = createIdFilter(reqData.documentId);
|
||||
const result = await collection.deleteOne(filter);
|
||||
|
||||
return {
|
||||
success: result.deletedCount > 0,
|
||||
deletedCount: result.deletedCount,
|
||||
};
|
||||
} catch (err) {
|
||||
console.error('Error deleting document:', err);
|
||||
return { success: false, deletedCount: 0 };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Run aggregation pipeline
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_RunAggregation>(
|
||||
'runAggregation',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const results = await collection.aggregate(reqData.pipeline).toArray();
|
||||
|
||||
// Convert ObjectIds to strings
|
||||
const serializedResults = results.map((doc: any) => {
|
||||
if (doc._id) {
|
||||
doc._id = doc._id.toString();
|
||||
}
|
||||
return doc;
|
||||
});
|
||||
|
||||
return { results: serializedResults };
|
||||
} catch (err) {
|
||||
console.error('Error running aggregation:', err);
|
||||
return { results: [] };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// List indexes
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_ListIndexes>(
|
||||
'listIndexes',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const indexesInfo = await collection.indexes();
|
||||
|
||||
const indexes: interfaces.IMongoIndex[] = indexesInfo.map((idx: any) => ({
|
||||
name: idx.name,
|
||||
keys: idx.key,
|
||||
unique: idx.unique || false,
|
||||
sparse: idx.sparse || false,
|
||||
}));
|
||||
|
||||
return { indexes };
|
||||
} catch (err) {
|
||||
console.error('Error listing indexes:', err);
|
||||
return { indexes: [] };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Create index
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_CreateIndex>(
|
||||
'createIndex',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const indexName = await collection.createIndex(reqData.keys, reqData.options || {});
|
||||
|
||||
return { indexName };
|
||||
} catch (err) {
|
||||
console.error('Error creating index:', err);
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Drop index
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_DropIndex>(
|
||||
'dropIndex',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
await collection.dropIndex(reqData.indexName);
|
||||
|
||||
return { success: true };
|
||||
} catch (err) {
|
||||
console.error('Error dropping index:', err);
|
||||
return { success: false };
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Get collection stats
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_GetCollectionStats>(
|
||||
'getCollectionStats',
|
||||
async (reqData) => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const db = client.db(reqData.databaseName);
|
||||
const collection = db.collection(reqData.collectionName);
|
||||
|
||||
const stats = await db.command({ collStats: reqData.collectionName });
|
||||
const indexCount = (await collection.indexes()).length;
|
||||
|
||||
return {
|
||||
stats: {
|
||||
count: stats.count || 0,
|
||||
size: stats.size || 0,
|
||||
avgObjSize: stats.avgObjSize || 0,
|
||||
storageSize: stats.storageSize || 0,
|
||||
indexCount,
|
||||
},
|
||||
};
|
||||
} catch (err) {
|
||||
console.error('Error getting collection stats:', err);
|
||||
return {
|
||||
stats: {
|
||||
count: 0,
|
||||
size: 0,
|
||||
avgObjSize: 0,
|
||||
storageSize: 0,
|
||||
indexCount: 0,
|
||||
},
|
||||
};
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
|
||||
// Get server status
|
||||
typedrouter.addTypedHandler(
|
||||
new plugins.typedrequest.TypedHandler<interfaces.IReq_GetServerStatus>(
|
||||
'getServerStatus',
|
||||
async () => {
|
||||
try {
|
||||
const client = await getMongoClient();
|
||||
const adminDb = client.db().admin();
|
||||
const serverInfo = await adminDb.serverInfo();
|
||||
const serverStatus = await adminDb.serverStatus();
|
||||
|
||||
return {
|
||||
version: serverInfo.version || 'unknown',
|
||||
uptime: serverStatus.uptime || 0,
|
||||
connections: {
|
||||
current: serverStatus.connections?.current || 0,
|
||||
available: serverStatus.connections?.available || 0,
|
||||
},
|
||||
};
|
||||
} catch (err) {
|
||||
console.error('Error getting server status:', err);
|
||||
return {
|
||||
version: 'unknown',
|
||||
uptime: 0,
|
||||
connections: { current: 0, available: 0 },
|
||||
};
|
||||
}
|
||||
}
|
||||
)
|
||||
);
|
||||
}
|
||||
Reference in New Issue
Block a user