231 lines
		
	
	
	
		
			8.2 KiB
			
		
	
	
	
		
			CoffeeScript
		
	
	
	
	
	
			
		
		
	
	
			231 lines
		
	
	
	
		
			8.2 KiB
			
		
	
	
	
		
			CoffeeScript
		
	
	
	
	
	
path = require 'path'
 | 
						|
 | 
						|
electron = require 'electron'
 | 
						|
{ipcMain} = electron
 | 
						|
objectsRegistry = require './objects-registry'
 | 
						|
 | 
						|
v8Util = process.atomBinding 'v8_util'
 | 
						|
{IDWeakMap} = process.atomBinding 'id_weak_map'
 | 
						|
 | 
						|
# Convert a real value into meta data.
 | 
						|
valueToMeta = (sender, value, optimizeSimpleObject=false) ->
 | 
						|
  meta = type: typeof value
 | 
						|
 | 
						|
  meta.type = 'buffer' if Buffer.isBuffer value
 | 
						|
  meta.type = 'value' if value is null
 | 
						|
  meta.type = 'array' if Array.isArray value
 | 
						|
  meta.type = 'error' if value instanceof Error
 | 
						|
  meta.type = 'date' if value instanceof Date
 | 
						|
  meta.type = 'promise' if value?.constructor.name is 'Promise'
 | 
						|
 | 
						|
  # Treat simple objects as value.
 | 
						|
  if optimizeSimpleObject and meta.type is 'object' and v8Util.getHiddenValue value, 'simple'
 | 
						|
    meta.type = 'value'
 | 
						|
 | 
						|
  # Treat the arguments object as array.
 | 
						|
  meta.type = 'array' if meta.type is 'object' and value.callee? and value.length?
 | 
						|
 | 
						|
  if meta.type is 'array'
 | 
						|
    meta.members = []
 | 
						|
    meta.members.push valueToMeta(sender, el) for el in value
 | 
						|
  else if meta.type is 'object' or meta.type is 'function'
 | 
						|
    meta.name = value.constructor.name
 | 
						|
 | 
						|
    # Reference the original value if it's an object, because when it's
 | 
						|
    # passed to renderer we would assume the renderer keeps a reference of
 | 
						|
    # it.
 | 
						|
    meta.id = objectsRegistry.add sender.getId(), value
 | 
						|
 | 
						|
    meta.members = ({name, type: typeof field} for name, field of value)
 | 
						|
  else if meta.type is 'buffer'
 | 
						|
    meta.value = Array::slice.call value, 0
 | 
						|
  else if meta.type is 'promise'
 | 
						|
    meta.then = valueToMeta sender, value.then.bind(value)
 | 
						|
  else if meta.type is 'error'
 | 
						|
    meta.members = plainObjectToMeta value
 | 
						|
    # Error.name is not part of own properties.
 | 
						|
    meta.members.push {name: 'name', value: value.name}
 | 
						|
  else if meta.type is 'date'
 | 
						|
    meta.value = value.getTime()
 | 
						|
  else
 | 
						|
    meta.type = 'value'
 | 
						|
    meta.value = value
 | 
						|
 | 
						|
  meta
 | 
						|
 | 
						|
# Convert object to meta by value.
 | 
						|
plainObjectToMeta = (obj) ->
 | 
						|
  Object.getOwnPropertyNames(obj).map (name) -> {name, value: obj[name]}
 | 
						|
 | 
						|
# Convert Error into meta data.
 | 
						|
exceptionToMeta = (error) ->
 | 
						|
  type: 'exception', message: error.message, stack: (error.stack || error)
 | 
						|
 | 
						|
# Convert array of meta data from renderer into array of real values.
 | 
						|
