RunRubyScriptAction.rb

#
#  RunRubyScriptAction.rb
#  RunRubyScriptAction
#
#  Created by Jason Foreman on 12/26/07.
#  Copyright (c) 2008 Jason Foreman. All rights reserved.
#

require 'osx/cocoa'

include OSX

require_framework 'ScriptingBridge'
require_framework 'Automator'
require 'RubyTextView'

class RunRubyScriptAction < AMBundleAction

	def typeForDescriptor(desc)
		desc[0].to_i << 24 | desc[1].to_i << 16 | desc[2].to_i << 8 | desc[3].to_i
	end

	def getSBApplication(appname)
		bundleId = LSWrapper.bundleIdentifierForApplicationWithName(appname)
		return SBApplication.applicationWithBundleIdentifier(bundleId)
	end
	
	def getObjectSource(source)
		case source.descriptorType()
		when typeForDescriptor('obj ')
			return convertObject(source)
		when typeForDescriptor('psn ')
			return getSBApplication(source.stringValue())
		end
	end
	
	def convertObject(object)
		from = object.descriptorForKeyword_(typeForDescriptor("from"))
		source = @sources[from.description.to_s] ||= getObjectSource(from)
		want = object.descriptorForKeyword_(typeForDescriptor('want'))
		elements = source.elementArrayWithCode(want.typeCodeValue())
		seld = elements.objectWithID_(object.descriptorForKeyword_(typeForDescriptor('seld')))
		return seld
	end
	
	def convertInput(input)
		if input.class == NSAppleEventDescriptor then
			case input.descriptorType
			when typeForDescriptor('list')
				ret = []
				1.upto(input.numberOfItems()) do |i|
					ret << convertInput(input.descriptorAtIndex(i))
				end
				return ret
			when typeForDescriptor('obj ')
				return convertObject(input)
			when typeForDescriptor('utxt')
				return input.stringValue()
			else
				return input
			end
		else
			return input
		end
	end
	
	def convertOutput(output)
		case output
		when Array, NSArray
			list = NSAppleEventDescriptor.listDescriptor()
			output.each_with_index do |o, i|
				list.insertDescriptor_atIndex_(convertOutput(o), i)
			end
			return list
		when SBObject
			return output.qualifiedSpecifier()
		when String, NSString
			return NSAppleEventDescriptor.descriptorWithString_(output)
		else
			NSLog "warning: cannot convert output of type #{output.class}"
			return NSNull # TODO return nil, output, something else?
		end
	end

	def runWithInput_fromAction_error(input, action, error)

		begin
			@sources = {}

			scriptSource = self.parameters['ruby script source']

			input = convertInput(input)

			klass = Class.new
			klass.class_eval(scriptSource)
			if klass.method_defined? :performAction then
				result = klass.new.performAction(input)
			else
				result = nil
			end

			return convertOutput(result)
		rescue
			e = {
				:OSAScriptErrorNumber => -1,
				:OSAScriptErrorMessage => 'A Ruby exception occured.',
			}
			error.assign(NSDictionary.dictionaryWithDictionary_(e))
		end
	end

end