###
# Copyright 2013-2016  Zaid Abdulla
#
# GenieACS is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# GenieACS is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with GenieACS.  If not, see <http://www.gnu.org/licenses/>.
#
# This file incorporates work covered by the following copyright and
# permission notice:
#
# Copyright 2013 Fanoos Telecom
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
###

util = require 'util'
zlib = require 'zlib'
crypto = require 'crypto'
mongodb = require 'mongodb'

config = require './config'
common = require './common'
soap = require './soap'
session = require './session'
query = require './query'
device = require './device'
cache = require './cache'
db = require './db'


writeResponse = (currentRequest, res) ->
  if config.get('DEBUG', currentRequest.sessionData.deviceId)
    dump = "# RESPONSE #{new Date(Date.now())}\n" + JSON.stringify(res.headers) + "\n#{res.data}\n\n"
    require('fs').appendFile("./debug/#{currentRequest.sessionData.deviceId}.dump", dump, (err) ->
      throw err if err
    )

  # respond using the same content-encoding as the request
  if currentRequest.httpRequest.headers['content-encoding']? and res.data.length > 0
    switch currentRequest.httpRequest.headers['content-encoding']
      when 'gzip'
        res.headers['Content-Encoding'] = 'gzip'
        compress = zlib.gzip
      when 'deflate'
        res.headers['Content-Encoding'] = 'deflate'
        compress = zlib.deflate

  if compress?
    compress(res.data, (err, data) ->
      throw err if err
      res.headers['Content-Length'] = data.length
      currentRequest.httpResponse.writeHead(res.code, res.headers)
      currentRequest.httpResponse.end(data)
    )
  else
    res.headers['Content-Length'] = res.data.length
    currentRequest.httpResponse.writeHead(res.code, res.headers)
    currentRequest.httpResponse.end(res.data)


inform = (currentRequest, cwmpRequest) ->
  if config.get('LOG_INFORMS', currentRequest.sessionData.deviceId)
    util.log("#{currentRequest.sessionData.deviceId}: Inform (#{cwmpRequest.methodRequest.event}); retry count #{cwmpRequest.methodRequest.retryCount}")

  session.inform(currentRequest.sessionData, cwmpRequest.methodRequest, (err, rpcResponse) ->
    throw err if err
    res = soap.response({
      id : cwmpRequest.id,
      methodResponse : rpcResponse,
      cwmpVersion : currentRequest.sessionData.cwmpVersion
    })

    if !!cookiesPath = config.get('COOKIES_PATH', currentRequest.sessionData.deviceId)
      res.headers['Set-Cookie'] = "session=#{currentRequest.sessionData.sessionId}; Path=#{cookiesPath}"
    else
      res.headers['Set-Cookie'] = "session=#{currentRequest.sessionData.sessionId}"

    writeResponse(currentRequest, res)
  )


transferComplete = (currentRequest, cwmpRequest) ->
  session.transferComplete(currentRequest.sessionData, cwmpRequest.methodRequest, (err, rpcResponse, faults) ->
    throw err if err

    for channel, fault of faults
      if not fault.retries? and Object.keys(faults).length == 1
        fault.retries = 0

      currentRequest.sessionData.faults[channel] = fault
      util.log("#{currentRequest.sessionData.deviceId}: Fault response for channel #{channel} (retries: #{fault.retries ? 0})")

    res = soap.response({
      id : cwmpRequest.id,
      methodResponse : rpcResponse,
      cwmpVersion : currentRequest.sessionData.cwmpVersion
    })
    writeResponse(currentRequest, res)
  )


