codecombat/app/lib/simulator/Simulator.coffee

SuperModel = require 'models/SuperModel'
LevelLoader = require 'lib/LevelLoader'
GoalManager = require 'lib/world/GoalManager'

module.exports = class Simulator
  constructor: ->
    @retryDelayInSeconds = 10
    @taskURL = "/queue/scoring"


  fetchAndSimulateTask: =>
    $.ajax
      url: @taskURL
      type: "GET"
      error: @handleFetchTaskError
      success: @setupSimulationAndLoadLevel

  cleanupSimulation: ->


  handleFetchTaskError: (errorData) =>
    console.log "There were no games to score. Error: #{JSON.stringify errorData}"
    console.log "Retrying in #{@retryDelayInSeconds}"
    _.delay @fetchAndSimulateTask, @retryDelayInSeconds * 1000


  setupSimulationAndLoadLevel: (taskData) =>
    @task = new SimulationTask(taskData)
    @superModel = new SuperModel()
    @god = new God()
    @levelLoader = new LevelLoader @task.getLevelName(), @superModel, @task.getFirstSessionID()

    @levelLoader.once 'loaded-all', @simulateGame



  simulateGame: =>
    @assignWorldAndLevelFromLevelLoaderAndDestroyIt()
    @setupGod()
    try
      @commenceSimulationAndSetupCallback()
    catch e
      console.log "There was an error in simulation. Trying again in #{retryDelayInSeconds} seconds"
      console.log "Error #{e}"
      _.delay @fetchAndSimulateTask, @retryDelayInSeconds * 1000

  assignWorldAndLevelFromLevelLoaderAndDestroyIt: ->
    @world = @levelLoader.world
    @level = @levelLoader.level
    @levelLoader.destroy()


  setupGod: ->
    @god.level = @level.serialize @supermodel
    @god.worldClassMap = world.classMap
    @setupGoalManager()
    @setupGodSpells()


  setupGoalManager: ->
    @god.goalManager = new GoalManager @world
    @god.goalManager.goals = @fetchGoalsFromWorldNoteChain()
    @god.goalManager.goalStates = @manuallyGenerateGoalStates()

  commenceSimulationAndSetupCallback: ->
    @god.createWorld()
    Backbone.Mediator.subscribeOnce 'god:new-world-created', @processResults, @


  processResults: (simulationResults) ->
    taskResults = @formTaskResultsObject simulationResults
    sendResultsBackToServer taskResults

  sendResultsBackToServer: (results) =>
    $.ajax
      url: @taskURL
      data: results
      type: "PUT"
      success: @handleTaskResultsTransferSuccess
      error: @handleTaskResultsTransferError
      complete: @cleanupAndSimulateAnotherTask()

  handleTaskResultsTransferSuccess: (result) ->
    console.log "Task registration result: #{JSON.stringify result}"

  handleTaskResultsTransferError: (error) ->
    console.log "Task registration error: #{JSON.stringify error}"

  cleanupAndSimulateAnotherTask: =>
    @cleanupSimulation()
    @fetchAndSimulateTask()



  formTaskResultsObject: (simulationResults) ->
    taskResults =
      taskID: @task.getTaskID()
      receiptHandle: @task.getReceiptHandle()
      calculationTime: 500
      sessions: []

    for session in @task.sessions
      sessionResult =
        sessionID: session.sessionID
        sessionChangedTime: session.sessionChangedTime
        metrics:
          rank: @calculateSessionRank session.sessionID, simulationResults.goalStates

      taskResults.sessions.push sessionResult

    return taskResults


  calculateSessionRank: (sessionID, goalStates) ->
    humansDestroyed = goalStates["destroy-humans"].status is "success"
    ogresDestroyed = goalStates["destroy-ogres"].status is "success"
    console.log "Humans destroyed:#{humansDestroyed}"
    console.log "Ogres destroyed:#{ogresDestroyed}"
    console.log "Team Session Map: #{JSON.stringify @teamSessionMap}"
    if humansDestroyed is ogresDestroyed
      return 0
    else if humansDestroyed and @teamSessionMap["ogres"] is sessionID
      return 0
    else if humansDestroyed and @teamSessionMap["ogres"] isnt sessionID
      return 1
    else if ogresDestroyed and @teamSessionMap["humans"] is sessionID
      return 0
    else
      return 1



  fetchGoalsFromWorldNoteChain: -> return @god.goalManager.world.scripts[0].noteChain[0].goals.add


  manuallyGenerateGoalStates: ->
    goalStates =
      "destroy-humans":
        keyFrame: 0
        killed:
          "Human Base": false
        status: "incomplete"
      "destroy-ogres":
        keyFrame:0
        killed:
          "Ogre Base": false
        status: "incomplete"



  setupGodSpells: ->
    @generateSpellsObject()
    @god.spells = @spells


  generateSpellsObject: (currentUserCodeMap) ->
    @userCodeMap = currentUserCodeMap
    @spells = {}
    for thang in @level.attributes.thangs
      continue if @thangIsATemplate thang
      @generateSpellKeyToSourceMapPropertiesFromThang thang


  thangIsATemplate: (thang) ->
    for component in thang.components
      continue unless @componentHasProgrammableMethods component
      for methodName, method of component.config.programmableMethods
        return true if methodBelongsToTemplateThang method

    return false


  componentHasProgrammableMethods: (component) -> component.config? and _.has component.config, 'programmableMethods'

  methodBelongsToTemplateThang: (method) -> typeof method is 'string'

  generateSpellKeyToSourceMapPropertiesFromThang: (thang) =>
    for component in thang.components
      continue unless @componentHasProgrammableMethods component

      for methodName, method of component.config.programmableMethods
        spellKey = @generateSpellKeyFromThangIDAndMethodName thang.id, methodName

        @createSpellAndAssignName spellKey, methodName
        @createSpellThang thang, method, spellKey
        @transpileSpell thang, spellKey, methodName




  generateSpellKeyFromThangIDAndMethodName: (thang, methodName) ->
    spellKeyComponents = [thang.id, methodName]
    pathComponents[0] = _.string.slugify pathComponents[0]
    pathComponents.join '/'

  createSpellAndAssignName: (spellKey, spellName) ->
    @spells[spellKey] ?= {}
    @spells[spellKey].name = methodName

  createSpellThang: (thang, method, spellKey) ->
    @spells[spellKey].thangs ?= {}
    @spells[spellKey].thangs[thang.id] ?= {}
    @spells[spellKey].thangs[thang.id].aether = @createAether @spells[spellKey].name, method

  transpileSpell: (thang, spellKey, methodName) ->
    slugifiedThangID = _.string.slugify thang.id
    source = @currentUserCodeMap[slugifiedThangID]?[methodName] ? ""
    @spells[spellKey].thangs[thang.id].aether.transpile source







  createAether: (methodName, method) ->
    aetherOptions =
      functionName: methodName
      protectAPI: false
      includeFlow: false
    return new Aether aetherOptions









class SimulationTask
  constructor: (@rawData) ->


  getLevelName: ->
    levelName =  @rawData.sessions?[0]?.levelID
    return levelName if levelName?
    @throwMalformedTaskError "The level name couldn't be deduced from the task."

  generateTeamToSessionMap: ->
    teamSessionMap = {}
    for session in @rawData.sessions
      @throwMalformedTaskError "Two players share the same team" if teamSessionMap[session.team]?
      teamSessionMap[session.team] = session.sessionID

    teamSessionMap

  throwMalformedTaskError: (errorString) ->
    throw new Error "The task was malformed, reason: #{errorString}"

  getFirstSessionID: -> @rawData.sessions[0].sessionID

  getTaskID: -> @rawData.taskID

  getReceiptHandle: -> @rawData.receiptHandle

  getSessions: -> @rawData.sessions

  generateSpellKeyToSourceMap: ->
    spellKeyToSourceMap = {}

    for session in @rawData.sessions
      teamSpells = session.teamSpells[session.team]
      _.merge spellKeyToSourceMap, _.pick(session.code, teamSpells)

      commonSpells = session.teamSpells["common"]
      _.merge spellKeyToSourceMap, _.pick(session.code, commonSpells) if commonSpells?

    spellKeyToSourceMap