1 files changed, 36 insertions, 8 deletions
diff --git a/models/project.js b/models/project.js
index 2424694..f29c7f8 100644
--- a/models/project.js
+++ b/models/project.js
@@ -6,21 +6,17 @@
 var mongoose = require('mongoose')
   , Schema = mongoose.Schema;
 
-
-
-
-
 var ProjectSchema = new Schema({
-    user: { type: String, ref: 'User' },
+    user: { type: Schema.ObjectId, ref: 'User' },
     name: { type: String, default: '', trim: true },
     description: {type: String, default: '', trim: true },
     currency: { type: String, default: 'kr', trim: true },
     public: { type: String, default: 'invite-only' },
     created: { type: Date, default: Date.now },
-    updated: { type: Date, default: Date.now }
+    updated: { type: Date, default: Date.now },
+    shortURL: { type: String, unique: true }
 });
 
-
 // the four validations below only apply if you are signing up traditionally
 
 ProjectSchema.path('name').validate(function(name) {
@@ -28,12 +24,44 @@ ProjectSchema.path('name').validate(function(name) {
     return name.length;
 }, 'Project name cannot be blank');
 
-ProjectSchema.methods = {
+
+ProjectSchema.pre('save', function(next) {
+    if (this.shortURL !== undefined) return next();
+    var chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890";
+    this.shortURL = '';
+    for (var i = 0; i < 6; i++) {
+        var x = Math.floor(Math.random() * chars.length);
+        this.shortURL += chars.charAt(x);
+    }
+    console.log('SHORT: ' + this.shortURL);
+    next();
+});
+
+
+
+ProjectSchema.statics = {
 
     log: function() {
         console.log('wat. wat logged this');
+    },
+
+
+
+   /**
+    * Find project by id
+    *
+    * @param {ObjectId} id
+    * @param {Function} callback
+    * @api private
+    */
+
+    load: function(id, callback) {
+        this.findOne({ _id: id })
+          .populate('user')
+          .exec(callback);
     }
 
+
 }
 
 mongoose.model('Project', ProjectSchema);