testSchedule = (schedule, timestamp) ->
  range = schedule.schedule.nextRange(10, new Date(timestamp))
  prev = schedule.schedule.prevRange(1, new Date(timestamp), new Date(timestamp - schedule.duration))

  if prev
    first = +(prev[0] ? 0)
    last = +(prev[1] ? 0) + schedule.duration
  else
    first = +(range[0][0] ? 0)
    last = +(range[0][1] ? 0) + schedule.duration

  for r in range
    if last < r[0]
      break
    last = +(r[1] ? 0) + schedule.duration

  if timestamp >= first
    return [true, Math.max(0, last - timestamp)]
  else
    return [false, first - timestamp]


applyPresets = (currentRequest) ->
  cache.getPresets((err, presetsHash, presets) ->
    throw err if err

    # Filter presets based on existing faults
    blackList = null
    whiteList = null
    whiteListProvisions = null

    RETRY_DELAY = config.get('RETRY_DELAY', currentRequest.sessionData.deviceId)

    if currentRequest.sessionData.faults?
      for channel, fault of currentRequest.sessionData.faults
        if fault.retries?
          retryTimestamp = fault.timestamp + (RETRY_DELAY * Math.pow(2, fault.retries)) * 1000
        else
          retryTimestamp = 0

        if retryTimestamp <= currentRequest.sessionData.timestamp
          whiteList = channel
          whiteListProvisions = fault.provisions
          break

        blackList ?= []
        blackList.push(channel)

    currentRequest.sessionData.presetsHash = presetsHash

    deviceEvents = {}
    iter = currentRequest.sessionData.deviceData.paths.subset(['Events', '*'])
    while (p = iter.next().value)
      if currentRequest.sessionData.timestamp <= currentRequest.sessionData.deviceData.attributes.get(p)?.value?[1][0]
        deviceEvents[p[1]] = true

    parameters = {}
    filteredPresets = []

    for p in presets
      if whiteList?
        continue if p.channel != whiteList
      else if blackList and p.channel in blackList
        continue

      eventsMatch = true
      for k, v of p.events
        if (!v) != (!deviceEvents[k.replace(' ', '_')])
          eventsMatch = false
          break

      continue if (not eventsMatch) or (p.schedule and not
        (p.schedule.schedule and testSchedule(p.schedule, currentRequest.sessionData.timestamp)[0]))

      filteredPresets.push(p)
      for k of p.precondition
        p = k.split(/([^a-zA-Z0-9\-\_\.].*)/, 1)[0]
        parameters[p] = common.parsePath(p)

    declarations = []
    for k, v of parameters
      declarations.push([v, 1, {value: 1}])

    session.rpcRequest(currentRequest.sessionData, declarations, (err, faults, id, rpcRequest) ->
      throw err if err

      if faults?
        throw new Error('A fault occured while trying to discover parameters to test preset preconditions: ' + JSON.stringify(faults))

      if rpcRequest?
        return sendRpcRequest(currentRequest, id, rpcRequest)

      for k, v of parameters
        unpacked = device.unpack(currentRequest.sessionData.deviceData, v)
        if unpacked[0] and (vv = currentRequest.sessionData.deviceData.attributes.get(unpacked[0]).value?[1])?
          parameters[k] = vv[0]
        else
          delete parameters[k]

      if whiteList?
        session.addProvisions(currentRequest.sessionData, whiteList, whiteListProvisions)

      for p in filteredPresets
        if query.testFilter(parameters, p.precondition)
          session.addProvisions(currentRequest.sessionData, p.channel, p.provisions)

      session.rpcRequest(currentRequest.sessionData, null, (err, faults, id, rpcRequest, doneProvisions) ->
        throw err if err

        if faults?
          for channel, fault of faults
            if not fault.retries? and Object.keys(faults).length == 1
              fault.retries = 0
            currentRequest.sessionData.faults[channel] = fault
            util.log("#{currentRequest.sessionData.deviceId}: Fault response for channel #{channel} (retries: #{fault.retries ? 0})")

          return nextRpc(currentRequest)

        for p, i in doneProvisions or []
          delete currentRequest.sessionData.faults[currentRequest.sessionData.channels[i]]

        sendRpcRequest(currentRequest, id, rpcRequest)
      )
    )
  )