unwrapArgs = (sender, args) ->
 | 
						|
  metaToValue = (meta) ->
 | 
						|
    switch meta.type
 | 
						|
      when 'value' then meta.value
 | 
						|
      when 'remote-object' then objectsRegistry.get meta.id
 | 
						|
      when 'array' then unwrapArgs sender, meta.value
 | 
						|
      when 'buffer' then new Buffer(meta.value)
 | 
						|
      when 'date' then new Date(meta.value)
 | 
						|
      when 'promise' then Promise.resolve(then: metaToValue(meta.then))
 | 
						|
      when 'object'
 | 
						|
        ret = v8Util.createObjectWithName meta.name
 | 
						|
        for member in meta.members
 | 
						|
          ret[member.name] = metaToValue(member.value)
 | 
						|
        ret
 | 
						|
      when 'function-with-return-value'
 | 
						|
        returnValue = metaToValue meta.value
 | 
						|
        -> returnValue
 | 
						|
      when 'function'
 | 
						|
        # Cache the callbacks in renderer.
 | 
						|
        unless sender.callbacks
 | 
						|
          sender.callbacks = new IDWeakMap
 | 
						|
          sender.on 'render-view-deleted', ->
 | 
						|
            sender.callbacks.clear()
 | 
						|
        return sender.callbacks.get meta.id if sender.callbacks.has meta.id
 | 
						|
 | 
						|
        rendererReleased = false
 | 
						|
        objectsRegistry.once "clear-#{sender.getId()}", ->
 | 
						|
          rendererReleased = true
 | 
						|
 | 
						|
        ret = ->
 | 
						|
          if rendererReleased
 | 
						|
            throw new Error("Attempting to call a function in a renderer window
 | 
						|
              that has been closed or released. Function provided here: #{meta.location}.")
 | 
						|
          sender.send 'ATOM_RENDERER_CALLBACK', meta.id, valueToMeta(sender, arguments)
 | 
						|
        v8Util.setDestructor ret, ->
 | 
						|
          return if rendererReleased
 | 
						|
          sender.callbacks.remove meta.id
 | 
						|
          sender.send 'ATOM_RENDERER_RELEASE_CALLBACK', meta.id
 | 
						|
        sender.callbacks.set meta.id, ret
 | 
						|
        ret
 | 
						|
      else throw new TypeError("Unknown type: #{meta.type}")
 | 
						|
 | 
						|
  args.map metaToValue
 | 
						|
 | 
						|
# Call a function and send reply asynchronously if it's a an asynchronous
 | 
						|
# style function and the caller didn't pass a callback.
 | 
						|
callFunction = (event, func, caller, args) ->
 | 
						|
  funcMarkedAsync = v8Util.getHiddenValue(func, 'asynchronous')
 | 
						|
  funcPassedCallback = typeof args[args.length - 1] is 'function'
 | 
						|
 | 
						|
  try
 | 
						|
    if funcMarkedAsync and not funcPassedCallback
 | 
						|
      args.push (ret) ->
 | 
						|
        event.returnValue = valueToMeta event.sender, ret, true
 | 
						|
      func.apply caller, args
 | 
						|
    else
 | 
						|
      ret = func.apply caller, args
 | 
						|
      event.returnValue = valueToMeta event.sender, ret, true
 | 
						|
  catch e
 | 
						|
    # Catch functions thrown further down in function invocation and wrap
 | 
						|
    # them with the function name so it's easier to trace things like
 | 
						|
    # `Error processing argument -1.`
 | 
						|
    funcName = func.name ? "anonymous"
 | 
						|
    throw new Error("Could not call remote function `#{funcName}`.
 | 
						|
                     Check that the function signature is correct.
 | 
						|
                     Underlying error: #{e.message}")
 | 
						|
 | 
						|
# Send by BrowserWindow when its render view is deleted.
 | 
						|
process.on 'ATOM_BROWSER_RELEASE_RENDER_VIEW', (id) ->
 | 
						|
  objectsRegistry.clear id
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_REQUIRE', (event, module) ->
 | 
						|
  try
 | 
						|
    event.returnValue = valueToMeta event.sender, process.mainModule.require(module)
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_GET_BUILTIN', (event, module) ->
 | 
						|
  try
 | 
						|
    event.returnValue = valueToMeta event.sender, electron[module]
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_GLOBAL', (event, name) ->
 | 
						|
  try
 | 
						|
    event.returnValue = valueToMeta event.sender, global[name]
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_CURRENT_WINDOW', (event) ->
 | 
						|
  try
 | 
						|
    event.returnValue = valueToMeta event.sender, event.sender.getOwnerBrowserWindow()
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_CURRENT_WEB_CONTENTS', (event) ->
 | 
						|
  event.returnValue = valueToMeta event.sender, event.sender
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_CONSTRUCTOR', (event, id, args) ->
 | 
						|
  try
 | 
						|
    args = unwrapArgs event.sender, args
 | 
						|
    constructor = objectsRegistry.get id
 | 
						|
    # Call new with array of arguments.
 | 
						|
    # http://stackoverflow.com/questions/1606797/use-of-apply-with-new-operator-is-this-possible
 | 
						|
    obj = new (Function::bind.apply(constructor, [null].concat(args)))
 | 
						|
    event.returnValue = valueToMeta event.sender, obj
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_FUNCTION_CALL', (event, id, args) ->
 | 
						|
  try
 | 
						|
    args = unwrapArgs event.sender, args
 | 
						|
    func = objectsRegistry.get id
 | 
						|
    callFunction event, func, global, args
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_MEMBER_CONSTRUCTOR', (event, id, method, args) ->
 | 
						|
  try
 | 
						|
    args = unwrapArgs event.sender, args
 | 
						|
    constructor = objectsRegistry.get(id)[method]
 | 
						|
    # Call new with array of arguments.
 | 
						|
    obj = new (Function::bind.apply(constructor, [null].concat(args)))
 | 
						|
    event.returnValue = valueToMeta event.sender, obj
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_MEMBER_CALL', (event, id, method, args) ->
 | 
						|
  try
 | 
						|
    args = unwrapArgs event.sender, args
 | 
						|
    obj = objectsRegistry.get id
 | 
						|
    callFunction event, obj[method], obj, args
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_MEMBER_SET', (event, id, name, value) ->
 | 
						|
  try
 | 
						|
    obj = objectsRegistry.get id
 | 
						|
    obj[name] = value
 | 
						|
    event.returnValue = null
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_MEMBER_GET', (event, id, name) ->
 | 
						|
  try
 | 
						|
    obj = objectsRegistry.get id
 | 
						|
    event.returnValue = valueToMeta event.sender, obj[name]
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_DEREFERENCE', (event, id) ->
 | 
						|
  objectsRegistry.remove event.sender.getId(), id
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_GUEST_WEB_CONTENTS', (event, guestInstanceId) ->
 | 
						|
  try
 | 
						|
    guestViewManager = require './guest-view-manager'
 | 
						|
    event.returnValue = valueToMeta event.sender, guestViewManager.getGuest(guestInstanceId)
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 | 
						|
 | 
						|
ipcMain.on 'ATOM_BROWSER_ASYNC_CALL_TO_GUEST_VIEW', (event, guestInstanceId, method, args...) ->
 | 
						|
  try
 | 
						|
    guestViewManager = require './guest-view-manager'
 | 
						|
    guest = guestViewManager.getGuest(guestInstanceId)
 | 
						|
    guest[method].apply(guest, args)
 | 
						|
  catch e
 | 
						|
    event.returnValue = exceptionToMeta e
 |