ground/src/ground/GroundFileModel.coffee

A framework work with sails and backbone.

# Author: andyzhau@nodeswork.com (Andy Zhau)

fs = require 'fs'
mongodb = require 'mongodb'
groundDb = require './ground_db'
GroundModel = require './GroundModel'

class GroundFileModel extends GroundModel

  @collection: 'fs.files'

  @methods:

    read: (options..., cb) ->
      groundDb.mongodbConnection @ground.adapter, (err, db) =>
        mongodb.GridStore.read db, new mongodb.ObjectID(@id), cb

  @statics:

    # options:
    #   id: fileId
    writeFile: (file, options..., cb) ->
      options = options[0] ? {}
      options.id ?= new mongodb.ObjectID()
      groundDb.mongodbConnection @_instanceMethods.ground.adapter, (err, db) =>
        gridStore = new mongodb.GridStore(db, options.id, 'w')
        gridStore.writeFile file, (err, gridStore) ->
          if err? then cb err
          else gridStore.close cb

    writeBuffer: (buf, options..., cb) ->
      options = options[0] ? {}
      options.id ?= new mongodb.ObjectID()
      groundDb.mongodbConnection @_instanceMethods.ground.adapter, (err, db) =>
        gridStore = new mongodb.GridStore(db, options.id, 'w')
        gridStore.open ->
          gridStore.write buf, ->
            gridStore.close cb

module.exports = GroundFileModel