nextRpc = (currentRequest) ->
  session.rpcRequest(currentRequest.sessionData, null, (err, faults, id, rpcRequest, doneProvisions) ->
    throw err if err

    if faults?
      for channel, fault of faults
        if channel.startsWith('_task_')
          taskId = channel.slice(6)
          for t, j in currentRequest.sessionData.tasks
            if t._id == taskId
              currentRequest.sessionData.tasks.splice(j, 1)
              fault.expiry = t.expiry if t.expiry?
              break

        if not fault.retries? and Object.keys(faults).length == 1
          fault.retries = 0
        currentRequest.sessionData.faults[channel] = fault
        util.log("#{currentRequest.sessionData.deviceId}: Fault response for channel #{channel} (retries: #{fault.retries ? 0})")

      return nextRpc(currentRequest)

    if rpcRequest?
      return sendRpcRequest(currentRequest, id, rpcRequest)

    for p, i in doneProvisions or []
      delete currentRequest.sessionData.faults[currentRequest.sessionData.channels[i]]

      if p[0] == '_task'
        currentRequest.sessionData.doneTasks ?= []
        currentRequest.sessionData.doneTasks.push(p[1])
        for t, j in currentRequest.sessionData.tasks
          if String(t._id) == p[1]
            currentRequest.sessionData.tasks.splice(j, 1)
            break

    for task in currentRequest.sessionData.tasks
      # Delete if expired
      if task.expiry <= currentRequest.sessionData.timestamp
        util.log("#{currentRequest.sessionData.deviceId}: Task is expired #{task.name}(#{task._id})")
        currentRequest.sessionData.doneTasks ?= []
        currentRequest.sessionData.doneTasks.push(String(task._id))
        continue

      switch task.name
        when 'getParameterValues'
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [['_task', task._id, 'getParameterValues'].concat(task.parameterNames)])
        when 'setParameterValues'
          t = ['_task', task._id, 'setParameterValues']
          for p in task.parameterValues
            t.push(p[0])
            t.push(p[1])
            t.push(p[2] ? '')
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [t])
        when 'refreshObject'
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [['_task', task._id, 'refreshObject', task.objectName]])
        when 'reboot'
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [['_task', task._id, 'reboot']])
        when 'factoryReset'
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [['_task', task._id, 'factoryReset']])
        when 'download'
          session.addProvisions(currentRequest.sessionData, "_task_#{task._id}", [['_task', task._id, 'download', task.fileType, task.fileName, task.targetFileName]])
        else
          throw new Error('Task name not recognized')

      return nextRpc(currentRequest)

    return applyPresets(currentRequest)
  )


sendRpcRequest = (currentRequest, id, rpcRequest) ->
  if not rpcRequest?
    session.end(currentRequest.sessionData, (err, isNew) ->
      throw err if err

      delete currentSessions.get(currentRequest.httpRequest.connection)[currentRequest.sessionData.sessionId]
      if isNew
        util.log("#{currentRequest.sessionData.deviceId}: New device registered")

      db.clearTasks(currentRequest.sessionData.doneTasks, (err) ->
        throw err if err

        if not currentRequest.sessionData.faults?
          return writeResponse(currentRequest, soap.response(null))

        db.syncFaults(currentRequest.sessionData.deviceId, currentRequest.sessionData.faults, (err) ->
          throw err if err
          writeResponse(currentRequest, soap.response(null))
        )
      )
    )
    return

  if rpcRequest.type is 'Download'
    if not rpcRequest.url?
      FS_PORT = config.get('FS_PORT')
      FS_IP = config.get('FS_IP')
      FS_SSL = config.get('FS_SSL')
      rpcRequest.url = if FS_SSL then 'https://' else 'http://'
      rpcRequest.url += FS_IP
      rpcRequest.url += ":#{FS_PORT}" if FS_PORT != 80
      rpcRequest.url += "/#{encodeURIComponent(rpcRequest.fileName)}"

    if not rpcRequest.fileSize?
      return cache.getFiles((err, hash, files) ->
        throw err if err

        if rpcRequest.fileName of files
          rpcRequest.fileSize = files[rpcRequest.fileName].length
        else
          rpcRequest.fileSize = 0

        return sendRpcRequest(currentRequest, id, rpcRequest)
      )

  util.log("#{currentRequest.sessionData.deviceId}: #{rpcRequest.type} (#{id})")

  res = soap.response({
    id : id,
    methodRequest : rpcRequest,
    cwmpVersion : currentRequest.sessionData.cwmpVersion
  })

  writeResponse(currentRequest, res)


