###

dnschain
http://dnschain.net

Copyright (c) 2014 okTurtles Foundation

This Source Code Form is subject to the terms of the Mozilla Public
License, v. 2.0. If a copy of the MPL was not distributed with this
file, You can obtain one at http://mozilla.org/MPL/2.0/.

###

if process.env.TEST_DNSCHAIN and not process.env.TEST_REAL_REDIS
    redis = require 'fakeredis'
else
    redis = require 'redis'

module.exports = (dnschain) ->
    # expose these into our namespace
    for k of dnschain.globals
        eval "var #{k} = dnschain.globals.#{k};"

    class ResolverCache
        constructor: (@dnschain) ->
            @log = gNewLogger 'Redis'
            gFillWithRunningChecks @

        start: ->
            @startCheck (cb) =>
                @blockchainEnabled = gConf.get 'redis:blockchain:enabled'
                @oldDNSEnabled = gConf.get 'redis:oldDNS:enabled'
                @oldDNSTTL = gConf.get 'redis:oldDNS:ttl'

                if !@blockchainEnabled and !@oldDNSEnabled
                    @log.info "cache not enabled".bold.yellow
                else
                    @log.info "cache is enabled".bold
                    [host,port] = gConf.get('redis:socket').split(':')
                    @cache =
                        if port?
                            if isNaN(portNum = parseInt port)
                                gErr "Redis port is NaN: #{port}"
                            redis.createClient portNum, host
                        else
                            redis.createClient host
                    @cache.on 'error', (err) =>
                        @log.error "cache errored: #{err.message}", err
                        @shutdown()
                cb()

        shutdown: ->
            @shutdownCheck (cb) =>
                @cache?.end()
                cb()

        get: (key, valueRetriever, valueDoer) ->
            @cache.get key, (err, result) =>
                if err
                    @log.error gLineInfo('caching error'), {err: err}
                if result?
                    @log.debug gLineInfo('resolved from cache'), {key: key}
                    valueDoer null, key, JSON.parse result
                    return
                valueRetriever key, (err2, ttl, value) =>
                    @cache.setex(key, ttl, JSON.stringify value) if not err2
                    valueDoer err, key, value

        resolveResource: (datastore, requestFn, serialization, cb) ->
            if @blockchainEnabled and datastore.cacheTTL?
                retriever = (key, callback) =>
                    requestFn (err, result) =>
                        callback err, datastore.cacheTTL, result
                doer = (err, key, result) =>
                    cb err, result
                @get serialization, retriever, doer
            else
                requestFn cb

        resolveOldDNS: (req, cb) ->
            if @oldDNSEnabled
                q = req.question[0]
                retriever = (key, callback) =>
                    f = (err, result) =>
                        ttl =
                            if result.answer[0]?.ttl?
                                Math.min result.answer[0].ttl, @oldDNSTTL
                            else
                                @oldDNSTTL
                        callback err, ttl, result
                    @dnschain.dns.oldDNSLookup req, f
                doer = (err, key, result) =>
                    cb err, result
                @get "oldDNS:#{q.name}:#{q.type}", retriever, doer
            else
                @dnschain.dns.oldDNSLookup req, cb