main.js

var apejs = require('apejs.js')
var googlestore = require('googlestore.js')
var select = require('select.js')
var mustache = require('./common/mustache.js')
var gdrive = require('./gdrive.js')
var blobstore = require('./blobstore.js')
var images = require('./images.js')
var url = require('./url.js')

var blobService = blobstore.service
var blobInfoFactory = blobstore.blobInfoFactory

apejs.urls = {
  '/': {
    get: function(req, res) {
      var html = mustache.to_html(render('skins/index.html'), {}, {})
      print(res).html(html)
    }
  },
  '/tests': {
    get: function(req, res) {
      // array of modules with tests we wanna test
      var tests = [
        //images, 
        url
      ]

      for(var i in tests) {
        var curTest = tests[i]
        for(var x in curTest.tests) {
          // run the test function
          curTest.tests[x]()
        }
      }
    }
  },
  '/read-headers': {
    get: function(req, res) {
      var headerNames = req.getHeaderNames()
      while(headerNames.hasMoreElements()) {
        var headerName = headerNames.nextElement()
        print(res).html(''+headerName+': '+ req.getHeader(headerName) +'<br>')
      }
    }
  },

  '/fileupload': {
    get: function(req, res) {
      var html = mustache.to_html(render('skins/fileupload.html'), {}, {})
      print(res).html(html)
    }
  },
  '/get-files': {
    get: function(req, res) {
      var files = [];
      select('file')
        .find()
        .sort('createdAt', 'DESC')
        .limit(9)
        .each(function(id) { 
          var obj = this
          if(this.blobKeyString) {
            try {
              var blobKey = new BlobKey(this.blobKeyString)
              var options = ServingUrlOptions.Builder.withBlobKey(blobKey)
              var url = images.service.getServingUrl(options)

              obj.thumbUrl = ''+url+'=s500'
            } catch(e) {
              // it's not an image, it's a file!
              obj.thumbUrl = 'http://cdn1.iconfinder.com/data/icons/CrystalClear/128x128/mimetypes/unknown.png'
            }
          } else if(this.url) { // it's a url
            obj.thumbUrl = '/rest/file/datastore/' + id
          }

          files.push(obj)
        })
      
      print(res).json(files)
    }
  },
  '/submit-link': {
    post: function(req, res) {
      var p = param(req)
      var u = p('url')
      var title = p('title')

      var thumbUrl = images.getThumb(u)
      if(!thumbUrl) { // maybe service is down or URL is invalid
        return Err(res, 'Image download from service didn\'t work')
      }
      // let's download the image and save it to datastore
      // (easier than blobstore)
      var bytes = url.fetchUrl(thumbUrl)

      select('file')
        .add({
          createdAt: new java.util.Date(),
          url: u,
          title: title,
          image: new Blob(bytes),
          thumbUrl: thumbUrl
        })
    }
  },
  /**
   * This service loads data from a Google Drive.
   * Shouldn't be a Public API
   */
  '/get-gdrive-files': {
    get: function(req, res) {
      var g = new gdrive
      print(res).html(g.createShortUrl())
    }
  },

  /**
   * REST API for fileupload
   */
  '/rest/file/url': {
    get: function(req, res) {
      var url = blobService.createUploadUrl("/rest/file")
      print(res).json(''+url)
    }
  },
  '/rest/file': {
    post: function(req, res) {
      var blobs = blobService.getUploadedBlobs(req)
      var blobKey = blobs.get('file')
      var blobKeyString = blobKey.getKeyString()

      // add it to datastore
      var e = googlestore.entity('file', {
        'createdAt': new java.util.Date(),
        'blobKeyString': blobKeyString
      })
      googlestore.put(e)

      res.sendRedirect("/rest/file/" + blobKeyString + "/meta")
    }
  },
  '/rest/file/(.*)/meta': {
    get: function(req, res, matches) {
      var key = matches[1]
      var blobKey = new BlobKey(key)
      var info = blobInfoFactory.loadBlobInfo(blobKey);

      var ret = [{
        name: ''+info.getFilename(),
        size: ''+info.getSize(),
        url: "/rest/file/" + key
      }]
      print(res).json(ret)
    }
  },
  '/rest/file/datastore/(.*)': {
    get: function(req, res, matches) {
      var key = googlestore.createKey('file', parseInt(matches[1], 10))
      var file = googlestore.get(key)
      var image = file.getProperty('image')
        
      res.setContentType('image/jpg');
      res.getOutputStream().write(image.getBytes());
    }
  },
  '/rest/file/(.*)': {
    get: function(req, res, matches) {
      var key = matches[1]
      var blobKey = new BlobKey(key)
      var blobInfo = blobInfoFactory.loadBlobInfo(blobKey)
      res.setHeader("Content-Disposition", "attachment; filename=" + blobInfo.getFilename())
      blobService.serve(blobKey, res)
    }
  }
};

// simple syntax sugar
function print(response) {
  response.setCharacterEncoding("UTF-8");
  return {
    html: function(html) {
      if(response == null) return;
      response.setContentType("text/html");
      response.getWriter().println(html);
    },
    json: function(j) {
      if(response == null) return;
      var jsonString = JSON.stringify(j);
      response.setContentType("application/json");
      response.getWriter().println(jsonString);
    }
  };
}
function param(request) {
  return function(par) {
    var p = request.getParameter(par);
    if(p) return p;
    else return false;
  }
}
function Err(res, msg) {
  res.sendError(res.SC_BAD_REQUEST, msg)
}