class AWS::SimpleWorkflow::ActivityTask

def cancel! options = {}
  respond :canceled, options
end

def complete! options = {}
  respond :completed, options
end

def fail! options = {}
  respond :failed, options
end

def initialize domain, data, options = {}
  @domain = domain
  @task_token = data['taskToken']
  @activity_id = data['activityId']
  @started_event_id = data['startedEventId']
  @input = data['input']
  name = data['activityType']['name']
  version = data['activityType']['version']
  @activity_type = domain.activity_types[name,version]
  workflow_id = data['workflowExecution']['workflowId']
  run_id = data['workflowExecution']['runId']
  @workflow_execution = domain.workflow_executions[workflow_id,run_id]
  super
end

def record_heartbeat! options = {}
  client_opts = {}
  client_opts[:task_token] = task_token
  client_opts[:details] = options[:details] if options[:details]
  response = client.record_activity_task_heartbeat(client_opts)
  raise CancelRequestedError if response.data['cancelRequested']
  nil
end

def respond status, options
  raise "already responded" if responded?
  @responded = status
  options[:task_token] = task_token
  client.send("respond_activity_task_#{status}", options)
  nil
end

def responded?
  !!@responded
end