getSession = (httpRequest, callback) ->
  # Separation by comma is important as some devices don't comform to standard
  COOKIE_REGEX = /\s*([a-zA-Z0-9\-_]+?)\s*=\s*"?([a-zA-Z0-9\-_]*?)"?\s*(,|;|$)/g
  while match = COOKIE_REGEX.exec(httpRequest.headers.cookie)
    sessionId = match[2] if match[1] == 'session'

  return callback() if not sessionId?

  sessionData = currentSessions.get(httpRequest.connection)?[sessionId]

  if sessionData?
    return callback(null, sessionData)

  setTimeout(() ->
    db.redisClient.eval('local v=redis.call("get",KEYS[1]);redis.call("del",KEYS[1]);return v;', 1, "session_#{sessionId}", (err, sessionDataString) ->
      return callback(err) if err or not sessionDataString
      session.deserialize(sessionDataString, (err, sessionData) ->
        return callback(err) if err
        currentSessions.get(httpRequest.connection)[sessionId] = sessionData
        callback(null, sessionData)
      )
    )
  , 100
  )


currentSessions = new WeakMap()

# When socket closes, store active sessions in redis
onConnection = (socket) ->
  currentSessions.set(socket, {})
  socket.on('close', () ->
    sessions = currentSessions.get(socket)
    for sessionId, sessionData of sessions
      session.serialize(sessionData, (err, sessionDataString) ->
        throw err if err
        # TODO don't set if process is shutting down
        db.redisClient.setex("session_#{sessionId}", sessionData.timeout, sessionDataString, (err) ->
          throw err if err
        )
      )
  )


