p5.js-web-editor/server/models/project.js

64 lines
1.6 KiB
JavaScript
Raw Normal View History

import mongoose from 'mongoose';
const Schema = mongoose.Schema;
const ObjectIdSchema = Schema.ObjectId;
const ObjectId = mongoose.Types.ObjectId;
2016-06-17 20:40:13 +00:00
import shortid from 'shortid';
const defaultSketch = `function setup() {
2016-06-29 16:52:16 +00:00
createCanvas(400, 400);
}
function draw() {
2016-06-29 16:52:16 +00:00
background(220);
}`
const defaultHTML =
`
<!DOCTYPE html>
<html>
<head>
</head>
<body>
<script src="sketch.js"></script>
</body>
</html>
`
2016-06-17 18:11:52 +00:00
const fileSchema = new Schema({
2016-06-23 22:29:55 +00:00
name: { type: String, default: 'sketch.js' },
2016-06-29 16:52:16 +00:00
content: { type: String, default: defaultSketch }
}, { timestamps: true, _id: true });
2016-06-17 18:11:52 +00:00
fileSchema.virtual('id').get(function(){
return this._id.toHexString();
});
fileSchema.set('toJSON', {
virtuals: true
});
const projectSchema = new Schema({
2016-06-23 22:29:55 +00:00
name: { type: String, default: "Hello p5.js, it's the server" },
user: { type: Schema.Types.ObjectId, ref: 'User' },
files: {type: [ fileSchema ], default: [{ name: 'sketch.js', content: defaultSketch, _id: new ObjectId() }, { name: 'index.html', content: defaultHTML, _id: new ObjectId() }]},
2016-07-08 18:57:22 +00:00
_id: { type: String, default: shortid.generate },
selectedFile: Schema.Types.ObjectId
2016-06-23 22:29:55 +00:00
}, { timestamps: true });
projectSchema.virtual('id').get(function(){
return this._id;
});
projectSchema.set('toJSON', {
virtuals: true
});
2016-07-08 18:57:22 +00:00
projectSchema.pre('save', function createSelectedFile(next) {
const project = this;
if (!project.selectedFile) {
project.selectedFile = project.files[0]._id; // eslint-disable-line no-underscore-dangle
return next();
}
});
2016-06-23 22:29:55 +00:00
export default mongoose.model('Project', projectSchema);