feat: Migrate Street and Task models from MongoDB to CouchDB
- Replace Street model with CouchDB-based implementation - Replace Task model with CouchDB-based implementation - Update routes to use new model interfaces - Handle geospatial queries with CouchDB design documents - Maintain adoption functionality and middleware - Use denormalized document structure with embedded data - Update test files to work with new models - Ensure API compatibility while using CouchDB underneath 🤖 Generated with [AI Assistant] Co-Authored-By: AI Assistant <noreply@ai-assistant.com>
This commit is contained in:
William Valentin
@@ -1,62 +1,312 @@
|
||||
const mongoose = require("mongoose");
|
||||
const couchdbService = require("../services/couchdbService");
|
||||
|
||||
const TaskSchema = new mongoose.Schema(
|
||||
{
|
||||
street: {
|
||||
type: mongoose.Schema.Types.ObjectId,
|
||||
ref: "Street",
|
||||
required: true,
|
||||
index: true,
|
||||
},
|
||||
description: {
|
||||
type: String,
|
||||
required: true,
|
||||
},
|
||||
completedBy: {
|
||||
type: mongoose.Schema.Types.ObjectId,
|
||||
ref: "User",
|
||||
index: true,
|
||||
},
|
||||
status: {
|
||||
type: String,
|
||||
enum: ["pending", "completed"],
|
||||
default: "pending",
|
||||
index: true,
|
||||
},
|
||||
},
|
||||
{
|
||||
timestamps: true,
|
||||
},
|
||||
);
|
||||
|
||||
// Compound indexes for common queries
|
||||
TaskSchema.index({ street: 1, status: 1 });
|
||||
TaskSchema.index({ completedBy: 1, status: 1 });
|
||||
|
||||
// Update user relationship when task is completed
|
||||
TaskSchema.post("save", async function (doc) {
|
||||
if (doc.completedBy && doc.status === "completed") {
|
||||
const User = mongoose.model("User");
|
||||
|
||||
// Add task to user's completedTasks if not already there
|
||||
await User.updateOne(
|
||||
{ _id: doc.completedBy },
|
||||
{ $addToSet: { completedTasks: doc._id } }
|
||||
);
|
||||
class Task {
|
||||
constructor(data) {
|
||||
this._id = data._id || null;
|
||||
this._rev = data._rev || null;
|
||||
this.type = "task";
|
||||
this.street = data.street || null;
|
||||
this.description = data.description;
|
||||
this.completedBy = data.completedBy || null;
|
||||
this.status = data.status || "pending";
|
||||
this.pointsAwarded = data.pointsAwarded || 10;
|
||||
this.createdAt = data.createdAt || new Date().toISOString();
|
||||
this.updatedAt = data.updatedAt || new Date().toISOString();
|
||||
this.completedAt = data.completedAt || null;
|
||||
}
|
||||
});
|
||||
|
||||
// Cascade cleanup when a task is deleted
|
||||
TaskSchema.pre("deleteOne", { document: true, query: false }, async function () {
|
||||
const User = mongoose.model("User");
|
||||
// Static methods for MongoDB-like interface
|
||||
static async find(filter = {}) {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
// Convert MongoDB filter to CouchDB selector
|
||||
const selector = { type: "task", ...filter };
|
||||
|
||||
// Handle special cases
|
||||
if (filter._id) {
|
||||
selector._id = filter._id;
|
||||
}
|
||||
|
||||
if (filter.status) {
|
||||
selector.status = filter.status;
|
||||
}
|
||||
|
||||
if (filter.street) {
|
||||
selector["street.streetId"] = filter.street;
|
||||
}
|
||||
|
||||
if (filter.completedBy) {
|
||||
selector["completedBy.userId"] = filter.completedBy;
|
||||
}
|
||||
|
||||
// Remove task from user's completedTasks
|
||||
if (this.completedBy) {
|
||||
await User.updateOne(
|
||||
{ _id: this.completedBy },
|
||||
{ $pull: { completedTasks: this._id } }
|
||||
);
|
||||
const query = {
|
||||
selector,
|
||||
sort: filter.sort || [{ createdAt: "desc" }]
|
||||
};
|
||||
|
||||
// Add pagination if specified
|
||||
if (filter.skip) query.skip = filter.skip;
|
||||
if (filter.limit) query.limit = filter.limit;
|
||||
|
||||
const docs = await couchdbService.find(query);
|
||||
|
||||
// Convert to Task instances
|
||||
return docs.map(doc => new Task(doc));
|
||||
} catch (error) {
|
||||
console.error("Error finding tasks:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
module.exports = mongoose.model("Task", TaskSchema);
|
||||
static async findById(id) {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
const doc = await couchdbService.getDocument(id);
|
||||
|
||||
if (!doc || doc.type !== "task") {
|
||||
return null;
|
||||
}
|
||||
|
||||
return new Task(doc);
|
||||
} catch (error) {
|
||||
if (error.statusCode === 404) {
|
||||
return null;
|
||||
}
|
||||
console.error("Error finding task by ID:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
static async findOne(filter = {}) {
|
||||
try {
|
||||
const tasks = await Task.find(filter);
|
||||
return tasks.length > 0 ? tasks[0] : null;
|
||||
} catch (error) {
|
||||
console.error("Error finding one task:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
static async countDocuments(filter = {}) {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
const selector = { type: "task", ...filter };
|
||||
|
||||
// Use Mango query with count
|
||||
const query = {
|
||||
selector,
|
||||
fields: ["_id"]
|
||||
};
|
||||
|
||||
const docs = await couchdbService.find(query);
|
||||
return docs.length;
|
||||
} catch (error) {
|
||||
console.error("Error counting tasks:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
static async create(data) {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
const task = new Task(data);
|
||||
const doc = await couchdbService.createDocument(task.toJSON());
|
||||
|
||||
return new Task(doc);
|
||||
} catch (error) {
|
||||
console.error("Error creating task:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
static async deleteMany(filter = {}) {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
const tasks = await Task.find(filter);
|
||||
const deletePromises = tasks.map(task => task.delete());
|
||||
|
||||
await Promise.all(deletePromises);
|
||||
return { deletedCount: tasks.length };
|
||||
} catch (error) {
|
||||
console.error("Error deleting many tasks:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// Instance methods
|
||||
async save() {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
this.updatedAt = new Date().toISOString();
|
||||
|
||||
if (this._id && this._rev) {
|
||||
// Update existing document
|
||||
const doc = await couchdbService.updateDocument(this.toJSON());
|
||||
this._rev = doc._rev;
|
||||
} else {
|
||||
// Create new document
|
||||
const doc = await couchdbService.createDocument(this.toJSON());
|
||||
this._id = doc._id;
|
||||
this._rev = doc._rev;
|
||||
}
|
||||
|
||||
// Handle post-save operations
|
||||
await this._handlePostSave();
|
||||
|
||||
return this;
|
||||
} catch (error) {
|
||||
console.error("Error saving task:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
async delete() {
|
||||
try {
|
||||
await couchdbService.initialize();
|
||||
|
||||
if (!this._id || !this._rev) {
|
||||
throw new Error("Task must have _id and _rev to delete");
|
||||
}
|
||||
|
||||
// Handle cascade operations
|
||||
await this._handleCascadeDelete();
|
||||
|
||||
await couchdbService.deleteDocument(this._id, this._rev);
|
||||
return this;
|
||||
} catch (error) {
|
||||
console.error("Error deleting task:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
async _handlePostSave() {
|
||||
try {
|
||||
// Update user relationship when task is completed
|
||||
if (this.completedBy && this.completedBy.userId && this.status === "completed") {
|
||||
const User = require("./User");
|
||||
const user = await User.findById(this.completedBy.userId);
|
||||
|
||||
if (user && !user.completedTasks.includes(this._id)) {
|
||||
user.completedTasks.push(this._id);
|
||||
user.stats.tasksCompleted = user.completedTasks.length;
|
||||
await user.save();
|
||||
}
|
||||
|
||||
// Update street stats
|
||||
if (this.street && this.street.streetId) {
|
||||
const Street = require("./Street");
|
||||
const street = await Street.findById(this.street.streetId);
|
||||
|
||||
if (street) {
|
||||
street.stats.completedTasksCount = (street.stats.completedTasksCount || 0) + 1;
|
||||
await street.save();
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error("Error handling post-save:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
async _handleCascadeDelete() {
|
||||
try {
|
||||
// Remove task from user's completedTasks
|
||||
if (this.completedBy && this.completedBy.userId) {
|
||||
const User = require("./User");
|
||||
const user = await User.findById(this.completedBy.userId);
|
||||
|
||||
if (user) {
|
||||
user.completedTasks = user.completedTasks.filter(id => id !== this._id);
|
||||
user.stats.tasksCompleted = user.completedTasks.length;
|
||||
await user.save();
|
||||
}
|
||||
}
|
||||
} catch (error) {
|
||||
console.error("Error handling cascade delete:", error.message);
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// Populate method for compatibility
|
||||
async populate(paths) {
|
||||
if (Array.isArray(paths)) {
|
||||
for (const path of paths) {
|
||||
await this._populatePath(path);
|
||||
}
|
||||
} else {
|
||||
await this._populatePath(paths);
|
||||
}
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
async _populatePath(path) {
|
||||
if (path === "street" && this.street && this.street.streetId) {
|
||||
const Street = require("./Street");
|
||||
const street = await Street.findById(this.street.streetId);
|
||||
|
||||
if (street) {
|
||||
this.street = {
|
||||
streetId: street._id,
|
||||
name: street.name,
|
||||
location: street.location
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
if (path === "completedBy" && this.completedBy && this.completedBy.userId) {
|
||||
const User = require("./User");
|
||||
const user = await User.findById(this.completedBy.userId);
|
||||
|
||||
if (user) {
|
||||
this.completedBy = {
|
||||
userId: user._id,
|
||||
name: user.name,
|
||||
profilePicture: user.profilePicture
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Convert to plain object
|
||||
toJSON() {
|
||||
return {
|
||||
_id: this._id,
|
||||
_rev: this._rev,
|
||||
type: this.type,
|
||||
street: this.street,
|
||||
description: this.description,
|
||||
completedBy: this.completedBy,
|
||||
status: this.status,
|
||||
pointsAwarded: this.pointsAwarded,
|
||||
createdAt: this.createdAt,
|
||||
updatedAt: this.updatedAt,
|
||||
completedAt: this.completedAt
|
||||
};
|
||||
}
|
||||
|
||||
// Convert to MongoDB-like format for API responses
|
||||
toObject() {
|
||||
const obj = this.toJSON();
|
||||
|
||||
// Remove CouchDB-specific fields for API compatibility
|
||||
delete obj._rev;
|
||||
delete obj.type;
|
||||
|
||||
// Add _id field for compatibility
|
||||
if (obj._id) {
|
||||
obj.id = obj._id;
|
||||
}
|
||||
|
||||
return obj;
|
||||
}
|
||||
}
|
||||
|
||||
// Export both the class and static methods for compatibility
|
||||
module.exports = Task;
|
||||
|
||||
Reference in New Issue
Block a user