listener = (httpRequest, httpResponse) ->
  if httpRequest.method != 'POST'
    httpResponse.writeHead 405, {'Allow': 'POST'}
    httpResponse.end('405 Method Not Allowed')
    return

  if httpRequest.headers['content-encoding']?
    switch httpRequest.headers['content-encoding']
      when 'gzip'
        stream = httpRequest.pipe(zlib.createGunzip())
      when 'deflate'
        stream = httpRequest.pipe(zlib.createInflate())
      else
        httpResponse.writeHead(415)
        httpResponse.end('415 Unsupported Media Type')
        return
  else
    stream = httpRequest

  chunks = []
  bytes = 0

  stream.on('data', (chunk) ->
    chunks.push(chunk)
    bytes += chunk.length
  )

  httpRequest.getBody = () ->
    # Write all chunks into a Buffer
    body = new Buffer(bytes)
    offset = 0
    chunks.forEach((chunk) ->
      chunk.copy(body, offset, 0, chunk.length)
      offset += chunk.length
    )
    return body

  stream.on('end', () ->
    cwmpRequest = null
    getSession(httpRequest, f = (err, sessionData) ->
      throw err if err
      cwmpRequest ?= soap.request(httpRequest, sessionData?.cwmpVersion)
      if not sessionData?
        if cwmpRequest.methodRequest?.type isnt 'Inform'
          httpResponse.writeHead(400)
          httpResponse.end('Session is expired')
          return

        deviceId = common.generateDeviceId(cwmpRequest.methodRequest.deviceId)
        return session.init(deviceId, cwmpRequest.cwmpVersion, cwmpRequest.sessionTimeout ? config.get('SESSION_TIMEOUT', deviceId), (err, sessionData) ->
          throw err if err

          sessionData.sessionId = crypto.randomBytes(8).toString('hex')

          currentSessions.get(httpRequest.connection)[sessionData.sessionId] = sessionData

          httpRequest.connection.setTimeout(sessionData.timeout * 1000)

          db.getDueTasksAndFaultsAndOperations(deviceId, sessionData.timestamp, (err, dueTasks, faults, operations) ->
            throw err if err
            sessionData.tasks = dueTasks
            sessionData.faults = faults
            sessionData.operations = operations

            # Delete expired faults
            for k, v of sessionData.faults
              delete sessionData.faults[k] if v.expiry >= sessionData.timestamp

            f(null, sessionData)
          )
        )

      currentRequest = {
        httpRequest : httpRequest,
        httpResponse : httpResponse,
        sessionData : sessionData
      }

      if config.get('DEBUG', currentRequest.sessionData.deviceId)
        dump = "# REQUEST #{new Date(Date.now())}\n" + JSON.stringify(httpRequest.headers) + "\n#{httpRequest.getBody()}\n\n"
        require('fs').appendFile("./debug/#{currentRequest.sessionData.deviceId}.dump", dump, (err) ->
          throw err if err
        )

      if cwmpRequest.methodRequest?
        if cwmpRequest.methodRequest.type is 'Inform'
          inform(currentRequest, cwmpRequest)
        else if cwmpRequest.methodRequest.type is 'TransferComplete'
          transferComplete(currentRequest, cwmpRequest)
        else if cwmpRequest.methodRequest.type is 'GetRPCMethods'
          util.log("#{currentRequest.sessionData.deviceId}: GetRPCMethods")
          res = soap.response({
            id : cwmpRequest.id,
            methodResponse : {type : 'GetRPCMethodsResponse', methodList : ['Inform', 'GetRPCMethods', 'TransferComplete', 'RequestDownload']},
            cwmpVersion : currentRequest.sessionData.cwmpVersion
          })
          writeResponse(currentRequest, res)
        else if cwmpRequest.methodRequest.type is 'TransferComplete'
          throw new Error('ACS method not supported')
      else if cwmpRequest.methodResponse?
        session.rpcResponse(currentRequest.sessionData, cwmpRequest.id, cwmpRequest.methodResponse, (err) ->
          throw err if err
          nextRpc(currentRequest)
        )
      else if cwmpRequest.fault?
        session.rpcFault(currentRequest.sessionData, cwmpRequest.id, cwmpRequest.fault, (err, faults) ->
          throw err if err

          for channel, fault of faults
            if channel.startsWith('_task_')
              taskId = channel.slice(6)
              for t, j in currentRequest.sessionData.tasks
                if t._id == taskId
                  currentRequest.sessionData.tasks.splice(j, 1)
                  fault.expiry = t.expiry if t.expiry?
                  break

            if not fault.retries? and Object.keys(faults).length == 1
              fault.retries = 0

            currentRequest.sessionData.faults[channel] = fault
            util.log("#{currentRequest.sessionData.deviceId}: Fault response for channel #{channel} (retries: #{fault.retries ? 0})")

          nextRpc(currentRequest)
        )
      else # CPE sent empty response
        session.timeoutOperations(currentRequest.sessionData, (err, faults) ->
          throw err if err

          for channel, fault of faults
            if not fault.retries? and Object.keys(faults).length == 1
              fault.retries = 0

            currentRequest.sessionData.faults[channel] = fault
            util.log("#{currentRequest.sessionData.deviceId}: Fault response for channel #{channel} (retries: #{fault.retries ? 0})")

          nextRpc(currentRequest)
        )
    )
  )


exports.listener = listener
exports.onConnection = onConnection