64 lines
1.6 KiB
JavaScript
64 lines
1.6 KiB
JavaScript
import mongoose from 'mongoose';
|
|
const Schema = mongoose.Schema;
|
|
const ObjectIdSchema = Schema.ObjectId;
|
|
const ObjectId = mongoose.Types.ObjectId;
|
|
import shortid from 'shortid';
|
|
|
|
const defaultSketch = `function setup() {
|
|
createCanvas(400, 400);
|
|
}
|
|
function draw() {
|
|
background(220);
|
|
}`
|
|
|
|
const defaultHTML =
|
|
`
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/0.5.0/p5.min.js"></script>
|
|
</head>
|
|
<body>
|
|
<script src="sketch.js"></script>
|
|
</body>
|
|
</html>
|
|
`
|
|
|
|
const fileSchema = new Schema({
|
|
name: { type: String, default: 'sketch.js' },
|
|
content: { type: String, default: defaultSketch }
|
|
}, { timestamps: true, _id: true });
|
|
|
|
fileSchema.virtual('id').get(function(){
|
|
return this._id.toHexString();
|
|
});
|
|
|
|
fileSchema.set('toJSON', {
|
|
virtuals: true
|
|
});
|
|
|
|
const projectSchema = new Schema({
|
|
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() }]},
|
|
_id: { type: String, default: shortid.generate },
|
|
selectedFile: Schema.Types.ObjectId
|
|
}, { timestamps: true });
|
|
|
|
projectSchema.virtual('id').get(function(){
|
|
return this._id;
|
|
});
|
|
|
|
projectSchema.set('toJSON', {
|
|
virtuals: true
|
|
});
|
|
|
|
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();
|
|
}
|
|
});
|
|
|
|
export default mongoose.model('Project', projectSchema);
|